Improve error handling (#4)

lib/coherentui.go

@@ -4,9 +4,17 @@ import (
 	"github.com/mitchellh/go-ps"
 	"log"
 	"os"
+	"errors"
 )
 
-func killCoherentUI() {
+func killCoherentUI() (err error) {
+
+	defer func() {
+		if r := recover(); r != nil {
+			err = errors.New("Unable to kill 'CoherentUI_Host.exe', need admin rights!'")
+		}
+	}()
+
 	// Find process(es)
 	chp, err := ps.Processes()
 	if err != nil {
@@ -30,4 +38,6 @@ func killCoherentUI() {
 			proc.Kill()
 		}
 	}
+
+	return err
 }

lib/config.go

@@ -5,6 +5,7 @@ import (
 	"io/ioutil"
 	"os"
 	"gopkg.in/yaml.v2"
+	"errors"
 )
 
 // Telegram and program settings (config.yml)
@@ -22,7 +23,13 @@ type Config struct {
 	KillCoherentUI bool
 }
 
-func Read_Settings(ex string, err error) Config {
+func Read_Settings(ex string) (config Config, err error) {
+
+	defer func() {
+		if r := recover(); r != nil {
+			err = errors.New("Unable to find config.yml, created a new one.")
+		}
+	}()
 
 	////  SETTINGS
 	//--------------------------------------------------------------------------------------------------------------
@@ -31,7 +38,6 @@ func Read_Settings(ex string, err error) Config {
 	// This is necessary for dynamic builds in Jetbrains Gogland IDE
 	//newex = strings.Replace(ex, "Application.exe", "config.yml", -1)
 
-	var config Config
 	source, err := ioutil.ReadFile(newex)
 
 	if err != nil {
@@ -62,5 +68,5 @@ func Read_Settings(ex string, err error) Config {
 		panic(err)
 	}
 
-	return(config)
+	return config, err
 }

lib/watchdog.go

@@ -29,7 +29,10 @@ func Watchdog(
 
 	// KILL CoherentUI_Host.exe
 	if config.KillCoherentUI {
-		killCoherentUI()
+		err := killCoherentUI()
+		if err != nil {
+			label_Status.SetText(err.Error())
+		}
 	}
 
 	// INFINITE MAIN LOOP

main.go

@@ -9,12 +9,17 @@ import (
 
 // Variables
 const VERSION = "0.1.5"
+var errmsg string
 
 func main() {
 
 
 	// SETTINGS
-	config := Read_Settings(os.Executable())
+	exp, _ := os.Executable()
+	config, err := Read_Settings(exp)
+	if err != nil {
+		errmsg = err.Error()
+	}
 
 	//// GUI
 	//--------------------------------------------------------------------------------------------------------------
@@ -23,8 +28,8 @@ func main() {
 
 		label_Process := ui.NewLabel("  Process: " + config.Process)
 		label_Status := ui.NewLabel("  Initializing...")
-		label_PID := ui.NewLabel("-")
-		label_Connection := ui.NewLabel("-")
+		label_PID := ui.NewLabel("")
+		label_Connection := ui.NewLabel("")
 		label_Update := ui.NewLabel("")
 
 		//box := ui.NewVerticalBox()
@@ -88,7 +93,12 @@ func main() {
 			return true
 		})
 		window.Show()
-		go Watchdog(config, label_Status, label_PID, label_Connection, label_Update, pb)
+
+		if (errmsg) != "" {
+			label_Status.SetText("  " + errmsg)
+		} else {
+			go Watchdog(config, label_Status, label_PID, label_Connection, label_Update, pb)
+		}
 	})
 	if ui != nil {
 		panic(ui)