Add Command Functionality to Discord Package & Implement Feedback Webhooks

2023-06-16 01:24:25 +00:00
parent 204803cd0b
commit 996155205d
10 changed files with 324 additions and 20 deletions
--- a/discord/command.go
+++ b/discord/command.go
@@ -0,0 +1,112 @@
+package discord
+
+import (
+	"log"
+
+	"github.com/bwmarrin/discordgo"
+	"github.com/yeslayla/birdbot/common"
+)
+
+type CommandConfiguration struct {
+	Description       string
+	EphemeralResponse bool
+	Options           map[string]CommandOption
+}
+
+type CommandOption struct {
+	Description string
+	Type        CommandOptionType
+	Required    bool
+}
+
+type CommandOptionType uint64
+
+const (
+	CommandTypeString CommandOptionType = CommandOptionType(discordgo.ApplicationCommandOptionString)
+	CommandTypeInt    CommandOptionType = CommandOptionType(discordgo.ApplicationCommandOptionInteger)
+	CommandTypeBool   CommandOptionType = CommandOptionType(discordgo.ApplicationCommandOptionBoolean)
+	CommandTypeFloat  CommandOptionType = CommandOptionType(discordgo.ApplicationCommandOptionNumber)
+)
+
+// RegisterCommand creates an new command that can be used to interact with bird bot
+func (discord *Discord) RegisterCommand(name string, config CommandConfiguration, handler func(common.User, map[string]any) string) {
+	command := &discordgo.ApplicationCommand{
+		Name:        name,
+		Description: config.Description,
+	}
+
+	// Convert options to discordgo objects
+	command.Options = make([]*discordgo.ApplicationCommandOption, len(config.Options))
+	index := 0
+	for name, option := range config.Options {
+		command.Options[index] = &discordgo.ApplicationCommandOption{
+			Name:        name,
+			Description: option.Description,
+			Required:    option.Required,
+			Type:        discordgo.ApplicationCommandOptionType(option.Type),
+		}
+		index++
+	}
+
+	// Register handler
+	discord.commandHandlers[name] = func(session *discordgo.Session, r *discordgo.InteractionCreate) {
+		if r.Interaction.Type != discordgo.InteractionApplicationCommand {
+			return
+		}
+
+		cmdOptions := r.ApplicationCommandData().Options
+
+		// Parse option types
+		optionsMap := make(map[string]any, len(cmdOptions))
+		for _, opt := range cmdOptions {
+			switch config.Options[opt.Name].Type {
+			case CommandTypeString:
+				optionsMap[opt.Name] = opt.StringValue()
+			case CommandTypeInt:
+				optionsMap[opt.Name] = opt.IntValue()
+			case CommandTypeBool:
+				optionsMap[opt.Name] = opt.BoolValue()
+			case CommandTypeFloat:
+				optionsMap[opt.Name] = opt.FloatValue()
+			default:
+				optionsMap[opt.Name] = opt.Value
+			}
+		}
+
+		result := handler(NewUser(r.Member.User), optionsMap)
+
+		if result != "" {
+			// Handle response
+			responseData := &discordgo.InteractionResponseData{
+				Content: result,
+			}
+
+			if config.EphemeralResponse {
+				responseData.Flags = discordgo.MessageFlagsEphemeral
+			}
+
+			session.InteractionRespond(r.Interaction, &discordgo.InteractionResponse{
+				Type: discordgo.InteractionResponseChannelMessageWithSource,
+				Data: responseData,
+			})
+		} else {
+			log.Printf("Command '%s' did not return a response: %v", name, optionsMap)
+		}
+	}
+
+	cmd, err := discord.session.ApplicationCommandCreate(discord.applicationID, discord.guildID, command)
+	if err != nil {
+		log.Fatalf("Cannot create command '%s': %v", name, err)
+	}
+	discord.commands[name] = cmd
+}
+
+// ClearCommands deregisters all commands from the discord API
+func (discord *Discord) ClearCommands() {
+	for _, v := range discord.commands {
+		err := discord.session.ApplicationCommandDelete(discord.session.State.User.ID, discord.guildID, v.ID)
+		if err != nil {
+			log.Fatalf("Cannot delete command '%s': %v", v.Name, err)
+		}
+	}
+}