main.go

package main

import (
	"flag"
	"fmt"
	"log"
	"os"
	"os/signal"
	"syscall"

	"github.com/bwmarrin/discordgo"
	"github.com/yapms/yapbot/poll"
	"github.com/yapms/yapbot/yesno"
)

var (
	Token    string
	GuildID  string
	Discord  *discordgo.Session
	commands = []*discordgo.ApplicationCommand{
		poll.GetCommand(),
		yesno.GetCommand(),
	}
	commandHandlers = map[string]func(session *discordgo.Session, interaction *discordgo.InteractionCreate){
		"poll":  poll.GetHandler(),
		"yesno": yesno.GetHandler(),
	}
)

func parseArguments() {
	flag.StringVar(&Token, "t", "", "Bot Token")
	flag.StringVar(&GuildID, "g", "", "Guild Token")
	flag.Parse()
}

func initializeDiscord() {
	var err error
	Discord, err = discordgo.New("Bot " + Token)
	if err != nil {
		log.Panic(err)
		return
	}
	Discord.Open()
	Discord.Identify.Intents = discordgo.IntentGuildMessages
}

func createCommands() {
	for _, command := range commands {
		fmt.Printf("creating command: %+v\n", command)
		_, err := Discord.ApplicationCommandCreate(Discord.State.User.ID, GuildID, command)
		if err != nil {
			log.Panic(err)
		}
	}
}

func addHandler() {
	Discord.AddHandler(func(session *discordgo.Session, interaction *discordgo.InteractionCreate) {
		name := interaction.ApplicationCommandData().Name
		command, okay := commandHandlers[name]
		if okay {
			command(session, interaction)
		}
	})
}

func cleanupCommands() {
	commands, err := Discord.ApplicationCommands(Discord.State.User.ID, GuildID)
	if err != nil {
		log.Panic(err)
	}

	for _, command := range commands {
		err = Discord.ApplicationCommandDelete(Discord.State.User.ID, GuildID, command.ID)
		if err != nil {
			log.Panic(err)
		}
	}
}

func cleanup() {
	cleanupCommands()
	Discord.Close()
}

func main() {
	parseArguments()
	initializeDiscord()
	createCommands()
	addHandler()

	fmt.Println("Bot is now running. Press CTRL-C to exit.")
	stop := make(chan os.Signal, 1)
	signal.Notify(stop, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
	<-stop

	cleanup()
}