cmd/geas: improve error message when more than one arg given

fjl · Sep 29, 2023 · d61ac65 · d61ac65
1 parent b7b7bc2
commit d61ac65
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/cmd/geas/geas.go b/cmd/geas/geas.go
@@ -33,10 +33,15 @@ func main() {
 	)
 	flag.Parse()
 
-	if flag.NArg() != 1 {
+	var file string
+	switch flag.NArg() {
+	case 0:
 		exit(fmt.Errorf("need filename as argument"))
+	case 1:
+		file = flag.Arg(0)
+	default:
+		exit(fmt.Errorf("too many arguments"))
 	}
-	file := flag.Arg(0)
 
 	c := asm.NewCompiler(os.DirFS("."))
 	c.SetUsePush0(!*noPush0)