@superblocksteam/vite-plugin-file-sync 1.99.99-next.2 → 2.0.0-SNAPSHOT.2143e2e95d0107ab

package/dist/ai-service/agent/tools2/tools/deploy-service.js

@@ -0,0 +1,167 @@
+import { cp, mkdir, mkdtemp, rm } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { getLogger } from "../../../../util/logger.js";
+const SB_WORKTREE_SNAPSHOTS_DIR = join(tmpdir(), "sb-worktree-snapshots");
+const BUILD_POLL_INTERVAL_MS = 5_000;
+const BUILD_POLL_TIMEOUT_MS = 5 * 60 * 1_000;
+const TERMINAL_BUILD_STATUSES = new Set([
+    "SUCCESS",
+    "BUILD_STATUS_SUCCESS",
+    "FAILED",
+    "BUILD_STATUS_FAILED",
+    "TIMED_OUT",
+    "BUILD_STATUS_TIMED_OUT",
+]);
+function isSuccessStatus(status) {
+    return status === "SUCCESS" || status === "BUILD_STATUS_SUCCESS";
+}
+function isFailedStatus(status) {
+    return status === "FAILED" || status === "BUILD_STATUS_FAILED";
+}
+function isTimedOutStatus(status) {
+    return status === "TIMED_OUT" || status === "BUILD_STATUS_TIMED_OUT";
+}
+async function serverPost(superblocksBaseUrl, jwt, path, body, options = {}) {
+    const url = `${superblocksBaseUrl}/api/${path}`;
+    const response = await fetch(url, {
+        method: "POST",
+        headers: {
+            Authorization: `Bearer ${jwt}`,
+            "Content-Type": "application/json",
+        },
+        body: JSON.stringify(body),
+        signal: options.abortSignal,
+    });
+    if (!response.ok) {
+        const text = await response.text().catch(() => "");
+        throw new Error(`Deploy API call to ${path} failed (${response.status}): ${text}`);
+    }
+    return (await response.json());
+}
+function throwIfAborted(abortSignal) {
+    if (abortSignal?.aborted) {
+        throw new DOMException("Aborted", "AbortError");
+    }
+}
+function abortableDelay(ms, abortSignal) {
+    return new Promise((resolve, reject) => {
+        if (abortSignal?.aborted) {
+            reject(new DOMException("Aborted", "AbortError"));
+            return;
+        }
+        const onAbort = () => {
+            clearTimeout(timeoutId);
+            reject(new DOMException("Aborted", "AbortError"));
+        };
+        const timeoutId = setTimeout(() => {
+            abortSignal?.removeEventListener("abort", onAbort);
+            resolve();
+        }, ms);
+        abortSignal?.addEventListener("abort", onAbort, { once: true });
+    });
+}
+export function createDeployService(deps) {
+    const { applicationId, gitService, getJwt, snapshotUploader, superblocksBaseUrl, } = deps;
+    const logger = getLogger();
+    function requireJwt() {
+        const jwt = getJwt();
+        if (!jwt) {
+            throw new Error("No authentication token available for deploy");
+        }
+        return jwt;
+    }
+    return {
+        async prepareMainCommit(sha, options = {}) {
+            throwIfAborted(options.abortSignal);
+            const worktreeBase = await mkdtemp(join(tmpdir(), "sb-main-deploy-"));
+            const worktreePath = join(worktreeBase, "worktree");
+            try {
+                throwIfAborted(options.abortSignal);
+                await gitService.raw([
+                    "worktree",
+                    "add",
+                    "--detach",
+                    worktreePath,
+                    sha,
+                ]);
+                throwIfAborted(options.abortSignal);
+                await rm(join(worktreePath, ".git"), {
+                    force: true,
+                    recursive: true,
+                });
+                throwIfAborted(options.abortSignal);
+                const { directoryHash } = await snapshotUploader.uploadSnapshotToDbfs(worktreePath);
+                throwIfAborted(options.abortSignal);
+                const snapshotPath = join(SB_WORKTREE_SNAPSHOTS_DIR, encodeURIComponent(directoryHash));
+                await mkdir(SB_WORKTREE_SNAPSHOTS_DIR, { recursive: true });
+                await rm(snapshotPath, { recursive: true, force: true });
+                await cp(worktreePath, snapshotPath, { recursive: true });
+                return { directoryHash, sha };
+            }
+            finally {
+                try {
+                    await gitService.raw(["worktree", "remove", "--force", worktreePath]);
+                }
+                catch {
+                    try {
+                        await gitService.raw(["worktree", "prune"]);
+                    }
+                    catch (pruneErr) {
+                        logger.warn("[deploy-service] worktree prune failed", {
+                            worktreePath,
+                            error: pruneErr instanceof Error ? pruneErr.message : String(pruneErr),
+                        });
+                    }
+                }
+                await rm(worktreeBase, { recursive: true, force: true });
+            }
+        },
+        async syncMainToDbfs(directoryHash, sha, options = {}) {
+            throwIfAborted(options.abortSignal);
+            const jwt = requireJwt();
+            const result = await serverPost(superblocksBaseUrl, jwt, `v3/applications/${applicationId}/sync-main-to-dbfs`, { directoryHash, sha }, options);
+            const commitId = result.data?.commitId;
+            if (!commitId) {
+                throw new Error("Server did not return a commitId");
+            }
+            return { commitId };
+        },
+        async createBuild(directoryHash, commitId, options = {}) {
+            throwIfAborted(options.abortSignal);
+            const jwt = requireJwt();
+            await serverPost(superblocksBaseUrl, jwt, `v3/applications/${applicationId}/${encodeURIComponent(directoryHash)}/builds`, { commitId }, options);
+        },
+        async pollBuildUntilComplete(directoryHash, options = {}) {
+            const startTime = Date.now();
+            while (Date.now() - startTime < BUILD_POLL_TIMEOUT_MS) {
+                throwIfAborted(options.abortSignal);
+                const jwt = requireJwt();
+                const result = await serverPost(superblocksBaseUrl, jwt, `v3/applications/${applicationId}/build-bulk-status`, { directoryHashes: [directoryHash] }, options);
+                const entry = result.data?.directoryHashToStatus?.[directoryHash];
+                if (entry && TERMINAL_BUILD_STATUSES.has(entry.status)) {
+                    if (isSuccessStatus(entry.status)) {
+                        return { succeeded: true };
+                    }
+                    if (isFailedStatus(entry.status)) {
+                        return { succeeded: false, error: "Build failed" };
+                    }
+                    if (isTimedOutStatus(entry.status)) {
+                        return { succeeded: false, error: "Build timed out" };
+                    }
+                }
+                await abortableDelay(BUILD_POLL_INTERVAL_MS, options.abortSignal);
+            }
+            return {
+                succeeded: false,
+                error: "Build polling timed out after 5 minutes",
+            };
+        },
+        async deployCommit(commitId, deployMessage, options = {}) {
+            throwIfAborted(options.abortSignal);
+            const jwt = requireJwt();
+            await serverPost(superblocksBaseUrl, jwt, `v2/applications/${applicationId}/deployments`, { commitId, deployMessage }, options);
+        },
+    };
+}
+//# sourceMappingURL=deploy-service.js.map

package/dist/ai-service/agent/tools2/tools/deploy-service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"deploy-service.js","sourceRoot":"","sources":["../../../../../src/ai-service/agent/tools2/tools/deploy-service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAGjC,OAAO,EAAE,SAAS,EAAE,MAAM,4BAA4B,CAAC;AAGvD,MAAM,yBAAyB,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,uBAAuB,CAAC,CAAC;AAC1E,MAAM,sBAAsB,GAAG,KAAK,CAAC;AACrC,MAAM,qBAAqB,GAAG,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;AAE7C,MAAM,uBAAuB,GAAwB,IAAI,GAAG,CAAC;IAC3D,SAAS;IACT,sBAAsB;IACtB,QAAQ;IACR,qBAAqB;IACrB,WAAW;IACX,wBAAwB;CACzB,CAAC,CAAC;AAEH,SAAS,eAAe,CAAC,MAAc;IACrC,OAAO,MAAM,KAAK,SAAS,IAAI,MAAM,KAAK,sBAAsB,CAAC;AACnE,CAAC;AAED,SAAS,cAAc,CAAC,MAAc;IACpC,OAAO,MAAM,KAAK,QAAQ,IAAI,MAAM,KAAK,qBAAqB,CAAC;AACjE,CAAC;AAED,SAAS,gBAAgB,CAAC,MAAc;IACtC,OAAO,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,wBAAwB,CAAC;AACvE,CAAC;AAgBD,KAAK,UAAU,UAAU,CACvB,kBAA0B,EAC1B,GAAW,EACX,IAAY,EACZ,IAAa,EACb,UAAqC,EAAE;IAEvC,MAAM,GAAG,GAAG,GAAG,kBAAkB,QAAQ,IAAI,EAAE,CAAC;IAChD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE;QAChC,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,aAAa,EAAE,UAAU,GAAG,EAAE;YAC9B,cAAc,EAAE,kBAAkB;SACnC;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;QAC1B,MAAM,EAAE,OAAO,CAAC,WAAW;KAC5B,CAAC,CAAC;IAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QACnD,MAAM,IAAI,KAAK,CACb,sBAAsB,IAAI,YAAY,QAAQ,CAAC,MAAM,MAAM,IAAI,EAAE,CAClE,CAAC;IACJ,CAAC;IAED,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAM,CAAC;AACtC,CAAC;AAED,SAAS,cAAc,CAAC,WAAyB;IAC/C,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;QACzB,MAAM,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC;IAClD,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,EAAU,EAAE,WAAyB;IAC3D,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACrC,IAAI,WAAW,EAAE,OAAO,EAAE,CAAC;YACzB,MAAM,CAAC,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;YAClD,OAAO;QACT,CAAC;QAED,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,YAAY,CAAC,SAAS,CAAC,CAAC;YACxB,MAAM,CAAC,IAAI,YAAY,CAAC,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;QACpD,CAAC,CAAC;QACF,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,EAAE;YAChC,WAAW,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACnD,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,EAAE,CAAC,CAAC;QACP,WAAW,EAAE,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;AACL,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,IAAuB;IACzD,MAAM,EACJ,aAAa,EACb,UAAU,EACV,MAAM,EACN,gBAAgB,EAChB,kBAAkB,GACnB,GAAG,IAAI,CAAC;IACT,MAAM,MAAM,GAAG,SAAS,EAAE,CAAC;IAE3B,SAAS,UAAU;QACjB,MAAM,GAAG,GAAG,MAAM,EAAE,CAAC;QACrB,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;QAClE,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC;IAED,OAAO;QACL,KAAK,CAAC,iBAAiB,CACrB,GAAW,EACX,UAAqC,EAAE;YAEvC,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,YAAY,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;YACtE,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC;YACpD,IAAI,CAAC;gBACH,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;gBACpC,MAAM,UAAU,CAAC,GAAG,CAAC;oBACnB,UAAU;oBACV,KAAK;oBACL,UAAU;oBACV,YAAY;oBACZ,GAAG;iBACJ,CAAC,CAAC;gBAEH,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;gBACpC,MAAM,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,MAAM,CAAC,EAAE;oBACnC,KAAK,EAAE,IAAI;oBACX,SAAS,EAAE,IAAI;iBAChB,CAAC,CAAC;gBAEH,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;gBACpC,MAAM,EAAE,aAAa,EAAE,GACrB,MAAM,gBAAgB,CAAC,oBAAoB,CAAC,YAAY,CAAC,CAAC;gBAE5D,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;gBACpC,MAAM,YAAY,GAAG,IAAI,CACvB,yBAAyB,EACzB,kBAAkB,CAAC,aAAa,CAAC,CAClC,CAAC;gBACF,MAAM,KAAK,CAAC,yBAAyB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC5D,MAAM,EAAE,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzD,MAAM,EAAE,CAAC,YAAY,EAAE,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAE1D,OAAO,EAAE,aAAa,EAAE,GAAG,EAAE,CAAC;YAChC,CAAC;oBAAS,CAAC;gBACT,IAAI,CAAC;oBACH,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,QAAQ,EAAE,SAAS,EAAE,YAAY,CAAC,CAAC,CAAC;gBACxE,CAAC;gBAAC,MAAM,CAAC;oBACP,IAAI,CAAC;wBACH,MAAM,UAAU,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC,CAAC;oBAC9C,CAAC;oBAAC,OAAO,QAAQ,EAAE,CAAC;wBAClB,MAAM,CAAC,IAAI,CAAC,wCAAwC,EAAE;4BACpD,YAAY;4BACZ,KAAK,EACH,QAAQ,YAAY,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC;yBAClE,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBACD,MAAM,EAAE,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3D,CAAC;QACH,CAAC;QAED,KAAK,CAAC,cAAc,CAClB,aAAqB,EACrB,GAAW,EACX,UAAqC,EAAE;YAEvC,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;YACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAG7B,kBAAkB,EAClB,GAAG,EACH,mBAAmB,aAAa,oBAAoB,EACpD,EAAE,aAAa,EAAE,GAAG,EAAE,EACtB,OAAO,CACR,CAAC;YACF,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC;YACvC,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACtD,CAAC;YACD,OAAO,EAAE,QAAQ,EAAE,CAAC;QACtB,CAAC;QAED,KAAK,CAAC,WAAW,CACf,aAAqB,EACrB,QAAgB,EAChB,UAAqC,EAAE;YAEvC,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;YACzB,MAAM,UAAU,CACd,kBAAkB,EAClB,GAAG,EACH,mBAAmB,aAAa,IAAI,kBAAkB,CAAC,aAAa,CAAC,SAAS,EAC9E,EAAE,QAAQ,EAAE,EACZ,OAAO,CACR,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,sBAAsB,CAC1B,aAAqB,EACrB,UAAqC,EAAE;YAEvC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAE7B,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,GAAG,qBAAqB,EAAE,CAAC;gBACtD,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;gBACpC,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,MAAM,UAAU,CAK7B,kBAAkB,EAClB,GAAG,EACH,mBAAmB,aAAa,oBAAoB,EACpD,EAAE,eAAe,EAAE,CAAC,aAAa,CAAC,EAAE,EACpC,OAAO,CACR,CAAC;gBAEF,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,EAAE,qBAAqB,EAAE,CAAC,aAAa,CAAC,CAAC;gBAClE,IAAI,KAAK,IAAI,uBAAuB,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;oBACvD,IAAI,eAAe,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;wBAClC,OAAO,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;oBAC7B,CAAC;oBACD,IAAI,cAAc,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;wBACjC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,cAAc,EAAE,CAAC;oBACrD,CAAC;oBACD,IAAI,gBAAgB,CAAC,KAAK,CAAC,MAAM,CAAC,EAAE,CAAC;wBACnC,OAAO,EAAE,SAAS,EAAE,KAAK,EAAE,KAAK,EAAE,iBAAiB,EAAE,CAAC;oBACxD,CAAC;gBACH,CAAC;gBAED,MAAM,cAAc,CAAC,sBAAsB,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;YACpE,CAAC;YAED,OAAO;gBACL,SAAS,EAAE,KAAK;gBAChB,KAAK,EAAE,yCAAyC;aACjD,CAAC;QACJ,CAAC;QAED,KAAK,CAAC,YAAY,CAChB,QAAgB,EAChB,aAAqB,EACrB,UAAqC,EAAE;YAEvC,cAAc,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACpC,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;YACzB,MAAM,UAAU,CACd,kBAAkB,EAClB,GAAG,EACH,mBAAmB,aAAa,cAAc,EAC9C,EAAE,QAAQ,EAAE,aAAa,EAAE,EAC3B,OAAO,CACR,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}

package/dist/ai-service/agent/tools2/tools/download-attachments.d.ts

@@ -0,0 +1,33 @@
+import { type AttachmentArchiveType } from "@superblocksteam/shared";
+import type { Clark, ClarkStateHandlerParams } from "../../../state-machine/clark-fsm.js";
+type DownloadResult = {
+    attachmentId: number;
+    fileName: string;
+    mediaType: string;
+    scopeType: "org" | "app";
+    archiveType?: AttachmentArchiveType;
+    downloadPath: string;
+    extracted: boolean;
+    extractedFilesCount?: number;
+    workingPath: string;
+};
+type DownloadFailure = {
+    attachmentId?: number;
+    error: string;
+    fileName?: string;
+    storageKey?: string;
+};
+export declare const downloadAttachmentsToolFactory: import("../types.js").ToolFactory<{
+    clark: Clark;
+    services: ClarkStateHandlerParams;
+}, import("../types.js").Tool<{
+    attachmentIds?: number[] | undefined;
+    storageKeys?: string[] | undefined;
+}, {
+    downloads: DownloadResult[];
+    failures: DownloadFailure[];
+    count: number;
+    rootPath: string;
+}>>;
+export {};
+//# sourceMappingURL=download-attachments.d.ts.map

package/dist/ai-service/agent/tools2/tools/download-attachments.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"download-attachments.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/agent/tools2/tools/download-attachments.ts"],"names":[],"mappings":"AAUA,OAAO,EAGL,KAAK,qBAAqB,EAC3B,MAAM,yBAAyB,CAAC;AAOjC,OAAO,KAAK,EACV,KAAK,EACL,uBAAuB,EACxB,MAAM,qCAAqC,CAAC;AAoC7C,KAAK,cAAc,GAAG;IACpB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,MAAM,CAAC;IACjB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,KAAK,GAAG,KAAK,CAAC;IACzB,WAAW,CAAC,EAAE,qBAAqB,CAAC;IACpC,YAAY,EAAE,MAAM,CAAC;IACrB,SAAS,EAAE,OAAO,CAAC;IACnB,mBAAmB,CAAC,EAAE,MAAM,CAAC;IAC7B,WAAW,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,KAAK,eAAe,GAAG;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,MAAM,CAAC;IACd,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB,CAAC;AAEF,eAAO,MAAM,8BAA8B;WAMhC,KAAK;cACF,uBAAuB;;;;;eAkIR,cAAc,EAAE;;;;GA4P5C,CAAC"}

package/dist/ai-service/agent/tools2/tools/download-attachments.js

@@ -0,0 +1,309 @@
+import { randomUUID } from "node:crypto";
+import { createWriteStream } from "node:fs";
+import { mkdir } from "node:fs/promises";
+import path from "node:path";
+import { Readable } from "node:stream";
+import { pipeline } from "node:stream/promises";
+import { z } from "zod";
+import { resolveAttachmentFileName } from "@superblocksteam/library-shared/types";
+import { detectAttachmentArchiveType, } from "@superblocksteam/shared";
+import { getLogger, getPrefixedLogger } from "../../../../util/logger.js";
+import { AttachmentsStore } from "../../../attachments/index.js";
+import { getCanonicalAttachmentUrl } from "../../../attachments/uploaded-content-part.js";
+import { getActiveContext } from "../../../llm/context-v2/utils/index.js";
+import { createByteCapTransform, extractGzipArchive, extractTgzArchive, extractZipArchive, MAX_ARCHIVE_EXTRACTED_BYTES, sanitizeFileName, } from "../../../util/archive-extractors.js";
+import { getScopedToken } from "../../../util/scoped-token-utils.js";
+import { createToolFactory, PermissionLevel, ToolCategory, ToolExecutionError, } from "../types.js";
+const DOWNLOAD_SCRATCH_ROOT = path.join(".superblocks", "scratch", "attachments");
+const MAX_ATTACHMENT_IDS = 20;
+const LIST_ATTACHMENTS_HINT = "Call listAttachments to see valid ids and storageKeys for this session.";
+/**
+ * Cap on the compressed-on-disk size of a single downloaded attachment. Even
+ * before decompression, a malicious signed URL could stream an arbitrarily
+ * large body and fill the scratch volume, so we bound it at 2x the decompressed
+ * cap. This is generous enough for legitimate zip/tgz uploads but bounds the
+ * blast radius of a single bad attachment.
+ */
+const MAX_DOWNLOAD_BYTES = 2 * MAX_ARCHIVE_EXTRACTED_BYTES;
+export const downloadAttachmentsToolFactory = createToolFactory("downloadAttachments", ({ clark, services, }) => {
+    return {
+        // Categorized as DEBUG so it is available in PLAN mode: Clark needs
+        // to materialize uploaded archives into the private scratch dir in
+        // order to inspect their contents before exitPlanMode when an
+        // attachment's classification is `reference_only`.
+        category: ToolCategory.DEBUG,
+        defaultPermission: PermissionLevel.ALLOW,
+        // This tool writes downloaded + extracted files into the app scratch
+        // directory, so it is not read-only.
+        readOnly: false,
+        description: `Download uploaded session attachments into a fixed scratch directory.
+
+Use listAttachments first to discover attachment IDs and/or storage keys.
+
+Behavior:
+- Downloads uploaded attachments.
+- Automatically extracts supported archive formats: zip, tgz, gz.
+- Returns local working paths for further file operations (ls/read/glob/etc).`,
+        inputSchema: z
+            .object({
+            attachmentIds: z
+                .array(z.number().int().positive())
+                .max(MAX_ATTACHMENT_IDS)
+                .optional()
+                .describe(`Session attachment IDs from listAttachments (max ${MAX_ATTACHMENT_IDS})`),
+            storageKeys: z
+                .array(z.string().min(1))
+                .max(MAX_ATTACHMENT_IDS)
+                .optional()
+                .describe(`Uploaded attachment storage keys from listAttachments (max ${MAX_ATTACHMENT_IDS})`),
+        })
+            .refine((input) => (input.attachmentIds?.length ?? 0) > 0 ||
+            (input.storageKeys?.length ?? 0) > 0, {
+            message: "Provide at least one attachmentId or storageKey",
+        }),
+        execute: async ({ attachmentIds, storageKeys }) => {
+            const logger = getPrefixedLogger("[downloadAttachments]");
+            const context = getActiveContext(clark, services);
+            if (!context) {
+                throw new ToolExecutionError("Context not available", {
+                    context: {},
+                    input: { attachmentIds, storageKeys },
+                    severity: "warn",
+                    toolName: "downloadAttachments",
+                });
+            }
+            const uniqueIds = Array.from(new Set(attachmentIds ?? [])).sort((a, b) => a - b);
+            const uniqueStorageKeys = Array.from(new Set((storageKeys ?? [])
+                .map((storageKey) => storageKey.trim())
+                .filter((storageKey) => storageKey.length > 0))).sort();
+            const store = new AttachmentsStore();
+            const attachmentsById = store.getAttachmentsByIds(context, uniqueIds);
+            const attachmentsByStorageKey = store.getUploadedAttachmentsByStorageKeys(context, uniqueStorageKeys);
+            const foundIds = new Set(attachmentsById.map((attachment) => attachment.id));
+            const foundStorageKeys = new Set(attachmentsByStorageKey
+                .map((attachment) => attachment.storageKey?.trim())
+                .filter((storageKey) => !!storageKey));
+            const mergedAttachmentsById = new Map();
+            for (const attachment of attachmentsById) {
+                mergedAttachmentsById.set(attachment.id, attachment);
+            }
+            for (const attachment of attachmentsByStorageKey) {
+                mergedAttachmentsById.set(attachment.id, attachment);
+            }
+            const storedAttachments = Array.from(mergedAttachmentsById.values()).sort((left, right) => left.id - right.id);
+            const failures = [];
+            const missingIds = uniqueIds.filter((id) => !foundIds.has(id));
+            for (const missingId of missingIds) {
+                failures.push({
+                    attachmentId: missingId,
+                    error: `Attachment ID not found in current session. ${LIST_ATTACHMENTS_HINT}`,
+                });
+            }
+            const missingStorageKeys = uniqueStorageKeys.filter((storageKey) => !foundStorageKeys.has(storageKey));
+            for (const missingStorageKey of missingStorageKeys) {
+                failures.push({
+                    error: `Attachment storageKey not found in current session. ${LIST_ATTACHMENTS_HINT}`,
+                    storageKey: missingStorageKey,
+                });
+            }
+            const uploadedAttachments = storedAttachments.filter((attachment) => {
+                if (attachment.type === "uploaded") {
+                    return true;
+                }
+                failures.push({
+                    attachmentId: attachment.id,
+                    error: "Attachment is not an uploaded file reference. Inline attachments without a storageKey cannot be downloaded; ask the user to re-attach via the upload button if you need bytes on disk.",
+                    fileName: attachment.fileName,
+                });
+                return false;
+            });
+            if (uploadedAttachments.length === 0) {
+                return {
+                    downloads: [],
+                    failures,
+                    count: 0,
+                    rootPath: DOWNLOAD_SCRATCH_ROOT,
+                };
+            }
+            const appScopeAttachments = uploadedAttachments.filter((attachment) => attachment.scopeType !== "org");
+            let scopedToken = "";
+            if (appScopeAttachments.length > 0) {
+                if (!clark.context.jwt) {
+                    throw new Error("Cannot resolve app-scoped attachments without a user JWT");
+                }
+                scopedToken = await getScopedToken({
+                    applicationId: services.applicationId,
+                    jwt: clark.context.jwt,
+                    superblocksBaseUrl: services.superblocksBaseUrl,
+                });
+            }
+            const leaseReferences = uploadedAttachments.map((attachment) => {
+                const resolvedFileName = resolveAttachmentFileName({
+                    fileName: attachment.fileName,
+                    mediaType: attachment.mediaType,
+                    type: "uploaded",
+                });
+                const scopeType = attachment.scopeType === "org" ? "org" : "app";
+                return {
+                    applicationId: scopeType === "app"
+                        ? (attachment.applicationId ?? services.applicationId)
+                        : undefined,
+                    fileName: resolvedFileName,
+                    mediaType: attachment.mediaType,
+                    scopeType,
+                    storageKey: attachment.storageKey,
+                    url: getCanonicalAttachmentUrl(attachment.url),
+                };
+            });
+            const leaseResponse = await services.rpcClient.call(async (client) => {
+                return await client.call.v1.ai.resolveAttachmentLeases({
+                    attachments: leaseReferences,
+                    scopedToken,
+                    superblocksBaseUrl: services.superblocksBaseUrl,
+                });
+            });
+            const resolvedAttachments = leaseResponse?.data?.attachments;
+            if (!Array.isArray(resolvedAttachments)) {
+                throw new ToolExecutionError("Download failed. Attachment lease response missing attachments array.", {
+                    context: {
+                        attachmentCount: leaseReferences.length,
+                    },
+                    input: { attachmentIds, storageKeys },
+                    severity: "error",
+                    toolName: "downloadAttachments",
+                });
+            }
+            const resolvedByUrl = new Map(resolvedAttachments
+                .filter((attachment) => typeof attachment?.url === "string" &&
+                typeof attachment?.signedUrl === "string")
+                .map((attachment) => [
+                getCanonicalAttachmentUrl(attachment.url),
+                attachment,
+            ]));
+            const runId = `${Date.now()}-${randomUUID().slice(0, 8)}`;
+            const rootRelativePath = path.join(DOWNLOAD_SCRATCH_ROOT, runId);
+            const rootAbsolutePath = path.join(services.appShell.appRootDirPath, rootRelativePath);
+            await mkdir(rootAbsolutePath, { recursive: true });
+            const downloads = [];
+            for (const attachment of uploadedAttachments) {
+                const canonicalUrl = getCanonicalAttachmentUrl(attachment.url);
+                const resolved = resolvedByUrl.get(canonicalUrl);
+                const resolvedFileName = resolveAttachmentFileName({
+                    fileName: attachment.fileName,
+                    mediaType: attachment.mediaType,
+                    type: "uploaded",
+                });
+                if (!resolved) {
+                    failures.push({
+                        attachmentId: attachment.id,
+                        error: "Lease resolution failed",
+                        fileName: resolvedFileName,
+                    });
+                    continue;
+                }
+                try {
+                    const response = await fetch(resolved.signedUrl);
+                    if (!response.ok || !response.body) {
+                        throw new Error(`Failed to download attachment (${response.status})`);
+                    }
+                    const safeName = sanitizeFileName(resolvedFileName);
+                    const relativeDownloadPath = path.join(rootRelativePath, `${attachment.id}-${safeName}`);
+                    const absoluteDownloadPath = path.join(services.appShell.appRootDirPath, relativeDownloadPath);
+                    await mkdir(path.dirname(absoluteDownloadPath), {
+                        recursive: true,
+                    });
+                    // Stream the response body directly to disk instead of buffering
+                    // the entire file in memory via arrayBuffer(). Large attachments
+                    // (zip/tgz up to 200 MiB) would otherwise OOM the vite-plugin process.
+                    // Cap the on-disk size so a malicious signed URL cannot stream
+                    // arbitrary bytes and fill the scratch volume before decompression
+                    // limits get a chance to run.
+                    await pipeline(Readable.fromWeb(response.body), createByteCapTransform(MAX_DOWNLOAD_BYTES, `Attachment download exceeds maximum size (${MAX_DOWNLOAD_BYTES} bytes)`), createWriteStream(absoluteDownloadPath));
+                    const archiveType = detectAttachmentArchiveType({
+                        fileName: resolvedFileName,
+                        mimeType: attachment.mediaType,
+                    });
+                    if (!archiveType) {
+                        downloads.push({
+                            attachmentId: attachment.id,
+                            downloadPath: relativeDownloadPath,
+                            extracted: false,
+                            fileName: resolvedFileName,
+                            mediaType: attachment.mediaType,
+                            scopeType: attachment.scopeType === "org" ? "org" : "app",
+                            workingPath: relativeDownloadPath,
+                        });
+                        continue;
+                    }
+                    const extractionDirRelative = path.join(rootRelativePath, `${attachment.id}-${safeName}-extracted`);
+                    const extractionDirAbsolute = path.join(services.appShell.appRootDirPath, extractionDirRelative);
+                    await mkdir(extractionDirAbsolute, { recursive: true });
+                    let extractedFilesCount = 0;
+                    let workingPath = extractionDirRelative;
+                    if (archiveType === "zip") {
+                        extractedFilesCount = await extractZipArchive({
+                            archivePath: absoluteDownloadPath,
+                            destinationDir: extractionDirAbsolute,
+                        });
+                    }
+                    else if (archiveType === "tgz") {
+                        extractedFilesCount = await extractTgzArchive({
+                            archivePath: absoluteDownloadPath,
+                            destinationDir: extractionDirAbsolute,
+                        });
+                    }
+                    else {
+                        const extracted = await extractGzipArchive({
+                            archivePath: absoluteDownloadPath,
+                            destinationDir: extractionDirAbsolute,
+                            fileName: resolvedFileName,
+                        });
+                        extractedFilesCount = extracted.extractedCount;
+                        workingPath = path.relative(services.appShell.appRootDirPath, extracted.outputPath);
+                    }
+                    downloads.push({
+                        archiveType,
+                        attachmentId: attachment.id,
+                        downloadPath: relativeDownloadPath,
+                        extracted: true,
+                        extractedFilesCount,
+                        fileName: resolvedFileName,
+                        mediaType: attachment.mediaType,
+                        scopeType: attachment.scopeType === "org" ? "org" : "app",
+                        workingPath,
+                    });
+                }
+                catch (error) {
+                    getLogger().warn("[downloadAttachments] Failed to download", {
+                        attachmentId: attachment.id,
+                        error: error instanceof Error ? error.message : String(error),
+                    });
+                    failures.push({
+                        attachmentId: attachment.id,
+                        error: error instanceof Error ? error.message : "Download failed",
+                        fileName: resolvedFileName,
+                    });
+                }
+            }
+            logger.info(`Downloaded ${downloads.length} attachment(s), ${failures.length} failure(s)`);
+            return {
+                downloads,
+                failures,
+                count: downloads.length,
+                rootPath: rootRelativePath,
+            };
+        },
+        getSummary: (_input, output) => {
+            const count = typeof output === "object" &&
+                output &&
+                "count" in output &&
+                typeof output.count === "number"
+                ? output.count
+                : 0;
+            return {
+                output: `${count} attachment${count === 1 ? "" : "s"} downloaded`,
+            };
+        },
+    };
+});
+//# sourceMappingURL=download-attachments.js.map

package/dist/ai-service/agent/tools2/tools/download-attachments.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"download-attachments.js","sourceRoot":"","sources":["../../../../../src/ai-service/agent/tools2/tools/download-attachments.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,EAAE,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACzC,OAAO,IAAI,MAAM,WAAW,CAAC;AAC7B,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AACvC,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAEhD,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,yBAAyB,EAAE,MAAM,uCAAuC,CAAC;AAClF,OAAO,EACL,2BAA2B,GAG5B,MAAM,yBAAyB,CAAC;AAEjC,OAAO,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,4BAA4B,CAAC;AAC1E,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AAEjE,OAAO,EAAE,yBAAyB,EAAE,MAAM,+CAA+C,CAAC;AAC1F,OAAO,EAAE,gBAAgB,EAAE,MAAM,wCAAwC,CAAC;AAK1E,OAAO,EACL,sBAAsB,EACtB,kBAAkB,EAClB,iBAAiB,EACjB,iBAAiB,EACjB,2BAA2B,EAC3B,gBAAgB,GACjB,MAAM,qCAAqC,CAAC;AAC7C,OAAO,EAAE,cAAc,EAAE,MAAM,qCAAqC,CAAC;AACrE,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,YAAY,EACZ,kBAAkB,GACnB,MAAM,aAAa,CAAC;AAErB,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CACrC,cAAc,EACd,SAAS,EACT,aAAa,CACd,CAAC;AAEF,MAAM,kBAAkB,GAAG,EAAE,CAAC;AAC9B,MAAM,qBAAqB,GACzB,yEAAyE,CAAC;AAE5E;;;;;;GAMG;AACH,MAAM,kBAAkB,GAAG,CAAC,GAAG,2BAA2B,CAAC;AAqB3D,MAAM,CAAC,MAAM,8BAA8B,GAAG,iBAAiB,CAC7D,qBAAqB,EACrB,CAAC,EACC,KAAK,EACL,QAAQ,GAIT,EAAE,EAAE;IACH,OAAO;QACL,oEAAoE;QACpE,mEAAmE;QACnE,8DAA8D;QAC9D,mDAAmD;QACnD,QAAQ,EAAE,YAAY,CAAC,KAAK;QAC5B,iBAAiB,EAAE,eAAe,CAAC,KAAK;QACxC,qEAAqE;QACrE,qCAAqC;QACrC,QAAQ,EAAE,KAAK;QACf,WAAW,EAAE;;;;;;;8EAO2D;QACxE,WAAW,EAAE,CAAC;aACX,MAAM,CAAC;YACN,aAAa,EAAE,CAAC;iBACb,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE,CAAC;iBAClC,GAAG,CAAC,kBAAkB,CAAC;iBACvB,QAAQ,EAAE;iBACV,QAAQ,CACP,oDAAoD,kBAAkB,GAAG,CAC1E;YACH,WAAW,EAAE,CAAC;iBACX,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;iBACxB,GAAG,CAAC,kBAAkB,CAAC;iBACvB,QAAQ,EAAE;iBACV,QAAQ,CACP,8DAA8D,kBAAkB,GAAG,CACpF;SACJ,CAAC;aACD,MAAM,CACL,CAAC,KAAK,EAAE,EAAE,CACR,CAAC,KAAK,CAAC,aAAa,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC;YACtC,CAAC,KAAK,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,EACtC;YACE,OAAO,EAAE,iDAAiD;SAC3D,CACF;QACH,OAAO,EAAE,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE,EAAE,EAAE;YAChD,MAAM,MAAM,GAAG,iBAAiB,CAAC,uBAAuB,CAAC,CAAC;YAC1D,MAAM,OAAO,GAAG,gBAAgB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;YAClD,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,MAAM,IAAI,kBAAkB,CAAC,uBAAuB,EAAE;oBACpD,OAAO,EAAE,EAAE;oBACX,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE;oBACrC,QAAQ,EAAE,MAAM;oBAChB,QAAQ,EAAE,qBAAqB;iBAChC,CAAC,CAAC;YACL,CAAC;YAED,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,GAAG,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAC7D,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAChB,CAAC;YACF,MAAM,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAClC,IAAI,GAAG,CACL,CAAC,WAAW,IAAI,EAAE,CAAC;iBAChB,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;iBACtC,MAAM,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CACjD,CACF,CAAC,IAAI,EAAE,CAAC;YACT,MAAM,KAAK,GAAG,IAAI,gBAAgB,EAAE,CAAC;YACrC,MAAM,eAAe,GAAG,KAAK,CAAC,mBAAmB,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;YACtE,MAAM,uBAAuB,GAC3B,KAAK,CAAC,mCAAmC,CAAC,OAAO,EAAE,iBAAiB,CAAC,CAAC;YAExE,MAAM,QAAQ,GAAG,IAAI,GAAG,CACtB,eAAe,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CACnD,CAAC;YACF,MAAM,gBAAgB,GAAG,IAAI,GAAG,CAC9B,uBAAuB;iBACpB,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC;iBAClD,MAAM,CAAC,CAAC,UAAU,EAAwB,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,CAC9D,CAAC;YAEF,MAAM,qBAAqB,GAAG,IAAI,GAAG,EAGlC,CAAC;YACJ,KAAK,MAAM,UAAU,IAAI,eAAe,EAAE,CAAC;gBACzC,qBAAqB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACvD,CAAC;YACD,KAAK,MAAM,UAAU,IAAI,uBAAuB,EAAE,CAAC;gBACjD,qBAAqB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YACvD,CAAC;YACD,MAAM,iBAAiB,GAAG,KAAK,CAAC,IAAI,CAClC,qBAAqB,CAAC,MAAM,EAAE,CAC/B,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC,CAAC;YAE5C,MAAM,QAAQ,GAAsB,EAAE,CAAC;YACvC,MAAM,UAAU,GAAG,SAAS,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/D,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,QAAQ,CAAC,IAAI,CAAC;oBACZ,YAAY,EAAE,SAAS;oBACvB,KAAK,EAAE,+CAA+C,qBAAqB,EAAE;iBAC9E,CAAC,CAAC;YACL,CAAC;YACD,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,MAAM,CACjD,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC,gBAAgB,CAAC,GAAG,CAAC,UAAU,CAAC,CAClD,CAAC;YACF,KAAK,MAAM,iBAAiB,IAAI,kBAAkB,EAAE,CAAC;gBACnD,QAAQ,CAAC,IAAI,CAAC;oBACZ,KAAK,EAAE,uDAAuD,qBAAqB,EAAE;oBACrF,UAAU,EAAE,iBAAiB;iBAC9B,CAAC,CAAC;YACL,CAAC;YAED,MAAM,mBAAmB,GAAG,iBAAiB,CAAC,MAAM,CAClD,CAAC,UAAU,EAA0C,EAAE;gBACrD,IAAI,UAAU,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBACnC,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,QAAQ,CAAC,IAAI,CAAC;oBACZ,YAAY,EAAE,UAAU,CAAC,EAAE;oBAC3B,KAAK,EACH,wLAAwL;oBAC1L,QAAQ,EAAE,UAAU,CAAC,QAAQ;iBAC9B,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC;YACf,CAAC,CACF,CAAC;YAEF,IAAI,mBAAmB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACrC,OAAO;oBACL,SAAS,EAAE,EAAsB;oBACjC,QAAQ;oBACR,KAAK,EAAE,CAAC;oBACR,QAAQ,EAAE,qBAAqB;iBAChC,CAAC;YACJ,CAAC;YAED,MAAM,mBAAmB,GAAG,mBAAmB,CAAC,MAAM,CACpD,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,SAAS,KAAK,KAAK,CAC/C,CAAC;YACF,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACnC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC;oBACvB,MAAM,IAAI,KAAK,CACb,0DAA0D,CAC3D,CAAC;gBACJ,CAAC;gBACD,WAAW,GAAG,MAAM,cAAc,CAAC;oBACjC,aAAa,EAAE,QAAQ,CAAC,aAAa;oBACrC,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG;oBACtB,kBAAkB,EAAE,QAAQ,CAAC,kBAAkB;iBAChD,CAAC,CAAC;YACL,CAAC;YAED,MAAM,eAAe,GACnB,mBAAmB,CAAC,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE;gBACrC,MAAM,gBAAgB,GAAG,yBAAyB,CAAC;oBACjD,QAAQ,EAAE,UAAU,CAAC,QAAQ;oBAC7B,SAAS,EAAE,UAAU,CAAC,SAAS;oBAC/B,IAAI,EAAE,UAAU;iBACjB,CAAC,CAAC;gBACH,MAAM,SAAS,GACb,UAAU,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC;gBAEjD,OAAO;oBACL,aAAa,EACX,SAAS,KAAK,KAAK;wBACjB,CAAC,CAAC,CAAC,UAAU,CAAC,aAAa,IAAI,QAAQ,CAAC,aAAa,CAAC;wBACtD,CAAC,CAAC,SAAS;oBACf,QAAQ,EAAE,gBAAgB;oBAC1B,SAAS,EAAE,UAAU,CAAC,SAAS;oBAC/B,SAAS;oBACT,UAAU,EAAE,UAAU,CAAC,UAAU;oBACjC,GAAG,EAAE,yBAAyB,CAAC,UAAU,CAAC,GAAG,CAAC;iBAC/C,CAAC;YACJ,CAAC,CAAC,CAAC;YAEL,MAAM,aAAa,GAAG,MAAM,QAAQ,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE;gBACnE,OAAO,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC,uBAAuB,CAAC;oBACrD,WAAW,EAAE,eAAe;oBAC5B,WAAW;oBACX,kBAAkB,EAAE,QAAQ,CAAC,kBAAkB;iBAChD,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;YAEH,MAAM,mBAAmB,GAAG,aAAa,EAAE,IAAI,EAAE,WAAW,CAAC;YAC7D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,mBAAmB,CAAC,EAAE,CAAC;gBACxC,MAAM,IAAI,kBAAkB,CAC1B,uEAAuE,EACvE;oBACE,OAAO,EAAE;wBACP,eAAe,EAAE,eAAe,CAAC,MAAM;qBACxC;oBACD,KAAK,EAAE,EAAE,aAAa,EAAE,WAAW,EAAE;oBACrC,QAAQ,EAAE,OAAO;oBACjB,QAAQ,EAAE,qBAAqB;iBAChC,CACF,CAAC;YACJ,CAAC;YAED,MAAM,aAAa,GAAG,IAAI,GAAG,CAC3B,mBAAmB;iBAChB,MAAM,CACL,CAAC,UAAU,EAAqD,EAAE,CAChE,OAAO,UAAU,EAAE,GAAG,KAAK,QAAQ;gBACnC,OAAO,UAAU,EAAE,SAAS,KAAK,QAAQ,CAC5C;iBACA,GAAG,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC;gBACnB,yBAAyB,CAAC,UAAU,CAAC,GAAG,CAAC;gBACzC,UAAU;aACX,CAAC,CACL,CAAC;YAEF,MAAM,KAAK,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,UAAU,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YAC1D,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAAC;YACjE,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAChC,QAAQ,CAAC,QAAQ,CAAC,cAAc,EAChC,gBAAgB,CACjB,CAAC;YACF,MAAM,KAAK,CAAC,gBAAgB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAEnD,MAAM,SAAS,GAAqB,EAAE,CAAC;YAEvC,KAAK,MAAM,UAAU,IAAI,mBAAmB,EAAE,CAAC;gBAC7C,MAAM,YAAY,GAAG,yBAAyB,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;gBAC/D,MAAM,QAAQ,GAAG,aAAa,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;gBACjD,MAAM,gBAAgB,GAAG,yBAAyB,CAAC;oBACjD,QAAQ,EAAE,UAAU,CAAC,QAAQ;oBAC7B,SAAS,EAAE,UAAU,CAAC,SAAS;oBAC/B,IAAI,EAAE,UAAU;iBACjB,CAAC,CAAC;gBAEH,IAAI,CAAC,QAAQ,EAAE,CAAC;oBACd,QAAQ,CAAC,IAAI,CAAC;wBACZ,YAAY,EAAE,UAAU,CAAC,EAAE;wBAC3B,KAAK,EAAE,yBAAyB;wBAChC,QAAQ,EAAE,gBAAgB;qBAC3B,CAAC,CAAC;oBACH,SAAS;gBACX,CAAC;gBAED,IAAI,CAAC;oBACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;oBACjD,IAAI,CAAC,QAAQ,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;wBACnC,MAAM,IAAI,KAAK,CACb,kCAAkC,QAAQ,CAAC,MAAM,GAAG,CACrD,CAAC;oBACJ,CAAC;oBAED,MAAM,QAAQ,GAAG,gBAAgB,CAAC,gBAAgB,CAAC,CAAC;oBACpD,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CACpC,gBAAgB,EAChB,GAAG,UAAU,CAAC,EAAE,IAAI,QAAQ,EAAE,CAC/B,CAAC;oBACF,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CACpC,QAAQ,CAAC,QAAQ,CAAC,cAAc,EAChC,oBAAoB,CACrB,CAAC;oBACF,MAAM,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,oBAAoB,CAAC,EAAE;wBAC9C,SAAS,EAAE,IAAI;qBAChB,CAAC,CAAC;oBAEH,iEAAiE;oBACjE,iEAAiE;oBACjE,uEAAuE;oBACvE,+DAA+D;oBAC/D,mEAAmE;oBACnE,8BAA8B;oBAC9B,MAAM,QAAQ,CACZ,QAAQ,CAAC,OAAO,CACd,QAAQ,CAAC,IAA8C,CACxD,EACD,sBAAsB,CACpB,kBAAkB,EAClB,6CAA6C,kBAAkB,SAAS,CACzE,EACD,iBAAiB,CAAC,oBAAoB,CAAC,CACxC,CAAC;oBAEF,MAAM,WAAW,GAAG,2BAA2B,CAAC;wBAC9C,QAAQ,EAAE,gBAAgB;wBAC1B,QAAQ,EAAE,UAAU,CAAC,SAAS;qBAC/B,CAAC,CAAC;oBAEH,IAAI,CAAC,WAAW,EAAE,CAAC;wBACjB,SAAS,CAAC,IAAI,CAAC;4BACb,YAAY,EAAE,UAAU,CAAC,EAAE;4BAC3B,YAAY,EAAE,oBAAoB;4BAClC,SAAS,EAAE,KAAK;4BAChB,QAAQ,EAAE,gBAAgB;4BAC1B,SAAS,EAAE,UAAU,CAAC,SAAS;4BAC/B,SAAS,EAAE,UAAU,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK;4BACzD,WAAW,EAAE,oBAAoB;yBAClC,CAAC,CAAC;wBACH,SAAS;oBACX,CAAC;oBAED,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CACrC,gBAAgB,EAChB,GAAG,UAAU,CAAC,EAAE,IAAI,QAAQ,YAAY,CACzC,CAAC;oBACF,MAAM,qBAAqB,GAAG,IAAI,CAAC,IAAI,CACrC,QAAQ,CAAC,QAAQ,CAAC,cAAc,EAChC,qBAAqB,CACtB,CAAC;oBACF,MAAM,KAAK,CAAC,qBAAqB,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAExD,IAAI,mBAAmB,GAAG,CAAC,CAAC;oBAC5B,IAAI,WAAW,GAAG,qBAAqB,CAAC;oBAExC,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;wBAC1B,mBAAmB,GAAG,MAAM,iBAAiB,CAAC;4BAC5C,WAAW,EAAE,oBAAoB;4BACjC,cAAc,EAAE,qBAAqB;yBACtC,CAAC,CAAC;oBACL,CAAC;yBAAM,IAAI,WAAW,KAAK,KAAK,EAAE,CAAC;wBACjC,mBAAmB,GAAG,MAAM,iBAAiB,CAAC;4BAC5C,WAAW,EAAE,oBAAoB;4BACjC,cAAc,EAAE,qBAAqB;yBACtC,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,MAAM,SAAS,GAAG,MAAM,kBAAkB,CAAC;4BACzC,WAAW,EAAE,oBAAoB;4BACjC,cAAc,EAAE,qBAAqB;4BACrC,QAAQ,EAAE,gBAAgB;yBAC3B,CAAC,CAAC;wBACH,mBAAmB,GAAG,SAAS,CAAC,cAAc,CAAC;wBAC/C,WAAW,GAAG,IAAI,CAAC,QAAQ,CACzB,QAAQ,CAAC,QAAQ,CAAC,cAAc,EAChC,SAAS,CAAC,UAAU,CACrB,CAAC;oBACJ,CAAC;oBAED,SAAS,CAAC,IAAI,CAAC;wBACb,WAAW;wBACX,YAAY,EAAE,UAAU,CAAC,EAAE;wBAC3B,YAAY,EAAE,oBAAoB;wBAClC,SAAS,EAAE,IAAI;wBACf,mBAAmB;wBACnB,QAAQ,EAAE,gBAAgB;wBAC1B,SAAS,EAAE,UAAU,CAAC,SAAS;wBAC/B,SAAS,EAAE,UAAU,CAAC,SAAS,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK;wBACzD,WAAW;qBACZ,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,SAAS,EAAE,CAAC,IAAI,CAAC,0CAA0C,EAAE;wBAC3D,YAAY,EAAE,UAAU,CAAC,EAAE;wBAC3B,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;qBAC9D,CAAC,CAAC;oBACH,QAAQ,CAAC,IAAI,CAAC;wBACZ,YAAY,EAAE,UAAU,CAAC,EAAE;wBAC3B,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,iBAAiB;wBACjE,QAAQ,EAAE,gBAAgB;qBAC3B,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;YAED,MAAM,CAAC,IAAI,CACT,cAAc,SAAS,CAAC,MAAM,mBAAmB,QAAQ,CAAC,MAAM,aAAa,CAC9E,CAAC;YAEF,OAAO;gBACL,SAAS;gBACT,QAAQ;gBACR,KAAK,EAAE,SAAS,CAAC,MAAM;gBACvB,QAAQ,EAAE,gBAAgB;aAC3B,CAAC;QACJ,CAAC;QACD,UAAU,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE;YAC7B,MAAM,KAAK,GACT,OAAO,MAAM,KAAK,QAAQ;gBAC1B,MAAM;gBACN,OAAO,IAAI,MAAM;gBACjB,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;gBAC9B,CAAC,CAAC,MAAM,CAAC,KAAK;gBACd,CAAC,CAAC,CAAC,CAAC;YACR,OAAO;gBACL,MAAM,EAAE,GAAG,KAAK,cAAc,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,aAAa;aAClE,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC,CACF,CAAC"}

package/dist/ai-service/agent/tools2/tools/edit.d.ts

@@ -0,0 +1,17 @@
+import type { AppShell } from "../../../app-interface/shell.js";
+/**
+ * Edit tool - Performs exact string replacements in files.
+ */
+export declare const editToolFactory: import("../types.js").ToolFactory<{
+    appShell: AppShell;
+}, import("../types.js").Tool<{
+    file_path: string;
+    old_string: string;
+    new_string: string;
+    replace_all?: boolean | undefined;
+}, {
+    success: boolean;
+    message: string;
+    replacements: number;
+}>>;
+//# sourceMappingURL=edit.d.ts.map

package/dist/ai-service/agent/tools2/tools/edit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit.d.ts","sourceRoot":"","sources":["../../../../../src/ai-service/agent/tools2/tools/edit.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iCAAiC,CAAC;AAQhE;;GAEG;AACH,eAAO,MAAM,eAAe;cAEH,QAAQ;;;;;;;;;;GAgEhC,CAAC"}

package/dist/ai-service/agent/tools2/tools/edit.js

@@ -0,0 +1,67 @@
+import { z } from "zod";
+import { createToolFactory, PermissionLevel, ToolCategory, ToolExecutionError, } from "../types.js";
+/**
+ * Edit tool - Performs exact string replacements in files.
+ */
+export const editToolFactory = createToolFactory("edit", (services) => ({
+    description: "Performs exact string replacements in files.",
+    category: ToolCategory.FILE,
+    defaultPermission: PermissionLevel.ALLOW,
+    inputSchema: z.object({
+        file_path: z.string().describe("Relative path to file"),
+        old_string: z.string().describe("Text to replace"),
+        new_string: z.string().describe("Text to replace with"),
+        replace_all: z
+            .boolean()
+            .optional()
+            .default(false)
+            .describe("Replace all occurrences (default: false)"),
+    }),
+    execute: async ({ file_path, old_string, new_string, replace_all }) => {
+        try {
+            const content = await services.appShell.readFile(file_path);
+            let newContent;
+            if (replace_all) {
+                newContent = content.split(old_string).join(new_string);
+            }
+            else {
+                const index = content.indexOf(old_string);
+                if (index === -1) {
+                    throw new Error(`String not found in file: ${old_string}`);
+                }
+                const secondIndex = content.indexOf(old_string, index + 1);
+                if (secondIndex !== -1) {
+                    throw new Error("String is not unique in file. Use replace_all=true or provide more context.");
+                }
+                newContent =
+                    content.substring(0, index) +
+                        new_string +
+                        content.substring(index + old_string.length);
+            }
+            await services.appShell.writeFile(file_path, newContent);
+            return {
+                success: true,
+                message: `File edited: ${file_path}`,
+                replacements: replace_all ? content.split(old_string).length - 1 : 1,
+            };
+        }
+        catch (error) {
+            if (!(error instanceof Error)) {
+                throw new Error("Failed to edit file: Unknown error");
+            }
+            throw new ToolExecutionError(`Failed to edit file: ${error.message}`, {
+                toolName: "",
+                input: {},
+                context: {},
+                severity: "warn",
+            });
+        }
+    },
+    getSummary: (_input, output) => {
+        const count = output.replacements ?? 1;
+        return {
+            output: `${count} replacement${count === 1 ? "" : "s"}`,
+        };
+    },
+}));
+//# sourceMappingURL=edit.js.map

package/dist/ai-service/agent/tools2/tools/edit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"edit.js","sourceRoot":"","sources":["../../../../../src/ai-service/agent/tools2/tools/edit.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,OAAO,EACL,iBAAiB,EACjB,eAAe,EACf,YAAY,EACZ,kBAAkB,GACnB,MAAM,aAAa,CAAC;AAErB;;GAEG;AACH,MAAM,CAAC,MAAM,eAAe,GAAG,iBAAiB,CAC9C,MAAM,EACN,CAAC,QAAgC,EAAE,EAAE,CAAC,CAAC;IACrC,WAAW,EAAE,8CAA8C;IAC3D,QAAQ,EAAE,YAAY,CAAC,IAAI;IAC3B,iBAAiB,EAAE,eAAe,CAAC,KAAK;IACxC,WAAW,EAAE,CAAC,CAAC,MAAM,CAAC;QACpB,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;QACvD,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;QAClD,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,sBAAsB,CAAC;QACvD,WAAW,EAAE,CAAC;aACX,OAAO,EAAE;aACT,QAAQ,EAAE;aACV,OAAO,CAAC,KAAK,CAAC;aACd,QAAQ,CAAC,0CAA0C,CAAC;KACxD,CAAC;IACF,OAAO,EAAE,KAAK,EAAE,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,EAAE,EAAE,EAAE;QACpE,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;YAE5D,IAAI,UAAkB,CAAC;YACvB,IAAI,WAAW,EAAE,CAAC;gBAChB,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YAC1D,CAAC;iBAAM,CAAC;gBACN,MAAM,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;gBAC1C,IAAI,KAAK,KAAK,CAAC,CAAC,EAAE,CAAC;oBACjB,MAAM,IAAI,KAAK,CAAC,6BAA6B,UAAU,EAAE,CAAC,CAAC;gBAC7D,CAAC;gBACD,MAAM,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC,UAAU,EAAE,KAAK,GAAG,CAAC,CAAC,CAAC;gBAC3D,IAAI,WAAW,KAAK,CAAC,CAAC,EAAE,CAAC;oBACvB,MAAM,IAAI,KAAK,CACb,6EAA6E,CAC9E,CAAC;gBACJ,CAAC;gBACD,UAAU;oBACR,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC;wBAC3B,UAAU;wBACV,OAAO,CAAC,SAAS,CAAC,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC;YACjD,CAAC;YAED,MAAM,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;YAEzD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE,gBAAgB,SAAS,EAAE;gBACpC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;aACrE,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;YACxD,CAAC;YACD,MAAM,IAAI,kBAAkB,CAAC,wBAAwB,KAAK,CAAC,OAAO,EAAE,EAAE;gBACpE,QAAQ,EAAE,EAAE;gBACZ,KAAK,EAAE,EAAE;gBACT,OAAO,EAAE,EAAE;gBACX,QAAQ,EAAE,MAAM;aACjB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IACD,UAAU,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE;QAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,IAAI,CAAC,CAAC;QACvC,OAAO;YACL,MAAM,EAAE,GAAG,KAAK,eAAe,KAAK,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE;SACxD,CAAC;IACJ,CAAC;CACF,CAAC,CACH,CAAC"}