Add flag version

kubernetes-csi · Mar 1, 2019 · e18d8b6 · e18d8b6
1 parent c44a77d
commit e18d8b6
Showing 1 changed file with 12 additions and 3 deletions.
diff --git a/cmd/hostpathplugin/main.go b/cmd/hostpathplugin/main.go
@@ -20,6 +20,7 @@ import (
 	"flag"
 	"fmt"
 	"os"
+	"path"
 
 	"github.com/kubernetes-csi/csi-driver-host-path/pkg/hostpath"
 )
@@ -29,14 +30,22 @@ func init() {
 }
 
 var (
-	endpoint   = flag.String("endpoint", "unix://tmp/csi.sock", "CSI endpoint")
-	driverName = flag.String("drivername", "csi-hostpath", "name of the driver")
-	nodeID     = flag.String("nodeid", "", "node id")
+	endpoint    = flag.String("endpoint", "unix://tmp/csi.sock", "CSI endpoint")
+	driverName  = flag.String("drivername", "csi-hostpath", "name of the driver")
+	nodeID      = flag.String("nodeid", "", "node id")
+	showVersion = flag.Bool("version", false, "Show version.")
+	version     = ""
 )
 
 func main() {
 	flag.Parse()
 
+	if *showVersion {
+		baseName := path.Base(os.Args[0])
+		fmt.Println(baseName, version)
+		return
+	}
+
 	handle()
 	os.Exit(0)
 }