mirror of
https://github.com/Xevion/r2park.git
synced 2025-12-06 07:15:57 -06:00
Module setup, initial command handling & bot initialization
This commit is contained in:
11
go.mod
Normal file
11
go.mod
Normal 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
144
main.go
Normal 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.")
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user