Move command definition/handler into commands.go

This commit is contained in:
2023-12-12 01:39:04 -06:00
parent 2f593aeb6a
commit c6b83b883c
2 changed files with 93 additions and 117 deletions

93
commands.go Normal file
View File

@@ -0,0 +1,93 @@
package main
import (
"fmt"
"strconv"
"time"
"github.com/bwmarrin/discordgo"
)
var RegisterCommandDefinition = &discordgo.ApplicationCommand{
Name: "register",
Description: "Register a vehicle for parking",
Options: []*discordgo.ApplicationCommandOption{
{
Type: discordgo.ApplicationCommandOptionString,
Name: "location",
Description: "The complex to register with",
Required: true,
Autocomplete: true,
},
{
Type: discordgo.ApplicationCommandOptionString,
Name: "make",
Description: "Make of Vehicle (e.g. Honda)",
Required: true,
},
{
Type: discordgo.ApplicationCommandOptionString,
Name: "model",
Description: "Model of Vehicle (e.g. Civic)",
Required: true,
},
{
Type: discordgo.ApplicationCommandOptionString,
Name: "plate",
Description: "License Plate Number (e.g. 123ABC)",
Required: true,
},
},
}
func RegisterCommandHandler(session *discordgo.Session, interaction *discordgo.InteractionCreate) {
switch interaction.Type {
case discordgo.InteractionApplicationCommand:
session.InteractionRespond(interaction.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Embeds: []*discordgo.MessageEmbed{
{
Footer: &discordgo.MessageEmbedFooter{
Text: fmt.Sprintf("Fetched at %s", time.Now().Format("Monday, January 2, 2006 at 3:04:05PM")),
},
Description: "",
Fields: []*discordgo.MessageEmbedField{},
},
},
AllowedMentions: &discordgo.MessageAllowedMentions{},
},
})
case discordgo.InteractionApplicationCommandAutocomplete:
data := interaction.ApplicationCommandData()
var choices []*discordgo.ApplicationCommandOptionChoice
if data.Options[0].Focused {
// Seed value is based on the user ID + a 15 minute interval)
user_id, _ := strconv.Atoi(interaction.Member.User.ID)
seed_value := int64(user_id) + (time.Now().Unix() / 15 * 60)
locations := FilterLocations(GetLocations(), data.Options[0].StringValue(), 25, seed_value)
choices = make([]*discordgo.ApplicationCommandOptionChoice, len(locations))
for i, location := range locations {
choices[i] = &discordgo.ApplicationCommandOptionChoice{
Name: location.name,
Value: strconv.Itoa(int(location.id)),
}
}
} else {
choices = []*discordgo.ApplicationCommandOptionChoice{}
}
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)
}
}
}

117
main.go
View File

@@ -1,12 +1,9 @@
package main
import (
"fmt"
"log"
"os"
"os/signal"
"strconv"
"time"
"github.com/bwmarrin/discordgo"
"github.com/joho/godotenv"
@@ -20,120 +17,6 @@ var (
}
)
var RegisterCommandDefinition = &discordgo.ApplicationCommand{
Name: "register",
Description: "Register a vehicle for parking",
Options: []*discordgo.ApplicationCommandOption{
{
Type: discordgo.ApplicationCommandOptionString,
Name: "location",
Description: "The complex to register with",
Required: true,
Autocomplete: true,
},
{
Type: discordgo.ApplicationCommandOptionString,
Name: "make",
Description: "Make of Vehicle (e.g. Honda)",
Required: true,
},
{
Type: discordgo.ApplicationCommandOptionString,
Name: "model",
Description: "Model of Vehicle (e.g. Civic)",
Required: true,
},
{
Type: discordgo.ApplicationCommandOptionString,
Name: "plate",
Description: "License Plate Number (e.g. 123ABC)",
Required: true,
},
},
}
func RegisterCommandHandler(session *discordgo.Session, interaction *discordgo.InteractionCreate) {
switch interaction.Type {
case discordgo.InteractionApplicationCommand:
session.InteractionRespond(interaction.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Embeds: []*discordgo.MessageEmbed{
{
Footer: &discordgo.MessageEmbedFooter{
Text: fmt.Sprintf("Fetched at %s", time.Now().Format("Monday, January 2, 2006 at 3:04:05PM")),
},
Description: "",
Fields: []*discordgo.MessageEmbedField{},
},
},
AllowedMentions: &discordgo.MessageAllowedMentions{},
},
})
case discordgo.InteractionApplicationCommandAutocomplete:
data := interaction.ApplicationCommandData()
var choices []*discordgo.ApplicationCommandOptionChoice
if data.Options[0].Focused {
// Seed value is based on the user ID + a 15 minute interval)
user_id, _ := strconv.Atoi(interaction.Member.User.ID)
seed_value := int64(user_id) + (time.Now().Unix() / 15 * 60)
locations := FilterLocations(GetLocations(), data.Options[0].StringValue(), 25, seed_value)
choices = make([]*discordgo.ApplicationCommandOptionChoice, len(locations))
for i, location := range locations {
choices[i] = &discordgo.ApplicationCommandOptionChoice{
Name: location.name,
Value: strconv.Itoa(int(location.id)),
}
}
} else {
choices = []*discordgo.ApplicationCommandOptionChoice{}
}
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)
}
}
}
// func test() {
// body := EncodeForm(
// map[string]string{
// "propertyIdSelected": "22167",
// "propertySource": "parking-snap",
// "guestCode": code,
// })
// req, err := http.NewRequest("POST", url, strings.NewReader(body))
// if err != nil {
// log.Fatal(err)
// }
// req.Header.Set("Content-Type", "application/x-www-form-urlencoded; charset=UTF-8")
// req.Header.Set("Referer", "https://www.register2park.com/register?key=678zv9zzylvw")
// req.Header.Set("User-Agent", "Mozilla/5.0 (X11; Linux x86_64; rv:121.0) Gecko/20100101 Firefox/121.0")
// req.Header.Set("X-Requested-With", "XMLHttpRequest")
// req.Header.Set("Host", "www.register2park.com")
// req.Header.Set("Cookie", "PHPSESSID=dbc956tgnapqv6l81ue56uf1ng")
// resp, err := client.Do(req)
// if err != nil {
// log.Fatal(err)
// }
// if resp.StatusCode == 200 {
// }
// }
func main() {
// Load environment variables
if err := godotenv.Load(); err != nil {