Merge pull request #46 from daichirata/feature/version

Add version command

cmd/version.go

@@ -0,0 +1,32 @@
+package cmd
+
+import (
+	"fmt"
+	"runtime/debug"
+
+	"github.com/spf13/cobra"
+)
+
+var Version string
+
+var (
+	versionCmd = &cobra.Command{
+		Use:   "version",
+		Short: "Display version",
+		Run: func(cmd *cobra.Command, args []string) {
+			if Version != "" {
+				fmt.Println(Version)
+				return
+			}
+			if buildInfo, ok := debug.ReadBuildInfo(); ok {
+				fmt.Println(buildInfo.Main.Version)
+				return
+			}
+			fmt.Println("(unknown)")
+		},
+	}
+)
+
+func init() {
+	rootCmd.AddCommand(versionCmd)
+}

main.go

@@ -6,7 +6,11 @@ import (
 	"github.com/daichirata/hammer/cmd"
 )
 
+var version string
+
 func main() {
+	cmd.Version = version
+
 	if err := cmd.Execute(); err != nil {
 		os.Exit(1)
 	}