devbonzai 2.0.8 → 2.1.0

package/cli.js

@@ -93,6 +93,7 @@ app.get('/', (req, res) => {
       'POST /open-cursor': 'Open Cursor (body: {path, line?})',
       'POST /analyze_prompt': 'Analyze what files would be modified (body: {prompt})',
       'POST /prompt_agent': 'Execute cursor-agent command (body: {prompt})',
+      'POST /prompt_agent_stream': 'Execute cursor-agent with SSE streaming (body: {prompt})',
       'POST /revert_job': 'Revert to a previous commit (body: {beforeCommit})',
       'POST /shutdown': 'Gracefully shutdown the server'
     },
@@ -1566,6 +1567,179 @@ app.post('/prompt_agent', (req, res) => {
   });
 });
 
+// Streaming version of prompt_agent using Server-Sent Events
+app.post('/prompt_agent_stream', (req, res) => {
+  console.log('🔵 [prompt_agent_stream] Endpoint hit');
+  const { prompt } = req.body;
+  console.log('🔵 [prompt_agent_stream] Received prompt:', prompt ? \`\${prompt.substring(0, 50)}...\` : 'none');
+
+  if (!prompt || typeof prompt !== 'string') {
+    console.log('❌ [prompt_agent_stream] Error: prompt required');
+    return res.status(400).json({ error: 'prompt required' });
+  }
+
+  // Set up SSE headers
+  res.setHeader('Content-Type', 'text/event-stream');
+  res.setHeader('Cache-Control', 'no-cache');
+  res.setHeader('Connection', 'keep-alive');
+  res.setHeader('Access-Control-Allow-Origin', '*');
+  res.flushHeaders();
+
+  // Helper to send SSE events
+  const sendEvent = (type, data) => {
+    res.write(\`data: \${JSON.stringify({ type, ...data })}\\n\\n\`);
+  };
+
+  // Capture beforeCommit
+  let beforeCommit = '';
+  try {
+    beforeCommit = execSync('git rev-parse HEAD', { cwd: ROOT }).toString().trim();
+    console.log('🔵 [prompt_agent_stream] beforeCommit:', beforeCommit);
+  } catch (e) {
+    console.log('⚠️ [prompt_agent_stream] Could not get beforeCommit:', e.message);
+  }
+
+  // Capture initial state of modified files
+  const initiallyModifiedFiles = new Set();
+  try {
+    const initialStatus = execSync('git status --short', { cwd: ROOT }).toString();
+    initialStatus.split('\\n').filter(Boolean).forEach(line => {
+      const filePath = line.substring(3).trim();
+      if (filePath) initiallyModifiedFiles.add(filePath);
+    });
+  } catch (e) {
+    // Ignore
+  }
+
+  // Send starting event
+  sendEvent('start', { beforeCommit });
+
+  // Set up file change tracking with real-time updates
+  const changedFiles = new Set();
+  const pollInterval = setInterval(() => {
+    try {
+      const status = execSync('git status --short', { cwd: ROOT }).toString();
+      status.split('\\n').filter(Boolean).forEach(line => {
+        const filePath = line.substring(3).trim();
+        if (filePath && !initiallyModifiedFiles.has(filePath)) {
+          if (!changedFiles.has(filePath)) {
+            changedFiles.add(filePath);
+            console.log('📁 [prompt_agent_stream] File changed:', filePath);
+            // Send real-time update to client
+            sendEvent('file_changed', { path: filePath });
+          }
+        }
+      });
+    } catch (e) {
+      // Ignore git status errors
+    }
+  }, 500);
+
+  const timeoutMs = parseInt(req.body.timeout) || 5 * 60 * 1000;
+  let timeoutId = null;
+  let responseSent = false;
+
+  const args = ['--print', '--force', '--workspace', '.', prompt];
+
+  console.log('🔵 [prompt_agent_stream] Spawning cursor-agent process...');
+  const proc = spawn(
+    'cursor-agent',
+    args,
+    {
+      cwd: ROOT,
+      env: process.env,
+      stdio: ['ignore', 'pipe', 'pipe']
+    }
+  );
+
+  console.log('🔵 [prompt_agent_stream] Process spawned, PID:', proc.pid);
+
+  let stdout = '';
+  let stderr = '';
+
+  timeoutId = setTimeout(() => {
+    if (!responseSent && proc && !proc.killed) {
+      console.log('⏱️ [prompt_agent_stream] Timeout reached');
+      clearInterval(pollInterval);
+      proc.kill('SIGTERM');
+      
+      setTimeout(() => {
+        if (!proc.killed) proc.kill('SIGKILL');
+      }, 5000);
+      
+      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();
+      }
+    }
+  }, timeoutMs);
+
+  proc.stdout.on('data', (d) => {
+    stdout += d.toString();
+  });
+
+  proc.stderr.on('data', (d) => {
+    stderr += d.toString();
+  });
+
+  proc.on('error', (error) => {
+    console.log('❌ [prompt_agent_stream] Process error:', error.message);
+    clearInterval(pollInterval);
+    if (timeoutId) clearTimeout(timeoutId);
+    if (!responseSent) {
+      responseSent = true;
+      sendEvent('error', { error: error.message });
+      res.end();
+    }
+  });
+
+  proc.on('close', (code, signal) => {
+    console.log('🔵 [prompt_agent_stream] Process closed with code:', code);
+    clearInterval(pollInterval);
+    if (timeoutId) clearTimeout(timeoutId);
+    
+    let afterCommit = '';
+    try {
+      afterCommit = execSync('git rev-parse HEAD', { cwd: ROOT }).toString().trim();
+    } catch (e) {
+      // Ignore
+    }
+    
+    if (!responseSent) {
+      responseSent = true;
+      sendEvent('complete', {
+        code,
+        stdout,
+        stderr,
+        changedFiles: Array.from(changedFiles),
+        beforeCommit,
+        afterCommit
+      });
+      res.end();
+    }
+  });
+
+  // Handle client disconnect - DON'T kill the process, let it complete
+  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;
+  });
+});
+
 // Revert job endpoint to reset to a previous commit
 app.post('/revert_job', (req, res) => {
   console.log('🔵 [revert_job] Endpoint hit');

package/package.json

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