Birthday Feature

commands/enabled.go

@@ -31,6 +31,7 @@ func populateSlashCommands(ctx ddtrace.SpanContext) {
 	SlashCommands["update"] = slash.Update
 	SlashCommands["query"] = slash.Query
 	SlashCommands["scoreboard"] = slash.Scoreboard
+	SlashCommands["birthday"] = slash.Birthday
 }
 
 // populateHandlers populates the Handlers map with all of the handlers
@@ -68,4 +69,5 @@ func populateScheduledEvents(ctx ddtrace.SpanContext) {
 	ScheduledEvents["heartbeat"] = scheduled.Heartbeat
 	ScheduledEvents["status"] = scheduled.Status
 	ScheduledEvents["update"] = scheduled.Update
+	ScheduledEvents["birthday"] = scheduled.Birthday
 }

commands/scheduled/birthday.go

@@ -0,0 +1,116 @@
+package scheduled
+
+import (
+	"time"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/ritsec/ops-bot-iii/config"
+	"github.com/ritsec/ops-bot-iii/data"
+	"github.com/ritsec/ops-bot-iii/helpers"
+	"github.com/ritsec/ops-bot-iii/logging"
+	"github.com/robfig/cron"
+	"github.com/sirupsen/logrus"
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace"
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
+)
+
+var (
+	birthdayRoleID string = config.GetString("commands.birthday.role_id")
+)
+
+// removeBirthday removes the birthday role from a user
+func removeBirthday(s *discordgo.Session, UserID string, ctx ddtrace.SpanContext) error {
+	span := tracer.StartSpan(
+		"commands.scheduled.birthday:removeBirthday",
+		tracer.ResourceName("Scheduled.Birthday.removeBirthday"),
+		tracer.ChildOf(ctx),
+	)
+	defer span.Finish()
+
+	return s.GuildMemberRoleRemove(config.GuildID, UserID, birthdayRoleID)
+}
+
+// addBirthday adds the birthday role to a user
+func addBirthday(s *discordgo.Session, UserID string, ctx ddtrace.SpanContext) error {
+	span := tracer.StartSpan(
+		"commands.scheduled.birthday:addBirthday",
+		tracer.ResourceName("Scheduled.Birthday.addBirthday"),
+		tracer.ChildOf(ctx),
+	)
+	defer span.Finish()
+
+	return s.GuildMemberRoleAdd(config.GuildID, UserID, birthdayRoleID)
+}
+
+// Birthday is a scheduled event that runs at midnight to remove existing birthday roles and add new ones
+func Birthday(s *discordgo.Session, quit chan interface{}) error {
+	span := tracer.StartSpan(
+		"commands.scheduled.birthday:Birthday",
+		tracer.ResourceName("Scheduled.Birthday"),
+	)
+	defer span.Finish()
+
+	// Set the cron job to run at EST
+	est, err := time.LoadLocation("America/New_York")
+	if err != nil {
+		logging.Error(s, err.Error(), nil, span)
+		return err
+	}
+
+	c := cron.NewWithLocation(est)
+
+	// Run at midnight
+	err = c.AddFunc("0 0 0 * * *", func() {
+		internalSpan := tracer.StartSpan(
+			"commands.scheduled.birthday:Birthday.Cron",
+			tracer.ResourceName("Scheduled.Birthday.Cron"),
+			tracer.ChildOf(span.Context()),
+		)
+		defer internalSpan.Finish()
+
+		today := time.Now()
+		yesterday := today.Add(-24 * time.Hour)
+
+		// Get yesterday's birthdays
+		entRemoveBirthdays, err := data.Birthday.GetBirthdays(yesterday.Day(), int(yesterday.Month()), internalSpan.Context())
+		if err != nil {
+			logging.Error(s, "failed to get yesterday's birthdays", nil, span, logrus.Fields{"error": err})
+			return
+		}
+
+		// Remove yesterday's birthdays
+		for _, entRemoveBirthday := range entRemoveBirthdays {
+			err = removeBirthday(s, entRemoveBirthday.Edges.User.ID, internalSpan.Context())
+			if err != nil {
+				logging.Error(s, "failed to remove birthday for "+helpers.AtUser(entRemoveBirthday.Edges.User.ID), nil, span, logrus.Fields{"error": err})
+				return
+			}
+		}
+
+		// Get today's birthdays
+		entAddBirthday, err := data.Birthday.GetBirthdays(today.Day(), int(today.Month()), internalSpan.Context())
+		if err != nil {
+			logging.Error(s, "failed to get today's birthdays", nil, span, logrus.Fields{"error": err})
+			return
+		}
+
+		// Add today's birthdays
+		for _, entAddBirthday := range entAddBirthday {
+			err = addBirthday(s, entAddBirthday.Edges.User.ID, internalSpan.Context())
+			if err != nil {
+				logging.Error(s, "failed to add birthday for "+helpers.AtUser(entAddBirthday.Edges.User.ID), nil, span, logrus.Fields{"error": err})
+				return
+			}
+		}
+	})
+	if err != nil {
+		logging.Error(s, "failed to create cron job", nil, span, logrus.Fields{"error": err})
+		return err
+	}
+
+	c.Start()
+	<-quit
+	c.Stop()
+
+	return nil
+}

commands/slash/birthday.go

@@ -0,0 +1,235 @@
+package slash
+
+import (
+	"fmt"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/ritsec/ops-bot-iii/commands/slash/permission"
+	"github.com/ritsec/ops-bot-iii/data"
+	"github.com/ritsec/ops-bot-iii/logging"
+	"github.com/sirupsen/logrus"
+	"gopkg.in/DataDog/dd-trace-go.v1/ddtrace/tracer"
+)
+
+// Birthday is a slash command that allows users to add or edit their birthday
+func Birthday() (*discordgo.ApplicationCommand, func(s *discordgo.Session, i *discordgo.InteractionCreate)) {
+	var minValue float64 = 1
+	return &discordgo.ApplicationCommand{
+			Name:                     "birthday",
+			Description:              "Add or edit birthday",
+			DefaultMemberPermissions: &permission.Member,
+			Options: []*discordgo.ApplicationCommandOption{
+				{
+					Type:     discordgo.ApplicationCommandOptionInteger,
+					Name:     "month",
+					Required: true,
+					Choices: []*discordgo.ApplicationCommandOptionChoice{
+						{
+							Name:  "January",
+							Value: 1,
+						},
+						{
+							Name:  "February",
+							Value: 2,
+						},
+						{
+							Name:  "March",
+							Value: 3,
+						},
+						{
+							Name:  "April",
+							Value: 4,
+						},
+						{
+							Name:  "May",
+							Value: 5,
+						},
+						{
+							Name:  "June",
+							Value: 6,
+						},
+						{
+							Name:  "July",
+							Value: 7,
+						},
+						{
+							Name:  "August",
+							Value: 8,
+						},
+						{
+							Name:  "September",
+							Value: 9,
+						},
+						{
+							Name:  "October",
+							Value: 10,
+						},
+						{
+							Name:  "November",
+							Value: 11,
+						},
+						{
+							Name:  "December",
+							Value: 12,
+						},
+					},
+				},
+				{
+					Type:     discordgo.ApplicationCommandOptionInteger,
+					Name:     "day",
+					Required: true,
+					MinValue: &minValue,
+					MaxValue: 31,
+				},
+			},
+		},
+		func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+			span := tracer.StartSpan(
+				"commands.slash.birthday:Birthday",
+				tracer.ResourceName("/birthday"),
+			)
+			defer span.Finish()
+
+			// get user parameters
+			month := int(i.ApplicationCommandData().Options[0].IntValue())
+			day := int(i.ApplicationCommandData().Options[1].IntValue())
+
+			// TODO: check if birthday is current date and prevent edit
+
+			// check is birthday already exists
+			exists, err := data.Birthday.Exists(i.Member.User.ID, span.Context())
+			if err != nil {
+				logging.Error(s, "encounted error when checking if birthday exists", i.Member.User, span, logrus.Fields{"err": err.Error()})
+			}
+
+			if exists {
+				// birthday exists, update existing one
+
+				// get current birthday
+				entBirthday, err := data.Birthday.Get(i.Member.User.ID, span.Context())
+				if err != nil {
+					logging.Error(s, "encounted error when getting birthday from user", i.Member.User, span, logrus.Fields{"err": err.Error()})
+				}
+
+				// update birthday
+				_, err = entBirthday.Update().SetDay(day).SetMonth(month).Save(data.Ctx)
+				if err != nil {
+					logging.Error(s, "encounted error when updating birthday", i.Member.User, span, logrus.Fields{"err": err.Error()})
+				}
+
+				// log that birthday has been updated
+				logging.Debug(s, "Updated birthday for "+i.Member.User.Username, i.Member.User, span)
+
+				// send user message that birthday was successfully updated
+				err = s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
+					Type: discordgo.InteractionResponseChannelMessageWithSource,
+					Data: &discordgo.InteractionResponseData{
+						Content: fmt.Sprintf("Changed birtday to %d/%d", month, day),
+						Flags:   discordgo.MessageFlagsEphemeral,
+					},
+				})
+				if err != nil {
+					logging.Error(s, "encounted error when responding to user", i.Member.User, span, logrus.Fields{"err": err.Error()})
+				}
+
+			} else {
+				// birthday does not exist create new one
+
+				// create new birthday for user
+				_, err = data.Birthday.Create(i.Member.User.ID, day, month, span.Context())
+				if err != nil {
+					logging.Error(s, "encounted error when creating birthday", i.Member.User, span, logrus.Fields{"err": err.Error()})
+				}
+
+				// log that birthday has been updated
+				logging.Debug(s, "Created birthday for "+i.Member.User.Username, i.Member.User, span)
+
+				// send user message that birthday was successfully updated
+				err = s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
+					Type: discordgo.InteractionResponseChannelMessageWithSource,
+					Data: &discordgo.InteractionResponseData{
+						Content: "Created birthday for " + i.Member.User.Username + " ",
+						Flags:   discordgo.MessageFlagsEphemeral,
+					},
+				})
+				if err != nil {
+					logging.Error(s, "encounted error when responding to user", i.Member.User, span, logrus.Fields{"err": err.Error()})
+				}
+			}
+		}
+}
+
+// BirthdayRemove is a slash command that allows users to remove their birthday
+func BirthdayRemove() (*discordgo.ApplicationCommand, func(s *discordgo.Session, i *discordgo.InteractionCreate)) {
+	return &discordgo.ApplicationCommand{
+			Name:                     "birthday_remove",
+			Description:              "Remove a birthday",
+			DefaultMemberPermissions: &permission.Member,
+		},
+		func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+			span := tracer.StartSpan(
+				"commands.slash.birthday:BirthdayRemove",
+				tracer.ResourceName("/birthday_remove"),
+			)
+			defer span.Finish()
+
+			// TODO: check if birthday is current date and prevent removal
+
+			// check if birthday exists
+			exists, err := data.Birthday.Exists(i.Member.User.ID, span.Context())
+			if err != nil {
+				logging.Error(s, "Birthday has been removed", i.Member.User, span)
+			}
+
+			if exists {
+				// birthday exists, remove it
+
+				// remove birthday
+				_, err = data.Birthday.Delete(i.Member.User.ID, span.Context())
+				if err != nil {
+					// respond to user that error occured when removing birthday
+					err = s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
+						Type: discordgo.InteractionResponseChannelMessageWithSource,
+						Data: &discordgo.InteractionResponseData{
+							Content: "Error occured when removing birthday",
+							Flags:   discordgo.MessageFlagsEphemeral,
+						},
+					})
+					if err != nil {
+						logging.Error(s, "encounted error when responding to user", i.Member.User, span, logrus.Fields{"err": err.Error()})
+					}
+
+					logging.Error(s, "Birthday has been removed", i.Member.User, span)
+
+				}
+
+				// respond to user that birthday has been removed
+				err = s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
+					Type: discordgo.InteractionResponseChannelMessageWithSource,
+					Data: &discordgo.InteractionResponseData{
+						Content: "Birthday has been removed",
+						Flags:   discordgo.MessageFlagsEphemeral,
+					},
+				})
+				if err != nil {
+					logging.Error(s, "encounted error when responding to user", i.Member.User, span, logrus.Fields{"err": err.Error()})
+				}
+
+			} else {
+				// birthday does not exist
+
+				// respond to user that birthday does not exist
+				err = s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
+					Type: discordgo.InteractionResponseChannelMessageWithSource,
+					Data: &discordgo.InteractionResponseData{
+						Content: "Birthday does not exist",
+						Flags:   discordgo.MessageFlagsEphemeral,
+					},
+				})
+				if err != nil {
+					logging.Error(s, "encounted error when responding to user", i.Member.User, span, logrus.Fields{"err": err.Error()})
+				}
+			}
+
+		}
+}