feat: auto-focus terminal after sending selection and optimize test performance

Change-Id: Iacfd6193a2d87da88df1679994a740f0d12e9c8f
Signed-off-by: Thomas Kosiewski <[email protected]>

Author: ThomasK33

lua/claudecode/init.lua

@@ -275,6 +275,15 @@ function M._create_commands()
       if sent_successfully then
         vim.api.nvim_feedkeys(vim.api.nvim_replace_termcodes("<Esc>", true, false, true), "n", false)
         logger.debug("command", "ClaudeCodeSend: Exited visual mode after successful send.")
+
+        -- Focus the Claude Code terminal after sending selection
+        local terminal_ok, terminal = pcall(require, "claudecode.terminal")
+        if terminal_ok then
+          terminal.open({}) -- Open/focus the terminal
+          logger.debug("command", "ClaudeCodeSend: Focused Claude Code terminal after selection send.")
+        else
+          logger.warn("command", "ClaudeCodeSend: Failed to load terminal module for focusing.")
+        end
       end
     else
       logger.error("command", "ClaudeCodeSend: Failed to load selection module.")

lua/claudecode/terminal.lua

@@ -357,7 +357,7 @@ function M.open(opts_override)
       managed_snacks_terminal:focus()
       local term_buf_id = managed_snacks_terminal.buf
       if term_buf_id and vim.api.nvim_buf_get_option(term_buf_id, "buftype") == "terminal" then
-        vim.api.nvim_win_call(managed_snacks_terminal.winid, function()
+        vim.api.nvim_win_call(managed_snacks_terminal.win, function()
           vim.cmd("startinsert")
         end)
       end

tests/mocks/vim.lua

@@ -438,18 +438,19 @@ local vim = {
 
   -- Additional missing vim functions
   wait = function(timeout, condition, interval, fast_only)
-    -- Mock implementation that immediately returns true if condition is met
-    -- For tests, we'll assume the condition is met quickly
+    -- Optimized mock implementation for faster test execution
     local start_time = os.clock()
-    interval = interval or 200
+    interval = interval or 10 -- Reduced from 200ms to 10ms for faster polling
     timeout = timeout or 1000
 
     while (os.clock() - start_time) * 1000 < timeout do
       if condition and condition() then
         return true
       end
-      -- In a real implementation this would yield, but for tests we'll just continue
+      -- Add a small sleep to prevent busy-waiting and reduce CPU usage
+      os.execute("sleep 0.001") -- 1ms sleep
     end
+
     return false
   end,
 

tests/unit/diff_mcp_spec.lua

@@ -49,7 +49,7 @@ describe("MCP-compliant diff operations", function()
       end)
 
       -- Wait for resolution
-      vim.wait(1000, function()
+      vim.wait(100, function() -- Reduced from 1000ms to 100ms
         return coroutine.status(co) == "dead"
       end)
 
@@ -78,7 +78,7 @@ describe("MCP-compliant diff operations", function()
       end)
 
       -- Wait for resolution
-      vim.wait(1000, function()
+      vim.wait(100, function() -- Reduced from 1000ms to 100ms
         return coroutine.status(co) == "dead"
       end)
 
@@ -125,7 +125,7 @@ describe("MCP-compliant diff operations", function()
         diff._resolve_diff_as_rejected(test_tab_name)
       end)
 
-      vim.wait(1000, function()
+      vim.wait(100, function() -- Reduced from 1000ms to 100ms
         return coroutine.status(co2) == "dead"
       end)
     end)
@@ -225,7 +225,7 @@ describe("MCP-compliant diff operations", function()
         diff._resolve_diff_as_rejected(tab_name_2)
       end)
 
-      vim.wait(1000, function()
+      vim.wait(100, function() -- Reduced from 1000ms to 100ms
         return coroutine.status(co1) == "dead" and coroutine.status(co2) == "dead"
       end)
     end)

tests/unit/tools/open_diff_mcp_spec.lua

@@ -127,7 +127,7 @@ describe("openDiff tool MCP compliance", function()
       end)
 
       -- Wait for resolution
-      vim.wait(1000, function()
+      vim.wait(100, function() -- Reduced from 1000ms to 100ms
         return coroutine.status(co) == "dead"
       end)
 
@@ -164,7 +164,7 @@ describe("openDiff tool MCP compliance", function()
       end)
 
       -- Wait for resolution
-      vim.wait(1000, function()
+      vim.wait(100, function() -- Reduced from 1000ms to 100ms
         return coroutine.status(co) == "dead"
       end)