From 5b0f84bf08c39f64868729ba914b2a0315ad5a9c Mon Sep 17 00:00:00 2001
From: Shiv Jha-Mathur <shivjm@users.noreply.github.com>
Date: Tue, 9 Nov 2021 02:42:16 +0530
Subject: [PATCH] feat: add version command

---
 cmd/root.go    |  6 +++++-
 cmd/version.go | 34 ++++++++++++++++++++++++++++++++++
 2 files changed, 39 insertions(+), 1 deletion(-)
 create mode 100644 cmd/version.go

diff --git a/cmd/root.go b/cmd/root.go
index 2a06418..47b6e59 100644
--- a/cmd/root.go
+++ b/cmd/root.go
@@ -4,11 +4,15 @@ import (
 	"github.com/spf13/cobra"
 )
 
+const (
+	ProgramName = "dockerfile-image-tags"
+)
+
 var (
 	unknownMarker string
 
 	rootCmd = &cobra.Command{
-		Use:   "dockerfile-image-tags",
+		Use:   ProgramName,
 		Short: "List or query images and tags used in a Dockerfile.",
 	}
 )
diff --git a/cmd/version.go b/cmd/version.go
new file mode 100644
index 0000000..3cbc868
--- /dev/null
+++ b/cmd/version.go
@@ -0,0 +1,34 @@
+package cmd
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/spf13/cobra"
+)
+
+var Version = "development"
+var Revision = ""
+
+var (
+	versionCmd = &cobra.Command{
+		Use:   "version",
+		Short: "Prints version",
+		Long:  "Prints the program’s version information.",
+		Args:  cobra.ExactArgs(0),
+		Run: func(cmd *cobra.Command, args []string) {
+			git := ""
+			r := strings.TrimSpace(Revision)
+
+			if len(r) > 0 {
+				git = fmt.Sprintf(" (r%s)", r)
+			}
+
+			fmt.Printf("%s %s%s", ProgramName, Version, git)
+		},
+	}
+)
+
+func init() {
+	rootCmd.AddCommand(versionCmd)
+}