Merge branch 'develop' of github.com:warrensbox/terraform-switcher in…

…to develop

lib/symlink.go

@@ -20,12 +20,12 @@ func RemoveSymlink(symlinkPath string) {
 
 	_, err := os.Lstat(symlinkPath)
 	if err != nil {
-		log.Fatal("Unable to remove symlink. You must have SUDO privileges")
+		log.Fatalf("Unable to find symlink. You must have SUDO privileges - %v \n", err)
 		panic(err)
 	} else {
 		errRemove := os.Remove(symlinkPath)
 		if errRemove != nil {
-			log.Fatal("Unable to remove symlink. You must have SUDO privileges")
+			log.Fatalf("Unable to remove symlink. You must have SUDO privileges - %v \n", err)
 			panic(errRemove)
 		}
 	}

main.go

@@ -144,8 +144,15 @@ func main() {
 
 		/* if selected version already exist, */
 		if fileExist {
-			/* remove current symlink and set new symlink to desired version */
-			lib.RemoveSymlink(installedBinPath)
+			/* remove current symlink if exist*/
+			exist := lib.CheckFileExist(installedBinPath)
+
+			if !exist {
+				fmt.Println("Symlink does not exist")
+			} else {
+				fmt.Printf("Symlink exist")
+				lib.RemoveSymlink(installedBinPath)
+			}
 
 			/* set symlink to desired version */
 			lib.CreateSymlink(installLocation+installVersion+version, installedBinPath)
@@ -176,8 +183,15 @@ func main() {
 		/* remove zipped file to clear clutter */
 		lib.RemoveFiles(installLocation + installVersion + version + "_" + goos + "_" + goarch + ".zip")
 
-		/* remove current symlink and set new symlink to desired version  */
-		lib.RemoveSymlink(installedBinPath)
+		/* remove current symlink if exist*/
+		exist := lib.CheckFileExist(installedBinPath)
+
+		if !exist {
+			fmt.Println("Symlink does not exist")
+		} else {
+			fmt.Printf("Symlink exist")
+			lib.RemoveSymlink(installedBinPath)
+		}
 
 		/* set symlink to desired version */
 		lib.CreateSymlink(installLocation+installVersion+version, installedBinPath)