r2park/commands.go

package main

import (
	"fmt"
	"regexp"
	"strconv"
	"time"

	"github.com/bwmarrin/discordgo"
	"github.com/davecgh/go-spew/spew"
	"github.com/samber/lo"
	"github.com/sirupsen/logrus"
	"github.com/zekroTJA/timedmap"
)

// In order for the modal submission to be useful, the context for it's initial request must be stored.
var SubmissionContexts = timedmap.New(5 * time.Minute)

var codePattern = regexp.MustCompile(`^[a-zA-Z0-9]{4,12}$`)

var CodeCommandDefinition = &discordgo.ApplicationCommand{
	Name:        "code",
	Description: "Set the guest code for a given location",
	Options: []*discordgo.ApplicationCommandOption{
		{
			Type:         discordgo.ApplicationCommandOptionString,
			Name:         "location",
			Description:  "The complex to set the code for",
			Required:     true,
			Autocomplete: true,
		},
		{
			Type:        discordgo.ApplicationCommandOptionString,
			Name:        "code",
			Description: "The new code to set",
			Required:    true,
		},
	},
}

func CodeCommandHandler(session *discordgo.Session, interaction *discordgo.InteractionCreate) {
	log := logrus.WithFields(logrus.Fields{
		"interaction": interaction.ID,
		"user":        interaction.Member.User.ID,
		"command":     "code",
	})

	switch interaction.Type {

	case discordgo.InteractionApplicationCommand:
		data := interaction.ApplicationCommandData()

		locationId, _ := strconv.Atoi(data.Options[0].StringValue())
		code := data.Options[1].StringValue()
		userId, _ := strconv.Atoi(interaction.Member.User.ID)

		// Validate that the location exists
		if !LocationExists(int64(locationId)) {
			HandleError(session, interaction, nil, "The location provided does not exist.")
			return
		}

		// Validate that the code has no invalid characters
		if !codePattern.MatchString(code) {
			HandleError(session, interaction, nil, "The code provided contains invalid characters.")
			return
		}

		alreadySet := StoreCode(code, int64(locationId), userId)
		responseText := "Your guest code at \"%s\" has been set."
		if alreadySet {
			responseText = "Your guest code at \"%s\" has been updated."
		}

		location := cachedLocationsMap[uint(locationId)]

		session.InteractionRespond(interaction.Interaction, &discordgo.InteractionResponse{
			Type: discordgo.InteractionResponseChannelMessageWithSource,
			Data: &discordgo.InteractionResponseData{
				Embeds: []*discordgo.MessageEmbed{
					{
						Footer: &discordgo.MessageEmbedFooter{
							Text: GetFooterText(),
						},
						Description: fmt.Sprintf(responseText, location.name),
						Fields:      []*discordgo.MessageEmbedField{},
					},
				},
				AllowedMentions: &discordgo.MessageAllowedMentions{},
			},
		})

	case discordgo.InteractionApplicationCommandAutocomplete:
		data := interaction.ApplicationCommandData()
		var choices []*discordgo.ApplicationCommandOptionChoice

		LocationOption := data.Options[0]

		switch {
		case LocationOption.Focused:
			// Seed value is based on the user ID + a 15 minute interval)
			userId, _ := strconv.Atoi(interaction.Member.User.ID)
			seedValue := int64(userId) + (time.Now().Unix() / 15 * 60)
			locations := FilterLocations(GetLocations(), data.Options[0].StringValue(), 25, seedValue)

			// Convert the locations to choices
			choices = make([]*discordgo.ApplicationCommandOptionChoice, len(locations))
			for i, location := range locations {
				choices[i] = &discordgo.ApplicationCommandOptionChoice{
					Name:  location.name,
					Value: strconv.Itoa(int(location.id)),
				}
			}
		default:
			// An option was focused, but it does not have a handler.
			var focusedOption *discordgo.ApplicationCommandInteractionDataOption
			focusedIndex := 0
			for i, option := range data.Options {
				if option.Focused {
					focusedOption = option
					focusedIndex = i
					break
				}
			}
			log.WithFields(logrus.Fields{"focusedIndex": focusedIndex, "focusedOption": focusedOption.Name, "focusedOption.value": focusedOption.Value}).Warn("Unhandled autocomplete option")
			return
		}

		err := session.InteractionRespond(interaction.Interaction, &discordgo.InteractionResponse{
			Type: discordgo.InteractionApplicationCommandAutocompleteResult,
			Data: &discordgo.InteractionResponseData{
				Choices: choices,
			},
		})
		if err != nil {
			panic(err)
		}
	}
}

var (
	LocationOption = &discordgo.ApplicationCommandOption{
		Type:         discordgo.ApplicationCommandOptionString,
		Name:         "location",
		Description:  "The complex to register with",
		Required:     true,
		Autocomplete: true,
	}
	GuestCodeOption = &discordgo.ApplicationCommandOption{
		Type:        discordgo.ApplicationCommandOptionString,
		Name:        "code",
		Description: "The guest code, if required",
		Required:    false,
	}
	RegisterCommandDefinition = &discordgo.ApplicationCommand{
		Name:        "register",
		Description: "Register a vehicle for parking",
		Options: []*discordgo.ApplicationCommandOption{
			LocationOption,
			GuestCodeOption,
		},
	}
)

func RegisterCommandHandler(session *discordgo.Session, interaction *discordgo.InteractionCreate) {
	log := logrus.WithFields(logrus.Fields{
		"interaction": interaction.ID,
		"user":        interaction.Member.User.ID,
		"command":     "code",
	})

	switch interaction.Type {

	case discordgo.InteractionApplicationCommand:
		data := interaction.ApplicationCommandData()

		locationId, parseErr := strconv.Atoi(data.Options[0].StringValue())
		if parseErr != nil {
			HandleError(session, interaction, parseErr, "Error occurred while parsing location id")
			return
		}

		var useStoredCode bool
		var code string

		// Check if a guest code was provided (and set it)
		_, guestCodeProvided := lo.Find(data.Options, func(option *discordgo.ApplicationCommandInteractionDataOption) bool {
			code = option.StringValue()
			return option.Name == GuestCodeOption.Name
		})

		userId, parseErr := strconv.Atoi(interaction.Member.User.ID)
		if parseErr != nil {
			HandleError(session, interaction, parseErr, "Error occurred while parsing user id")
			return
		}

		// Check if a guest code is required for this location
		guestCodeCondition := GetCodeRequirement(int64(locationId))

		// TODO: Add case for when guest code is provided but not required

		// Circumstance under which error is certain
		if !guestCodeProvided && guestCodeCondition == GuestCodeNotRequired {
			// A guest code could be stored, so check for it.
			log.WithField("location", locationId).Debug("No guest code provided for location, but one is not required. Checking for stored code.")
			code = GetCode(int64(locationId), int(userId))

			if code == "" {
				// No code was stored, error out.
				HandleError(session, interaction, nil, ":x: This location requires a guest code.")
				return
			} else {
				// Code available, use it.
				log.WithFields(logrus.Fields{
					"locationId": locationId,
					"code":       code,
				}).Debug("Using stored code for location")
				guestCodeProvided = true
				useStoredCode = true
			}
		}

		// Get the form for the location
		var form GetFormResult
		if guestCodeProvided {
			form = GetVipForm(uint(locationId), code)

			// requireGuestCode being returned for a VIP form indicates an invalid code.
			if form.requireGuestCode {
				// Handling is the same for both cases, but the message differs & the code is removed if it was stored.
				if useStoredCode {
					HandleError(session, interaction, nil, ":x: This location requires a guest code and the one stored was not valid (and subsequently deleted).")
					RemoveCode(int64(locationId), int(userId))
				} else {
					HandleError(session, interaction, nil, ":x: This location requires a guest code and the one provided was not valid.")
				}
				return
			}
		} else {
			form = GetForm(uint(locationId))

			if form.requireGuestCode {
				// The code ended up being required, so we mark it as such.
				if guestCodeCondition == Unknown {
					log.WithFields(logrus.Fields{
						"locationId": locationId,
					}).Debug("Marking location as requiring a guest code")
					SetCodeRequirement(int64(locationId), true)
				}
				HandleError(session, interaction, nil, ":x: This location requires a guest code.")
				return
			}
		}

		// Convert the form into message components for a modal presented to the user
		registrationFormComponents := FormToModalComponents(form)

		// The ID of the original interaction is used as the identifier for the registration context (uint64)
		registerIdentifier, parseErr := strconv.ParseUint(interaction.ID, 10, 64)
		if parseErr != nil {
			HandleError(session, interaction, parseErr, "Error occurred while parsing interaction message identifier")
		}

		// Parse resident profile ID
		residentProfileId, parseErr := strconv.ParseUint(form.residentProfileId, 10, 64)
		if parseErr != nil {
			HandleError(session, interaction, parseErr, "Error occurred while parsing resident profile identifier")
		}

		// Log the registration context at debug
		log.WithFields(logrus.Fields{
			"registerIdentifier": registerIdentifier,
			"propertyId":         locationId,
			"residentId":         form.residentProfileId,
		})

		// Store the registration context for later use
		SubmissionContexts.Set(registerIdentifier, &RegisterContext{
			hiddenKeys: form.hiddenInputs,
			propertyId: uint(locationId),
			requiredFormKeys: lo.Map(form.fields, func(field Field, _ int) string {
				return field.id
			}),
			residentId: uint(residentProfileId),
		}, time.Hour)

		registrationFormComponents = append(registrationFormComponents, discordgo.ActionsRow{
			Components: []discordgo.MessageComponent{
				discordgo.TextInput{
					CustomID:  "email",
					Label:     "Email Address (for confirmation)",
					Style:     discordgo.TextInputShort,
					Required:  false,
					MinLength: 1,
				},
			},
		})

		response := discordgo.InteractionResponse{
			Type: discordgo.InteractionResponseModal,
			Data: &discordgo.InteractionResponseData{
				CustomID:   "register:" + interaction.ID,
				Title:      "Vehicle Registration",
				Components: registrationFormComponents,
			},
		}

		err := session.InteractionRespond(interaction.Interaction, &response)
		if err != nil {
			log.WithField("dump", spew.Sdump(response)).Error(err)
		}

	// Autocomplete is used to provide the user with a list of locations to choose from
	case discordgo.InteractionApplicationCommandAutocomplete:
		data := interaction.ApplicationCommandData()
		var choices []*discordgo.ApplicationCommandOptionChoice

		// Find the focused option
		focusedOption, _ := lo.Find(data.Options, func(option *discordgo.ApplicationCommandInteractionDataOption) bool {
			return option.Focused
		})

		switch focusedOption.Name {
		case LocationOption.Name:
			// Seed value is based on the user ID + a 15 minute interval)
			userId, _ := strconv.Atoi(interaction.Member.User.ID)
			seedValue := int64(userId) + (time.Now().Unix() / 15 * 60)
			locations := FilterLocations(GetLocations(), data.Options[0].StringValue(), 25, seedValue)

			// Convert the locations to choices
			choices = make([]*discordgo.ApplicationCommandOptionChoice, len(locations))
			for i, location := range locations {
				choices[i] = &discordgo.ApplicationCommandOptionChoice{
					Name:  location.name,
					Value: strconv.Itoa(int(location.id)),
				}
			}

		default:
			// An option was focused, but it does not have a handler.
			log.WithFields(logrus.Fields{"focusedOption": focusedOption.Name, "focusedOption.value": focusedOption.Value}).Warn("Unhandled autocomplete option")
		}

		err := session.InteractionRespond(interaction.Interaction, &discordgo.InteractionResponse{
			Type: discordgo.InteractionApplicationCommandAutocompleteResult,
			Data: &discordgo.InteractionResponseData{
				Choices: choices, // This is basically the whole purpose of autocomplete interaction - return custom options to the user.
			},
		})
		if err != nil {
			panic(err)
		}
	}
}