From 629a2666909aad1bb2f3279da4bee7d6002ae453 Mon Sep 17 00:00:00 2001
From: Marcus Kok <marcusk1699@gmail.com>
Date: Fri, 5 Apr 2024 12:57:10 -0400
Subject: [PATCH] support adding single files to config

---
 cmd/add.go | 36 +++++++++++++++++++++++++++++++-----
 1 file changed, 31 insertions(+), 5 deletions(-)

diff --git a/cmd/add.go b/cmd/add.go
index a36ef3c..0f218c1 100644
--- a/cmd/add.go
+++ b/cmd/add.go
@@ -8,6 +8,7 @@ import (
 
 	"github.com/Marcusk19/dotctl/tools"
 	"github.com/manifoldco/promptui"
+	"github.com/spf13/afero"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
 )
@@ -66,11 +67,36 @@ func runAddCommand(cmd *cobra.Command, args []string) {
     }
     overwrite, _ := confirm.Run()
     if strings.ToUpper(overwrite) == "Y" {
-      err = tools.CopyDir(fs, configSrc, dotfileDest)
-      if err != nil {
-        log.Fatal(err)
-      }
-      fmt.Printf("Copied %s -> %s\n", configSrc, dotfileDest)
+      addConfigToDir(fs, configSrc, dotfileDest)
     }
+  } else {
+    addConfigToDir(fs, configSrc, dotfileDest)
   }
 }
+
+func addConfigToDir(fs afero.Fs, configSrc, dotfileDest string) {
+  configFile, err := fs.Open(configSrc)
+  if err != nil {
+    log.Fatal(err)
+  }
+  
+  defer configFile.Close()
+
+  fileInfo, err := configFile.Stat()
+  if err != nil {
+    log.Fatal(err)
+  }
+
+  if fileInfo.IsDir() {
+    err = tools.CopyDir(fs, configSrc, dotfileDest)
+  } else {
+    err = tools.CopyFile(fs, configSrc, dotfileDest)
+  }
+
+  if err != nil {
+    log.Fatal(err)
+  }
+
+  fmt.Printf("Copied %s -> %s\n", configSrc, dotfileDest)
+}
+