Merge pull request #317 from gaohan085/dev

update & fix go program flag usage

README.md

@@ -13,13 +13,16 @@ make && make build
 #### Usage
 
 ```bash
-videoserver-0.7.0.1-linux-amd64 usage -ffmpeg -pro # As ffmpeg server in production mode
+videoserver-0.7.0.1-linux-amd64 -usage=ffmpeg # As ffmpeg server in production mode
 ```
 
 ```bash
-videoserver-0.7.0.1-linux-amd64 usage -pro # As main video server in production mode
+videoserver-0.7.0.1-linux-amd64 -usage=pro # As main video server in production mode
 ```
 
 ```bash
-videoserver-0.7.0.1-linux-amd64 usage -dev # As main video server in development mode
+videoserver-0.7.0.1-linux-amd64 -usage=dev -file=remote # As main video server in development mode and get file information from remote server
+```
+```bash
+videoserver-0.7.0.1-linux-amd64 -usage=dev -file=local # As main video server in development mode and get file information from local path
 ```

main.go

@@ -27,55 +27,13 @@ var content embed.FS
 // main usage -ffmpeg -pro
 
 func Parseflag() {
-	usage := flag.NewFlagSet("usage", flag.ExitOnError)
-	usageFFmpeg := usage.Bool("ffmpeg", false, "-ffmpeg use as ffmpeg server, not provide as main server")
-	usageIsDev := usage.Bool("dev", false, "'-dev' use as development server and '-pro' use as production server")
-	usageIsPro := usage.Bool("pro", false, "'-dev' use as development server and '-pro' use as production server")
-	usageIsLocal := usage.Bool("local", false, "-local to show local file")
-	usageIsRemote := usage.Bool("remote", false, "-remote to get file info from remote")
-
-	if len(os.Args) <= 1 {
-		log.Panic("expected 'usage' commands")
-	}
+	usage := flag.String("usage", "dev", "Web server usage, can be 'pro' 'dev' or 'ffmpeg'. Default: dev")
+	file := flag.String("file", "local", "Where to get file information, can be 'local' or 'remote'. Default: local")
 
-	switch os.Args[1] {
-	case "usage":
-		usage.Parse(os.Args[2:])
-		if *usageFFmpeg {
-			os.Setenv("USAGE", "ffmpeg")
-		} else {
-			os.Setenv("USAGE", "main")
-		}
-
-		if *usageIsDev && *usageIsPro {
-			log.Panic("'-dev' and '-pro' cannot be provided at same time.")
-		}
-
-		if *usageIsDev {
-			os.Setenv("ENV", "dev")
-
-			if !*usageIsLocal && !*usageIsRemote {
-				log.Panic("Please provide file location '-local' or '-remote'")
-			}
-		}
-
-		if *usageIsPro {
-			os.Setenv("ENV", "pro")
-		}
-
-		if *usageIsLocal && *usageIsRemote {
-			log.Panic("'-local' and '-remote' cannot provide at same time.")
-		}
-
-		if *usageIsLocal {
-			os.Setenv("FILELOCATION", "local")
-		}
-		if *usageIsRemote {
-			os.Setenv("FILELOCATION", "remote")
-		}
-	default:
-		log.Panic("expect command 'usage'")
-	}
+	flag.Parse()
+
+	os.Setenv("USAGE", *usage)
+	os.Setenv("FileLocation", *file)
 }
 
 func main() {

routes.go

@@ -33,7 +33,7 @@ func SetRoutes(app *fiber.App) {
 	api.Get("/actress/:name", handlers.GetVideosByActress)
 	api.Post("/convert", handlers.ConvertHandler)
 
-	switch os.Getenv("FILELOCATION") {
+	switch os.Getenv("FileLocation") {
 	case "local":
 		api.Get("/*", handlers.FileReaderHandler)