Feature: remove alias

cmd/unalias.go

@@ -0,0 +1,33 @@
+package cmd
+
+import (
+	"github.com/spf13/cobra"
+
+	"github.com/tfversion/tfversion/pkg/helpers"
+	"github.com/tfversion/tfversion/pkg/unalias"
+)
+
+const (
+	unaliasExample = "# Un-alias a Terraform version\n" +
+		"tfversion unalias default\n" +
+		"tfversion unalias legacy"
+)
+
+var (
+	unaliasCmd = &cobra.Command{
+		Use:     "unalias",
+		Short:   "Un-alias a Terraform version",
+		Example: unaliasExample,
+		Run: func(cmd *cobra.Command, args []string) {
+			if len(args) != 1 {
+				err := helpers.ErrorWithHelp("tfversion unalias -h")
+				helpers.ExitWithError("provide an alias name", err)
+			}
+			unalias.UnaliasVersion(args[0])
+		},
+	}
+)
+
+func init() {
+	rootCmd.AddCommand(unaliasCmd)
+}

pkg/alias/alias.go

@@ -24,7 +24,7 @@ func AliasVersion(alias string, version string) {
 	if err == nil {
 		err = os.RemoveAll(aliasPath)
 		if err != nil {
-			helpers.ExitWithError("error removing symlink", err)
+			helpers.ExitWithError("removing symlink", err)
 		}
 	}
 
@@ -42,14 +42,14 @@ func AliasVersion(alias string, version string) {
 func GetAliasLocation() string {
 	user, err := os.UserHomeDir()
 	if err != nil {
-		helpers.ExitWithError("error getting user home directory", err)
+		helpers.ExitWithError("getting user home directory", err)
 	}
 
 	aliasLocation := filepath.Join(user, download.ApplicationDir, download.AliasesDir)
 	if _, err := os.Stat(aliasLocation); os.IsNotExist(err) {
 		err := os.Mkdir(aliasLocation, 0755)
 		if err != nil {
-			helpers.ExitWithError("error creating alias directory", err)
+			helpers.ExitWithError("creating alias directory", err)
 		}
 	}
 

pkg/unalias/unalias.go

@@ -0,0 +1,27 @@
+package unalias
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/tfversion/tfversion/pkg/alias"
+	"github.com/tfversion/tfversion/pkg/helpers"
+)
+
+// UnaliasVersion removes the symlink for the specified alias.
+func UnaliasVersion(aliasName string) {
+	aliasLocation := alias.GetAliasLocation()
+	aliasPath := filepath.Join(aliasLocation, aliasName)
+	_, err := os.Lstat(aliasPath)
+	if err != nil {
+		helpers.ExitWithError("removing symlink", err)
+	}
+
+	err = os.RemoveAll(aliasPath)
+	if err != nil {
+		helpers.ExitWithError("removing symlink", err)
+	}
+
+	fmt.Printf("Removed alias %s\n", helpers.ColoredVersion(aliasName))
+}