karaokards/main.go

package main

import (
	"encoding/json"
	"io/ioutil"
	"math/rand"
	"os"
	"path/filepath"
	"time"

	builtins "git.martyn.berlin/martyn/karaokards/internal/builtins"
	irc "git.martyn.berlin/martyn/karaokards/internal/irc"
	webserver "git.martyn.berlin/martyn/karaokards/internal/webserver"
	rgb "github.com/foresthoffman/rgblog"
	scribble "github.com/nanobox-io/golang-scribble"
)

type customStringsStruct struct {
	Strings []string `json:"strings,omitempty"`
}

var selectablePrompts []string

var customStrings customStringsStruct

var config irc.ConfigStruct

func readConfig() {
	var data []byte
	var err error
	configFile := ""
	if os.Getenv("KARAOKARDS_CONFIGFILE") != "" {
		if _, err := os.Stat(os.Getenv("KARAOKARDS_CONFIGFILE")); os.IsNotExist(err) {
			rgb.RPrintf("[%s] Error, KARAOKARDS_CONFIGFILE env var set and '%s' doesn't exist!\n", irc.TimeStamp(), os.Getenv("KARAOKARDS_CONFIGFILE"))
			os.Exit(1)
		}
		configFile = os.Getenv("KARAOKARDS_CONFIGFILE")
	} else {
		ex, err := os.Executable()
		if err != nil {
			rgb.YPrintf("[%s] Warning, KARAOKARDS_CONFIGFILE env var unset and cannot find executable!\n", irc.TimeStamp())
		}
		exPath := filepath.Dir(ex)
		if _, err := os.Stat(exPath + "/config.json"); os.IsNotExist(err) {
			rgb.YPrintf("[%s] Warning, KARAOKARDS_CONFIGFILE env var unset and `config.json` not alongside executable!\n", irc.TimeStamp())
			if _, err := os.Stat("/etc/karaokards/config.json"); os.IsNotExist(err) {
				rgb.RPrintf("[%s] Error, KARAOKARDS_CONFIGFILE env var unset and neither '%s' nor '%s' exist!\n", irc.TimeStamp(), exPath+"/config.json", "/etc/karaokards/config.json")
				os.Exit(1)
			} else {
				configFile = "/etc/karaokards/config.json"
			}
		} else {
			configFile = exPath + "/config.json"
		}
	}
	data, err = ioutil.ReadFile(configFile)
	if err != nil {
		rgb.RPrintf("[%s] Could not read `%s`. File reading error: %s\n", irc.TimeStamp(), configFile, err)
		os.Exit(1)
	}
	err = json.Unmarshal(data, &config)
	if err != nil {
		rgb.RPrintf("[%s] Could not unmarshal `%s`. Unmarshal error: %s\n", irc.TimeStamp(), configFile, err)
		os.Exit(1)
	}
	rgb.YPrintf("[%s] Read config file from `%s`\n", irc.TimeStamp(), configFile)
	rgb.YPrintf("[%s] config %v\n", irc.TimeStamp(), config)
	return
}

//openDatabase "database" in this sense being a scribble db
func openDatabase() *scribble.Driver {
	dataPath := ""
	if config.DataPath == "" {
		if os.Getenv("KARAOKARDS_DATA_FOLDER") != "" {
			if _, err := os.Stat(os.Getenv("KARAOKARDS_DATA_FOLDER")); os.IsNotExist(err) {
				rgb.RPrintf("[%s] Error, KARAOKARDS_DATA_FOLDER env var set and '%s' doesn't exist!\n", irc.TimeStamp(), os.Getenv("KARAOKARDS_DATA_FOLDER"))
				os.Exit(1)
			}
			dataPath = os.Getenv("KARAOKARDS_DATA_FOLDER")
		} else {
			ex, err := os.Executable()
			if err != nil {
				rgb.RPrintf("[%s] Error, KARAOKARDS_DATA_FOLDER env var unset and cannot find executable!\n", irc.TimeStamp())
				os.Exit(1)
			}
			exPath := filepath.Dir(ex)
			if _, err := os.Stat(exPath + "/data"); os.IsNotExist(err) {
				rgb.YPrintf("[%s] Warning %s doesn't exist, trying to create it.\n", irc.TimeStamp(), exPath+"/data")
				err = os.Mkdir(exPath+"/data", 0770)
				if err != nil {
					rgb.RPrintf("[%s] Error cannot create %s: %s!\n", irc.TimeStamp(), exPath+"/data", err)
					os.Exit(1)
				}
			}
			dataPath = exPath + "/data"
		}
	} else {
		if _, err := os.Stat(config.DataPath); os.IsNotExist(err) {
			rgb.RPrintf("[%s] Error, config-specified path '%s' doesn't exist!\n", irc.TimeStamp(), config.DataPath)
			os.Exit(1)
		}
		dataPath = config.DataPath
	}
	db, err := scribble.New(dataPath, nil)
	if err != nil {
		rgb.RPrintf("[%s] Error opening database in '%s' : %s\n", irc.TimeStamp(), dataPath, err)
		os.Exit(1)
	}
	return db
}

func readBonusStrings() []string {
	var data []byte
	var err error
	if config.StringPath == "" {
		ex, err := os.Executable()
		if err != nil {
			rgb.YPrintf("[%s] Could not read `strings.json`, will only have builtin prompts. File reading error: %s\n", irc.TimeStamp(), err)
			return []string{}
		}
		exPath := filepath.Dir(ex)
		data, err = ioutil.ReadFile(exPath + "/strings.json")
		if err != nil {
			rgb.YPrintf("[%s] Could not read `strings.json`, will only have builtin prompts. File reading error: %s\n", irc.TimeStamp(), err)
			return []string{}
		}
	} else {
		data, err = ioutil.ReadFile(config.StringPath)
		if err != nil {
			rgb.YPrintf("[%s] Could not read `strings.json`, will only have builtin prompts. File reading error: %s\n", irc.TimeStamp(), err)
			return []string{}
		}
	}
	err = json.Unmarshal(data, &customStrings)
	if err != nil {
		rgb.YPrintf("[%s] Could not unmarshal `strings.json`, will only have builtin prompts. Unmarshal error: %s\n", irc.TimeStamp(), err)
		return []string{}
	}
	rgb.YPrintf("[%s] Read %d prompts from `strings.json`\n", irc.TimeStamp(), len(customStrings.Strings))
	return customStrings.Strings
}

var buildDate string

func main() {
	rgb.YPrintf("[%s] starting karaokard bot build %s\n", irc.TimeStamp(), buildDate)
	readConfig()
	rand.Seed(time.Now().UnixNano())
	for _, val := range builtins.Karaokards {
		selectablePrompts = append(selectablePrompts, val)
	}
	for _, val := range readBonusStrings() {
		selectablePrompts = append(selectablePrompts, val)
	}
	persistentData := openDatabase()
	var dbGlobalPrompts []string
	if err := persistentData.Read("prompts", "global", &dbGlobalPrompts); err != nil {
		persistentData.Write("prompts", "common", dbGlobalPrompts)
	}
	selectablePrompts := append(selectablePrompts, dbGlobalPrompts...)

	rgb.YPrintf("[%s] %d prompts available.\n", irc.TimeStamp(), len(selectablePrompts))
	ircOauthPath := ""
	if config.IrcOAuthPath == "" {
		if os.Getenv("TWITCH_OAUTH_JSON") != "" {
			if _, err := os.Stat(os.Getenv("TWITCH_OAUTH_JSON")); os.IsNotExist(err) {
				os.Exit(1)
			}
			ircOauthPath = os.Getenv("TWITCH_OAUTH_JSON")
		} else {
			if _, err := os.Stat(os.Getenv("HOME") + "/.twitch/ircoauth.json"); os.IsNotExist(err) {
				rgb.YPrintf("[%s] Warning %s doesn't exist, trying %s next!\n", irc.TimeStamp(), os.Getenv("HOME")+"/.twitch/ircoauth.json", "/etc/twitch/ircoauth.json")
				if _, err := os.Stat("/etc/twitch/ircoauth.json"); os.IsNotExist(err) {
					rgb.YPrintf("[%s] Error %s doesn't exist either, bailing!\n", irc.TimeStamp(), "/etc/twitch/ircoauth.json")
					os.Exit(1)
				}
				ircOauthPath = "/etc/twitch/ircoauth.json"
			} else {
				ircOauthPath = os.Getenv("HOME") + "/.twitch/ircoauth.json"
			}
		}
	} else {
		if _, err := os.Stat(config.IrcOAuthPath); os.IsNotExist(err) {
			rgb.YPrintf("[%s] Error config-specified oauth file %s doesn't exist, bailing!\n", irc.TimeStamp(), config.IrcOAuthPath)
			os.Exit(1)
		}
		ircOauthPath = config.IrcOAuthPath
	}
	appOauthPath := ""
	if config.AppOAuthPath == "" {
		if os.Getenv("TWITCH_OAUTH_JSON") != "" {
			if _, err := os.Stat(os.Getenv("TWITCH_OAUTH_JSON")); os.IsNotExist(err) {
				os.Exit(1)
			}
			appOauthPath = os.Getenv("TWITCH_OAUTH_JSON")
		} else {
			if _, err := os.Stat(os.Getenv("HOME") + "/.twitch/appoauth.json"); os.IsNotExist(err) {
				rgb.YPrintf("[%s] Warning %s doesn't exist, trying %s next!\n", irc.TimeStamp(), os.Getenv("HOME")+"/.twitch/appoauth.json", "/etc/twitch/appoauth.json")
				if _, err := os.Stat("/etc/twitch/appoauth.json"); os.IsNotExist(err) {
					rgb.YPrintf("[%s] Error %s doesn't exist either, bailing!\n", irc.TimeStamp(), "/etc/twitch/appoauth.json")
					os.Exit(1)
				}
				appOauthPath = "/etc/twitch/appoauth.json"
			} else {
				appOauthPath = os.Getenv("HOME") + "/.twitch/appoauth.json"
			}
		}
	} else {
		if _, err := os.Stat(config.AppOAuthPath); os.IsNotExist(err) {
			rgb.YPrintf("[%s] Error config-specified oauth file %s doesn't exist, bailing!\n", irc.TimeStamp(), config.AppOAuthPath)
			os.Exit(1)
		}
		appOauthPath = config.AppOAuthPath
	}

	// Replace the channel name, bot name, and the path to the private directory with your respective
	// values.
	myBot := irc.KardBot{
		Channel:     "karaokards",
		MsgRate:     time.Duration(20/30) * time.Millisecond,
		Name:        "Karaokards",
		Port:        "6667",
		IrcPrivatePath: ircOauthPath,
		AppPrivatePath: appOauthPath,
		Server:      "irc.chat.twitch.tv",
		Prompts:     selectablePrompts,
		Database:    *persistentData,
		Config:      config,
	}
	go func() {
		rgb.YPrintf("[%s] Starting webserver on port %s\n", irc.TimeStamp(), "5353")
		webserver.HandleHTTP(&myBot)
	}()
	myBot.Start()
}