✨ feat (diff): Add interactive staging for untracked files

Author: Abizrh

lua/commit-ai/utils.lua

@@ -2,6 +2,29 @@ local log = require("commit-ai.log")
 local M = {}
 
 function M.get_git_diff()
+  local status_lines = vim.fn.systemlist('git status --porcelain')
+  local untracked_files = vim.tbl_filter(function(line)
+    return line:match("^%?%?")
+  end, status_lines)
+
+  if #untracked_files > 0 then
+    for _, line in ipairs(untracked_files) do
+      local file = line:sub(4)
+      log.info("‼️ Untracked file: " .. file)
+    end
+    local answer = vim.fn.input("Do you want to add them to the staging area? (y/n) ")
+    if answer:lower() == "y" then
+      for _, line in ipairs(untracked_files) do
+        local file = line:sub(4)
+        vim.fn.system("git add " .. file)
+        log.info("📝 Added " .. file .. " to the staging area")
+      end
+    else
+      log.error("Aborting")
+      return nil
+    end
+    return nil
+  end
   local staged_diff = vim.fn.system('git diff --cached --no-color')
   if staged_diff == "" then
     staged_diff = vim.fn.system('git diff --no-color')