From e63ad28f74624db82031b8a5a691ae8850e8dd57 Mon Sep 17 00:00:00 2001
From: Albert S <mail@quitesimple.org>
Date: Mon, 22 Mar 2021 19:50:27 +0100
Subject: [PATCH] Print usage also when an invalid command is supplied

---
 main.go | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/main.go b/main.go
index fb0ece1..1f29009 100644
--- a/main.go
+++ b/main.go
@@ -119,6 +119,10 @@ func exists(path string) (bool, error) {
 	return false, err
 }
 
+func printUsage() {
+	fmt.Fprintf(os.Stderr, "Usage:   ./hs9001 <add/search/init/import>\n")
+}
+
 func main() {
 	var ret int
 	flag.IntVar(&ret, "ret", 0, "Return value of the command to add")
@@ -127,7 +131,7 @@ func main() {
 	argslen := len(args)
 
 	if argslen < 1 {
-		fmt.Fprintf(os.Stderr, "Usage:   ./hs9001 <add/search/init/import>\n")
+		printUsage()
 		return
 	}
 
@@ -164,5 +168,9 @@ func main() {
 		initDatabase(conn)
 	} else if cmd == "import" {
 		importFromStdin(conn)
+	} else {
+		fmt.Fprint(os.Stderr, "Error: Unknown command supplied\n\n")
+		printUsage()
+		return
 	}
 }