Less hard-coding, more "database"

Signed-off-by: Martyn Ranyard <m@rtyn.berlin>

configs/config.json

@@ -0,0 +1,3 @@
+{
+    "channels": ["iMartynOnTwitch"]
+}

main.go

@@ -2,7 +2,6 @@ package main
 
 import (
 	"encoding/json"
-	"fmt"
 	"io/ioutil"
 	"math/rand"
 	"os"
@@ -13,8 +12,16 @@ import (
 	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 configStruct struct {
+	InitialChannels []string `json:"channels"`
+	OAuthPath       string   `json:"oauthpath,omitempty"`
+	StringPath      string   `json:"authpath,omitempty"`
+	DataPath        string   `json:"datapath,omitempty"`
+}
+
 type customStringsStruct struct {
 	Strings []string `json:"strings,omitempty"`
 }
@@ -23,30 +30,125 @@ var selectablePrompts []string
 
 var customStrings customStringsStruct
 
-func readBonusStrings() []string {
+var config 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 {
-		fmt.Println("Could not read `strings.json`, will only have builtin prompts. File reading error", err)
+			rgb.YPrintf("[%s] Warning, KARAOKARDS_CONFIGFILE env var unset and cannot find executable!\n", irc.TimeStamp())
-		return []string{}
 		}
 		exPath := filepath.Dir(ex)
-	data, err := ioutil.ReadFile(exPath + "/strings.json")
+		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 {
-		fmt.Println("Could not read `strings.json`, will only have builtin prompts. File reading error", err)
+		rgb.RPrintf("[%s] Could not read `%s`. File reading error: %s\n", irc.TimeStamp(), configFile, err)
-		return []string{}
+		os.Exit(1)
 	}
 	err = json.Unmarshal(data, &customStrings)
 	if err != nil {
-		fmt.Println("Could not unmarshal `strings.json`, will only have builtin prompts. Unmarshal error", err)
+		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)
+	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.OAuthPath); os.IsNotExist(err) {
+			rgb.RPrintf("[%s] Error, config-specified path '%s' doesn't exist!\n", irc.TimeStamp(), os.Getenv("KARAOKARDS_DATA_FOLDER"))
+			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{}
 		}
-	fmt.Println("Read ", len(customStrings.Strings), " prompts from `strings.json`")
+		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
 }
 
 func main() {
-
+	readConfig()
 	rand.Seed(time.Now().UnixNano())
 	for _, val := range builtins.Karaokards {
 		selectablePrompts = append(selectablePrompts, val)
@@ -54,8 +156,9 @@ func main() {
 	for _, val := range readBonusStrings() {
 		selectablePrompts = append(selectablePrompts, val)
 	}
-	fmt.Println(len(selectablePrompts), " prompts available.")
+	rgb.YPrintf("[%s] %d prompts available.\n", irc.TimeStamp(), len(selectablePrompts))
 	oauthPath := ""
+	if config.OAuthPath == "" {
 		if os.Getenv("TWITCH_OAUTH_JSON") != "" {
 			if _, err := os.Stat(os.Getenv("TWITCH_OAUTH_JSON")); os.IsNotExist(err) {
 				os.Exit(1)
@@ -73,6 +176,16 @@ func main() {
 				oauthPath = os.Getenv("HOME") + "/.twitch/oauth.json"
 			}
 		}
+	} else {
+		if _, err := os.Stat(config.OAuthPath); os.IsNotExist(err) {
+			rgb.YPrintf("[%s] Error config-specified oauth file %s doesn't exist, bailing!\n", irc.TimeStamp(), config.OAuthPath)
+			os.Exit(1)
+		}
+		oauthPath = config.OAuthPath
+	}
+	persistentData := openDatabase()
+	persistentData.Write("prompts", "common", selectablePrompts)
+
 	// Replace the channel name, bot name, and the path to the private directory with your respective
 	// values.
 	myBot := irc.KardBot{