@nodetool-ai/websocket 0.7.0-rc.23 → 0.8.2

package/dist/lib/zip-stream.js

@@ -0,0 +1,140 @@
+/**
+ * Streaming zip output for the export surfaces (timeline and storyboard zips).
+ *
+ * An export used to load every asset into memory and build the archive with
+ * `zipSync`, so a timeline naming a two-hour video held the video, and then
+ * the whole archive, in memory. Here each entry is written as it is read: a
+ * local file (a managed asset under the local store, or an external asset in
+ * place) is streamed from disk, and only a backend with no local file hands
+ * over bytes. The archive itself goes out as a stream.
+ *
+ * fflate writes no Zip64 records, so an entry is limited to 4 GiB; callers
+ * check {@link MAX_ZIP_ENTRY_BYTES} and leave a larger file out.
+ */
+import { createHash } from "node:crypto";
+import { createReadStream } from "node:fs";
+import { PassThrough } from "node:stream";
+import { Zip, ZipDeflate, ZipPassThrough } from "fflate";
+/** Largest entry fflate can write without Zip64. */
+export const MAX_ZIP_ENTRY_BYTES = 0xffff_ffff;
+export function isLocalFileSource(source) {
+    return !(source instanceof Uint8Array);
+}
+export function exportSourceSize(source) {
+    return isLocalFileSource(source) ? source.size : source.byteLength;
+}
+/** Chunks of `source`, read from disk for a local file. */
+async function* readSource(source) {
+    if (!isLocalFileSource(source)) {
+        yield source;
+        return;
+    }
+    for await (const chunk of createReadStream(source.path)) {
+        yield chunk;
+    }
+}
+/** Hex sha256 of `source`, streaming a local file. */
+export async function sha256OfSource(source) {
+    const hash = createHash("sha256");
+    for await (const chunk of readSource(source)) {
+        hash.update(chunk);
+    }
+    return hash.digest("hex");
+}
+/**
+ * Writes zip entries one after another to a sink, waiting on the sink between
+ * chunks so a slow reader holds back the file reads.
+ */
+export class ZipStreamWriter {
+    sink;
+    zip;
+    pending = [];
+    failure = null;
+    constructor(sink) {
+        this.sink = sink;
+        this.zip = new Zip((err, chunk) => {
+            if (err) {
+                this.failure = err;
+                return;
+            }
+            this.pending.push(chunk);
+        });
+    }
+    async flush() {
+        if (this.failure)
+            throw this.failure;
+        while (this.pending.length > 0) {
+            const chunk = this.pending.shift();
+            await this.sink(chunk);
+        }
+    }
+    /**
+     * Add one entry. `compress` deflates it, which suits text; media is already
+     * compressed and is stored as-is.
+     */
+    async add(name, source, options = {}) {
+        const entry = options.compress
+            ? new ZipDeflate(name, { level: 6 })
+            : new ZipPassThrough(name);
+        this.zip.add(entry);
+        for await (const chunk of readSource(source)) {
+            entry.push(chunk, false);
+            await this.flush();
+        }
+        entry.push(new Uint8Array(0), true);
+        await this.flush();
+    }
+    /** Write the central directory. No entry may be added afterwards. */
+    async finish() {
+        this.zip.end();
+        await this.flush();
+    }
+}
+/** Run `write` against a writer that collects the whole archive in memory. */
+export async function collectZip(write) {
+    const chunks = [];
+    const writer = new ZipStreamWriter(async (chunk) => {
+        chunks.push(chunk);
+    });
+    await write(writer);
+    await writer.finish();
+    return new Uint8Array(Buffer.concat(chunks));
+}
+/**
+ * A readable stream of the archive `write` produces. Writing starts at once
+ * and waits whenever the reader falls behind; a failure or a reader that goes
+ * away destroys the stream, which ends the download early.
+ */
+export function streamZip(write) {
+    const out = new PassThrough();
+    const sink = (chunk) => new Promise((resolve, reject) => {
+        if (out.destroyed) {
+            reject(new Error("zip reader went away"));
+            return;
+        }
+        if (out.write(chunk)) {
+            resolve();
+            return;
+        }
+        const onDrain = () => {
+            out.off("close", onClose);
+            resolve();
+        };
+        const onClose = () => {
+            out.off("drain", onDrain);
+            reject(new Error("zip reader went away"));
+        };
+        out.once("drain", onDrain);
+        out.once("close", onClose);
+    });
+    const writer = new ZipStreamWriter(sink);
+    void (async () => {
+        await write(writer);
+        await writer.finish();
+        out.end();
+    })().catch((err) => {
+        out.destroy(err instanceof Error ? err : new Error(String(err)));
+    });
+    return out;
+}
+//# sourceMappingURL=zip-stream.js.map

package/dist/lib/zip-stream.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"zip-stream.js","sourceRoot":"","sources":["../../src/lib/zip-stream.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;GAYG;AACH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,OAAO,EAAE,WAAW,EAAY,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,GAAG,EAAE,UAAU,EAAE,cAAc,EAAE,MAAM,QAAQ,CAAC;AAWzD,oDAAoD;AACpD,MAAM,CAAC,MAAM,mBAAmB,GAAG,WAAW,CAAC;AAE/C,MAAM,UAAU,iBAAiB,CAC/B,MAAoB;IAEpB,OAAO,CAAC,CAAC,MAAM,YAAY,UAAU,CAAC,CAAC;AACzC,CAAC;AAED,MAAM,UAAU,gBAAgB,CAAC,MAAoB;IACnD,OAAO,iBAAiB,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC;AACrE,CAAC;AAED,2DAA2D;AAC3D,KAAK,SAAS,CAAC,CAAC,UAAU,CAAC,MAAoB;IAC7C,IAAI,CAAC,iBAAiB,CAAC,MAAM,CAAC,EAAE,CAAC;QAC/B,MAAM,MAAM,CAAC;QACb,OAAO;IACT,CAAC;IACD,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;QACxD,MAAM,KAAe,CAAC;IACxB,CAAC;AACH,CAAC;AAED,sDAAsD;AACtD,MAAM,CAAC,KAAK,UAAU,cAAc,CAAC,MAAoB;IACvD,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IAClC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC;IACD,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;AAC5B,CAAC;AAKD;;;GAGG;AACH,MAAM,OAAO,eAAe;IAKG,IAAI;IAJhB,GAAG,CAAM;IAClB,OAAO,GAAiB,EAAE,CAAC;IAC3B,OAAO,GAAiB,IAAI,CAAC;IAErC,YAA6B,IAAkB;oBAAlB,IAAI;QAC/B,IAAI,CAAC,GAAG,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE;YAChC,IAAI,GAAG,EAAE,CAAC;gBACR,IAAI,CAAC,OAAO,GAAG,GAAG,CAAC;gBACnB,OAAO;YACT,CAAC;YACD,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,KAAK;QACjB,IAAI,IAAI,CAAC,OAAO;YAAE,MAAM,IAAI,CAAC,OAAO,CAAC;QACrC,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAG,CAAC;YACpC,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,GAAG,CACP,IAAY,EACZ,MAAoB,EACpB,OAAO,GAA2B,EAAE;QAEpC,MAAM,KAAK,GAAgC,OAAO,CAAC,QAAQ;YACzD,CAAC,CAAC,IAAI,UAAU,CAAC,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;YACpC,CAAC,CAAC,IAAI,cAAc,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACpB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YAC7C,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACzB,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;QACpC,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;IAED,qEAAqE;IACrE,KAAK,CAAC,MAAM;QACV,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;QACf,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;IACrB,CAAC;CACF;AAED,8EAA8E;AAC9E,MAAM,CAAC,KAAK,UAAU,UAAU,CAC9B,KAAiD;IAEjD,MAAM,MAAM,GAAiB,EAAE,CAAC;IAChC,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;QACjD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrB,CAAC,CAAC,CAAC;IACH,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC;IACpB,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;IACtB,OAAO,IAAI,UAAU,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC;AAC/C,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,SAAS,CACvB,KAAiD;IAEjD,MAAM,GAAG,GAAG,IAAI,WAAW,EAAE,CAAC;IAC9B,MAAM,IAAI,GAAiB,CAAC,KAAK,EAAE,EAAE,CACnC,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QACpC,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YAClB,MAAM,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC,CAAC;YAC1C,OAAO;QACT,CAAC;QACD,IAAI,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;YACrB,OAAO,EAAE,CAAC;YACV,OAAO;QACT,CAAC;QACD,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1B,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC;QACF,MAAM,OAAO,GAAG,GAAS,EAAE;YACzB,GAAG,CAAC,GAAG,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC1B,MAAM,CAAC,IAAI,KAAK,CAAC,sBAAsB,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC;QACF,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAC3B,GAAG,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAC7B,CAAC,CAAC,CAAC;IACL,MAAM,MAAM,GAAG,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC;IACzC,KAAK,CAAC,KAAK,IAAI,EAAE;QACf,MAAM,KAAK,CAAC,MAAM,CAAC,CAAC;QACpB,MAAM,MAAM,CAAC,MAAM,EAAE,CAAC;QACtB,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,GAAY,EAAE,EAAE;QAC1B,GAAG,CAAC,OAAO,CAAC,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;IACnE,CAAC,CAAC,CAAC;IACH,OAAO,GAAG,CAAC;AACb,CAAC"}

package/dist/mcp-agent-tools.d.ts

@@ -0,0 +1,56 @@
+/**
+ * Bridges the canonical agent tools (`@nodetool-ai/agents`) onto the MCP server
+ * so external agents (Claude Code, ChatGPT, …) get the same surface the in-app
+ * chat agent runs on.
+ *
+ * That surface is CodeAct, not a flat catalog. `execute_code` is built by the
+ * same {@link createChatCodeActSession} the chat runner uses. `view_image` and
+ * asset transfer stay direct because pixels and file bytes need MCP channels.
+ * Every other capability lives inside the sandbox as `tools.<name>()` and the
+ * `nodetool.*` object model, found with `nodetool.searchTools()` and catalogued
+ * on `nodetool://capabilities` and `nodetool://sandbox`.
+ *
+ * The bridged set is *derived*, not hand-listed: `getBuiltinTools()` plus
+ * `getAllMcpTools()` plus the Google Workspace tools are exactly what
+ * `websocket-client-session` assembles for a chat turn, so a tool added to
+ * either catalog reaches the sandbox belt with no edit here. Only tools whose
+ * constructor needs something the catalogs can't supply (a timeline loader, the
+ * lazily probed provider map) are named
+ * individually below.
+ */
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { ProcessingContext } from "@nodetool-ai/runtime";
+import { type PermissionGateOptions, type CapabilityRun, MCP_SANDBOX_RESOURCE_URI } from "@nodetool-ai/agents";
+import type { McpServerOptions } from "./mcp-server.js";
+/**
+ * Per-user workspace directory the bridged file tools (read_file, write_file,
+ * glob, grep, …) are rooted at. Kept under the NodeTool data dir rather than a
+ * temp dir because an MCP session is long-lived and spans many calls — a file
+ * written on one call has to still be there on the next. One directory per
+ * user, so a multi-user mount cannot let one caller read another's files.
+ */
+declare function mcpWorkspaceDir(userId: string): string;
+/** Test-only: expose the workspace path rule for isolation assertions. */
+export declare const __mcpWorkspaceDirForTests: typeof mcpWorkspaceDir;
+/** Test-only: expose the per-user context construction for scope assertions. */
+export declare const __buildAgentToolContextForTests: (userId: string) => ProcessingContext;
+/** URI of the structured capability catalog this mount publishes. */
+export declare const MCP_CAPABILITIES_RESOURCE_URI = "nodetool://capabilities";
+/** URI of the guest-JS surface this mount publishes. Re-exported for callers. */
+export { MCP_SANDBOX_RESOURCE_URI };
+/**
+ * Register the agent surface on `server`: CodeAct, image viewing, asset
+ * transfer, and the promoted direct tools.
+ *
+ * `gate` defaults to {@link mcpSessionGate} — `auto` with an approver that
+ * denies, because an MCP session has no user of its own to ask. A host that
+ * does have an approval channel passes its own gate instead.
+ *
+ * Everything else the catalogs offer becomes the sandbox belt instead of an MCP
+ * tool. It stays fully reachable — `tools.<name>()`, the `nodetool.*` object
+ * model, `await nodetool.searchTools("query")` — and the `execute_code` description
+ * carries the same contract and catalog the chat runner puts in its system
+ * prompt, because MCP has no system prompt to put it in.
+ */
+export declare function registerAgentMcpTools(server: McpServer, options: McpServerOptions, gate?: PermissionGateOptions): CapabilityRun;
+//# sourceMappingURL=mcp-agent-tools.d.ts.map

package/dist/mcp-agent-tools.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp-agent-tools.d.ts","sourceRoot":"","sources":["../src/mcp-agent-tools.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;GAmBG;AAEH,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AAGzE,OAAO,KAAK,EAAc,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AAU1E,OAAO,EAkBL,KAAK,qBAAqB,EAC1B,KAAK,aAAa,EAGlB,wBAAwB,EAKzB,MAAM,qBAAqB,CAAC;AAgC7B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AAkBxD;;;;;;GAMG;AACH,iBAAS,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,MAAM,CAK/C;AAED,0EAA0E;AAC1E,eAAO,MAAM,yBAAyB,wBAAkB,CAAC;AA0EzD,gFAAgF;AAChF,eAAO,MAAM,+BAA+B,WAClC,MAAM,KACb,iBAAkD,CAAC;AAuKtD,qEAAqE;AACrE,eAAO,MAAM,6BAA6B,4BAA4B,CAAC;AAEvE,iFAAiF;AACjF,OAAO,EAAE,wBAAwB,EAAE,CAAC;AA+cpC;;;;;;;;;;;;;GAaG;AACH,wBAAgB,qBAAqB,CACnC,MAAM,EAAE,SAAS,EACjB,OAAO,EAAE,gBAAgB,EACzB,IAAI,GAAE,qBAAwC,GAC7C,aAAa,CAyhBf"}