102 lines
2.8 KiB
Go
102 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log/slog"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/disgoorg/disgo/bot"
|
|
"github.com/disgoorg/disgolink/v4/disgolink"
|
|
"github.com/joho/godotenv"
|
|
|
|
"discord-music-bot/internal/commands"
|
|
"discord-music-bot/internal/config"
|
|
"discord-music-bot/internal/discord"
|
|
"discord-music-bot/internal/lavalink"
|
|
"discord-music-bot/internal/logging"
|
|
"discord-music-bot/internal/music"
|
|
)
|
|
|
|
func main() {
|
|
_ = godotenv.Load()
|
|
|
|
cfg, err := config.Load()
|
|
if err != nil {
|
|
slog.Error("invalid configuration", slog.Any("err", err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
logger := logging.New(cfg.Debug)
|
|
|
|
ctx, stop := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
|
defer stop()
|
|
|
|
discordClient, err := discord.NewClient(cfg.DiscordToken, logger)
|
|
if err != nil {
|
|
logger.Error("failed to build discord client", slog.Any("err", err))
|
|
os.Exit(1)
|
|
}
|
|
defer discordClient.Close(context.Background())
|
|
|
|
lavalinkClient, err := lavalink.Connect(ctx, discordClient.ApplicationID, logger, disgolink.NodeConfig{
|
|
Name: cfg.LavalinkNodeName,
|
|
Address: cfg.LavalinkAddress,
|
|
Password: cfg.LavalinkPassword,
|
|
Secure: cfg.LavalinkSecure,
|
|
})
|
|
if err != nil {
|
|
logger.Error("failed to connect to lavalink", slog.Any("err", err))
|
|
os.Exit(1)
|
|
}
|
|
defer lavalinkClient.Close()
|
|
|
|
musicManager := music.NewManager(lavalinkClient, logger)
|
|
lavalinkClient.AddListeners(disgolink.NewListenerFunc(musicManager.OnTrackEnd))
|
|
|
|
deps := &commands.Dependencies{
|
|
Discord: discordClient,
|
|
Lavalink: lavalinkClient,
|
|
Music: musicManager,
|
|
}
|
|
|
|
registry := commands.NewRegistry(
|
|
commands.NewPlayCommand(deps),
|
|
commands.NewSkipCommand(deps),
|
|
commands.NewStopCommand(deps),
|
|
commands.NewPauseCommand(deps),
|
|
commands.NewResumeCommand(deps),
|
|
commands.NewQueueCommand(deps),
|
|
commands.NewNowPlayingCommand(deps),
|
|
commands.NewSeekCommand(deps),
|
|
commands.NewVolumeCommand(deps),
|
|
commands.NewLoopCommand(deps),
|
|
commands.NewShuffleCommand(deps),
|
|
commands.NewRemoveCommand(deps),
|
|
)
|
|
|
|
router := discord.NewRouter(registry, logger)
|
|
voiceForwarder := discord.NewVoiceForwarder(discordClient, lavalinkClient, musicManager, logger)
|
|
|
|
discordClient.AddEventListeners(
|
|
bot.NewListenerFunc(router.OnApplicationCommand),
|
|
bot.NewListenerFunc(voiceForwarder.OnVoiceStateUpdate),
|
|
bot.NewListenerFunc(voiceForwarder.OnVoiceServerUpdate),
|
|
)
|
|
|
|
if err := discord.RegisterCommands(discordClient, registry.Definitions()); err != nil {
|
|
logger.Error("failed to register commands", slog.Any("err", err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
if err := discordClient.OpenGateway(ctx); err != nil {
|
|
logger.Error("failed to open gateway", slog.Any("err", err))
|
|
os.Exit(1)
|
|
}
|
|
|
|
logger.Info("bot is running", slog.String("application_id", discordClient.ApplicationID.String()))
|
|
<-ctx.Done()
|
|
logger.Info("shutting down")
|
|
}
|