Merge pull request #26 from intercube/bugfix/DEV-320

cmd/map.go

@@ -21,6 +21,7 @@ import (
 	"os"
 )
 
+var override = false
 var mapCmd = &cobra.Command{
 	Use:   "map",
 	Short: "Maps files based on config yaml file",
@@ -33,18 +34,35 @@ var mapCmd = &cobra.Command{
 
 func init() {
 	rootCmd.AddCommand(mapCmd)
+	mapCmd.PersistentFlags().BoolVarP(&override, "override", "o", false, "Overrides existing destination file")
 }
 
 func symlink(from string, to string) {
-	if !fileExists(to) {
-		err := os.Symlink(from, to)
-		if err != nil {
-			panic(fmt.Errorf("Unable to map: %s \n", err))
+	if !fileExists(from) {
+		fmt.Print(fmt.Errorf("Origin file %v does not exists\n", to))
+	} else {
+		if !fileExists(to) || override {
+			if override {
+				if destination, err := os.Lstat(to); err == nil {
+					if destination.IsDir() {
+						os.RemoveAll(to)
+						fmt.Printf("Removed directory %v\n", to)
+					} else {
+						os.Remove(to)
+						fmt.Printf("Removed file %v\n", to)
+					}
+				}
+			}
+
+			err := os.Symlink(from, to)
+			if err != nil {
+				panic(fmt.Errorf("Unable to map: %s \n", err))
+			} else {
+				fmt.Printf("Mapped %v to %v\n", from, to)
+			}
 		} else {
-			fmt.Printf("Mapped %v to %v\n", from, to)
+			fmt.Print(fmt.Errorf("Destination file %v already exists\n", to))
 		}
-	} else {
-		fmt.Print(fmt.Errorf("Destination file %v already exists\n", to))
 	}
 }