Initial Commit

internal/bot/bot.go

@@ -0,0 +1,59 @@
+package bot
+
+import (
+	"log"
+	"velox-bot/internal/commands"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+type Bot struct {
+	Session            *discordgo.Session
+	AppID              string
+	GuildID            string
+	Commands           []*discordgo.ApplicationCommand
+	registeredCommands []*discordgo.ApplicationCommand
+}
+
+func NewBot(token, appID, guildID string, cmds []*discordgo.ApplicationCommand) (*Bot, error) {
+	session, err := discordgo.New("Bot " + token)
+	if err != nil {
+		return nil, err
+	}
+
+	return &Bot{
+		Session:  session,
+		AppID:    appID,
+		GuildID:  guildID,
+		Commands: cmds,
+	}, nil
+}
+
+func (b *Bot) Start() error {
+	if err := b.Session.Open(); err != nil {
+		return err
+	}
+
+	b.registeredCommands = make([]*discordgo.ApplicationCommand, len(b.Commands))
+	for _, cmd := range b.Commands {
+		created, err := b.Session.ApplicationCommandCreate(b.AppID, b.GuildID, cmd)
+		if err != nil {
+			log.Printf("Cannot create '%v' command: %v", cmd.Name, err)
+			continue
+		}
+		b.registeredCommands = append(b.registeredCommands, created)
+	}
+
+	b.Session.AddHandler(commands.HandleInteraction)
+	return nil
+
+}
+
+func (b *Bot) Close() error {
+	for _, cmd := range b.registeredCommands {
+		if err := b.Session.ApplicationCommandDelete(b.AppID, b.GuildID, cmd.ID); err != nil {
+			log.Printf("cannot delete '%s' command: %v", cmd.Name, err)
+		}
+	}
+	return b.Session.Close()
+}

internal/commands/commands.go

@@ -0,0 +1,21 @@
+package commands
+
+import (
+	"velox-bot/internal/commands/fun"
+
+	"github.com/bwmarrin/discordgo"
+)
+
+var AllCommands = []*discordgo.ApplicationCommand{
+	fun.PingCommand,
+}
+
+var handlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
+	"ping": fun.HandlePing,
+}
+
+func HandleInteraction(s *discordgo.Session, i *discordgo.InteractionCreate) {
+	if h, ok := handlers[i.ApplicationCommandData().Name]; ok {
+		h(s, i)
+	}
+}

internal/commands/fun/ping.go

@@ -0,0 +1,17 @@
+package fun
+
+import "github.com/bwmarrin/discordgo"
+
+var PingCommand = &discordgo.ApplicationCommand{
+	Name:        "ping",
+	Description: "Ping the bot",
+}
+
+func HandlePing(s *discordgo.Session, i *discordgo.InteractionCreate) {
+	s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
+		Type: discordgo.InteractionResponseChannelMessageWithSource,
+		Data: &discordgo.InteractionResponseData{
+			Content: "Pong!",
+		},
+	})
+}

internal/config/config.go

@@ -0,0 +1,41 @@
+package config
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/joho/godotenv"
+)
+
+type Config struct {
+	BotToken string
+	AppID    string
+	GuildID  string
+}
+
+func LoadConfig() (*Config, error) {
+	if err := godotenv.Load(); err != nil {
+		return nil, err
+	}
+
+	token := os.Getenv("BOT_TOKEN")
+	if token == "" {
+		return nil, fmt.Errorf("BOT_TOKEN is not set")
+	}
+
+	appID := os.Getenv("APP_ID")
+	if appID == "" {
+		return nil, fmt.Errorf("APP_ID is not set")
+	}
+
+	guildID := os.Getenv("GUILD_ID")
+	if guildID == "" {
+		return nil, fmt.Errorf("GUILD_ID is not set")
+	}
+
+	return &Config{
+		BotToken: token,
+		AppID:    appID,
+		GuildID:  guildID,
+	}, nil
+}