Module setup, initial command handling & bot initialization

This commit is contained in:
2023-12-11 20:25:17 -06:00
parent 225a2a78aa
commit 941e6ac531
2 changed files with 155 additions and 0 deletions

11
go.mod Normal file
View File

@@ -0,0 +1,11 @@
module park
go 1.21.3
require (
github.com/bwmarrin/discordgo v0.27.1 // indirect
github.com/gorilla/websocket v1.4.2 // indirect
github.com/joho/godotenv v1.5.1 // indirect
golang.org/x/crypto v0.0.0-20210421170649-83a5a9bb288b // indirect
golang.org/x/sys v0.0.0-20201119102817-f84b799fce68 // indirect
)

144
main.go Normal file
View File

@@ -0,0 +1,144 @@
package main
import (
"log"
"os"
"os/signal"
"github.com/bwmarrin/discordgo"
"github.com/joho/godotenv"
)
var (
session *discordgo.Session
commandDefinitions = []*discordgo.ApplicationCommand{RegisterCommandDefinition}
commandHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
"register": RegisterCommandHandler,
}
)
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(s *discordgo.Session, m *discordgo.InteractionCreate) {
}
// 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 {
log.Fatal("Error loading .env file")
}
session, err := discordgo.New("Bot " + os.Getenv("BOT_TOKEN"))
if err != nil {
log.Fatalf("Invalid bot parameters: %v", err)
}
session.AddHandler(func(s *discordgo.Session, r *discordgo.Ready) {
log.Printf("Logged in as: %v#%v", s.State.User.Username, s.State.User.Discriminator)
// Count serers
guilds := s.State.Guilds
log.Printf("Connected to %d server%s", len(guilds), Plural(len(guilds)))
})
err = session.Open()
if err != nil {
log.Fatalf("Cannot open the session: %v", err)
}
session.AddHandler(func(internalSession *discordgo.Session, interaction *discordgo.InteractionCreate) {
if handler, ok := commandHandlers[interaction.ApplicationCommandData().Name]; ok {
handler(internalSession, interaction)
}
})
log.Printf("Adding %d command%s...", len(commandDefinitions), Plural(len(commandDefinitions)))
registeredCommands := make([]*discordgo.ApplicationCommand, len(commandDefinitions))
for i, v := range commandDefinitions {
cmd, err := session.ApplicationCommandCreate(session.State.User.ID, os.Getenv("BOT_TARGET_GUILD"), v)
if err != nil {
log.Panicf("Cannot create '%v' command: %v", v.Name, err)
}
registeredCommands[i] = cmd
}
defer session.Close()
stop := make(chan os.Signal, 1)
signal.Notify(stop, os.Interrupt)
log.Println("Press Ctrl+C to exit")
<-stop
if true {
log.Printf("Removing %d command%s...\n", len(registeredCommands), Plural(len(registeredCommands)))
for _, v := range registeredCommands {
err := session.ApplicationCommandDelete(session.State.User.ID, os.Getenv("BOT_TARGET_GUILD"), v.ID)
if err != nil {
log.Panicf("Cannot delete '%v' command: %v", v.Name, err)
}
}
}
log.Println("Gracefully shutting down.")
}