core.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"log"
	"os"
	"os/signal"
	"path/filepath"
	"regexp"
	"runtime"
	"strconv"
	"syscall"
	"time"

	config "maquiaBot/config"
	handlers "maquiaBot/handlers"
	gencommands "maquiaBot/handlers/general-commands"
	osucommands "maquiaBot/handlers/osu-commands"
	osuapi "maquiaBot/osu-api"
	osutools "maquiaBot/osu-tools"
	structs "maquiaBot/structs"

	"github.com/bwmarrin/discordgo"
	_ "github.com/jinzhu/gorm/dialects/mysql"
)

func main() {
	// Create data folders and stuff
	if _, err := os.Stat("./data"); os.IsNotExist(err) {
		err = os.MkdirAll("./data", 0755)
		readErr(err)
		log.Println("Created data directory.")

		err = ioutil.WriteFile("./data/genitalRecords.json", []byte{}, 0644)
		readErr(err)
		log.Println("Created data/genitalRecords.json.")
		err = ioutil.WriteFile("./data/reminders.json", []byte{}, 0644)
		readErr(err)
		log.Println("Created data/reminders.json.")
		err = ioutil.WriteFile("./data/tasks.json", []byte{}, 0644)
		readErr(err)
		log.Println("Created data/tasks.json.")

		err = os.MkdirAll("./data/channelData", 0755)
		readErr(err)
		log.Println("Created data/channelData directory.")
		err = os.MkdirAll("./data/channelData", 0755)
		readErr(err)
		log.Println("Created data/channelData directory.")

		err = os.MkdirAll("./data/osuData", 0755)
		readErr(err)
		log.Println("Created data/osuData directory.")
		err = ioutil.WriteFile("./data/osuData/mapFarm.json", []byte{}, 0644)
		readErr(err)
		log.Println("Created data/osuData/mapFarm.json.")
		err = ioutil.WriteFile("./data/osuData/profileCache.json", []byte{}, 0644)
		readErr(err)
		log.Println("Created data/osuData/profileCache.json.")
	}

	// Obtain config
	config.NewConfig()
	osuAPI := osuapi.NewClient(config.Conf.OsuToken)
	osucommands.OsuAPI = osuAPI
	osutools.OsuAPI = osuAPI
	discord, err := discordgo.New("Bot " + config.Conf.DiscordToken)
	readErr(err)

	// Handle farm data
	go osutools.FarmUpdate(discord)

	// Add handlers
	discord.AddHandler(handlers.MessageHandler)
	discord.AddHandler(handlers.ReactAdd)
	discord.AddHandler(handlers.ServerJoin)
	discord.AddHandler(handlers.ServerLeave)

	// Open a websocket connection to Discord and begin listening
	for {
		err = discord.Open()
		if err == nil {
			break
		}
	}
	log.Println("Bot is now running in " + strconv.Itoa(len(discord.State.Guilds)) + " servers.")
	discord.UpdateStatus(0, "maquiahelp | maquiaprefix | "+strconv.Itoa(len(discord.State.Guilds))+" servers")

	// Resume all reminder timers
	reminders := []structs.Reminder{}
	_, err = os.Stat("./data/reminders.json")
	if err == nil {
		f, err := ioutil.ReadFile("./data/reminders.json")
		readErr(err)
		_ = json.Unmarshal(f, &reminders)
	} else {
		readErr(err)
	}
	reminderTimers := []structs.ReminderTimer{}
	for _, reminder := range reminders {
		reminderTimer := structs.ReminderTimer{
			Reminder: reminder,
			Timer:    *time.NewTimer(reminder.Target.Sub(time.Now().UTC())),
		}
		reminderTimers = append(reminderTimers, reminderTimer)
		go gencommands.RunReminder(discord, reminderTimer)
	}
	gencommands.ReminderTimers = reminderTimers

	// Resume all tasks
	tasks := []structs.Task{}
	_, err = os.Stat("./data/tasks.json")
	if err == nil {
		f, err := ioutil.ReadFile("./data/tasks.json")
		readErr(err)
		_ = json.Unmarshal(f, &tasks)
	} else {
		readErr(err)
	}
	taskCache := []structs.TaskCache{}
	for _, task := range tasks {
		activeTask := structs.TaskCache{
			Task:   task,
			Active: true,
		}
		taskCache = append(taskCache, activeTask)
		go gencommands.RunTask(discord, task)
	}
	gencommands.TaskCache = taskCache

	// Get osu! tracking data for channels
	var channels []string

	err = filepath.Walk("./data/channelData", func(path string, info os.FileInfo, err error) error {
		readErr(err)
		channels = append(channels, path)
		return nil
	})
	readErr(err)
	IDregex, _ := regexp.Compile(`(?i)(\d+)\.json`)
	for _, channel := range channels {
		if IDregex.MatchString(channel) {
			chID := IDregex.FindStringSubmatch(channel)[1]
			ch, err := discord.Channel(chID)
			if err == nil {
				go osutools.TrackPost(*ch, discord)
			}
		}
	}

	// Open DB
	// tools.DB, err = gorm.Open("mysql", config.Conf.Database.Username+":"+config.Conf.Database.Password+"@/"+config.Conf.Database.Name)
	// readErr(err)

	// Create a channel to keep the bot running until a prompt is given to close
	sc := make(chan os.Signal, 1)
	signal.Notify(sc, syscall.SIGINT, syscall.SIGTERM, os.Kill)
	<-sc

	// Close sessions
	discord.Close()
	// tools.DB.Close()
}

func readErr(err error) {
	if err != nil {
		pc, fn, line, _ := runtime.Caller(1)
		log.Fatalf("[error] in %s[%s:%d] %v\n", runtime.FuncForPC(pc).Name(), fn, line, err)
	}
}