From 5ae671f942e526719dbf4c539c756a7e31f5c976 Mon Sep 17 00:00:00 2001
From: Benny~ <benny.think@gmail.com>
Date: Sun, 23 Feb 2020 21:20:10 +0800
Subject: [PATCH] fix cli param, exit on file not found

---
 webp_server.go | 19 +++++++++++--------
 1 file changed, 11 insertions(+), 8 deletions(-)

diff --git a/webp_server.go b/webp_server.go
index 63dcbe0..75888bb 100644
--- a/webp_server.go
+++ b/webp_server.go
@@ -27,6 +27,8 @@ type Config struct {
 	AllowedTypes []string `json:"ALLOWED_TYPES"`
 }
 
+var configPath string
+
 func webpEncoder(p1, p2 string, quality float32) {
 	var buf bytes.Buffer
 	var img image.Image
@@ -47,18 +49,19 @@ func webpEncoder(p1, p2 string, quality float32) {
 	fmt.Println("Save output.webp ok")
 }
 
+func init() {
+	// Config Here
+	flag.StringVar(&configPath, "config", "config.json", "/path/to/config.json. (Default: ./config.json)")
+	flag.Parse()
+	//flag.PrintDefaults()
+}
+
 func main() {
 	app := fiber.New()
 	app.Banner = false
 	app.Server = "WebP Server Go"
 
-	// Config Here
-	var config_path string
-	flag.StringVar(&config_path, "config", "config.json", "/path/to/config.json. (Default: ./config.json)")
-	flag.Parse()
-	flag.PrintDefaults()
-
-	config := load_config(config_path)
+	config := load_config(configPath)
 
 	HOST := config.HOST
 	PORT := config.PORT
@@ -161,7 +164,7 @@ func load_config(path string) Config {
 	var config Config
 	jsonObject, err := os.Open(path)
 	if err != nil {
-		fmt.Println(err.Error())
+		log.Fatal(err)
 	}
 	defer jsonObject.Close()
 	decoder := json.NewDecoder(jsonObject)