main.go

package main

import (
	"log"
	"os"
	"os/signal"

	"github.com/bwmarrin/discordgo"
	"github.com/camvaz/go-discord-bot/handlers"
	"github.com/joho/godotenv"
)

var s *discordgo.Session

func init() {
	err := godotenv.Load()
	if err != nil {
		log.Fatal("Error loading .env file")
	}
}

func init() {
	var err error
	BotToken := os.Getenv("DISCORD_API_TOKEN")
	s, err = discordgo.New("Bot " + BotToken)
	if err != nil {
		log.Fatalf("Invalid bot parameters: %v", err)
	}
}

func main() {
	KingID := os.Getenv("DISCORD_KING_ID")
	VictimID := os.Getenv("DISCORD_VICTIM_ID")
	PolloID := os.Getenv("DISCORD_POLLO_ID")
	ChannelID := os.Getenv("DISCORD_GENERAL_CHANNEL")
	GuildID := os.Getenv("DISCORD_GUILD_ID")
	l := log.New(os.Stdout, "ndejous-bot", log.LstdFlags)
	bot := handlers.NewBot(l, KingID, VictimID, PolloID, ChannelID, GuildID, "$")

	s.AddHandler(func(s *discordgo.Session, r *discordgo.Ready) {
		bot.Log("Session ready")
	})
	s.AddHandler(bot.VoiceUpdateHandler)
	s.AddHandler(bot.MessageCreationHandler)

	err := s.Open()
	if err != nil {
		bot.FatalLog("Cannot open the session: %v", err)
	}

	defer s.Close()

	stop := make(chan os.Signal)
	signal.Notify(stop, os.Interrupt)
	signal.Notify(stop, os.Kill)
	<-stop
	bot.Log("Gracefully shutting down")
}