main.go

package main

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

	"github.com/diamondburned/arikawa/v2/gateway"
	"github.com/diamondburned/arikawa/v2/state"
	"github.com/diamondburned/arikawa/v2/utils/handler"
	"github.com/diamondburned/arikawa/v2/voice"
)

var token = flag.String("t", "", "Set the token (overrides $BOT_TOKEN)")
var botname = flag.String("n", "eto", "Set the bot's name")
var prefix = flag.String("p", "!", "Set the bot's prefix")
var gittoken = flag.String("T", "", "Set the git token (overrides $GIT_TOKEN)")

var s *state.State
var u *gateway.ReadyEvent

func main() {
	flag.Parse()
	if *token == "" {
		toke := os.Getenv("BOT_TOKEN")
		if toke == "" {
			fmt.Println("No BOT_TOKEN: Set $BOT_TOKEN or use '-t'")
			os.Exit(1)
		}
		token = &toke
	}
	if *gittoken == "" {
		toke := os.Getenv("GIT_TOKEN")
		if toke == "" {
			fmt.Println("No GIT_TOKEN: Set $GIT_TOKEN or use '-T'")
		}
		gittoken = &toke
	}

	var err error
	s, err = state.New(fmt.Sprintf("Bot %s", *token))
	if err != nil {
		fmt.Printf("Session failed: %s\n", err)
		os.Exit(1)
	}

	addPreHandlers()
	addHandlers()
	addIntents()

	if err := loggerSetup(); err != nil {
		fmt.Printf("Failed to setup logger: %s\n", err)
	}
	if err := basicSetup(); err != nil {
		fmt.Printf("%s\n", err)
	}
	// if *gittoken != "" {
	// 	if err := gitSetup(); err != nil {
	// 		fmt.Printf("%s\n", err)
	// 	}
	// }

	if err := s.Open(); err != nil {
		fmt.Printf("Failed to connect: %s\n", err)
		os.Exit(1)
	}
	defer s.Close()

	me, err := s.Me()
	if err != nil {
		fmt.Printf("Failed to get bot user: %s\n", err)
		os.Exit(1)
	}

	fmt.Printf("Started as %s\n", me.Username)

	sigChan := make(chan os.Signal)
	signal.Notify(sigChan, os.Interrupt)
	<-sigChan // block
}

func addHandlers() {
	// setup
	s.AddHandler(func(m *gateway.ReadyEvent) {
		u = m // info used globally
	})

	// logger
	s.AddHandler(sent)

	// basic
	s.AddHandler(help)
	s.AddHandler(setPrefix)

	// music
	s.AddHandler(play)
	s.AddHandler(kill)
	s.AddHandler(skip)
	s.AddHandler(queue)
}

func addPreHandlers() {
	// setup
	s.PreHandler = handler.New()
	s.PreHandler.Synchronous = true

	s.PreHandler.AddHandler(unsent)
}

func addIntents() {
	s.Gateway.AddIntents(gateway.IntentGuilds)
	// s.Gateway.AddIntents(gateway.IntentGuildMembers)
	// s.Gateway.AddIntents(gateway.IntentGuildBans)
	// s.Gateway.AddIntents(gateway.IntentGuildEmojis)
	// s.Gateway.AddIntents(gateway.IntentGuildIntegrations)
	// s.Gateway.AddIntents(gateway.IntentGuildWebhooks)
	// s.Gateway.AddIntents(gateway.IntentGuildInvites)
	s.Gateway.AddIntents(gateway.IntentGuildVoiceStates)
	// s.Gateway.AddIntents(gateway.IntentGuildPresences)
	s.Gateway.AddIntents(gateway.IntentGuildMessages)
	// s.Gateway.AddIntents(gateway.IntentGuildMessageReactions)
	// s.Gateway.AddIntents(gateway.IntentGuildMessageTyping)
	s.Gateway.AddIntents(gateway.IntentDirectMessages)
	// s.Gateway.AddIntents(gateway.IntentDirectMessageReactions)
	// s.Gateway.AddIntents(gateway.IntentDirectMessageTyping)

	voice.AddIntents(s.Gateway) // for voice
}