implement basic whitelist add and remove functionality

app/commands.go

@@ -0,0 +1,128 @@
+package app
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+var (
+	commands = []*discordgo.ApplicationCommand{
+		{
+			Name:        "whitelist-add",
+			Description: "Add yourself to the whitelist",
+			Options: []*discordgo.ApplicationCommandOption{
+				{
+					Type:        discordgo.ApplicationCommandOptionString,
+					Name:        "minecraft-username",
+					Description: "Minecraft Username",
+					Required:    true,
+				},
+			},
+		},
+		{
+			Name:        "whitelist-remove",
+			Description: "Remove yourself from the whitelist",
+			Options:     []*discordgo.ApplicationCommandOption{},
+		},
+		{
+			Name:        "whitelist-show",
+			Description: "Display your whitelist link",
+			Options:     []*discordgo.ApplicationCommandOption{},
+		},
+	}
+
+	commandHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
+		"whitelist-add": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+			// Access options in the order provided by the user.
+			options := i.ApplicationCommandData().Options
+
+			// Convert the slice into a map
+			optionMap := make(map[string]*discordgo.ApplicationCommandInteractionDataOption, len(options))
+			for _, opt := range options {
+				optionMap[opt.Name] = opt
+			}
+
+			requestedUsername := (optionMap["minecraft-username"]).StringValue()
+			invokingUserName := i.Member.User.Username
+			invokingUserID := i.Member.User.ID
+
+			// check requestedUsername validity
+			valid := checkMcUsernameValid(requestedUsername)
+			if !valid {
+				simpleResponse(s, i, fmt.Sprintf("%s is not a valid minecraft username", requestedUsername))
+				return
+			}
+
+			existingMCUsername, ok := db[invokingUserID]
+			if !ok { // invoking user does not currently have an mc username associated
+				// execute RCON
+				response, err := conn.Execute(fmt.Sprintf("whitelist add %s", requestedUsername))
+				if err != nil {
+					simpleResponse(s, i, fmt.Sprintf("Failed to add %s to whitelist: %s", requestedUsername, err.Error()))
+					return
+				}
+
+				// this can happen if the username is not a real player, OR if the user is already on the whitelist
+				// In both cases we want to exit early to avoid adding an invalid username to the db
+				if !strings.EqualFold(response, fmt.Sprintf("Added %s to the whitelist", requestedUsername)) {
+					simpleResponse(s, i, fmt.Sprintf("Failed to add %s to whitelist: %s", requestedUsername, response))
+					return
+				}
+
+				// save state to db
+				db[invokingUserID] = requestedUsername
+				SaveDB(dbPath, db)
+
+				// send response
+				simpleResponse(s, i, fmt.Sprintf("%s linked minecraft username %s", invokingUserName, requestedUsername))
+			} else {
+				simpleResponse(s, i, fmt.Sprintf("%s already has a linked minecraft username %s", invokingUserName, existingMCUsername))
+			}
+
+		},
+		"whitelist-remove": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+			invokingUserName := i.Member.User.Username
+			invokingUserID := i.Member.User.ID
+
+			existingMCUsername, ok := db[invokingUserID]
+			if !ok { // invoking user does not currently have an mc username associated
+				simpleResponse(s, i, fmt.Sprintf("%s does not have a linked minecraft username", invokingUserName))
+			} else {
+				// execute RCON
+				response, err := conn.Execute(fmt.Sprintf("whitelist remove %s", existingMCUsername))
+				if err != nil {
+					simpleResponse(s, i, fmt.Sprintf("Failed to remove %s from whitelist: %s", existingMCUsername, err.Error()))
+					return
+				}
+
+				// this can happen if the username is not a real player, OR if the user is already not on the whitelist
+				if !strings.EqualFold(response, fmt.Sprintf("Removed %s from the whitelist", existingMCUsername)) {
+					simpleResponse(s, i, fmt.Sprintf("Failed to remove %s from whitelist: %s", existingMCUsername, response))
+					return
+				}
+
+				// save state to db
+				delete(db, invokingUserID)
+				SaveDB(dbPath, db)
+
+				// send response
+				simpleResponse(s, i, fmt.Sprintf("%s unlinked minecraft username %s", invokingUserName, existingMCUsername))
+
+			}
+		},
+		"whitelist-show": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
+			invokingUserName := i.Member.User.Username
+			invokingUserID := i.Member.User.ID
+
+			existingMCUsername, ok := db[invokingUserID]
+
+			if ok {
+				simpleResponse(s, i, fmt.Sprintf("%s is %s", invokingUserName, existingMCUsername))
+			} else {
+				simpleResponse(s, i, fmt.Sprintf("%s does not have a linked minecraft username", invokingUserName))
+			}
+		},
+	}
+)