Added a new command to change version (#185)

* added update command to change version

Signed-off-by: Kartikay <[email protected]>

* added homedir.Dir()

Signed-off-by: Kartikay <[email protected]>

* added homedir

Signed-off-by: Kartikay <[email protected]>

---------

Signed-off-by: Kartikay <[email protected]>

Author: kartikaysaxena

pkg/cmd/root/root.go

@@ -25,6 +25,7 @@ import (
 
 	"github.com/litmuschaos/litmusctl/pkg/cmd/run"
 	"github.com/litmuschaos/litmusctl/pkg/cmd/save"
+	"github.com/litmuschaos/litmusctl/pkg/cmd/update"
 
 	"github.com/litmuschaos/litmusctl/pkg/cmd/connect"
 	"github.com/litmuschaos/litmusctl/pkg/cmd/delete"
@@ -77,6 +78,7 @@ func init() {
 	rootCmd.AddCommand(save.SaveCmd)
 	rootCmd.AddCommand(run.RunCmd)
 	rootCmd.AddCommand(list.ListCmd)
+	rootCmd.AddCommand(update.UpdateCmd)
 
 	// Here you will define your flags and configuration settings.
 	// Cobra supports persistent flags, which, if defined here,

pkg/cmd/update/update.go

@@ -0,0 +1,98 @@
+package update
+
+import (
+	"io"
+	"net/http"
+	"os"
+	"os/exec"
+	"runtime"
+
+	"github.com/litmuschaos/litmusctl/pkg/utils"
+	homedir "github.com/mitchellh/go-homedir"
+	"github.com/spf13/cobra"
+)
+
+var UpdateCmd = &cobra.Command{
+	Use:   "update",
+	Short: "Changes the version of litmusctl",
+	Args:  cobra.ExactArgs(1),
+	Long:  ``,
+	Run: func(cmd *cobra.Command, args []string) {
+		updateVersion := args[0]
+		homeDir, err := homedir.Dir()
+		if err != nil {
+			utils.PrintError(err)
+		}
+
+		var assetURL string = "https://litmusctl-production-bucket.s3.amazonaws.com/"
+		var binaryName string = "litmusctl"
+		switch runtime.GOOS {
+		case "windows":
+			if runtime.GOARCH == "386" {
+				binaryName += "-windows-386-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			} else if runtime.GOARCH == "amd64" {
+				binaryName += "-windows-amd64-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			} else {
+				binaryName += "-windows-arm64-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			}
+		case "linux":
+			if runtime.GOARCH == "arm64" {
+				binaryName += "-linux-arm64-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			} else if runtime.GOARCH == "amd64" {
+				binaryName += "-linux-amd64-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			} else if runtime.GOARCH == "arm" {
+				binaryName += "-linux-arm-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			} else {
+				binaryName += "-linux-386-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			}
+		case "darwin":
+			if runtime.GOARCH == "amd64" {
+				binaryName += "-darwin-amd64-" + updateVersion + ".tar.gz"
+				assetURL += binaryName
+			}
+		}
+
+		utils.White.Print("Downloading:\n")
+
+		resp2, err := http.Get(assetURL)
+		if err != nil {
+			utils.PrintError(err)
+		}
+
+		tempFile, err := os.CreateTemp("", binaryName)
+		if err != nil {
+			utils.PrintError(err)
+			return
+		}
+		defer os.Remove(tempFile.Name())
+
+		_, err = io.Copy(tempFile, resp2.Body)
+		if err != nil {
+			utils.PrintError(err)
+			return
+		}
+		utils.White_B.Print("OK\n")
+
+		tempFile.Close()
+
+		tarCmd := exec.Command("tar", "xzf", tempFile.Name(), "-C", homeDir)
+		tarCmd.Stdout = os.Stdout
+		tarCmd.Stderr = os.Stderr
+
+		utils.White_B.Print("Extracting binary...\n")
+		err = tarCmd.Run()
+		if err != nil {
+			utils.PrintError(err)
+			return
+		}
+
+		utils.White_B.Print("Binary extracted successfully\n")
+	},
+}