devbonzai 2.0.3 → 2.0.4

package/cli.js

@@ -64,7 +64,7 @@ const express = require('./node_modules/express');
 const cors = require('./node_modules/cors');
 const fs = require('fs');
 const path = require('path');
-const { exec, spawn } = require('child_process');
+const { exec, spawn, execSync } = require('child_process');
 let babelParser = null;
 try {
   babelParser = require('./node_modules/@babel/parser');
@@ -92,6 +92,7 @@ app.get('/', (req, res) => {
       'POST /move': 'Move file or folder (body: {source, destination})',
       'POST /open-cursor': 'Open Cursor (body: {path, line?})',
       'POST /prompt_agent': 'Execute cursor-agent command (body: {prompt})',
+      'POST /revert_job': 'Revert to a previous commit (body: {beforeCommit})',
       'POST /shutdown': 'Gracefully shutdown the server'
     },
     example: 'Try: /list or /read?path=README.md'
@@ -1284,6 +1285,32 @@ app.post('/prompt_agent', (req, res) => {
     return res.status(400).json({ error: 'prompt required' });
   }
 
+  // Capture beforeCommit
+  let beforeCommit = '';
+  try {
+    beforeCommit = execSync('git rev-parse HEAD', { cwd: ROOT }).toString().trim();
+    console.log('🔵 [prompt_agent] beforeCommit:', beforeCommit);
+  } catch (e) {
+    console.log('⚠️ [prompt_agent] Could not get beforeCommit:', e.message);
+  }
+
+  // Set up file change tracking
+  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(); // Remove status prefix (XY + space)
+        if (filePath) changedFiles.add(filePath);
+      });
+      if (changedFiles.size > 0) {
+        console.log('📁 [prompt_agent] Changed files:', Array.from(changedFiles));
+      }
+    } catch (e) {
+      // Ignore git status errors
+    }
+  }, 500);
+
   // Configurable timeout (default 5 minutes)
   const timeoutMs = parseInt(req.body.timeout) || 5 * 60 * 1000;
   let timeoutId = null;
@@ -1312,6 +1339,7 @@ app.post('/prompt_agent', (req, res) => {
   timeoutId = setTimeout(() => {
     if (!responseSent && proc && !proc.killed) {
       console.log('⏱️  [prompt_agent] Timeout reached, killing process...');
+      clearInterval(pollInterval);
       proc.kill('SIGTERM');
       
       // Force kill after a short grace period if SIGTERM doesn't work
@@ -1329,7 +1357,10 @@ app.post('/prompt_agent', (req, res) => {
           message: \`cursor-agent exceeded timeout of \${timeoutMs / 1000} seconds\`,
           code: -1,
           stdout,
-          stderr
+          stderr,
+          changedFiles: Array.from(changedFiles),
+          beforeCommit,
+          afterCommit: ''
         });
       }
     }
@@ -1349,6 +1380,7 @@ app.post('/prompt_agent', (req, res) => {
 
   proc.on('error', (error) => {
     console.log('❌ [prompt_agent] Process error:', error.message);
+    clearInterval(pollInterval);
     if (timeoutId) clearTimeout(timeoutId);
     if (!responseSent) {
       responseSent = true;
@@ -1361,8 +1393,19 @@ app.post('/prompt_agent', (req, res) => {
     console.log('🔵 [prompt_agent] stdout length:', stdout.length);
     console.log('🔵 [prompt_agent] stderr length:', stderr.length);
     
+    // Stop polling for file changes
+    clearInterval(pollInterval);
     if (timeoutId) clearTimeout(timeoutId);
     
+    // Capture afterCommit
+    let afterCommit = '';
+    try {
+      afterCommit = execSync('git rev-parse HEAD', { cwd: ROOT }).toString().trim();
+      console.log('🔵 [prompt_agent] afterCommit:', afterCommit);
+    } catch (e) {
+      console.log('⚠️ [prompt_agent] Could not get afterCommit:', e.message);
+    }
+    
     if (!responseSent) {
       responseSent = true;
       // Check if process was killed due to timeout
@@ -1372,19 +1415,52 @@ app.post('/prompt_agent', (req, res) => {
           message: signal === 'SIGTERM' ? 'Process was terminated due to timeout' : 'Process was force killed',
           code: code || -1,
           stdout,
-          stderr
+          stderr,
+          changedFiles: Array.from(changedFiles),
+          beforeCommit,
+          afterCommit
         });
       } else {
         res.json({
           code,
           stdout,
-          stderr
+          stderr,
+          changedFiles: Array.from(changedFiles),
+          beforeCommit,
+          afterCommit
         });
       }
     }
   });
 });
 
+// Revert job endpoint to reset to a previous commit
+app.post('/revert_job', (req, res) => {
+  console.log('🔵 [revert_job] Endpoint hit');
+  const { beforeCommit } = req.body;
+  
+  if (!beforeCommit || typeof beforeCommit !== 'string') {
+    console.log('❌ [revert_job] Error: beforeCommit required');
+    return res.status(400).json({ error: 'beforeCommit required' });
+  }
+  
+  // Validate commit hash format (basic sanitization to prevent command injection)
+  if (!/^[a-f0-9]{7,40}$/i.test(beforeCommit)) {
+    console.log('❌ [revert_job] Error: invalid commit hash format');
+    return res.status(400).json({ error: 'Invalid commit hash format' });
+  }
+  
+  try {
+    console.log('🔵 [revert_job] Resetting to commit:', beforeCommit);
+    execSync(\`git reset --hard \${beforeCommit}\`, { cwd: ROOT });
+    console.log('✅ [revert_job] Successfully reverted to commit:', beforeCommit);
+    res.json({ success: true });
+  } catch (e) {
+    console.log('❌ [revert_job] Error:', e.message);
+    res.status(500).json({ error: e.message });
+  }
+});
+
 // Shutdown endpoint to kill the server
 app.post('/shutdown', (req, res) => {
   console.log('🛑 Shutdown endpoint called - terminating server...');

package/package.json

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