remove leading '.' in file if adding to the config (#31)

cmd/add.go

@@ -37,16 +37,13 @@ func runAddCommand(cmd *cobra.Command, args []string) {
   configSrc := args[0]
   dirs := strings.Split(configSrc, "/")
   name := dirs[len(dirs) - 1] // take the last section of the path, this should be the name
+  if name[0] == '.' {
+    name = name[1:]
+  }
 
   links := viper.GetStringMap("links")
   links[name] = configSrc
   viper.Set("links", links)
-  if !testing {
-    err := viper.WriteConfig()
-    if err != nil {
-      fmt.Printf("Problem updating dotctl config %s", err)
-    }
-  }
 
   dotfilePath := viper.Get("dotfile-path").(string)
 
@@ -69,9 +66,17 @@ func runAddCommand(cmd *cobra.Command, args []string) {
     if strings.ToUpper(overwrite) == "Y" {
       addConfigToDir(fs, configSrc, dotfileDest)
     }
+    fmt.Printf("Just set up %s to link to %s\n", configSrc, dotfileDest)
   } else {
     addConfigToDir(fs, configSrc, dotfileDest)
   }
+
+  if !testing {
+    err := viper.WriteConfig()
+    if err != nil {
+      fmt.Printf("Problem updating dotctl config %s", err)
+    }
+  }
 }
 
 func addConfigToDir(fs afero.Fs, configSrc, dotfileDest string) {

cmd/remove.go

@@ -61,6 +61,14 @@ func runRemoveCommand(cmd *cobra.Command, args []string) {
     return
   }
 
+  delete(links, dotfile)
+  viper.Set("links", links)
+  err = viper.WriteConfig()
+  if err != nil {
+    fmt.Printf("ERROR: problem saving config: %s\n", err)
+    return
+  }
+  
   fmt.Printf("%s symlink removed, copied files over to %s\n", dotfile, dotfileConfigPath)
 }