fix: parse flags properly

savely-krasovsky · Jun 30, 2023 · ef61af3 · ef61af3
1 parent 30d38de
commit ef61af3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/cmd/filegenerator/main.go b/cmd/filegenerator/main.go
@@ -12,14 +12,16 @@ func main() {
 		output string
 	)
 
-	flag.StringVar(&output, "output", "", "Output file name")
-	generator := geosite_antizapret.NewGenerator()
+	flag.StringVar(&output, "output", "", "Output path")
+	flag.Parse()
 
 	if output == "" {
-		fmt.Println("Provide a name of file!")
+		fmt.Println("Provide path to file!")
 		return
 	}
 
+	generator := geosite_antizapret.NewGenerator()
+
 	if err := generator.GenerateAndWrite(output); err != nil {
 		log.Fatal(err)
 	}