discord-minecraft-whitelist-bot/app/commands.go

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))
			}
		},
	}
)