fixed a silly bug

0xsha · Sep 10, 2020 · 8a6b5b3 · 8a6b5b3
1 parent a05a6f3
commit 8a6b5b3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -72,7 +72,6 @@ func main() {
 
 	mode := parser.String("m", "mode",
 		&argparse.Options{
-			Default:  1,
 			Required: false,
 			Help:     "storage or app"})
 
@@ -116,6 +115,13 @@ func main() {
 		zerolog.SetGlobalLevel(zerolog.PanicLevel)
 	}
 
+	// check mode
+
+	if *mode != "storage" && *mode != "app" {
+
+		log.Fatal().Msg("Invalid mode use app or storage")
+	}
+
 	// initialize config.yaml
 	configPath := path.Join(*configFolder, "config.yaml")
 	providerPath := path.Join(*configFolder, "modules")