From 941e6ac53165dc1fbb6250b266737b39c919a78d Mon Sep 17 00:00:00 2001
From: Xevion <xevion@xevion.dev>
Date: Mon, 11 Dec 2023 20:25:17 -0600
Subject: [PATCH] Module setup, initial command handling & bot initialization

---
 go.mod  |  11 +++++
 main.go | 144 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 155 insertions(+)
 create mode 100644 go.mod
 create mode 100644 main.go

diff --git a/go.mod b/go.mod
new file mode 100644
index 0000000..c065770
--- /dev/null
+++ b/go.mod
@@ -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
+)
diff --git a/main.go b/main.go
new file mode 100644
index 0000000..436ad1b
--- /dev/null
+++ b/main.go
@@ -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.")
+}