add version flag

preved911 · Feb 17, 2020 · 9b99840 · 9b99840
1 parent 0a4f09a
commit 9b99840
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -1,15 +1,25 @@
 package main
 
 import (
+	"fmt"
 	flag "github.com/spf13/pflag"
 	"log"
+	"os"
 )
 
+const AppVersion = "0.0.11"
+
 func main() {
-	var configPath *string = flag.StringP("config", "c", "", "config file path")
+	var configPath *string = flag.StringP("config", "c", "", "specify config file path")
+	var version *bool = flag.BoolP("version", "v", false, "return tool version and exit")
 
 	flag.Parse()
 
+	if *version {
+		fmt.Printf("kube-certs-gen, version %s\n", AppVersion)
+		os.Exit(0)
+	}
+
 	cfg, err := parseConfig(configPath)
 	if err != nil {
 		log.Fatalln(err)