devbonzai 2.1.4 → 2.1.6

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "devbonzai",
-  "version": "2.1.4",
+  "version": "2.1.6",
   "description": "Quickly set up a local file server in any repository for browser-based file access",
   "main": "cli.js",
   "bin": {

package/templates/receiver.js

@@ -1525,9 +1525,21 @@ app.post('/prompt_agent_stream', (req, res) => {
   res.setHeader('Access-Control-Allow-Origin', '*');
   res.flushHeaders();
 
-  // Helper to send SSE events
+  // Helper to send SSE events with robustness checks
   const sendEvent = (type, data) => {
-    res.write(`data: ${JSON.stringify({ type, ...data })}\n\n`);
+    try {
+      // Check if response is still writable - try to send even if clientDisconnected flag is set
+      // because the response stream might still be open
+      if (res.destroyed || res.closed) {
+        console.log(`⚠️ [prompt_agent_stream] Response already closed, cannot send ${type} event`);
+        return false;
+      }
+      res.write(`data: ${JSON.stringify({ type, ...data })}\n\n`);
+      return true;
+    } catch (e) {
+      console.log(`⚠️ [prompt_agent_stream] Error sending ${type} event:`, e.message);
+      return false;
+    }
   };
 
   // Capture beforeCommit
@@ -1607,21 +1619,40 @@ app.post('/prompt_agent_stream', (req, res) => {
         if (!proc.killed) proc.kill('SIGKILL');
       }, 5000);
       
+      // Always try to send complete event when timeout occurs
+      // Only skip if we've already sent it
       if (!responseSent) {
-        responseSent = true;
-        sendEvent('error', {
-          error: 'Process timeout',
-          message: `cursor-agent exceeded timeout of ${timeoutMs / 1000} seconds`
-        });
-        sendEvent('complete', {
-          code: -1,
-          stdout,
-          stderr,
-          changedFiles: Array.from(changedFiles),
-          beforeCommit,
-          afterCommit: ''
-        });
-        res.end();
+        try {
+          // Check if response is still writable - try to send even if clientDisconnected flag is set
+          // because the response stream might still be open
+          if (res.destroyed || res.closed) {
+            console.log('⚠️ [prompt_agent_stream] Response already closed, cannot send timeout events');
+          } else {
+            responseSent = true;
+            sendEvent('error', {
+              error: 'Process timeout',
+              message: `cursor-agent exceeded timeout of ${timeoutMs / 1000} seconds`
+            });
+            sendEvent('complete', {
+              code: -1,
+              stdout,
+              stderr,
+              changedFiles: Array.from(changedFiles),
+              beforeCommit,
+              afterCommit: ''
+            });
+            // Send stop event after complete
+            sendEvent('stop', {});
+            res.end();
+            console.log('✅ [prompt_agent_stream] Sent timeout error, complete, and stop events');
+          }
+        } catch (e) {
+          console.log('⚠️ [prompt_agent_stream] Error sending timeout events:', e.message);
+          // Don't set responseSent = true on error, in case we can retry
+          // But realistically, if there's an error, the connection is probably dead
+        }
+      } else {
+        console.log('⚠️ [prompt_agent_stream] Timeout events already sent, skipping');
       }
     }
   }, timeoutMs);
@@ -1638,10 +1669,30 @@ app.post('/prompt_agent_stream', (req, res) => {
     console.log('❌ [prompt_agent_stream] Process error:', error.message);
     clearInterval(pollInterval);
     if (timeoutId) clearTimeout(timeoutId);
+    
+    // Always try to send error event when process error occurs
+    // Only skip if we've already sent it
     if (!responseSent) {
-      responseSent = true;
-      sendEvent('error', { error: error.message });
-      res.end();
+      try {
+        // Check if response is still writable - try to send even if clientDisconnected flag is set
+        // because the response stream might still be open
+        if (res.destroyed || res.closed) {
+          console.log('⚠️ [prompt_agent_stream] Response already closed, cannot send error event');
+        } else {
+          responseSent = true;
+          sendEvent('error', { error: error.message });
+          // Send stop event after error
+          sendEvent('stop', {});
+          res.end();
+          console.log('✅ [prompt_agent_stream] Sent error and stop events');
+        }
+      } catch (e) {
+        console.log('⚠️ [prompt_agent_stream] Error sending error event:', e.message);
+        // Don't set responseSent = true on error, in case we can retry
+        // But realistically, if there's an error, the connection is probably dead
+      }
+    } else {
+      console.log('⚠️ [prompt_agent_stream] Error event already sent, skipping');
     }
   });
 
@@ -1657,17 +1708,37 @@ app.post('/prompt_agent_stream', (req, res) => {
       // Ignore
     }
     
+    // Always try to send complete event when process finishes
+    // Check actual response stream state rather than relying on responseSent flag
+    // Only skip if we've already sent it (responseSent flag prevents duplicates)
     if (!responseSent) {
-      responseSent = true;
-      sendEvent('complete', {
-        code,
-        stdout,
-        stderr,
-        changedFiles: Array.from(changedFiles),
-        beforeCommit,
-        afterCommit
-      });
-      res.end();
+      try {
+        // Check if response is still writable - check actual stream state
+        if (res.destroyed || res.closed) {
+          console.log('⚠️ [prompt_agent_stream] Response already closed, cannot send complete event');
+        } else {
+          // Send events and only set flag after successful send
+          sendEvent('complete', {
+            code,
+            stdout,
+            stderr,
+            changedFiles: Array.from(changedFiles),
+            beforeCommit,
+            afterCommit
+          });
+          // Send stop event after complete
+          sendEvent('stop', {});
+          res.end();
+          responseSent = true;
+          console.log('✅ [prompt_agent_stream] Sent complete and stop events');
+        }
+      } catch (e) {
+        console.log('⚠️ [prompt_agent_stream] Error sending complete event:', e.message);
+        // Don't set responseSent = true on error, in case we can retry
+        // But realistically, if there's an error, the connection is probably dead
+      }
+    } else {
+      console.log('⚠️ [prompt_agent_stream] Complete event already sent, skipping');
     }
   });
 
@@ -1675,8 +1746,7 @@ app.post('/prompt_agent_stream', (req, res) => {
   req.on('close', () => {
     console.log('🔵 [prompt_agent_stream] Client disconnected (process continues in background)');
     // Don't kill the process - let it complete
-    // Just mark that we shouldn't try to send more events
-    responseSent = true;
+    // Don't set responseSent here - let proc.on('close') check actual stream state
   });
 });