Add arg check

Atlas-OS · Nov 21, 2021 · f0f5392 · f0f5392
1 parent d7463ee
commit f0f5392
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/multichoice/main.go b/multichoice/main.go
@@ -20,7 +20,10 @@ var outputChoices []string
 func main() {
 	log.SetFlags(log.LstdFlags | log.Lshortfile)
 	argsWithoutProg := os.Args[1:]
-
+	if len(argsWithoutProg) != 3 {
+		log.Println("Usage: multichoice Title Prompt Choices;here")
+		os.Exit(1)
+	}
 	title := argsWithoutProg[0]
 	prompt := argsWithoutProg[1]
 	inputChoices = strings.Split(argsWithoutProg[2], ";")