fix: improve connection reliability with better handshake detection (#133)

# Improve Claude connection reliability and queue processing

This PR enhances the connection handling and queue processing for Claude Code:

- Increases the connection wait delay from 200ms to 600ms to ensure Claude is fully ready before sending queued mentions
- Implements a more robust connection check that verifies WebSocket handshake completion
- Adds progressive polling after connection to ensure handshake is complete
- Increases the delay between sequential message sends to 25ms (from 10ms) to prevent overwhelming the WebSocket connection
- Adds more detailed logging for connection and queue processing states
- Implements a retry mechanism that polls until handshake completes when triggered by a new connection

These changes should significantly improve reliability when sending multiple mentions to Claude, especially immediately after connection.

Author: ThomasK33

lua/claudecode/config.lua

@@ -17,7 +17,7 @@ M.defaults = {
   -- When true, focus Claude terminal after a successful send while connected
   focus_after_send = false,
   visual_demotion_delay_ms = 50, -- Milliseconds to wait before demoting a visual selection
-  connection_wait_delay = 200, -- Milliseconds to wait after connection before sending queued @ mentions
+  connection_wait_delay = 600, -- Milliseconds to wait after connection before sending queued @ mentions
   connection_timeout = 10000, -- Maximum time to wait for Claude Code to connect (milliseconds)
   queue_timeout = 5000, -- Maximum time to keep @ mentions in queue (milliseconds)
   diff_opts = {

lua/claudecode/init.lua

@@ -47,7 +47,21 @@ function M.is_claude_connected()
 
   local server_module = require("claudecode.server.init")
   local status = server_module.get_status()
-  return status.running and status.client_count > 0
+  if not status.running then
+    return false
+  end
+
+  -- Prefer handshake-aware check when client info is available; otherwise fall back to client_count
+  if status.clients and #status.clients > 0 then
+    for _, info in ipairs(status.clients) do
+      if (info.state == "connected" or info.handshake_complete == true) and info.handshake_complete == true then
+        return true
+      end
+    end
+    return false
+  else
+    return status.client_count and status.client_count > 0
+  end
 end
 
 ---Clear the mention queue and stop any pending timer
@@ -149,8 +163,16 @@ function M.process_mention_queue(from_new_connection)
   end
 
   if not M.is_claude_connected() then
-    -- Still disconnected, wait for connection
-    logger.debug("queue", "Claude not connected, keeping " .. #M.state.mention_queue .. " mentions queued")
+    -- Still disconnected or handshake not complete yet, wait for readiness
+    logger.debug("queue", "Claude not ready (no handshake). Keeping ", #M.state.mention_queue, " mentions queued")
+
+    -- If triggered by a new connection, poll until handshake completes (bounded by connection_timeout timer)
+    if from_new_connection then
+      local retry_delay = math.max(50, math.floor((M.state.config.connection_wait_delay or 200) / 4))
+      vim.defer_fn(function()
+        M.process_mention_queue(true)
+      end, retry_delay)
+    end
     return
   end
 
@@ -173,7 +195,7 @@ function M.process_mention_queue(from_new_connection)
 
   logger.debug("queue", "Processing " .. #mentions_to_send .. " queued @ mentions")
 
-  -- Send mentions with 10ms delay between each to prevent WebSocket buffer overflow
+  -- Send mentions with a small delay between each to prevent WebSocket/extension overwhelm
   local function send_mention_sequential(index)
     if index > #mentions_to_send then
       logger.debug("queue", "All queued mentions sent successfully")
@@ -204,19 +226,22 @@ function M.process_mention_queue(from_new_connection)
 
     -- Process next mention with delay
     if index < #mentions_to_send then
+      local inter_message_delay = 25 -- ms
       vim.defer_fn(function()
         send_mention_sequential(index + 1)
-      end, 10) -- 10ms delay between mentions
+      end, inter_message_delay)
     end
   end
 
   -- Apply delay for new connections, send immediately for debounced processing
   if #mentions_to_send > 0 then
     if from_new_connection then
       -- Wait for connection_wait_delay when processing queue after new connection
+      local initial_delay = (M.state.config and M.state.config.connection_wait_delay) or 200
+      logger.debug("queue", "Waiting ", initial_delay, "ms after connect before flushing queue")
       vim.defer_fn(function()
         send_mention_sequential(1)
-      end, M.state.config.connection_wait_delay)
+      end, initial_delay)
     else
       -- Send immediately for debounced processing (Claude already connected)
       send_mention_sequential(1)