From 851f073e99e368984371c63d721a6ee39f4cdade Mon Sep 17 00:00:00 2001
From: Justin Hawkins <justin@hawkins.id.au>
Date: Sun, 7 Feb 2021 11:42:13 +1030
Subject: [PATCH] Read/Write config to $HOME

---
 config/config.go | 57 ++++++++++++++++++++++++++++++++++++++++++++++++
 data/config.html | 20 ++++++++++++++---
 dau.go           | 52 ++++++++++++++++++++-----------------------
 go.mod           |  1 +
 go.sum           |  2 ++
 web/server.go    |  9 ++++++++
 6 files changed, 110 insertions(+), 31 deletions(-)

diff --git a/config/config.go b/config/config.go
index 1e8e82a..fad5885 100644
--- a/config/config.go
+++ b/config/config.go
@@ -1,5 +1,13 @@
 package config
 
+import (
+  "github.com/mitchellh/go-homedir"
+  "log"
+  "os"
+  "encoding/json"
+  "io/ioutil"
+)
+
 // Config for the application
 var Config struct {
 	WebHookURL  string
@@ -11,3 +19,52 @@ var Config struct {
 }
 
 const CurrentVersion string = "0.6"
+
+// Load the current config or initialise with defaults
+func LoadOrInit() {
+  configPath := configPath()
+  log.Printf("Trying to load from %s\n", configPath)
+  _, err := os.Stat(configPath)
+  if os.IsNotExist(err) {
+    log.Printf("Initialising empty config")
+    Config.WebHookURL = ""
+    Config.Path = homeDir() + string(os.PathSeparator) + "screenshots"
+    Config.Watch = 10
+    SaveConfig()
+  } else {
+     LoadConfig()
+  }
+}
+
+func LoadConfig() {
+  path := configPath()
+  data, err := ioutil.ReadFile(path)
+  if err != nil {
+    log.Fatalf("cannot read config file %s: %s", path, err.Error())
+  }
+  err = json.Unmarshal([]byte(data), &Config)
+  if err != nil {
+    log.Fatalf("cannot decode config file %s: %s", path, err.Error())
+  }
+}
+
+func SaveConfig() {
+  log.Print("saving configuration")
+  path := configPath()
+  jsonString, _ := json.Marshal(Config)
+  err := ioutil.WriteFile(path, jsonString, os.ModePerm)
+  if (err != nil) {
+    log.Fatalf("Cannot save config %s: %s", path, err.Error())
+  }
+}
+
+func homeDir() string {
+  dir, err := homedir.Dir()
+  if (err != nil) { panic (err) }
+  return dir;
+}
+
+func configPath() string {
+  homeDir := homeDir()
+  return homeDir + string(os.PathSeparator) + ".dau.json"
+}
diff --git a/data/config.html b/data/config.html
index 2bbddc5..5cc2579 100644
--- a/data/config.html
+++ b/data/config.html
@@ -70,8 +70,9 @@
        </div>
        <div class="col-sm-4 my-1">
          <div class="custom-control custom-switch">
-           <input type="checkbox" class="custom-control-input rest-field rest-field-boolean" id="customSwitch1">
-           <label class="custom-control-label" for="customSwitch1"><span style="display: none;">😭</span></label>
+           <input type="checkbox" class="custom-control-input rest-field rest-field-boolean" id="input-nowatermark">
+           <label class="custom-control-label" for="input-nowatermark">&nbsp;</label>
+           <span id="sadness" style="">😭</span>
          </div>
        </div>
        <div class="col-auto my-1">
@@ -87,7 +88,18 @@
 
 
 <script>
+function update_sadness () {
+  if ($('#input-nowatermark').prop('checked')) {
+    $('#sadness').css('visibility','');
+  }
+  else {
+    $('#sadness').css('visibility','hidden');
+  }
+}
 $(document).ready(function() {
+
+  $('#input-nowatermark').on('click', function() { update_sadness(); });
+
   // populate each field
   $('.config-item').each(function() {
     let el = $(this);
@@ -102,6 +114,8 @@ $(document).ready(function() {
        else {
          this_el.val(data.Value);
        }
+       update_sadness();
+
      });
     });
 
@@ -109,7 +123,7 @@ $(document).ready(function() {
   $('.config-item button').on('click', function(e,f) {
     key = $(this).parents('.config-item').data('key');
     val = $(this).parents('.config-item').find('.rest-field').val();
-    if ($(this).parents('.config-item').find('.rest-field-boolean')) {
+    if ($(this).parents('.config-item').find('.rest-field-boolean').length) {
       val = $(this).parents('.config-item').find('.rest-field').prop('checked') ? 1 : 0;
     }
     $.post('/rest/config/'+key, { value: val })
diff --git a/dau.go b/dau.go
index d2fecae..4d6fa43 100644
--- a/dau.go
+++ b/dau.go
@@ -37,7 +37,6 @@ var newLastCheck = time.Now()
 func main() {
 
 	parseOptions()
-	checkPath(config.Config.Path)
 
 	// log.Print("Opening web browser")
 	// open.Start("http://localhost:9090")
@@ -48,26 +47,30 @@ func main() {
 	log.Print("Waiting for images to appear in ", config.Config.Path)
 	// wander the path, forever
 	for {
-		err := filepath.Walk(config.Config.Path,
-			func(path string, f os.FileInfo, err error) error { return checkFile(path, f, err) })
-		if err != nil {
-			log.Fatal("could not watch path", err)
+		if checkPath(config.Config.Path) {
+			err := filepath.Walk(config.Config.Path,
+				func(path string, f os.FileInfo, err error) error { return checkFile(path, f, err) })
+			if err != nil {
+				log.Fatal("could not watch path", err)
+			}
+			lastCheck = newLastCheck
 		}
-		lastCheck = newLastCheck
 		log.Print("sleeping before next check")
 		time.Sleep(time.Duration(config.Config.Watch) * time.Second)
 	}
 }
 
-func checkPath(path string) {
+func checkPath(path string) bool {
 	src, err := os.Stat(path)
 	if err != nil {
-		log.Fatal(err)
+		log.Println("path problem: ", err)
+		return false
 	}
 	if !src.IsDir() {
-		log.Fatal(path, " is not a directory")
-		os.Exit(1)
+		log.Println(path, " is not a directory")
+		return false
 	}
+	return true
 }
 
 func checkUpdates() {
@@ -111,12 +114,7 @@ func checkUpdates() {
 func parseOptions() {
 
 	// Declare the flags to be used
-	webhookFlag := getopt.StringLong("webhook", 'w', "", "discord webhook URL")
-	pathFlag := getopt.StringLong("directory", 'd', "", "directory to scan, optional, defaults to current directory")
-	watchFlag := getopt.Int16Long("watch", 's', 10, "time between scans")
-	usernameFlag := getopt.StringLong("username", 'u', "", "username for the bot upload")
 	excludeFlag := getopt.StringLong("exclude", 'x', "", "exclude files containing this string")
-	noWatermarkFlag := getopt.BoolLong("no-watermark", 'n', "do not put a watermark on images before uploading")
 	helpFlag := getopt.BoolLong("help", 'h', "help")
 	versionFlag := getopt.BoolLong("version", 'v', "show version")
 	getopt.SetParameters("")
@@ -134,22 +132,15 @@ func parseOptions() {
 		os.Exit(0)
 	}
 
-	if !getopt.IsSet("directory") {
-		*pathFlag = "./"
-		log.Println("Defaulting to current directory")
-	}
+	// if !getopt.IsSet("webhook") {
+	// 	log.Fatal("ERROR: You must specify a --webhook URL")
+	// }
 
-	if !getopt.IsSet("webhook") {
-		log.Fatal("ERROR: You must specify a --webhook URL")
-	}
+	// grab the config
+	config.LoadOrInit()
 
-	config.Config.Path = *pathFlag
-	config.Config.WebHookURL = *webhookFlag
-	config.Config.Watch = int(*watchFlag)
-	config.Config.Username = *usernameFlag
-	config.Config.NoWatermark = *noWatermarkFlag
+	// overrides from command line
 	config.Config.Exclude = *excludeFlag
-
 }
 
 func checkFile(path string, f os.FileInfo, err error) error {
@@ -190,6 +181,11 @@ func processFile(file string) {
 		file = mungeFile(file)
 	}
 
+	if config.Config.WebHookURL == "" {
+		log.Print("WebHookURL is not configured - cannot upload!")
+		return
+	}
+
 	log.Print("Uploading ", file)
 
 	extraParams := map[string]string{}
diff --git a/go.mod b/go.mod
index eee32e3..a1e753f 100644
--- a/go.mod
+++ b/go.mod
@@ -5,6 +5,7 @@ go 1.15
 require (
 	github.com/fogleman/gg v1.3.0
 	github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect
+	github.com/mitchellh/go-homedir v1.1.0
 	github.com/pborman/getopt v1.1.0
 	github.com/skratchdot/open-golang v0.0.0-20200116055534-eef842397966
 	golang.org/x/image v0.0.0-20201208152932-35266b937fa6
diff --git a/go.sum b/go.sum
index d8c49e0..e25f30d 100644
--- a/go.sum
+++ b/go.sum
@@ -2,6 +2,8 @@ github.com/fogleman/gg v1.3.0 h1:/7zJX8F6AaYQc57WQCyN9cAIz+4bCJGO9B+dyW29am8=
 github.com/fogleman/gg v1.3.0/go.mod h1:R/bRT+9gY/C5z7JzPU0zXsXHKM4/ayA+zqcVNZzPa1k=
 github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g=
 github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k=
+github.com/mitchellh/go-homedir v1.1.0 h1:lukF9ziXFxDFPkA1vsr5zpc1XuPDn/wFntq5mG+4E0Y=
+github.com/mitchellh/go-homedir v1.1.0/go.mod h1:SfyaCUpYCn1Vlf4IUYiD9fPX4A5wJrkLzIz1N1q0pr0=
 github.com/pborman/getopt v1.1.0 h1:eJ3aFZroQqq0bWmraivjQNt6Dmm5M0h2JcDW38/Azb0=
 github.com/pborman/getopt v1.1.0/go.mod h1:FxXoW1Re00sQG/+KIkuSqRL/LwQgSkv7uyac+STFsbk=
 github.com/skratchdot/open-golang v0.0.0-20200116055534-eef842397966 h1:JIAuq3EEf9cgbU6AtGPK4CTG3Zf6CKMNqf0MHTggAUA=
diff --git a/web/server.go b/web/server.go
index 361c571..d67e3ca 100644
--- a/web/server.go
+++ b/web/server.go
@@ -93,6 +93,7 @@ func getSetWebhook(w http.ResponseWriter, r *http.Request) {
 			log.Fatal(err)
 		}
 		config.Config.WebHookURL = r.PostForm.Get("value")
+		config.SaveConfig()
 		postResponse := valueStringResponse{Success: true, Value: config.Config.WebHookURL}
 
 		js, _ := json.Marshal(postResponse)
@@ -116,6 +117,8 @@ func getSetUsername(w http.ResponseWriter, r *http.Request) {
 			log.Fatal(err)
 		}
 		config.Config.Username = r.PostForm.Get("value")
+		config.SaveConfig()
+
 		postResponse := valueStringResponse{Success: true, Value: config.Config.Username}
 
 		js, _ := json.Marshal(postResponse)
@@ -155,6 +158,8 @@ func getSetWatch(w http.ResponseWriter, r *http.Request) {
 		}
 
 		config.Config.Watch = i
+		config.SaveConfig()
+
 		postResponse := valueStringResponse{Success: true, Value: strconv.Itoa(config.Config.Watch)}
 
 		js, _ := json.Marshal(postResponse)
@@ -191,6 +196,8 @@ func getSetNoWatermark(w http.ResponseWriter, r *http.Request) {
 		} else {
 			config.Config.NoWatermark = true
 		}
+		config.SaveConfig()
+
 		postResponse := valueBooleanResponse{Success: true, Value: config.Config.NoWatermark}
 
 		js, _ := json.Marshal(postResponse)
@@ -232,6 +239,8 @@ func getSetDirectory(w http.ResponseWriter, r *http.Request) {
 		}
 
 		config.Config.Path = newPath
+		config.SaveConfig()
+
 		postResponse := valueStringResponse{Success: true, Value: config.Config.Path}
 
 		js, _ := json.Marshal(postResponse)