From a302eeddc095ca6d856bc56034115211e8e5f888 Mon Sep 17 00:00:00 2001
From: n0vad3v <n0vad3v@riseup.net>
Date: Sun, 23 Feb 2020 20:44:06 +0800
Subject: [PATCH] Add broken flag

---
 webp_server.go | 15 +++++++++++----
 1 file changed, 11 insertions(+), 4 deletions(-)

diff --git a/webp_server.go b/webp_server.go
index 317fe68..63dcbe0 100644
--- a/webp_server.go
+++ b/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 {
@@ -51,7 +53,12 @@ func main() {
 	app.Server = "WebP Server Go"
 
 	// Config Here
-	config := load_config("config.json")
+	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)
 
 	HOST := config.HOST
 	PORT := config.PORT
@@ -125,7 +132,7 @@ func main() {
 
 		// Check for Safari users
 		UA := c.Get("User-Agent")
-		if strings.Contains(UA,"Safari") && !strings.Contains(UA,"Chrome") && !strings.Contains(UA,"Firefox"){
+		if strings.Contains(UA, "Safari") && !strings.Contains(UA, "Chrome") && !strings.Contains(UA, "Firefox") {
 			c.SendFile(ImgAbsolutePath)
 			return
 		}