From 48ec51d166bf36cbcd6815ae9dc3ca708f1acca5 Mon Sep 17 00:00:00 2001
From: ohdearaugustin <j-reindl@live.at>
Date: Sun, 25 Jul 2021 02:02:05 +0200
Subject: [PATCH] Refactor cmdVersion to cli package

---
 cmd/headscale/cli/version.go | 23 +++++++++++++++++++++++
 cmd/headscale/headscale.go   | 19 +------------------
 2 files changed, 24 insertions(+), 18 deletions(-)
 create mode 100644 cmd/headscale/cli/version.go

diff --git a/cmd/headscale/cli/version.go b/cmd/headscale/cli/version.go
new file mode 100644
index 0000000..4bdf92e
--- /dev/null
+++ b/cmd/headscale/cli/version.go
@@ -0,0 +1,23 @@
+package cli
+
+import (
+	"fmt"
+	"github.com/spf13/cobra"
+	"strings"
+)
+
+var version = "dev"
+
+var VersionCmd = &cobra.Command{
+	Use:   "version",
+	Short: "Print the version.",
+	Long:  "The version of headscale.",
+	Run: func(cmd *cobra.Command, args []string) {
+		o, _ := cmd.Flags().GetString("output")
+		if strings.HasPrefix(o, "json") {
+			JsonOutput(map[string]string{"version": version}, nil, o)
+			return
+		}
+		fmt.Println(version)
+	},
+}
diff --git a/cmd/headscale/headscale.go b/cmd/headscale/headscale.go
index 195cec0..c6e56fa 100644
--- a/cmd/headscale/headscale.go
+++ b/cmd/headscale/headscale.go
@@ -4,28 +4,11 @@ import (
 	"fmt"
 	"log"
 	"os"
-	"strings"
 
 	"github.com/juanfont/headscale/cmd/headscale/cli"
 	"github.com/spf13/cobra"
 )
 
-var version = "dev"
-
-var versionCmd = &cobra.Command{
-	Use:   "version",
-	Short: "Print the version.",
-	Long:  "The version of headscale.",
-	Run: func(cmd *cobra.Command, args []string) {
-		o, _ := cmd.Flags().GetString("output")
-		if strings.HasPrefix(o, "json") {
-			cli.JsonOutput(map[string]string{"version": version}, nil, o)
-			return
-		}
-		fmt.Println(version)
-	},
-}
-
 var headscaleCmd = &cobra.Command{
 	Use:   "headscale",
 	Short: "headscale - a Tailscale control server",
@@ -47,7 +30,7 @@ func main() {
 	headscaleCmd.AddCommand(cli.PreauthkeysCmd)
 	headscaleCmd.AddCommand(cli.RoutesCmd)
 	headscaleCmd.AddCommand(cli.ServeCmd)
-	headscaleCmd.AddCommand(versionCmd)
+	headscaleCmd.AddCommand(cli.VersionCmd)
 
 	cli.NodeCmd.PersistentFlags().StringP("namespace", "n", "", "Namespace")
 	err = cli.NodeCmd.MarkPersistentFlagRequired("namespace")