main.go

package main

import (
	"os"
	"os/signal"
	"syscall"
	"time"

	"github.com/juunini/palworld-discord-bot/src/bot"
	"github.com/juunini/palworld-discord-bot/src/cli"
	"github.com/juunini/palworld-discord-bot/src/config"
	"github.com/juunini/palworld-discord-bot/src/console_decoration"
	"github.com/juunini/palworld-discord-bot/src/i18n"
	"github.com/juunini/palworld-discord-bot/src/palworld/settings"
	"github.com/juunini/palworld-discord-bot/src/web"
)

func init() {
	if err := config.Load(); err != nil {
		config.LANGUAGE = askLanguage()
		i18n.SetLanguage(config.LANGUAGE)

		if err := config.Save(); err != nil {
			panic(err)
		}

		return
	}

	i18n.SetLanguage(config.LANGUAGE)
}

func main() {
	if config.WEB_SERVER_ENABLED {
		go web.Listen(config.WEB_SERVER_PORT)
	}

	if config.DISCORD_BOT_ENABLED {
		session := bot.New(config.DISCORD_BOT_TOKEN)
		session.AddHandler(bot.Handler)
		bot.Connect(session)
		console_decoration.PrintSuccess(i18n.BotRunningStart + "\n")

		if config.DISCORD_LOG_CHANNEL_ID != "" || config.DISCORD_DASHBOARD_CHANNEL_ID != "" {
			settings.RenewDashboard(session)

			go func() {
				for {
					bot.Watch(session)
					time.Sleep(10 * time.Second)
				}
			}()
		}
	}

	sc := make(chan os.Signal, 1)
	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Interrupt)
	<-sc

	web.Shutdown()
}

func askLanguage() string {
	languages := make([]string, len(i18n.Languages))
	for i, language := range i18n.Languages {
		languages[i] = language["name"]
	}
	language, err := cli.Choose("Choose your language", languages)
	if err != nil {
		panic(err)
	}

	for _, l := range i18n.Languages {
		if l["name"] == language {
			return l["code"]
		}
	}

	return "en"
}