Merge pull request #1 from n0vad3v/add-cli-config

Add cli config

webp_server.go

@@ -3,9 +3,8 @@ package main
 import (
 	"bytes"
 	"encoding/json"
+	"flag"
 	"fmt"
-	"github.com/chai2010/webp"
-	"github.com/gofiber/fiber"
 	"image"
 	"image/jpeg"
 	"image/png"
@@ -15,6 +14,9 @@ import (
 	"path"
 	"strconv"
 	"strings"
+
+	"github.com/chai2010/webp"
+	"github.com/gofiber/fiber"
 )
 
 type Config struct {
@@ -25,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
@@ -45,13 +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
-	config := load_config("config.json")
+	config := load_config(configPath)
 
 	HOST := config.HOST
 	PORT := config.PORT
@@ -154,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)