codebot-ai 1.0.2 → 1.1.1

package/dist/tools/edit.js

@@ -36,9 +36,13 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.EditFileTool = void 0;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
+const os = __importStar(require("os"));
+// Undo snapshot directory
+const UNDO_DIR = path.join(os.homedir(), '.codebot', 'undo');
+const MAX_UNDO = 50;
 class EditFileTool {
     name = 'edit_file';
-    description = 'Edit a file by replacing an exact string match with new content. The old_string must appear exactly once in the file.';
+    description = 'Edit a file by replacing an exact string match with new content. The old_string must appear exactly once in the file. Shows a diff preview and creates an undo snapshot.';
     permission = 'prompt';
     parameters = {
         type: 'object',
@@ -73,9 +77,120 @@ class EditFileTool {
         if (count > 1) {
             throw new Error(`String found ${count} times in ${filePath}. Provide more surrounding context to make it unique.`);
         }
+        // Save undo snapshot
+        this.saveSnapshot(filePath, content);
         const updated = content.replace(oldStr, newStr);
         fs.writeFileSync(filePath, updated, 'utf-8');
-        return `Edited ${filePath} (1 replacement)`;
+        // Generate diff preview
+        const diff = this.generateDiff(oldStr, newStr, content, filePath);
+        return diff;
+    }
+    generateDiff(oldStr, newStr, content, filePath) {
+        const lines = content.split('\n');
+        const matchIdx = content.indexOf(oldStr);
+        const linesBefore = content.substring(0, matchIdx).split('\n');
+        const startLine = linesBefore.length;
+        const oldLines = oldStr.split('\n');
+        const newLines = newStr.split('\n');
+        let diff = `Edited ${filePath}\n`;
+        // Show context (2 lines before)
+        const contextStart = Math.max(0, startLine - 3);
+        for (let i = contextStart; i < startLine - 1; i++) {
+            diff += `  ${i + 1} │ ${lines[i]}\n`;
+        }
+        // Show removed lines
+        for (const line of oldLines) {
+            diff += `  - │ ${line}\n`;
+        }
+        // Show added lines
+        for (const line of newLines) {
+            diff += `  + │ ${line}\n`;
+        }
+        // Show context (2 lines after)
+        const endLine = startLine - 1 + oldLines.length;
+        for (let i = endLine; i < Math.min(lines.length, endLine + 2); i++) {
+            diff += `  ${i + 1} │ ${lines[i]}\n`;
+        }
+        return diff.trimEnd();
+    }
+    /** Save a snapshot for undo */
+    saveSnapshot(filePath, content) {
+        try {
+            fs.mkdirSync(UNDO_DIR, { recursive: true });
+            const manifest = this.loadManifest();
+            const entry = {
+                file: filePath,
+                timestamp: Date.now(),
+                snapshotFile: `${Date.now()}-${path.basename(filePath)}`,
+            };
+            // Write snapshot content
+            fs.writeFileSync(path.join(UNDO_DIR, entry.snapshotFile), content);
+            manifest.push(entry);
+            // Prune old snapshots
+            while (manifest.length > MAX_UNDO) {
+                const old = manifest.shift();
+                try {
+                    fs.unlinkSync(path.join(UNDO_DIR, old.snapshotFile));
+                }
+                catch { /* ok */ }
+            }
+            fs.writeFileSync(path.join(UNDO_DIR, 'manifest.json'), JSON.stringify(manifest, null, 2));
+        }
+        catch {
+            // Best-effort, don't fail the edit
+        }
+    }
+    loadManifest() {
+        try {
+            const raw = fs.readFileSync(path.join(UNDO_DIR, 'manifest.json'), 'utf-8');
+            return JSON.parse(raw);
+        }
+        catch {
+            return [];
+        }
+    }
+    /** Undo the last edit to a file. Returns result message. */
+    static undo(filePath) {
+        try {
+            const manifestPath = path.join(UNDO_DIR, 'manifest.json');
+            if (!fs.existsSync(manifestPath))
+                return 'No undo history available.';
+            const manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf-8'));
+            if (manifest.length === 0)
+                return 'No undo history available.';
+            // Find the entry to undo
+            let entry;
+            if (filePath) {
+                const resolved = path.resolve(filePath);
+                for (let i = manifest.length - 1; i >= 0; i--) {
+                    if (manifest[i].file === resolved) {
+                        entry = manifest.splice(i, 1)[0];
+                        break;
+                    }
+                }
+                if (!entry)
+                    return `No undo history for ${filePath}`;
+            }
+            else {
+                entry = manifest.pop();
+            }
+            // Restore the snapshot
+            const snapshotPath = path.join(UNDO_DIR, entry.snapshotFile);
+            if (!fs.existsSync(snapshotPath))
+                return 'Snapshot file missing.';
+            const content = fs.readFileSync(snapshotPath, 'utf-8');
+            fs.writeFileSync(entry.file, content, 'utf-8');
+            // Cleanup
+            try {
+                fs.unlinkSync(snapshotPath);
+            }
+            catch { /* ok */ }
+            fs.writeFileSync(manifestPath, JSON.stringify(manifest, null, 2));
+            return `Restored ${entry.file} to state before last edit.`;
+        }
+        catch (err) {
+            return `Undo failed: ${err instanceof Error ? err.message : String(err)}`;
+        }
     }
 }
 exports.EditFileTool = EditFileTool;

package/dist/tools/execute.js

@@ -3,14 +3,42 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ExecuteTool = void 0;
 const child_process_1 = require("child_process");
 const BLOCKED_PATTERNS = [
+    // Destructive filesystem operations
     /rm\s+-rf\s+\//,
     /rm\s+-rf\s+~/,
     /rm\s+-rf\s+\*/,
+    /rm\s+-rf\s+\.\s/,
+    /rm\s+(-[a-z]*f[a-z]*\s+)?--no-preserve-root/,
+    // Disk/partition destruction
     /mkfs\./,
     /dd\s+if=.*of=\/dev\//,
+    /wipefs/,
+    /fdisk\s+\/dev\//,
+    /parted\s+\/dev\//,
+    // Fork bomb
     /:\(\)\s*\{[^}]*:\|:.*\}/,
+    // Permission escalation
     /chmod\s+-R\s+777\s+\//,
+    /chmod\s+777\s+\//,
+    /chown\s+-R\s+.*\s+\//,
+    // Windows destructive
     /format\s+c:/i,
+    /del\s+\/[sfq]\s+c:\\/i,
+    /rd\s+\/s\s+\/q\s+c:\\/i,
+    // Curl to shell pipes (common attack vector)
+    /curl\s+.*\|\s*(ba)?sh/,
+    /wget\s+.*\|\s*(ba)?sh/,
+    // History/log destruction
+    />\s*\/dev\/sda/,
+    /history\s+-c.*&&.*rm/,
+    // Shutdown/reboot
+    /shutdown\s+(-h\s+)?now/,
+    /reboot\b/,
+    /init\s+[06]/,
+    // Kernel module manipulation
+    /insmod\b/,
+    /rmmod\b/,
+    /modprobe\s+-r/,
 ];
 class ExecuteTool {
     name = 'execute';

package/dist/tools/index.d.ts

@@ -1,4 +1,5 @@
 import { Tool, ToolSchema } from '../types';
+export { EditFileTool } from './edit';
 export declare class ToolRegistry {
     private tools;
     constructor(projectRoot?: string);

package/dist/tools/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ToolRegistry = void 0;
+exports.ToolRegistry = exports.EditFileTool = void 0;
 const read_1 = require("./read");
 const write_1 = require("./write");
 const edit_1 = require("./edit");
@@ -11,12 +11,16 @@ const think_1 = require("./think");
 const memory_1 = require("./memory");
 const web_fetch_1 = require("./web-fetch");
 const browser_1 = require("./browser");
+const batch_edit_1 = require("./batch-edit");
+var edit_2 = require("./edit");
+Object.defineProperty(exports, "EditFileTool", { enumerable: true, get: function () { return edit_2.EditFileTool; } });
 class ToolRegistry {
     tools = new Map();
     constructor(projectRoot) {
         this.register(new read_1.ReadFileTool());
         this.register(new write_1.WriteFileTool());
         this.register(new edit_1.EditFileTool());
+        this.register(new batch_edit_1.BatchEditTool());
         this.register(new execute_1.ExecuteTool());
         this.register(new glob_1.GlobTool());
         this.register(new grep_1.GrepTool());

package/dist/tools/write.d.ts

@@ -18,5 +18,6 @@ export declare class WriteFileTool implements Tool {
         required: string[];
     };
     execute(args: Record<string, unknown>): Promise<string>;
+    private saveSnapshot;
 }
 //# sourceMappingURL=write.d.ts.map

package/dist/tools/write.js

@@ -36,9 +36,11 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.WriteFileTool = void 0;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
+const os = __importStar(require("os"));
+const UNDO_DIR = path.join(os.homedir(), '.codebot', 'undo');
 class WriteFileTool {
     name = 'write_file';
-    description = 'Create a new file or overwrite an existing file with the given content.';
+    description = 'Create a new file or overwrite an existing file with the given content. Automatically saves an undo snapshot for existing files.';
     permission = 'prompt';
     parameters = {
         type: 'object',
@@ -62,10 +64,45 @@ class WriteFileTool {
             fs.mkdirSync(dir, { recursive: true });
         }
         const existed = fs.existsSync(filePath);
+        // Save undo snapshot before overwriting
+        if (existed) {
+            try {
+                const oldContent = fs.readFileSync(filePath, 'utf-8');
+                this.saveSnapshot(filePath, oldContent);
+            }
+            catch { /* best effort */ }
+        }
         fs.writeFileSync(filePath, content, 'utf-8');
         const lines = content.split('\n').length;
         return `${existed ? 'Overwrote' : 'Created'} ${filePath} (${lines} lines, ${content.length} bytes)`;
     }
+    saveSnapshot(filePath, content) {
+        try {
+            fs.mkdirSync(UNDO_DIR, { recursive: true });
+            const manifestPath = path.join(UNDO_DIR, 'manifest.json');
+            let manifest = [];
+            try {
+                manifest = JSON.parse(fs.readFileSync(manifestPath, 'utf-8'));
+            }
+            catch { /* empty */ }
+            const entry = {
+                file: filePath,
+                timestamp: Date.now(),
+                snapshotFile: `${Date.now()}-${path.basename(filePath)}`,
+            };
+            fs.writeFileSync(path.join(UNDO_DIR, entry.snapshotFile), content);
+            manifest.push(entry);
+            while (manifest.length > 50) {
+                const old = manifest.shift();
+                try {
+                    fs.unlinkSync(path.join(UNDO_DIR, old.snapshotFile));
+                }
+                catch { /* ok */ }
+            }
+            fs.writeFileSync(manifestPath, JSON.stringify(manifest, null, 2));
+        }
+        catch { /* best effort */ }
+    }
 }
 exports.WriteFileTool = WriteFileTool;
 //# sourceMappingURL=write.js.map

package/package.json

@@ -1,11 +1,11 @@
 {
   "name": "codebot-ai",
-  "version": "1.0.2",
+  "version": "1.1.1",
   "description": "Local-first AI coding assistant. Zero dependencies. Works with Ollama, LM Studio, vLLM, Claude, GPT, Gemini, and more.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "bin": {
-    "codebot": "./bin/codebot"
+    "codebot": "bin/codebot"
   },
   "scripts": {
     "build": "tsc",
@@ -33,11 +33,11 @@
   "license": "MIT",
   "repository": {
     "type": "git",
-    "url": "git+https://github.com/AscendralSoftware/codebot-ai.git"
+    "url": "git+https://github.com/zanderone1980/codebot-ai.git"
   },
-  "homepage": "https://github.com/AscendralSoftware/codebot-ai#readme",
+  "homepage": "https://github.com/zanderone1980/codebot-ai#readme",
   "bugs": {
-    "url": "https://github.com/AscendralSoftware/codebot-ai/issues"
+    "url": "https://github.com/zanderone1980/codebot-ai/issues"
   },
   "engines": {
     "node": ">=18.0.0"