openvibe 0.63.2 → 0.63.3

package/dist/core/tools/batch-write.js

@@ -0,0 +1,267 @@
+import { Type } from "@sinclair/typebox";
+import { existsSync, mkdirSync, readFileSync, unlinkSync, writeFileSync } from "fs";
+import { homedir } from "os";
+import { basename, dirname, resolve } from "path";
+const BatchWriteFileSchema = Type.Object({
+    path: Type.String({ description: "File path to write" }),
+    content: Type.String({ description: "File content" }),
+});
+const BatchWriteOptionsSchema = Type.Object({
+    atomic: Type.Optional(Type.Boolean({ description: "Enable atomic write with rollback", default: true })),
+    backup: Type.Optional(Type.Boolean({ description: "Backup existing files", default: true })),
+    overwrite: Type.Optional(Type.Boolean({ description: "Overwrite existing files", default: true })),
+    createDirs: Type.Optional(Type.Boolean({ description: "Create parent directories", default: true })),
+    encoding: Type.Optional(Type.String({ description: "File encoding", default: "utf-8" })),
+});
+const BatchWriteSchema = Type.Object({
+    operation: Type.Literal("batch_write"),
+    files: Type.Array(BatchWriteFileSchema, { description: "Files to write" }),
+    options: Type.Optional(BatchWriteOptionsSchema),
+});
+function getBackupDir() {
+    const backupBase = resolve(homedir(), ".openvibe", "backups");
+    if (!existsSync(backupBase)) {
+        mkdirSync(backupBase, { recursive: true });
+    }
+    return backupBase;
+}
+function createBackup(filePath) {
+    if (!existsSync(filePath)) {
+        return null;
+    }
+    const backupDir = getBackupDir();
+    const timestamp = Date.now();
+    const fileName = basename(filePath);
+    const backupPath = resolve(backupDir, `${fileName}.${timestamp}.bak`);
+    try {
+        const content = readFileSync(filePath);
+        writeFileSync(backupPath, content);
+        return backupPath;
+    }
+    catch {
+        return null;
+    }
+}
+function restoreBackup(backupPath, originalPath) {
+    try {
+        if (!existsSync(backupPath)) {
+            return false;
+        }
+        const content = readFileSync(backupPath);
+        writeFileSync(originalPath, content);
+        unlinkSync(backupPath);
+        return true;
+    }
+    catch {
+        return false;
+    }
+}
+function cleanupBackup(backupPath) {
+    try {
+        if (existsSync(backupPath)) {
+            unlinkSync(backupPath);
+        }
+    }
+    catch { }
+}
+function writeSingleFile(filePath, content, options) {
+    const absolutePath = resolve(filePath);
+    const parentDir = dirname(absolutePath);
+    try {
+        const fileExists = existsSync(absolutePath);
+        if (fileExists && !options.overwrite) {
+            return {
+                path: absolutePath,
+                status: "skipped",
+                bytes: 0,
+            };
+        }
+        if (!existsSync(parentDir)) {
+            if (options.createDirs) {
+                mkdirSync(parentDir, { recursive: true });
+            }
+            else {
+                return {
+                    path: absolutePath,
+                    status: "failed",
+                    bytes: 0,
+                    error: `Parent directory does not exist: ${parentDir}`,
+                };
+            }
+        }
+        let backupPath;
+        if (fileExists && options.backup && options.atomic) {
+            const backup = createBackup(absolutePath);
+            if (backup) {
+                backupPath = backup;
+            }
+        }
+        writeFileSync(absolutePath, content, { encoding: options.encoding });
+        const bytes = Buffer.byteLength(content, options.encoding);
+        if (backupPath) {
+            cleanupBackup(backupPath);
+        }
+        return {
+            path: absolutePath,
+            status: fileExists ? "updated" : "created",
+            bytes,
+            backupPath,
+        };
+    }
+    catch (error) {
+        return {
+            path: absolutePath,
+            status: "failed",
+            bytes: 0,
+            error: error instanceof Error ? error.message : String(error),
+        };
+    }
+}
+function executeBatchWrite(input, onUpdate) {
+    const startTime = Date.now();
+    const options = {
+        atomic: input.options?.atomic ?? true,
+        backup: input.options?.backup ?? true,
+        overwrite: input.options?.overwrite ?? true,
+        createDirs: input.options?.createDirs ?? true,
+        encoding: input.options?.encoding ?? "utf-8",
+    };
+    const results = [];
+    const backups = new Map();
+    for (const file of input.files) {
+        const absolutePath = resolve(file.path);
+        if (existsSync(absolutePath) && options.backup && options.atomic) {
+            const backup = createBackup(absolutePath);
+            if (backup) {
+                backups.set(absolutePath, backup);
+            }
+        }
+    }
+    for (let i = 0; i < input.files.length; i++) {
+        const file = input.files[i];
+        const result = writeSingleFile(file.path, file.content, options);
+        results.push(result);
+        if (result.status === "created" || result.status === "updated") {
+            const backup = backups.get(result.path);
+            if (backup) {
+                cleanupBackup(backup);
+                backups.delete(result.path);
+            }
+        }
+        const statusIcon = result.status === "created"
+            ? "+"
+            : result.status === "updated"
+                ? "~"
+                : result.status === "skipped"
+                    ? "-"
+                    : "x";
+        onUpdate?.({
+            content: [
+                {
+                    type: "text",
+                    text: `[${statusIcon}] ${result.path} (${result.bytes} bytes)`,
+                },
+            ],
+            details: {
+                success: false,
+                total: input.files.length,
+                succeeded: results.filter((r) => r.status === "created" || r.status === "updated").length,
+                failed: results.filter((r) => r.status === "failed").length,
+                skipped: results.filter((r) => r.status === "skipped").length,
+                files: results,
+                duration: Date.now() - startTime,
+                currentIndex: i,
+                currentFile: result.path,
+            },
+        });
+    }
+    const failedCount = results.filter((r) => r.status === "failed").length;
+    if (failedCount > 0 && options.atomic) {
+        for (const [originalPath, backupPath] of backups) {
+            restoreBackup(backupPath, originalPath);
+        }
+        for (const result of results) {
+            if (result.status === "created") {
+                try {
+                    if (existsSync(result.path)) {
+                        unlinkSync(result.path);
+                    }
+                    result.status = "failed";
+                    result.error = "Rolled back due to batch failure";
+                }
+                catch { }
+            }
+            else if (result.status === "updated") {
+                const backup = backups.get(result.path);
+                if (backup) {
+                    restoreBackup(backup, result.path);
+                    result.status = "failed";
+                    result.error = "Rolled back due to batch failure";
+                }
+            }
+        }
+    }
+    const duration = Date.now() - startTime;
+    return {
+        success: failedCount === 0,
+        total: input.files.length,
+        succeeded: results.filter((r) => r.status === "created" || r.status === "updated").length,
+        failed: results.filter((r) => r.status === "failed").length,
+        skipped: results.filter((r) => r.status === "skipped").length,
+        files: results,
+        duration,
+    };
+}
+export const batchWriteTool = {
+    name: "batch_write",
+    label: "batch_write",
+    description: "Batch write multiple files with atomic guarantees. Automatically handles directory creation, backup, and rollback on failure.",
+    parameters: BatchWriteSchema,
+    execute: async (_id, input, _signal, onUpdate) => {
+        const result = executeBatchWrite(input, onUpdate);
+        const lines = [
+            `Batch Write Complete`,
+            `====================`,
+            `Total: ${result.total} | Success: ${result.succeeded} | Failed: ${result.failed} | Skipped: ${result.skipped}`,
+            `Duration: ${result.duration}ms`,
+        ];
+        return {
+            content: [
+                {
+                    type: "text",
+                    text: lines.join("\n"),
+                },
+            ],
+            details: result,
+        };
+    },
+};
+export function detectBatchWriteIntent(toolCalls) {
+    const writeCalls = toolCalls.filter((c) => c.name === "write" || c.name === "write_file" || c.name === "create_file");
+    if (writeCalls.length < 2) {
+        return null;
+    }
+    const files = [];
+    for (const call of writeCalls) {
+        const args = call.arguments;
+        const path = args.path || args.file_path;
+        const content = args.content;
+        if (path && content !== undefined) {
+            files.push({ path, content });
+        }
+    }
+    if (files.length < 2) {
+        return null;
+    }
+    return {
+        operation: "batch_write",
+        files,
+        options: {
+            atomic: true,
+            backup: true,
+            overwrite: true,
+            createDirs: true,
+        },
+    };
+}
+//# sourceMappingURL=batch-write.js.map

package/dist/core/tools/batch-write.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"batch-write.js","sourceRoot":"","sources":["../../../src/core/tools/batch-write.ts"],"names":[],"mappings":"AACA,OAAO,EAAe,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,YAAY,EAAE,UAAU,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACpF,OAAO,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAElD,MAAM,oBAAoB,GAAG,IAAI,CAAC,MAAM,CAAC;IACxC,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,oBAAoB,EAAE,CAAC;IACxD,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,cAAc,EAAE,CAAC;CACrD,CAAC,CAAC;AAEH,MAAM,uBAAuB,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3C,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,mCAAmC,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IACxG,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,uBAAuB,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5F,SAAS,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,0BAA0B,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IAClG,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,WAAW,EAAE,2BAA2B,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;IACpG,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;CACxF,CAAC,CAAC;AAEH,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC;IACpC,SAAS,EAAE,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC;IACtC,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,oBAAoB,EAAE,EAAE,WAAW,EAAE,gBAAgB,EAAE,CAAC;IAC1E,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,uBAAuB,CAAC;CAC/C,CAAC,CAAC;AAsBH,SAAS,YAAY,GAAW;IAC/B,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,EAAE,EAAE,WAAW,EAAE,SAAS,CAAC,CAAC;IAC9D,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7B,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC;IACD,OAAO,UAAU,CAAC;AAAA,CAClB;AAED,SAAS,YAAY,CAAC,QAAgB,EAAiB;IACtD,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IACjC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACpC,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,EAAE,GAAG,QAAQ,IAAI,SAAS,MAAM,CAAC,CAAC;IAEtE,IAAI,CAAC;QACJ,MAAM,OAAO,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACvC,aAAa,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;QACnC,OAAO,UAAU,CAAC;IACnB,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,IAAI,CAAC;IACb,CAAC;AAAA,CACD;AAED,SAAS,aAAa,CAAC,UAAkB,EAAE,YAAoB,EAAW;IACzE,IAAI,CAAC;QACJ,IAAI,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC7B,OAAO,KAAK,CAAC;QACd,CAAC;QACD,MAAM,OAAO,GAAG,YAAY,CAAC,UAAU,CAAC,CAAC;QACzC,aAAa,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QACrC,UAAU,CAAC,UAAU,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC;IACb,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,KAAK,CAAC;IACd,CAAC;AAAA,CACD;AAED,SAAS,aAAa,CAAC,UAAkB,EAAQ;IAChD,IAAI,CAAC;QACJ,IAAI,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YAC5B,UAAU,CAAC,UAAU,CAAC,CAAC;QACxB,CAAC;IACF,CAAC;IAAC,MAAM,CAAC,CAAA,CAAC;AAAA,CACV;AAED,SAAS,eAAe,CACvB,QAAgB,EAChB,OAAe,EACf,OAAyD,EAClC;IACvB,MAAM,YAAY,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IAExC,IAAI,CAAC;QACJ,MAAM,UAAU,GAAG,UAAU,CAAC,YAAY,CAAC,CAAC;QAE5C,IAAI,UAAU,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC;YACtC,OAAO;gBACN,IAAI,EAAE,YAAY;gBAClB,MAAM,EAAE,SAAS;gBACjB,KAAK,EAAE,CAAC;aACR,CAAC;QACH,CAAC;QAED,IAAI,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;YAC5B,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBACxB,SAAS,CAAC,SAAS,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3C,CAAC;iBAAM,CAAC;gBACP,OAAO;oBACN,IAAI,EAAE,YAAY;oBAClB,MAAM,EAAE,QAAQ;oBAChB,KAAK,EAAE,CAAC;oBACR,KAAK,EAAE,oCAAoC,SAAS,EAAE;iBACtD,CAAC;YACH,CAAC;QACF,CAAC;QAED,IAAI,UAA8B,CAAC;QAEnC,IAAI,UAAU,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACpD,MAAM,MAAM,GAAG,YAAY,CAAC,YAAY,CAAC,CAAC;YAC1C,IAAI,MAAM,EAAE,CAAC;gBACZ,UAAU,GAAG,MAAM,CAAC;YACrB,CAAC;QACF,CAAC;QAED,aAAa,CAAC,YAAY,EAAE,OAAO,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,QAA0B,EAAE,CAAC,CAAC;QACvF,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,QAA0B,CAAC,CAAC;QAE7E,IAAI,UAAU,EAAE,CAAC;YAChB,aAAa,CAAC,UAAU,CAAC,CAAC;QAC3B,CAAC;QAED,OAAO;YACN,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;YAC1C,KAAK;YACL,UAAU;SACV,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,OAAO;YACN,IAAI,EAAE,YAAY;YAClB,MAAM,EAAE,QAAQ;YAChB,KAAK,EAAE,CAAC;YACR,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;SAC7D,CAAC;IACH,CAAC;AAAA,CACD;AAED,SAAS,iBAAiB,CACzB,KAAsB,EACtB,QAAoD,EACjC;IACnB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC7B,MAAM,OAAO,GAAqD;QACjE,MAAM,EAAE,KAAK,CAAC,OAAO,EAAE,MAAM,IAAI,IAAI;QACrC,MAAM,EAAE,KAAK,CAAC,OAAO,EAAE,MAAM,IAAI,IAAI;QACrC,SAAS,EAAE,KAAK,CAAC,OAAO,EAAE,SAAS,IAAI,IAAI;QAC3C,UAAU,EAAE,KAAK,CAAC,OAAO,EAAE,UAAU,IAAI,IAAI;QAC7C,QAAQ,EAAE,KAAK,CAAC,OAAO,EAAE,QAAQ,IAAI,OAAO;KAC5C,CAAC;IAEF,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,MAAM,OAAO,GAAG,IAAI,GAAG,EAAkB,CAAC;IAE1C,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,KAAK,EAAE,CAAC;QAChC,MAAM,YAAY,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAExC,IAAI,UAAU,CAAC,YAAY,CAAC,IAAI,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YAClE,MAAM,MAAM,GAAG,YAAY,CAAC,YAAY,CAAC,CAAC;YAC1C,IAAI,MAAM,EAAE,CAAC;gBACZ,OAAO,CAAC,GAAG,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;YACnC,CAAC;QACF,CAAC;IACF,CAAC;IAED,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7C,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,MAAM,GAAG,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACjE,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAErB,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;YAChE,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YACxC,IAAI,MAAM,EAAE,CAAC;gBACZ,aAAa,CAAC,MAAM,CAAC,CAAC;gBACtB,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC7B,CAAC;QACF,CAAC;QAED,MAAM,UAAU,GACf,MAAM,CAAC,MAAM,KAAK,SAAS;YAC1B,CAAC,CAAC,GAAG;YACL,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS;gBAC5B,CAAC,CAAC,GAAG;gBACL,CAAC,CAAC,MAAM,CAAC,MAAM,KAAK,SAAS;oBAC5B,CAAC,CAAC,GAAG;oBACL,CAAC,CAAC,GAAG,CAAC;QAEV,QAAQ,EAAE,CAAC;YACV,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,IAAI,UAAU,KAAK,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC,KAAK,SAAS;iBAC9D;aACD;YACD,OAAO,EAAE;gBACR,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM;gBACzB,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,MAAM;gBACzF,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,MAAM;gBAC3D,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,MAAM;gBAC7D,KAAK,EAAE,OAAO;gBACd,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;gBAChC,YAAY,EAAE,CAAC;gBACf,WAAW,EAAE,MAAM,CAAC,IAAI;aACxB;SACD,CAAC,CAAC;IACJ,CAAC;IAED,MAAM,WAAW,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,MAAM,CAAC;IAExE,IAAI,WAAW,GAAG,CAAC,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;QACvC,KAAK,MAAM,CAAC,YAAY,EAAE,UAAU,CAAC,IAAI,OAAO,EAAE,CAAC;YAClD,aAAa,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;QACzC,CAAC;QAED,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC9B,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACjC,IAAI,CAAC;oBACJ,IAAI,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC7B,UAAU,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;oBACzB,CAAC;oBACD,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC;oBACzB,MAAM,CAAC,KAAK,GAAG,kCAAkC,CAAC;gBACnD,CAAC;gBAAC,MAAM,CAAC,CAAA,CAAC;YACX,CAAC;iBAAM,IAAI,MAAM,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;gBACxC,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;gBACxC,IAAI,MAAM,EAAE,CAAC;oBACZ,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;oBACnC,MAAM,CAAC,MAAM,GAAG,QAAQ,CAAC;oBACzB,MAAM,CAAC,KAAK,GAAG,kCAAkC,CAAC;gBACnD,CAAC;YACF,CAAC;QACF,CAAC;IACF,CAAC;IAED,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;IAExC,OAAO;QACN,OAAO,EAAE,WAAW,KAAK,CAAC;QAC1B,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,MAAM;QACzB,SAAS,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,MAAM;QACzF,MAAM,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,MAAM;QAC3D,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,MAAM;QAC7D,KAAK,EAAE,OAAO;QACd,QAAQ;KACR,CAAC;AAAA,CACF;AAED,MAAM,CAAC,MAAM,cAAc,GAAuC;IACjE,IAAI,EAAE,aAAa;IACnB,KAAK,EAAE,aAAa;IACpB,WAAW,EACV,+HAA+H;IAChI,UAAU,EAAE,gBAAgB;IAC5B,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,CAAC;QACjD,MAAM,MAAM,GAAG,iBAAiB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAElD,MAAM,KAAK,GAAa;YACvB,sBAAsB;YACtB,sBAAsB;YACtB,UAAU,MAAM,CAAC,KAAK,eAAe,MAAM,CAAC,SAAS,cAAc,MAAM,CAAC,MAAM,eAAe,MAAM,CAAC,OAAO,EAAE;YAC/G,aAAa,MAAM,CAAC,QAAQ,IAAI;SAChC,CAAC;QAEF,OAAO;YACN,OAAO,EAAE;gBACR;oBACC,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC;iBACtB;aACD;YACD,OAAO,EAAE,MAAM;SACf,CAAC;IAAA,CACF;CACD,CAAC;AAEF,MAAM,UAAU,sBAAsB,CAAC,SAAsD,EAA0B;IACtH,MAAM,UAAU,GAAG,SAAS,CAAC,MAAM,CAClC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,OAAO,IAAI,CAAC,CAAC,IAAI,KAAK,YAAY,IAAI,CAAC,CAAC,IAAI,KAAK,aAAa,CAChF,CAAC;IAEF,IAAI,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC;IACb,CAAC;IAED,MAAM,KAAK,GAA6C,EAAE,CAAC;IAE3D,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAoE,CAAC;QACvF,MAAM,IAAI,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,SAAS,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC;QAE7B,IAAI,IAAI,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;YACnC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,CAAC;QAC/B,CAAC;IACF,CAAC;IAED,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,OAAO,IAAI,CAAC;IACb,CAAC;IAED,OAAO;QACN,SAAS,EAAE,aAAa;QACxB,KAAK;QACL,OAAO,EAAE;YACR,MAAM,EAAE,IAAI;YACZ,MAAM,EAAE,IAAI;YACZ,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,IAAI;SAChB;KACD,CAAC;AAAA,CACF","sourcesContent":["import type { AgentTool, AgentToolUpdateCallback } from \"@boxiaolanya2008/pi-agent-core\";\nimport { type Static, Type } from \"@sinclair/typebox\";\nimport { existsSync, mkdirSync, readFileSync, unlinkSync, writeFileSync } from \"fs\";\nimport { homedir } from \"os\";\nimport { basename, dirname, resolve } from \"path\";\n\nconst BatchWriteFileSchema = Type.Object({\n\tpath: Type.String({ description: \"File path to write\" }),\n\tcontent: Type.String({ description: \"File content\" }),\n});\n\nconst BatchWriteOptionsSchema = Type.Object({\n\tatomic: Type.Optional(Type.Boolean({ description: \"Enable atomic write with rollback\", default: true })),\n\tbackup: Type.Optional(Type.Boolean({ description: \"Backup existing files\", default: true })),\n\toverwrite: Type.Optional(Type.Boolean({ description: \"Overwrite existing files\", default: true })),\n\tcreateDirs: Type.Optional(Type.Boolean({ description: \"Create parent directories\", default: true })),\n\tencoding: Type.Optional(Type.String({ description: \"File encoding\", default: \"utf-8\" })),\n});\n\nconst BatchWriteSchema = Type.Object({\n\toperation: Type.Literal(\"batch_write\"),\n\tfiles: Type.Array(BatchWriteFileSchema, { description: \"Files to write\" }),\n\toptions: Type.Optional(BatchWriteOptionsSchema),\n});\n\nexport type BatchWriteInput = Static<typeof BatchWriteSchema>;\nexport type BatchWriteFileResult = {\n\tpath: string;\n\tstatus: \"created\" | \"updated\" | \"skipped\" | \"failed\";\n\tbytes: number;\n\tbackupPath?: string;\n\terror?: string;\n};\nexport type BatchWriteResult = {\n\tsuccess: boolean;\n\ttotal: number;\n\tsucceeded: number;\n\tfailed: number;\n\tskipped: number;\n\tfiles: BatchWriteFileResult[];\n\tduration: number;\n\tcurrentIndex?: number;\n\tcurrentFile?: string;\n};\n\nfunction getBackupDir(): string {\n\tconst backupBase = resolve(homedir(), \".openvibe\", \"backups\");\n\tif (!existsSync(backupBase)) {\n\t\tmkdirSync(backupBase, { recursive: true });\n\t}\n\treturn backupBase;\n}\n\nfunction createBackup(filePath: string): string | null {\n\tif (!existsSync(filePath)) {\n\t\treturn null;\n\t}\n\n\tconst backupDir = getBackupDir();\n\tconst timestamp = Date.now();\n\tconst fileName = basename(filePath);\n\tconst backupPath = resolve(backupDir, `${fileName}.${timestamp}.bak`);\n\n\ttry {\n\t\tconst content = readFileSync(filePath);\n\t\twriteFileSync(backupPath, content);\n\t\treturn backupPath;\n\t} catch {\n\t\treturn null;\n\t}\n}\n\nfunction restoreBackup(backupPath: string, originalPath: string): boolean {\n\ttry {\n\t\tif (!existsSync(backupPath)) {\n\t\t\treturn false;\n\t\t}\n\t\tconst content = readFileSync(backupPath);\n\t\twriteFileSync(originalPath, content);\n\t\tunlinkSync(backupPath);\n\t\treturn true;\n\t} catch {\n\t\treturn false;\n\t}\n}\n\nfunction cleanupBackup(backupPath: string): void {\n\ttry {\n\t\tif (existsSync(backupPath)) {\n\t\t\tunlinkSync(backupPath);\n\t\t}\n\t} catch {}\n}\n\nfunction writeSingleFile(\n\tfilePath: string,\n\tcontent: string,\n\toptions: Required<Static<typeof BatchWriteOptionsSchema>>,\n): BatchWriteFileResult {\n\tconst absolutePath = resolve(filePath);\n\tconst parentDir = dirname(absolutePath);\n\n\ttry {\n\t\tconst fileExists = existsSync(absolutePath);\n\n\t\tif (fileExists && !options.overwrite) {\n\t\t\treturn {\n\t\t\t\tpath: absolutePath,\n\t\t\t\tstatus: \"skipped\",\n\t\t\t\tbytes: 0,\n\t\t\t};\n\t\t}\n\n\t\tif (!existsSync(parentDir)) {\n\t\t\tif (options.createDirs) {\n\t\t\t\tmkdirSync(parentDir, { recursive: true });\n\t\t\t} else {\n\t\t\t\treturn {\n\t\t\t\t\tpath: absolutePath,\n\t\t\t\t\tstatus: \"failed\",\n\t\t\t\t\tbytes: 0,\n\t\t\t\t\terror: `Parent directory does not exist: ${parentDir}`,\n\t\t\t\t};\n\t\t\t}\n\t\t}\n\n\t\tlet backupPath: string | undefined;\n\n\t\tif (fileExists && options.backup && options.atomic) {\n\t\t\tconst backup = createBackup(absolutePath);\n\t\t\tif (backup) {\n\t\t\t\tbackupPath = backup;\n\t\t\t}\n\t\t}\n\n\t\twriteFileSync(absolutePath, content, { encoding: options.encoding as BufferEncoding });\n\t\tconst bytes = Buffer.byteLength(content, options.encoding as BufferEncoding);\n\n\t\tif (backupPath) {\n\t\t\tcleanupBackup(backupPath);\n\t\t}\n\n\t\treturn {\n\t\t\tpath: absolutePath,\n\t\t\tstatus: fileExists ? \"updated\" : \"created\",\n\t\t\tbytes,\n\t\t\tbackupPath,\n\t\t};\n\t} catch (error) {\n\t\treturn {\n\t\t\tpath: absolutePath,\n\t\t\tstatus: \"failed\",\n\t\t\tbytes: 0,\n\t\t\terror: error instanceof Error ? error.message : String(error),\n\t\t};\n\t}\n}\n\nfunction executeBatchWrite(\n\tinput: BatchWriteInput,\n\tonUpdate?: AgentToolUpdateCallback<BatchWriteResult>,\n): BatchWriteResult {\n\tconst startTime = Date.now();\n\tconst options: Required<Static<typeof BatchWriteOptionsSchema>> = {\n\t\tatomic: input.options?.atomic ?? true,\n\t\tbackup: input.options?.backup ?? true,\n\t\toverwrite: input.options?.overwrite ?? true,\n\t\tcreateDirs: input.options?.createDirs ?? true,\n\t\tencoding: input.options?.encoding ?? \"utf-8\",\n\t};\n\n\tconst results: BatchWriteFileResult[] = [];\n\tconst backups = new Map<string, string>();\n\n\tfor (const file of input.files) {\n\t\tconst absolutePath = resolve(file.path);\n\n\t\tif (existsSync(absolutePath) && options.backup && options.atomic) {\n\t\t\tconst backup = createBackup(absolutePath);\n\t\t\tif (backup) {\n\t\t\t\tbackups.set(absolutePath, backup);\n\t\t\t}\n\t\t}\n\t}\n\n\tfor (let i = 0; i < input.files.length; i++) {\n\t\tconst file = input.files[i];\n\t\tconst result = writeSingleFile(file.path, file.content, options);\n\t\tresults.push(result);\n\n\t\tif (result.status === \"created\" || result.status === \"updated\") {\n\t\t\tconst backup = backups.get(result.path);\n\t\t\tif (backup) {\n\t\t\t\tcleanupBackup(backup);\n\t\t\t\tbackups.delete(result.path);\n\t\t\t}\n\t\t}\n\n\t\tconst statusIcon =\n\t\t\tresult.status === \"created\"\n\t\t\t\t? \"+\"\n\t\t\t\t: result.status === \"updated\"\n\t\t\t\t\t? \"~\"\n\t\t\t\t\t: result.status === \"skipped\"\n\t\t\t\t\t\t? \"-\"\n\t\t\t\t\t\t: \"x\";\n\n\t\tonUpdate?.({\n\t\t\tcontent: [\n\t\t\t\t{\n\t\t\t\t\ttype: \"text\",\n\t\t\t\t\ttext: `[${statusIcon}] ${result.path} (${result.bytes} bytes)`,\n\t\t\t\t},\n\t\t\t],\n\t\t\tdetails: {\n\t\t\t\tsuccess: false,\n\t\t\t\ttotal: input.files.length,\n\t\t\t\tsucceeded: results.filter((r) => r.status === \"created\" || r.status === \"updated\").length,\n\t\t\t\tfailed: results.filter((r) => r.status === \"failed\").length,\n\t\t\t\tskipped: results.filter((r) => r.status === \"skipped\").length,\n\t\t\t\tfiles: results,\n\t\t\t\tduration: Date.now() - startTime,\n\t\t\t\tcurrentIndex: i,\n\t\t\t\tcurrentFile: result.path,\n\t\t\t},\n\t\t});\n\t}\n\n\tconst failedCount = results.filter((r) => r.status === \"failed\").length;\n\n\tif (failedCount > 0 && options.atomic) {\n\t\tfor (const [originalPath, backupPath] of backups) {\n\t\t\trestoreBackup(backupPath, originalPath);\n\t\t}\n\n\t\tfor (const result of results) {\n\t\t\tif (result.status === \"created\") {\n\t\t\t\ttry {\n\t\t\t\t\tif (existsSync(result.path)) {\n\t\t\t\t\t\tunlinkSync(result.path);\n\t\t\t\t\t}\n\t\t\t\t\tresult.status = \"failed\";\n\t\t\t\t\tresult.error = \"Rolled back due to batch failure\";\n\t\t\t\t} catch {}\n\t\t\t} else if (result.status === \"updated\") {\n\t\t\t\tconst backup = backups.get(result.path);\n\t\t\t\tif (backup) {\n\t\t\t\t\trestoreBackup(backup, result.path);\n\t\t\t\t\tresult.status = \"failed\";\n\t\t\t\t\tresult.error = \"Rolled back due to batch failure\";\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\t}\n\n\tconst duration = Date.now() - startTime;\n\n\treturn {\n\t\tsuccess: failedCount === 0,\n\t\ttotal: input.files.length,\n\t\tsucceeded: results.filter((r) => r.status === \"created\" || r.status === \"updated\").length,\n\t\tfailed: results.filter((r) => r.status === \"failed\").length,\n\t\tskipped: results.filter((r) => r.status === \"skipped\").length,\n\t\tfiles: results,\n\t\tduration,\n\t};\n}\n\nexport const batchWriteTool: AgentTool<typeof BatchWriteSchema> = {\n\tname: \"batch_write\",\n\tlabel: \"batch_write\",\n\tdescription:\n\t\t\"Batch write multiple files with atomic guarantees. Automatically handles directory creation, backup, and rollback on failure.\",\n\tparameters: BatchWriteSchema,\n\texecute: async (_id, input, _signal, onUpdate) => {\n\t\tconst result = executeBatchWrite(input, onUpdate);\n\n\t\tconst lines: string[] = [\n\t\t\t`Batch Write Complete`,\n\t\t\t`====================`,\n\t\t\t`Total: ${result.total} | Success: ${result.succeeded} | Failed: ${result.failed} | Skipped: ${result.skipped}`,\n\t\t\t`Duration: ${result.duration}ms`,\n\t\t];\n\n\t\treturn {\n\t\t\tcontent: [\n\t\t\t\t{\n\t\t\t\t\ttype: \"text\",\n\t\t\t\t\ttext: lines.join(\"\\n\"),\n\t\t\t\t},\n\t\t\t],\n\t\t\tdetails: result,\n\t\t};\n\t},\n};\n\nexport function detectBatchWriteIntent(toolCalls: Array<{ name: string; arguments: unknown }>): BatchWriteInput | null {\n\tconst writeCalls = toolCalls.filter(\n\t\t(c) => c.name === \"write\" || c.name === \"write_file\" || c.name === \"create_file\",\n\t);\n\n\tif (writeCalls.length < 2) {\n\t\treturn null;\n\t}\n\n\tconst files: Array<{ path: string; content: string }> = [];\n\n\tfor (const call of writeCalls) {\n\t\tconst args = call.arguments as { path?: string; file_path?: string; content?: string };\n\t\tconst path = args.path || args.file_path;\n\t\tconst content = args.content;\n\n\t\tif (path && content !== undefined) {\n\t\t\tfiles.push({ path, content });\n\t\t}\n\t}\n\n\tif (files.length < 2) {\n\t\treturn null;\n\t}\n\n\treturn {\n\t\toperation: \"batch_write\",\n\t\tfiles,\n\t\toptions: {\n\t\t\tatomic: true,\n\t\t\tbackup: true,\n\t\t\toverwrite: true,\n\t\t\tcreateDirs: true,\n\t\t},\n\t};\n}\n"]}

package/dist/core/tools/edit.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"edit.d.ts","sourceRoot":"","sources":["../../../src/core/tools/edit.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,gCAAgC,CAAC;AAChE,OAAO,EAAE,KAAK,MAAM,EAAQ,MAAM,mBAAmB,CAAC;AAetD,QAAA,MAAM,UAAU;;;;EAId,CAAC;AACH,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,OAAO,UAAU,CAAC,CAAC;AACtD,MAAM,WAAW,eAAe;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC1B;AACD,MAAM,WAAW,cAAc;IAC9B,QAAQ,EAAE,CAAC,YAAY,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IACpD,SAAS,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpE,MAAM,EAAE,CAAC,YAAY,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAChD;AAMD,MAAM,WAAW,eAAe;IAC/B,UAAU,CAAC,EAAE,cAAc,CAAC;CAC5B;AACD,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC,OAAO,UAAU,CAAC,CAiInG;AACD,eAAO,MAAM,QAAQ;;;;QAAgC,CAAC","sourcesContent":["import type { AgentTool } from \"@boxiaolanya2008/pi-agent-core\";\nimport { type Static, Type } from \"@sinclair/typebox\";\nimport { constants } from \"fs\";\nimport { access as fsAccess } from \"fs/promises\";\nimport {\n\tdetectLineEnding,\n\tfuzzyFindText,\n\tgenerateDiffString,\n\tnormalizeForFuzzyMatch,\n\tnormalizeToLF,\n\trestoreLineEndings,\n\tstripBom,\n} from \"./edit-diff.js\";\nimport { acceleratedReadFile, acceleratedWriteFile } from \"./local-accelerator.js\";\nimport { resolveToCwd } from \"./path-utils.js\";\n\nconst editSchema = Type.Object({\n\tpath: Type.String({ description: \"Path to the file to edit (relative or absolute)\" }),\n\toldText: Type.String({ description: \"Exact text to find and replace (must match exactly)\" }),\n\tnewText: Type.String({ description: \"New text to replace the old text with\" }),\n});\nexport type EditToolInput = Static<typeof editSchema>;\nexport interface EditToolDetails {\n\tdiff: string;\n\tfirstChangedLine?: number;\n}\nexport interface EditOperations {\n\treadFile: (absolutePath: string) => Promise<Buffer>;\n\twriteFile: (absolutePath: string, content: string) => Promise<void>;\n\taccess: (absolutePath: string) => Promise<void>;\n}\nconst defaultEditOperations: EditOperations = {\n\treadFile: (path) => acceleratedReadFile(path),\n\twriteFile: (path, content) => acceleratedWriteFile(path, content),\n\taccess: (path) => fsAccess(path, constants.R_OK | constants.W_OK),\n};\nexport interface EditToolOptions {\n\toperations?: EditOperations;\n}\nexport function createEditTool(cwd: string, options?: EditToolOptions): AgentTool<typeof editSchema> {\n\tconst ops = options?.operations ?? defaultEditOperations;\n\treturn {\n\t\tname: \"edit\",\n\t\tlabel: \"edit\",\n\t\tdescription:\n\t\t\t\"Edit a file by replacing exact text. The oldText must match exactly (including whitespace). Use this for precise, surgical edits.\",\n\t\tparameters: editSchema,\n\t\texecute: async (\n\t\t\t_toolCallId: string,\n\t\t\t{ path, oldText, newText }: { path: string; oldText: string; newText: string },\n\t\t\tsignal?: AbortSignal,\n\t\t) => {\n\t\t\tconst absolutePath = resolveToCwd(path, cwd);\n\t\t\treturn new Promise<{\n\t\t\t\tcontent: Array<{ type: \"text\"; text: string }>;\n\t\t\t\tdetails: EditToolDetails | undefined;\n\t\t\t}>((resolve, reject) => {\n\t\t\t\tif (signal?.aborted) {\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tlet aborted = false;\n\t\t\t\tconst onAbort = () => {\n\t\t\t\t\taborted = true;\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t};\n\t\t\t\tif (signal) {\n\t\t\t\t\tsignal.addEventListener(\"abort\", onAbort, { once: true });\n\t\t\t\t}\n\t\t\t\t(async () => {\n\t\t\t\t\ttry {\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tawait ops.access(absolutePath);\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(new Error(`File not found: ${path}`));\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst buffer = await ops.readFile(absolutePath);\n\t\t\t\t\t\tconst rawContent = buffer.toString(\"utf-8\");\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst { bom, text: content } = stripBom(rawContent);\n\t\t\t\t\t\tconst originalEnding = detectLineEnding(content);\n\t\t\t\t\t\tconst normalizedContent = normalizeToLF(content);\n\t\t\t\t\t\tconst normalizedOldText = normalizeToLF(oldText);\n\t\t\t\t\t\tconst normalizedNewText = normalizeToLF(newText);\n\t\t\t\t\t\tconst matchResult = fuzzyFindText(normalizedContent, normalizedOldText);\n\t\t\t\t\t\tif (!matchResult.found) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Could not find the exact text in ${path}. The old text must match exactly including all whitespace and newlines.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst fuzzyContent = normalizeForFuzzyMatch(normalizedContent);\n\t\t\t\t\t\tconst fuzzyOldText = normalizeForFuzzyMatch(normalizedOldText);\n\t\t\t\t\t\tconst occurrences = fuzzyContent.split(fuzzyOldText).length - 1;\n\t\t\t\t\t\tif (occurrences > 1) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Found ${occurrences} occurrences of the text in ${path}. The text must be unique. Please provide more context to make it unique.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst baseContent = matchResult.contentForReplacement;\n\t\t\t\t\t\tconst newContent =\n\t\t\t\t\t\t\tbaseContent.substring(0, matchResult.index) +\n\t\t\t\t\t\t\tnormalizedNewText +\n\t\t\t\t\t\t\tbaseContent.substring(matchResult.index + matchResult.matchLength);\n\t\t\t\t\t\tif (baseContent === newContent) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`No changes made to ${path}. The replacement produced identical content. This might indicate an issue with special characters or the text not existing as expected.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst finalContent = bom + restoreLineEndings(newContent, originalEnding);\n\t\t\t\t\t\tawait ops.writeFile(absolutePath, finalContent);\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst diffResult = generateDiffString(baseContent, newContent);\n\t\t\t\t\t\tresolve({\n\t\t\t\t\t\t\tcontent: [\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\ttype: \"text\",\n\t\t\t\t\t\t\t\t\ttext: `Successfully replaced text in ${path}.`,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\tdetails: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },\n\t\t\t\t\t\t});\n\t\t\t\t\t} catch (error: any) {\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (!aborted) {\n\t\t\t\t\t\t\treject(error);\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t})();\n\t\t\t});\n\t\t},\n\t};\n}\nexport const editTool = createEditTool(process.cwd());\n"]}
+{"version":3,"file":"edit.d.ts","sourceRoot":"","sources":["../../../src/core/tools/edit.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAA2B,MAAM,gCAAgC,CAAC;AACzF,OAAO,EAAE,KAAK,MAAM,EAAQ,MAAM,mBAAmB,CAAC;AAetD,QAAA,MAAM,UAAU;;;;EAId,CAAC;AACH,MAAM,MAAM,aAAa,GAAG,MAAM,CAAC,OAAO,UAAU,CAAC,CAAC;AACtD,MAAM,WAAW,eAAe;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,gBAAgB,CAAC,EAAE,MAAM,CAAC;CAC1B;AACD,MAAM,WAAW,cAAc;IAC9B,QAAQ,EAAE,CAAC,YAAY,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;IACpD,SAAS,EAAE,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACpE,MAAM,EAAE,CAAC,YAAY,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CAChD;AAMD,MAAM,WAAW,eAAe;IAC/B,UAAU,CAAC,EAAE,cAAc,CAAC;CAC5B;AACD,wBAAgB,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,eAAe,GAAG,SAAS,CAAC,OAAO,UAAU,CAAC,CA8InG;AACD,eAAO,MAAM,QAAQ;;;;QAAgC,CAAC","sourcesContent":["import type { AgentTool, AgentToolUpdateCallback } from \"@boxiaolanya2008/pi-agent-core\";\nimport { type Static, Type } from \"@sinclair/typebox\";\nimport { constants } from \"fs\";\nimport { access as fsAccess } from \"fs/promises\";\nimport {\n\tdetectLineEnding,\n\tfuzzyFindText,\n\tgenerateDiffString,\n\tnormalizeForFuzzyMatch,\n\tnormalizeToLF,\n\trestoreLineEndings,\n\tstripBom,\n} from \"./edit-diff.js\";\nimport { acceleratedReadFile, acceleratedWriteFile } from \"./local-accelerator.js\";\nimport { resolveToCwd } from \"./path-utils.js\";\n\nconst editSchema = Type.Object({\n\tpath: Type.String({ description: \"Path to the file to edit (relative or absolute)\" }),\n\toldText: Type.String({ description: \"Exact text to find and replace (must match exactly)\" }),\n\tnewText: Type.String({ description: \"New text to replace the old text with\" }),\n});\nexport type EditToolInput = Static<typeof editSchema>;\nexport interface EditToolDetails {\n\tdiff: string;\n\tfirstChangedLine?: number;\n}\nexport interface EditOperations {\n\treadFile: (absolutePath: string) => Promise<Buffer>;\n\twriteFile: (absolutePath: string, content: string) => Promise<void>;\n\taccess: (absolutePath: string) => Promise<void>;\n}\nconst defaultEditOperations: EditOperations = {\n\treadFile: (path) => acceleratedReadFile(path),\n\twriteFile: (path, content) => acceleratedWriteFile(path, content),\n\taccess: (path) => fsAccess(path, constants.R_OK | constants.W_OK),\n};\nexport interface EditToolOptions {\n\toperations?: EditOperations;\n}\nexport function createEditTool(cwd: string, options?: EditToolOptions): AgentTool<typeof editSchema> {\n\tconst ops = options?.operations ?? defaultEditOperations;\n\treturn {\n\t\tname: \"edit\",\n\t\tlabel: \"edit\",\n\t\tdescription:\n\t\t\t\"Edit a file by replacing exact text. The oldText must match exactly (including whitespace). Use this for precise, surgical edits.\",\n\t\tparameters: editSchema,\n\t\texecute: async (\n\t\t\t_toolCallId: string,\n\t\t\t{ path, oldText, newText }: { path: string; oldText: string; newText: string },\n\t\t\tsignal?: AbortSignal,\n\t\t\tonUpdate?: AgentToolUpdateCallback<EditToolDetails | undefined>,\n\t\t) => {\n\t\t\tconst absolutePath = resolveToCwd(path, cwd);\n\n\t\t\tonUpdate?.({\n\t\t\t\tcontent: [{ type: \"text\", text: `Editing ${path}...` }],\n\t\t\t\tdetails: undefined,\n\t\t\t});\n\n\t\t\treturn new Promise<{\n\t\t\t\tcontent: Array<{ type: \"text\"; text: string }>;\n\t\t\t\tdetails: EditToolDetails | undefined;\n\t\t\t}>((resolve, reject) => {\n\t\t\t\tif (signal?.aborted) {\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tlet aborted = false;\n\t\t\t\tconst onAbort = () => {\n\t\t\t\t\taborted = true;\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t};\n\t\t\t\tif (signal) {\n\t\t\t\t\tsignal.addEventListener(\"abort\", onAbort, { once: true });\n\t\t\t\t}\n\t\t\t\t(async () => {\n\t\t\t\t\ttry {\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tawait ops.access(absolutePath);\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(new Error(`File not found: ${path}`));\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst buffer = await ops.readFile(absolutePath);\n\t\t\t\t\t\tconst rawContent = buffer.toString(\"utf-8\");\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst { bom, text: content } = stripBom(rawContent);\n\t\t\t\t\t\tconst originalEnding = detectLineEnding(content);\n\t\t\t\t\t\tconst normalizedContent = normalizeToLF(content);\n\t\t\t\t\t\tconst normalizedOldText = normalizeToLF(oldText);\n\t\t\t\t\t\tconst normalizedNewText = normalizeToLF(newText);\n\t\t\t\t\t\tconst matchResult = fuzzyFindText(normalizedContent, normalizedOldText);\n\t\t\t\t\t\tif (!matchResult.found) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Could not find the exact text in ${path}. The old text must match exactly including all whitespace and newlines.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst fuzzyContent = normalizeForFuzzyMatch(normalizedContent);\n\t\t\t\t\t\tconst fuzzyOldText = normalizeForFuzzyMatch(normalizedOldText);\n\t\t\t\t\t\tconst occurrences = fuzzyContent.split(fuzzyOldText).length - 1;\n\t\t\t\t\t\tif (occurrences > 1) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Found ${occurrences} occurrences of the text in ${path}. The text must be unique. Please provide more context to make it unique.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst baseContent = matchResult.contentForReplacement;\n\t\t\t\t\t\tconst newContent =\n\t\t\t\t\t\t\tbaseContent.substring(0, matchResult.index) +\n\t\t\t\t\t\t\tnormalizedNewText +\n\t\t\t\t\t\t\tbaseContent.substring(matchResult.index + matchResult.matchLength);\n\t\t\t\t\t\tif (baseContent === newContent) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`No changes made to ${path}. The replacement produced identical content. This might indicate an issue with special characters or the text not existing as expected.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst finalContent = bom + restoreLineEndings(newContent, originalEnding);\n\t\t\t\t\t\tawait ops.writeFile(absolutePath, finalContent);\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst diffResult = generateDiffString(baseContent, newContent);\n\n\t\t\t\t\t\tonUpdate?.({\n\t\t\t\t\t\t\tcontent: [{ type: \"text\", text: `[~] ${path} (1 replacement)` }],\n\t\t\t\t\t\t\tdetails: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },\n\t\t\t\t\t\t});\n\n\t\t\t\t\t\tresolve({\n\t\t\t\t\t\t\tcontent: [\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\ttype: \"text\",\n\t\t\t\t\t\t\t\t\ttext: `[~] ${path} (1 replacement)`,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\tdetails: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },\n\t\t\t\t\t\t});\n\t\t\t\t\t} catch (error: any) {\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (!aborted) {\n\t\t\t\t\t\t\treject(error);\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t})();\n\t\t\t});\n\t\t},\n\t};\n}\nexport const editTool = createEditTool(process.cwd());\n"]}

package/dist/core/tools/edit.js

@@ -21,8 +21,12 @@ export function createEditTool(cwd, options) {
         label: "edit",
         description: "Edit a file by replacing exact text. The oldText must match exactly (including whitespace). Use this for precise, surgical edits.",
         parameters: editSchema,
-        execute: async (_toolCallId, { path, oldText, newText }, signal) => {
+        execute: async (_toolCallId, { path, oldText, newText }, signal, onUpdate) => {
             const absolutePath = resolveToCwd(path, cwd);
+            onUpdate?.({
+                content: [{ type: "text", text: `Editing ${path}...` }],
+                details: undefined,
+            });
             return new Promise((resolve, reject) => {
                 if (signal?.aborted) {
                     reject(new Error("Operation aborted"));
@@ -102,11 +106,15 @@ export function createEditTool(cwd, options) {
                             signal.removeEventListener("abort", onAbort);
                         }
                         const diffResult = generateDiffString(baseContent, newContent);
+                        onUpdate?.({
+                            content: [{ type: "text", text: `[~] ${path} (1 replacement)` }],
+                            details: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },
+                        });
                         resolve({
                             content: [
                                 {
                                     type: "text",
-                                    text: `Successfully replaced text in ${path}.`,
+                                    text: `[~] ${path} (1 replacement)`,
                                 },
                             ],
                             details: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },

package/dist/core/tools/edit.js.map

@@ -1 +1 @@
-{"version":3,"file":"edit.js","sourceRoot":"","sources":["../../../src/core/tools/edit.ts"],"names":[],"mappings":"AACA,OAAO,EAAe,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,MAAM,IAAI,QAAQ,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EACN,gBAAgB,EAChB,aAAa,EACb,kBAAkB,EAClB,sBAAsB,EACtB,aAAa,EACb,kBAAkB,EAClB,QAAQ,GACR,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AACnF,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;IAC9B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,iDAAiD,EAAE,CAAC;IACrF,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,qDAAqD,EAAE,CAAC;IAC5F,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,uCAAuC,EAAE,CAAC;CAC9E,CAAC,CAAC;AAWH,MAAM,qBAAqB,GAAmB;IAC7C,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;IAC7C,SAAS,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC;IACjE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC;CACjE,CAAC;AAIF,MAAM,UAAU,cAAc,CAAC,GAAW,EAAE,OAAyB,EAAgC;IACpG,MAAM,GAAG,GAAG,OAAO,EAAE,UAAU,IAAI,qBAAqB,CAAC;IACzD,OAAO;QACN,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,MAAM;QACb,WAAW,EACV,mIAAmI;QACpI,UAAU,EAAE,UAAU;QACtB,OAAO,EAAE,KAAK,EACb,WAAmB,EACnB,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAsD,EAC9E,MAAoB,EACnB,EAAE,CAAC;YACJ,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAC7C,OAAO,IAAI,OAAO,CAGf,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC;gBACvB,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBACrB,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;oBACvC,OAAO;gBACR,CAAC;gBACD,IAAI,OAAO,GAAG,KAAK,CAAC;gBACpB,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC;oBACrB,OAAO,GAAG,IAAI,CAAC;oBACf,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;gBAAA,CACvC,CAAC;gBACF,IAAI,MAAM,EAAE,CAAC;oBACZ,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC3D,CAAC;gBACD,CAAC,KAAK,IAAI,EAAE,CAAC;oBACZ,IAAI,CAAC;wBACJ,IAAI,CAAC;4BACJ,MAAM,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;wBAChC,CAAC;wBAAC,MAAM,CAAC;4BACR,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,CAAC;4BAC7C,OAAO;wBACR,CAAC;wBACD,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;wBAChD,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;wBAC5C,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;wBACpD,MAAM,cAAc,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,WAAW,GAAG,aAAa,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;wBACxE,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;4BACxB,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CACL,IAAI,KAAK,CACR,oCAAoC,IAAI,0EAA0E,CAClH,CACD,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,MAAM,YAAY,GAAG,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;wBAC/D,MAAM,YAAY,GAAG,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;wBAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAChE,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;4BACrB,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CACL,IAAI,KAAK,CACR,SAAS,WAAW,+BAA+B,IAAI,2EAA2E,CAClI,CACD,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,MAAM,WAAW,GAAG,WAAW,CAAC,qBAAqB,CAAC;wBACtD,MAAM,UAAU,GACf,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,WAAW,CAAC,KAAK,CAAC;4BAC3C,iBAAiB;4BACjB,WAAW,CAAC,SAAS,CAAC,WAAW,CAAC,KAAK,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;wBACpE,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;4BAChC,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CACL,IAAI,KAAK,CACR,sBAAsB,IAAI,0IAA0I,CACpK,CACD,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,MAAM,YAAY,GAAG,GAAG,GAAG,kBAAkB,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;wBAC1E,MAAM,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;wBAChD,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,IAAI,MAAM,EAAE,CAAC;4BACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBAC9C,CAAC;wBACD,MAAM,UAAU,GAAG,kBAAkB,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;wBAC/D,OAAO,CAAC;4BACP,OAAO,EAAE;gCACR;oCACC,IAAI,EAAE,MAAM;oCACZ,IAAI,EAAE,iCAAiC,IAAI,GAAG;iCAC9C;6BACD;4BACD,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE,gBAAgB,EAAE,UAAU,CAAC,gBAAgB,EAAE;yBACjF,CAAC,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAU,EAAE,CAAC;wBACrB,IAAI,MAAM,EAAE,CAAC;4BACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBAC9C,CAAC;wBACD,IAAI,CAAC,OAAO,EAAE,CAAC;4BACd,MAAM,CAAC,KAAK,CAAC,CAAC;wBACf,CAAC;oBACF,CAAC;gBAAA,CACD,CAAC,EAAE,CAAC;YAAA,CACL,CAAC,CAAC;QAAA,CACH;KACD,CAAC;AAAA,CACF;AACD,MAAM,CAAC,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC","sourcesContent":["import type { AgentTool } from \"@boxiaolanya2008/pi-agent-core\";\nimport { type Static, Type } from \"@sinclair/typebox\";\nimport { constants } from \"fs\";\nimport { access as fsAccess } from \"fs/promises\";\nimport {\n\tdetectLineEnding,\n\tfuzzyFindText,\n\tgenerateDiffString,\n\tnormalizeForFuzzyMatch,\n\tnormalizeToLF,\n\trestoreLineEndings,\n\tstripBom,\n} from \"./edit-diff.js\";\nimport { acceleratedReadFile, acceleratedWriteFile } from \"./local-accelerator.js\";\nimport { resolveToCwd } from \"./path-utils.js\";\n\nconst editSchema = Type.Object({\n\tpath: Type.String({ description: \"Path to the file to edit (relative or absolute)\" }),\n\toldText: Type.String({ description: \"Exact text to find and replace (must match exactly)\" }),\n\tnewText: Type.String({ description: \"New text to replace the old text with\" }),\n});\nexport type EditToolInput = Static<typeof editSchema>;\nexport interface EditToolDetails {\n\tdiff: string;\n\tfirstChangedLine?: number;\n}\nexport interface EditOperations {\n\treadFile: (absolutePath: string) => Promise<Buffer>;\n\twriteFile: (absolutePath: string, content: string) => Promise<void>;\n\taccess: (absolutePath: string) => Promise<void>;\n}\nconst defaultEditOperations: EditOperations = {\n\treadFile: (path) => acceleratedReadFile(path),\n\twriteFile: (path, content) => acceleratedWriteFile(path, content),\n\taccess: (path) => fsAccess(path, constants.R_OK | constants.W_OK),\n};\nexport interface EditToolOptions {\n\toperations?: EditOperations;\n}\nexport function createEditTool(cwd: string, options?: EditToolOptions): AgentTool<typeof editSchema> {\n\tconst ops = options?.operations ?? defaultEditOperations;\n\treturn {\n\t\tname: \"edit\",\n\t\tlabel: \"edit\",\n\t\tdescription:\n\t\t\t\"Edit a file by replacing exact text. The oldText must match exactly (including whitespace). Use this for precise, surgical edits.\",\n\t\tparameters: editSchema,\n\t\texecute: async (\n\t\t\t_toolCallId: string,\n\t\t\t{ path, oldText, newText }: { path: string; oldText: string; newText: string },\n\t\t\tsignal?: AbortSignal,\n\t\t) => {\n\t\t\tconst absolutePath = resolveToCwd(path, cwd);\n\t\t\treturn new Promise<{\n\t\t\t\tcontent: Array<{ type: \"text\"; text: string }>;\n\t\t\t\tdetails: EditToolDetails | undefined;\n\t\t\t}>((resolve, reject) => {\n\t\t\t\tif (signal?.aborted) {\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tlet aborted = false;\n\t\t\t\tconst onAbort = () => {\n\t\t\t\t\taborted = true;\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t};\n\t\t\t\tif (signal) {\n\t\t\t\t\tsignal.addEventListener(\"abort\", onAbort, { once: true });\n\t\t\t\t}\n\t\t\t\t(async () => {\n\t\t\t\t\ttry {\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tawait ops.access(absolutePath);\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(new Error(`File not found: ${path}`));\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst buffer = await ops.readFile(absolutePath);\n\t\t\t\t\t\tconst rawContent = buffer.toString(\"utf-8\");\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst { bom, text: content } = stripBom(rawContent);\n\t\t\t\t\t\tconst originalEnding = detectLineEnding(content);\n\t\t\t\t\t\tconst normalizedContent = normalizeToLF(content);\n\t\t\t\t\t\tconst normalizedOldText = normalizeToLF(oldText);\n\t\t\t\t\t\tconst normalizedNewText = normalizeToLF(newText);\n\t\t\t\t\t\tconst matchResult = fuzzyFindText(normalizedContent, normalizedOldText);\n\t\t\t\t\t\tif (!matchResult.found) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Could not find the exact text in ${path}. The old text must match exactly including all whitespace and newlines.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst fuzzyContent = normalizeForFuzzyMatch(normalizedContent);\n\t\t\t\t\t\tconst fuzzyOldText = normalizeForFuzzyMatch(normalizedOldText);\n\t\t\t\t\t\tconst occurrences = fuzzyContent.split(fuzzyOldText).length - 1;\n\t\t\t\t\t\tif (occurrences > 1) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Found ${occurrences} occurrences of the text in ${path}. The text must be unique. Please provide more context to make it unique.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst baseContent = matchResult.contentForReplacement;\n\t\t\t\t\t\tconst newContent =\n\t\t\t\t\t\t\tbaseContent.substring(0, matchResult.index) +\n\t\t\t\t\t\t\tnormalizedNewText +\n\t\t\t\t\t\t\tbaseContent.substring(matchResult.index + matchResult.matchLength);\n\t\t\t\t\t\tif (baseContent === newContent) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`No changes made to ${path}. The replacement produced identical content. This might indicate an issue with special characters or the text not existing as expected.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst finalContent = bom + restoreLineEndings(newContent, originalEnding);\n\t\t\t\t\t\tawait ops.writeFile(absolutePath, finalContent);\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst diffResult = generateDiffString(baseContent, newContent);\n\t\t\t\t\t\tresolve({\n\t\t\t\t\t\t\tcontent: [\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\ttype: \"text\",\n\t\t\t\t\t\t\t\t\ttext: `Successfully replaced text in ${path}.`,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\tdetails: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },\n\t\t\t\t\t\t});\n\t\t\t\t\t} catch (error: any) {\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (!aborted) {\n\t\t\t\t\t\t\treject(error);\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t})();\n\t\t\t});\n\t\t},\n\t};\n}\nexport const editTool = createEditTool(process.cwd());\n"]}
+{"version":3,"file":"edit.js","sourceRoot":"","sources":["../../../src/core/tools/edit.ts"],"names":[],"mappings":"AACA,OAAO,EAAe,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAC;AAC/B,OAAO,EAAE,MAAM,IAAI,QAAQ,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EACN,gBAAgB,EAChB,aAAa,EACb,kBAAkB,EAClB,sBAAsB,EACtB,aAAa,EACb,kBAAkB,EAClB,QAAQ,GACR,MAAM,gBAAgB,CAAC;AACxB,OAAO,EAAE,mBAAmB,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AACnF,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAE/C,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;IAC9B,IAAI,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,iDAAiD,EAAE,CAAC;IACrF,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,qDAAqD,EAAE,CAAC;IAC5F,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,uCAAuC,EAAE,CAAC;CAC9E,CAAC,CAAC;AAWH,MAAM,qBAAqB,GAAmB;IAC7C,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,mBAAmB,CAAC,IAAI,CAAC;IAC7C,SAAS,EAAE,CAAC,IAAI,EAAE,OAAO,EAAE,EAAE,CAAC,oBAAoB,CAAC,IAAI,EAAE,OAAO,CAAC;IACjE,MAAM,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,GAAG,SAAS,CAAC,IAAI,CAAC;CACjE,CAAC;AAIF,MAAM,UAAU,cAAc,CAAC,GAAW,EAAE,OAAyB,EAAgC;IACpG,MAAM,GAAG,GAAG,OAAO,EAAE,UAAU,IAAI,qBAAqB,CAAC;IACzD,OAAO;QACN,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,MAAM;QACb,WAAW,EACV,mIAAmI;QACpI,UAAU,EAAE,UAAU;QACtB,OAAO,EAAE,KAAK,EACb,WAAmB,EACnB,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAsD,EAC9E,MAAoB,EACpB,QAA+D,EAC9D,EAAE,CAAC;YACJ,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;YAE7C,QAAQ,EAAE,CAAC;gBACV,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,WAAW,IAAI,KAAK,EAAE,CAAC;gBACvD,OAAO,EAAE,SAAS;aAClB,CAAC,CAAC;YAEH,OAAO,IAAI,OAAO,CAGf,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,CAAC;gBACvB,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;oBACrB,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;oBACvC,OAAO;gBACR,CAAC;gBACD,IAAI,OAAO,GAAG,KAAK,CAAC;gBACpB,MAAM,OAAO,GAAG,GAAG,EAAE,CAAC;oBACrB,OAAO,GAAG,IAAI,CAAC;oBACf,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,CAAC,CAAC,CAAC;gBAAA,CACvC,CAAC;gBACF,IAAI,MAAM,EAAE,CAAC;oBACZ,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC3D,CAAC;gBACD,CAAC,KAAK,IAAI,EAAE,CAAC;oBACZ,IAAI,CAAC;wBACJ,IAAI,CAAC;4BACJ,MAAM,GAAG,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;wBAChC,CAAC;wBAAC,MAAM,CAAC;4BACR,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,IAAI,EAAE,CAAC,CAAC,CAAC;4BAC7C,OAAO;wBACR,CAAC;wBACD,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;wBAChD,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;wBAC5C,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;wBACpD,MAAM,cAAc,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,iBAAiB,GAAG,aAAa,CAAC,OAAO,CAAC,CAAC;wBACjD,MAAM,WAAW,GAAG,aAAa,CAAC,iBAAiB,EAAE,iBAAiB,CAAC,CAAC;wBACxE,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,CAAC;4BACxB,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CACL,IAAI,KAAK,CACR,oCAAoC,IAAI,0EAA0E,CAClH,CACD,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,MAAM,YAAY,GAAG,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;wBAC/D,MAAM,YAAY,GAAG,sBAAsB,CAAC,iBAAiB,CAAC,CAAC;wBAC/D,MAAM,WAAW,GAAG,YAAY,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;wBAChE,IAAI,WAAW,GAAG,CAAC,EAAE,CAAC;4BACrB,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CACL,IAAI,KAAK,CACR,SAAS,WAAW,+BAA+B,IAAI,2EAA2E,CAClI,CACD,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,MAAM,WAAW,GAAG,WAAW,CAAC,qBAAqB,CAAC;wBACtD,MAAM,UAAU,GACf,WAAW,CAAC,SAAS,CAAC,CAAC,EAAE,WAAW,CAAC,KAAK,CAAC;4BAC3C,iBAAiB;4BACjB,WAAW,CAAC,SAAS,CAAC,WAAW,CAAC,KAAK,GAAG,WAAW,CAAC,WAAW,CAAC,CAAC;wBACpE,IAAI,WAAW,KAAK,UAAU,EAAE,CAAC;4BAChC,IAAI,MAAM,EAAE,CAAC;gCACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;4BAC9C,CAAC;4BACD,MAAM,CACL,IAAI,KAAK,CACR,sBAAsB,IAAI,0IAA0I,CACpK,CACD,CAAC;4BACF,OAAO;wBACR,CAAC;wBACD,MAAM,YAAY,GAAG,GAAG,GAAG,kBAAkB,CAAC,UAAU,EAAE,cAAc,CAAC,CAAC;wBAC1E,MAAM,GAAG,CAAC,SAAS,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;wBAChD,IAAI,OAAO,EAAE,CAAC;4BACb,OAAO;wBACR,CAAC;wBACD,IAAI,MAAM,EAAE,CAAC;4BACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBAC9C,CAAC;wBACD,MAAM,UAAU,GAAG,kBAAkB,CAAC,WAAW,EAAE,UAAU,CAAC,CAAC;wBAE/D,QAAQ,EAAE,CAAC;4BACV,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,IAAI,kBAAkB,EAAE,CAAC;4BAChE,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE,gBAAgB,EAAE,UAAU,CAAC,gBAAgB,EAAE;yBACjF,CAAC,CAAC;wBAEH,OAAO,CAAC;4BACP,OAAO,EAAE;gCACR;oCACC,IAAI,EAAE,MAAM;oCACZ,IAAI,EAAE,OAAO,IAAI,kBAAkB;iCACnC;6BACD;4BACD,OAAO,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE,gBAAgB,EAAE,UAAU,CAAC,gBAAgB,EAAE;yBACjF,CAAC,CAAC;oBACJ,CAAC;oBAAC,OAAO,KAAU,EAAE,CAAC;wBACrB,IAAI,MAAM,EAAE,CAAC;4BACZ,MAAM,CAAC,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;wBAC9C,CAAC;wBACD,IAAI,CAAC,OAAO,EAAE,CAAC;4BACd,MAAM,CAAC,KAAK,CAAC,CAAC;wBACf,CAAC;oBACF,CAAC;gBAAA,CACD,CAAC,EAAE,CAAC;YAAA,CACL,CAAC,CAAC;QAAA,CACH;KACD,CAAC;AAAA,CACF;AACD,MAAM,CAAC,MAAM,QAAQ,GAAG,cAAc,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC","sourcesContent":["import type { AgentTool, AgentToolUpdateCallback } from \"@boxiaolanya2008/pi-agent-core\";\nimport { type Static, Type } from \"@sinclair/typebox\";\nimport { constants } from \"fs\";\nimport { access as fsAccess } from \"fs/promises\";\nimport {\n\tdetectLineEnding,\n\tfuzzyFindText,\n\tgenerateDiffString,\n\tnormalizeForFuzzyMatch,\n\tnormalizeToLF,\n\trestoreLineEndings,\n\tstripBom,\n} from \"./edit-diff.js\";\nimport { acceleratedReadFile, acceleratedWriteFile } from \"./local-accelerator.js\";\nimport { resolveToCwd } from \"./path-utils.js\";\n\nconst editSchema = Type.Object({\n\tpath: Type.String({ description: \"Path to the file to edit (relative or absolute)\" }),\n\toldText: Type.String({ description: \"Exact text to find and replace (must match exactly)\" }),\n\tnewText: Type.String({ description: \"New text to replace the old text with\" }),\n});\nexport type EditToolInput = Static<typeof editSchema>;\nexport interface EditToolDetails {\n\tdiff: string;\n\tfirstChangedLine?: number;\n}\nexport interface EditOperations {\n\treadFile: (absolutePath: string) => Promise<Buffer>;\n\twriteFile: (absolutePath: string, content: string) => Promise<void>;\n\taccess: (absolutePath: string) => Promise<void>;\n}\nconst defaultEditOperations: EditOperations = {\n\treadFile: (path) => acceleratedReadFile(path),\n\twriteFile: (path, content) => acceleratedWriteFile(path, content),\n\taccess: (path) => fsAccess(path, constants.R_OK | constants.W_OK),\n};\nexport interface EditToolOptions {\n\toperations?: EditOperations;\n}\nexport function createEditTool(cwd: string, options?: EditToolOptions): AgentTool<typeof editSchema> {\n\tconst ops = options?.operations ?? defaultEditOperations;\n\treturn {\n\t\tname: \"edit\",\n\t\tlabel: \"edit\",\n\t\tdescription:\n\t\t\t\"Edit a file by replacing exact text. The oldText must match exactly (including whitespace). Use this for precise, surgical edits.\",\n\t\tparameters: editSchema,\n\t\texecute: async (\n\t\t\t_toolCallId: string,\n\t\t\t{ path, oldText, newText }: { path: string; oldText: string; newText: string },\n\t\t\tsignal?: AbortSignal,\n\t\t\tonUpdate?: AgentToolUpdateCallback<EditToolDetails | undefined>,\n\t\t) => {\n\t\t\tconst absolutePath = resolveToCwd(path, cwd);\n\n\t\t\tonUpdate?.({\n\t\t\t\tcontent: [{ type: \"text\", text: `Editing ${path}...` }],\n\t\t\t\tdetails: undefined,\n\t\t\t});\n\n\t\t\treturn new Promise<{\n\t\t\t\tcontent: Array<{ type: \"text\"; text: string }>;\n\t\t\t\tdetails: EditToolDetails | undefined;\n\t\t\t}>((resolve, reject) => {\n\t\t\t\tif (signal?.aborted) {\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t\treturn;\n\t\t\t\t}\n\t\t\t\tlet aborted = false;\n\t\t\t\tconst onAbort = () => {\n\t\t\t\t\taborted = true;\n\t\t\t\t\treject(new Error(\"Operation aborted\"));\n\t\t\t\t};\n\t\t\t\tif (signal) {\n\t\t\t\t\tsignal.addEventListener(\"abort\", onAbort, { once: true });\n\t\t\t\t}\n\t\t\t\t(async () => {\n\t\t\t\t\ttry {\n\t\t\t\t\t\ttry {\n\t\t\t\t\t\t\tawait ops.access(absolutePath);\n\t\t\t\t\t\t} catch {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(new Error(`File not found: ${path}`));\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst buffer = await ops.readFile(absolutePath);\n\t\t\t\t\t\tconst rawContent = buffer.toString(\"utf-8\");\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst { bom, text: content } = stripBom(rawContent);\n\t\t\t\t\t\tconst originalEnding = detectLineEnding(content);\n\t\t\t\t\t\tconst normalizedContent = normalizeToLF(content);\n\t\t\t\t\t\tconst normalizedOldText = normalizeToLF(oldText);\n\t\t\t\t\t\tconst normalizedNewText = normalizeToLF(newText);\n\t\t\t\t\t\tconst matchResult = fuzzyFindText(normalizedContent, normalizedOldText);\n\t\t\t\t\t\tif (!matchResult.found) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Could not find the exact text in ${path}. The old text must match exactly including all whitespace and newlines.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst fuzzyContent = normalizeForFuzzyMatch(normalizedContent);\n\t\t\t\t\t\tconst fuzzyOldText = normalizeForFuzzyMatch(normalizedOldText);\n\t\t\t\t\t\tconst occurrences = fuzzyContent.split(fuzzyOldText).length - 1;\n\t\t\t\t\t\tif (occurrences > 1) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`Found ${occurrences} occurrences of the text in ${path}. The text must be unique. Please provide more context to make it unique.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst baseContent = matchResult.contentForReplacement;\n\t\t\t\t\t\tconst newContent =\n\t\t\t\t\t\t\tbaseContent.substring(0, matchResult.index) +\n\t\t\t\t\t\t\tnormalizedNewText +\n\t\t\t\t\t\t\tbaseContent.substring(matchResult.index + matchResult.matchLength);\n\t\t\t\t\t\tif (baseContent === newContent) {\n\t\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t\t}\n\t\t\t\t\t\t\treject(\n\t\t\t\t\t\t\t\tnew Error(\n\t\t\t\t\t\t\t\t\t`No changes made to ${path}. The replacement produced identical content. This might indicate an issue with special characters or the text not existing as expected.`,\n\t\t\t\t\t\t\t\t),\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst finalContent = bom + restoreLineEndings(newContent, originalEnding);\n\t\t\t\t\t\tawait ops.writeFile(absolutePath, finalContent);\n\t\t\t\t\t\tif (aborted) {\n\t\t\t\t\t\t\treturn;\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tconst diffResult = generateDiffString(baseContent, newContent);\n\n\t\t\t\t\t\tonUpdate?.({\n\t\t\t\t\t\t\tcontent: [{ type: \"text\", text: `[~] ${path} (1 replacement)` }],\n\t\t\t\t\t\t\tdetails: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },\n\t\t\t\t\t\t});\n\n\t\t\t\t\t\tresolve({\n\t\t\t\t\t\t\tcontent: [\n\t\t\t\t\t\t\t\t{\n\t\t\t\t\t\t\t\t\ttype: \"text\",\n\t\t\t\t\t\t\t\t\ttext: `[~] ${path} (1 replacement)`,\n\t\t\t\t\t\t\t\t},\n\t\t\t\t\t\t\t],\n\t\t\t\t\t\t\tdetails: { diff: diffResult.diff, firstChangedLine: diffResult.firstChangedLine },\n\t\t\t\t\t\t});\n\t\t\t\t\t} catch (error: any) {\n\t\t\t\t\t\tif (signal) {\n\t\t\t\t\t\t\tsignal.removeEventListener(\"abort\", onAbort);\n\t\t\t\t\t\t}\n\t\t\t\t\t\tif (!aborted) {\n\t\t\t\t\t\t\treject(error);\n\t\t\t\t\t\t}\n\t\t\t\t\t}\n\t\t\t\t})();\n\t\t\t});\n\t\t},\n\t};\n}\nexport const editTool = createEditTool(process.cwd());\n"]}

package/dist/core/tools/fast-executor.d.ts

@@ -18,8 +18,13 @@ interface ExecuteToolsOptions {
     maxConcurrency?: number;
     priority?: number;
 }
+export interface ToolUpdateEvent {
+    toolCallId: string;
+    toolName: string;
+    partialResult: AgentToolResult<any>;
+}
 export declare function extractToolCalls(tools: AgentTool<any>[] | undefined, assistantMessage: AssistantMessage): ToolCallInfo[];
-export declare function executeToolsParallel(tools: AgentTool<any>[] | undefined, assistantMessage: AssistantMessage, signal?: AbortSignal, onProgress?: (result: ToolExecutionResult) => void, options?: ExecuteToolsOptions): Promise<ToolResultMessage[]>;
+export declare function executeToolsParallel(tools: AgentTool<any>[] | undefined, assistantMessage: AssistantMessage, signal?: AbortSignal, onProgress?: (result: ToolExecutionResult) => void, options?: ExecuteToolsOptions, onToolUpdate?: (event: ToolUpdateEvent) => void): Promise<ToolResultMessage[]>;
 export declare function canParallelize(tools: ToolCallInfo[]): boolean;
 export declare function groupToolsByDependency(tools: ToolCallInfo[]): ToolCallInfo[][];
 export declare function getExecutionStats(): Record<string, {

package/dist/core/tools/fast-executor.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"fast-executor.d.ts","sourceRoot":"","sources":["../../../src/core/tools/fast-executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjF,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAGlF,UAAU,YAAY;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC9B,IAAI,EAAE,SAAS,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;CACjC;AACD,UAAU,mBAAmB;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CACjB;AACD,UAAU,mBAAmB;IAC5B,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AACD,wBAAgB,gBAAgB,CAC/B,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,EACnC,gBAAgB,EAAE,gBAAgB,GAChC,YAAY,EAAE,CAQhB;AAoCD,wBAAsB,oBAAoB,CACzC,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,EACnC,gBAAgB,EAAE,gBAAgB,EAClC,MAAM,CAAC,EAAE,WAAW,EACpB,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,mBAAmB,KAAK,IAAI,EAClD,OAAO,GAAE,mBAAwB,GAC/B,OAAO,CAAC,iBAAiB,EAAE,CAAC,CAgB9B;AAYD,wBAAgB,cAAc,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,OAAO,CAE7D;AACD,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,YAAY,EAAE,EAAE,CAE9E;AACD,wBAAgB,iBAAiB,IAAI,MAAM,CAAC,MAAM,EAAE;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAE5G;AACD,wBAAgB,mBAAmB,IAAI,IAAI,CAE1C","sourcesContent":["import type { AgentTool, AgentToolResult } from \"@boxiaolanya2008/pi-agent-core\";\nimport type { AssistantMessage, ToolResultMessage } from \"@boxiaolanya2008/pi-ai\";\nimport { globalMetrics, parallelExecute } from \"./parallel-executor.js\";\n\ninterface ToolCallInfo {\n\tid: string;\n\tname: string;\n\targs: Record<string, unknown>;\n\ttool: AgentTool<any> | undefined;\n}\ninterface ToolExecutionResult {\n\ttoolCallId: string;\n\ttoolName: string;\n\tresult: AgentToolResult<any>;\n\tisError: boolean;\n\tduration: number;\n}\ninterface ExecuteToolsOptions {\n\tparallel?: boolean;\n\tmaxConcurrency?: number;\n\tpriority?: number;\n}\nexport function extractToolCalls(\n\ttools: AgentTool<any>[] | undefined,\n\tassistantMessage: AssistantMessage,\n): ToolCallInfo[] {\n\tconst toolCalls = assistantMessage.content.filter((c) => c.type === \"toolCall\");\n\treturn toolCalls.map((tc) => ({\n\t\tid: tc.id,\n\t\tname: tc.name,\n\t\targs: tc.arguments,\n\t\ttool: tools?.find((t) => t.name === tc.name),\n\t}));\n}\nasync function executeSingleTool(\n\ttoolCall: ToolCallInfo,\n\tsignal?: AbortSignal,\n\tonUpdate?: (partialResult: AgentToolResult<any>) => void,\n): Promise<ToolExecutionResult> {\n\tconst startTime = Date.now();\n\ttry {\n\t\tif (!toolCall.tool) {\n\t\t\tthrow new Error(`Tool ${toolCall.name} not found`);\n\t\t}\n\t\tconst result = await toolCall.tool.execute(toolCall.id, toolCall.args, signal, onUpdate);\n\t\tconst duration = Date.now() - startTime;\n\t\tglobalMetrics.record(`tool:${toolCall.name}`, duration);\n\t\treturn {\n\t\t\ttoolCallId: toolCall.id,\n\t\t\ttoolName: toolCall.name,\n\t\t\tresult,\n\t\t\tisError: false,\n\t\t\tduration,\n\t\t};\n\t} catch (error) {\n\t\tconst duration = Date.now() - startTime;\n\t\tglobalMetrics.record(`tool:${toolCall.name}:error`, duration);\n\t\treturn {\n\t\t\ttoolCallId: toolCall.id,\n\t\t\ttoolName: toolCall.name,\n\t\t\tresult: {\n\t\t\t\tcontent: [{ type: \"text\", text: error instanceof Error ? error.message : String(error) }],\n\t\t\t\tdetails: {},\n\t\t\t},\n\t\t\tisError: true,\n\t\t\tduration,\n\t\t};\n\t}\n}\nexport async function executeToolsParallel(\n\ttools: AgentTool<any>[] | undefined,\n\tassistantMessage: AssistantMessage,\n\tsignal?: AbortSignal,\n\tonProgress?: (result: ToolExecutionResult) => void,\n\toptions: ExecuteToolsOptions = {},\n): Promise<ToolResultMessage[]> {\n\tconst { parallel = true, maxConcurrency = 8 } = options;\n\tconst toolCalls = extractToolCalls(tools, assistantMessage);\n\tif (toolCalls.length === 0) {\n\t\treturn [];\n\t}\n\tif (toolCalls.length === 1 || !parallel) {\n\t\tconst result = await executeSingleTool(toolCalls[0], signal);\n\t\tonProgress?.(result);\n\t\treturn [toToolResultMessage(result)];\n\t}\n\tconst results = await parallelExecute(toolCalls, (tc) => executeSingleTool(tc, signal), {\n\t\tconcurrency: maxConcurrency,\n\t});\n\tfor (const result of results) onProgress?.(result);\n\treturn results.map(toToolResultMessage);\n}\nfunction toToolResultMessage(result: ToolExecutionResult): ToolResultMessage {\n\treturn {\n\t\trole: \"toolResult\",\n\t\ttoolCallId: result.toolCallId,\n\t\ttoolName: result.toolName,\n\t\tcontent: result.result.content,\n\t\tdetails: result.result.details,\n\t\tisError: result.isError,\n\t\ttimestamp: Date.now(),\n\t};\n}\nexport function canParallelize(tools: ToolCallInfo[]): boolean {\n\treturn tools.length > 1;\n}\nexport function groupToolsByDependency(tools: ToolCallInfo[]): ToolCallInfo[][] {\n\treturn [tools];\n}\nexport function getExecutionStats(): Record<string, { avg: number; min: number; max: number; count: number }> {\n\treturn globalMetrics.getAllStats();\n}\nexport function resetExecutionStats(): void {\n\tglobalMetrics.reset();\n}\n"]}
+{"version":3,"file":"fast-executor.d.ts","sourceRoot":"","sources":["../../../src/core/tools/fast-executor.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjF,OAAO,KAAK,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,MAAM,wBAAwB,CAAC;AAGlF,UAAU,YAAY;IACrB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAC9B,IAAI,EAAE,SAAS,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;CACjC;AACD,UAAU,mBAAmB;IAC5B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC;IAC7B,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,MAAM,CAAC;CACjB;AACD,UAAU,mBAAmB;IAC5B,QAAQ,CAAC,EAAE,OAAO,CAAC;IACnB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,eAAe;IAC/B,UAAU,EAAE,MAAM,CAAC;IACnB,QAAQ,EAAE,MAAM,CAAC;IACjB,aAAa,EAAE,eAAe,CAAC,GAAG,CAAC,CAAC;CACpC;AAkFD,wBAAgB,gBAAgB,CAC/B,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,EACnC,gBAAgB,EAAE,gBAAgB,GAChC,YAAY,EAAE,CAUhB;AAoCD,wBAAsB,oBAAoB,CACzC,KAAK,EAAE,SAAS,CAAC,GAAG,CAAC,EAAE,GAAG,SAAS,EACnC,gBAAgB,EAAE,gBAAgB,EAClC,MAAM,CAAC,EAAE,WAAW,EACpB,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,mBAAmB,KAAK,IAAI,EAClD,OAAO,GAAE,mBAAwB,EACjC,YAAY,CAAC,EAAE,CAAC,KAAK,EAAE,eAAe,KAAK,IAAI,GAC7C,OAAO,CAAC,iBAAiB,EAAE,CAAC,CA4B9B;AAYD,wBAAgB,cAAc,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,OAAO,CAE7D;AACD,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,YAAY,EAAE,GAAG,YAAY,EAAE,EAAE,CAE9E;AACD,wBAAgB,iBAAiB,IAAI,MAAM,CAAC,MAAM,EAAE;IAAE,GAAG,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC,CAE5G;AACD,wBAAgB,mBAAmB,IAAI,IAAI,CAE1C","sourcesContent":["import type { AgentTool, AgentToolResult } from \"@boxiaolanya2008/pi-agent-core\";\nimport type { AssistantMessage, ToolResultMessage } from \"@boxiaolanya2008/pi-ai\";\nimport { globalMetrics, parallelExecute } from \"./parallel-executor.js\";\n\ninterface ToolCallInfo {\n\tid: string;\n\tname: string;\n\targs: Record<string, unknown>;\n\ttool: AgentTool<any> | undefined;\n}\ninterface ToolExecutionResult {\n\ttoolCallId: string;\n\ttoolName: string;\n\tresult: AgentToolResult<any>;\n\tisError: boolean;\n\tduration: number;\n}\ninterface ExecuteToolsOptions {\n\tparallel?: boolean;\n\tmaxConcurrency?: number;\n\tpriority?: number;\n}\n\nexport interface ToolUpdateEvent {\n\ttoolCallId: string;\n\ttoolName: string;\n\tpartialResult: AgentToolResult<any>;\n}\n\nconst BATCH_WRITE_THRESHOLD = 2;\nconst WRITE_TOOL_NAMES = [\"write\", \"write_file\", \"create_file\", \"edit\"];\n\nfunction isWriteToolCall(tc: ToolCallInfo): boolean {\n\treturn WRITE_TOOL_NAMES.includes(tc.name);\n}\n\nfunction extractWriteFileData(tc: ToolCallInfo): { path: string; content: string } | null {\n\tconst args = tc.args as Record<string, unknown>;\n\tconst path = (args.path as string) || (args.file_path as string) || (args.filePath as string);\n\tconst content = (args.content as string) || (args.newText as string);\n\n\tif (!path || content === undefined) {\n\t\treturn null;\n\t}\n\n\treturn { path, content };\n}\n\nfunction optimizeToolCalls(toolCalls: ToolCallInfo[], tools: AgentTool<any>[] | undefined): ToolCallInfo[] {\n\tif (toolCalls.length < BATCH_WRITE_THRESHOLD) {\n\t\treturn toolCalls;\n\t}\n\n\tconst writeCalls: ToolCallInfo[] = [];\n\tconst otherCalls: ToolCallInfo[] = [];\n\n\tfor (const tc of toolCalls) {\n\t\tif (isWriteToolCall(tc)) {\n\t\t\twriteCalls.push(tc);\n\t\t} else {\n\t\t\totherCalls.push(tc);\n\t\t}\n\t}\n\n\tif (writeCalls.length < BATCH_WRITE_THRESHOLD) {\n\t\treturn toolCalls;\n\t}\n\n\tconst files: Array<{ path: string; content: string }> = [];\n\tconst writeCallIds: string[] = [];\n\n\tfor (const tc of writeCalls) {\n\t\tconst fileData = extractWriteFileData(tc);\n\t\tif (fileData) {\n\t\t\tfiles.push(fileData);\n\t\t\twriteCallIds.push(tc.id);\n\t\t} else {\n\t\t\totherCalls.push(tc);\n\t\t}\n\t}\n\n\tif (files.length < BATCH_WRITE_THRESHOLD) {\n\t\treturn toolCalls;\n\t}\n\n\tconst batchWriteTool = tools?.find((t) => t.name === \"batch_write\");\n\tif (!batchWriteTool) {\n\t\treturn toolCalls;\n\t}\n\n\tconst batchCall: ToolCallInfo = {\n\t\tid: `batch_${writeCallIds.join(\"_\")}`,\n\t\tname: \"batch_write\",\n\t\targs: {\n\t\t\toperation: \"batch_write\",\n\t\t\tfiles,\n\t\t\toptions: {\n\t\t\t\tatomic: true,\n\t\t\t\tbackup: true,\n\t\t\t\toverwrite: true,\n\t\t\t\tcreateDirs: true,\n\t\t\t},\n\t\t},\n\t\ttool: batchWriteTool,\n\t};\n\n\treturn [batchCall, ...otherCalls];\n}\n\nexport function extractToolCalls(\n\ttools: AgentTool<any>[] | undefined,\n\tassistantMessage: AssistantMessage,\n): ToolCallInfo[] {\n\tconst toolCalls = assistantMessage.content.filter((c) => c.type === \"toolCall\");\n\tconst rawCalls = toolCalls.map((tc) => ({\n\t\tid: tc.id,\n\t\tname: tc.name,\n\t\targs: tc.arguments,\n\t\ttool: tools?.find((t) => t.name === tc.name),\n\t}));\n\n\treturn optimizeToolCalls(rawCalls, tools);\n}\nasync function executeSingleTool(\n\ttoolCall: ToolCallInfo,\n\tsignal?: AbortSignal,\n\tonUpdate?: (partialResult: AgentToolResult<any>) => void,\n): Promise<ToolExecutionResult> {\n\tconst startTime = Date.now();\n\ttry {\n\t\tif (!toolCall.tool) {\n\t\t\tthrow new Error(`Tool ${toolCall.name} not found`);\n\t\t}\n\t\tconst result = await toolCall.tool.execute(toolCall.id, toolCall.args, signal, onUpdate);\n\t\tconst duration = Date.now() - startTime;\n\t\tglobalMetrics.record(`tool:${toolCall.name}`, duration);\n\t\treturn {\n\t\t\ttoolCallId: toolCall.id,\n\t\t\ttoolName: toolCall.name,\n\t\t\tresult,\n\t\t\tisError: false,\n\t\t\tduration,\n\t\t};\n\t} catch (error) {\n\t\tconst duration = Date.now() - startTime;\n\t\tglobalMetrics.record(`tool:${toolCall.name}:error`, duration);\n\t\treturn {\n\t\t\ttoolCallId: toolCall.id,\n\t\t\ttoolName: toolCall.name,\n\t\t\tresult: {\n\t\t\t\tcontent: [{ type: \"text\", text: error instanceof Error ? error.message : String(error) }],\n\t\t\t\tdetails: {},\n\t\t\t},\n\t\t\tisError: true,\n\t\t\tduration,\n\t\t};\n\t}\n}\nexport async function executeToolsParallel(\n\ttools: AgentTool<any>[] | undefined,\n\tassistantMessage: AssistantMessage,\n\tsignal?: AbortSignal,\n\tonProgress?: (result: ToolExecutionResult) => void,\n\toptions: ExecuteToolsOptions = {},\n\tonToolUpdate?: (event: ToolUpdateEvent) => void,\n): Promise<ToolResultMessage[]> {\n\tconst { parallel = true, maxConcurrency = 8 } = options;\n\tconst toolCalls = extractToolCalls(tools, assistantMessage);\n\tif (toolCalls.length === 0) {\n\t\treturn [];\n\t}\n\n\tconst createOnUpdate = (toolCall: ToolCallInfo) => {\n\t\tif (!onToolUpdate) return undefined;\n\t\treturn (partialResult: AgentToolResult<any>) => {\n\t\t\tonToolUpdate({\n\t\t\t\ttoolCallId: toolCall.id,\n\t\t\t\ttoolName: toolCall.name,\n\t\t\t\tpartialResult,\n\t\t\t});\n\t\t};\n\t};\n\n\tif (toolCalls.length === 1 || !parallel) {\n\t\tconst result = await executeSingleTool(toolCalls[0], signal, createOnUpdate(toolCalls[0]));\n\t\tonProgress?.(result);\n\t\treturn [toToolResultMessage(result)];\n\t}\n\tconst results = await parallelExecute(toolCalls, (tc) => executeSingleTool(tc, signal, createOnUpdate(tc)), {\n\t\tconcurrency: maxConcurrency,\n\t});\n\tfor (const result of results) onProgress?.(result);\n\treturn results.map(toToolResultMessage);\n}\nfunction toToolResultMessage(result: ToolExecutionResult): ToolResultMessage {\n\treturn {\n\t\trole: \"toolResult\",\n\t\ttoolCallId: result.toolCallId,\n\t\ttoolName: result.toolName,\n\t\tcontent: result.result.content,\n\t\tdetails: result.result.details,\n\t\tisError: result.isError,\n\t\ttimestamp: Date.now(),\n\t};\n}\nexport function canParallelize(tools: ToolCallInfo[]): boolean {\n\treturn tools.length > 1;\n}\nexport function groupToolsByDependency(tools: ToolCallInfo[]): ToolCallInfo[][] {\n\treturn [tools];\n}\nexport function getExecutionStats(): Record<string, { avg: number; min: number; max: number; count: number }> {\n\treturn globalMetrics.getAllStats();\n}\nexport function resetExecutionStats(): void {\n\tglobalMetrics.reset();\n}\n"]}

package/dist/core/tools/fast-executor.js

@@ -1,12 +1,80 @@
 import { globalMetrics, parallelExecute } from "./parallel-executor.js";
+const BATCH_WRITE_THRESHOLD = 2;
+const WRITE_TOOL_NAMES = ["write", "write_file", "create_file", "edit"];
+function isWriteToolCall(tc) {
+    return WRITE_TOOL_NAMES.includes(tc.name);
+}
+function extractWriteFileData(tc) {
+    const args = tc.args;
+    const path = args.path || args.file_path || args.filePath;
+    const content = args.content || args.newText;
+    if (!path || content === undefined) {
+        return null;
+    }
+    return { path, content };
+}
+function optimizeToolCalls(toolCalls, tools) {
+    if (toolCalls.length < BATCH_WRITE_THRESHOLD) {
+        return toolCalls;
+    }
+    const writeCalls = [];
+    const otherCalls = [];
+    for (const tc of toolCalls) {
+        if (isWriteToolCall(tc)) {
+            writeCalls.push(tc);
+        }
+        else {
+            otherCalls.push(tc);
+        }
+    }
+    if (writeCalls.length < BATCH_WRITE_THRESHOLD) {
+        return toolCalls;
+    }
+    const files = [];
+    const writeCallIds = [];
+    for (const tc of writeCalls) {
+        const fileData = extractWriteFileData(tc);
+        if (fileData) {
+            files.push(fileData);
+            writeCallIds.push(tc.id);
+        }
+        else {
+            otherCalls.push(tc);
+        }
+    }
+    if (files.length < BATCH_WRITE_THRESHOLD) {
+        return toolCalls;
+    }
+    const batchWriteTool = tools?.find((t) => t.name === "batch_write");
+    if (!batchWriteTool) {
+        return toolCalls;
+    }
+    const batchCall = {
+        id: `batch_${writeCallIds.join("_")}`,
+        name: "batch_write",
+        args: {
+            operation: "batch_write",
+            files,
+            options: {
+                atomic: true,
+                backup: true,
+                overwrite: true,
+                createDirs: true,
+            },
+        },
+        tool: batchWriteTool,
+    };
+    return [batchCall, ...otherCalls];
+}
 export function extractToolCalls(tools, assistantMessage) {
     const toolCalls = assistantMessage.content.filter((c) => c.type === "toolCall");
-    return toolCalls.map((tc) => ({
+    const rawCalls = toolCalls.map((tc) => ({
         id: tc.id,
         name: tc.name,
         args: tc.arguments,
         tool: tools?.find((t) => t.name === tc.name),
     }));
+    return optimizeToolCalls(rawCalls, tools);
 }
 async function executeSingleTool(toolCall, signal, onUpdate) {
     const startTime = Date.now();
@@ -40,18 +108,29 @@ async function executeSingleTool(toolCall, signal, onUpdate) {
         };
     }
 }
-export async function executeToolsParallel(tools, assistantMessage, signal, onProgress, options = {}) {
+export async function executeToolsParallel(tools, assistantMessage, signal, onProgress, options = {}, onToolUpdate) {
     const { parallel = true, maxConcurrency = 8 } = options;
     const toolCalls = extractToolCalls(tools, assistantMessage);
     if (toolCalls.length === 0) {
         return [];
     }
+    const createOnUpdate = (toolCall) => {
+        if (!onToolUpdate)
+            return undefined;
+        return (partialResult) => {
+            onToolUpdate({
+                toolCallId: toolCall.id,
+                toolName: toolCall.name,
+                partialResult,
+            });
+        };
+    };
     if (toolCalls.length === 1 || !parallel) {
-        const result = await executeSingleTool(toolCalls[0], signal);
+        const result = await executeSingleTool(toolCalls[0], signal, createOnUpdate(toolCalls[0]));
         onProgress?.(result);
         return [toToolResultMessage(result)];
     }
-    const results = await parallelExecute(toolCalls, (tc) => executeSingleTool(tc, signal), {
+    const results = await parallelExecute(toolCalls, (tc) => executeSingleTool(tc, signal, createOnUpdate(tc)), {
         concurrency: maxConcurrency,
     });
     for (const result of results)