comfyui-mcp 0.7.0 → 0.8.0

package/dist/services/storage/hf.js

@@ -0,0 +1,80 @@
+import { execFile } from "node:child_process";
+import { mkdtemp, rm, writeFile } from "node:fs/promises";
+import { tmpdir } from "node:os";
+import { basename, join } from "node:path";
+import { promisify } from "node:util";
+import { ModelError, ValidationError } from "../../utils/errors.js";
+import { withPrefix } from "./utils.js";
+const execFileAsync = promisify(execFile);
+const ASCII_CONTROL_RE = /[\x00-\x1F\x7F]/;
+async function findHfCli() {
+    for (const command of ["hf", "huggingface-cli"]) {
+        try {
+            await execFileAsync(command, ["--version"]);
+            return command;
+        }
+        catch {
+            // Try the next known CLI name.
+        }
+    }
+    throw new ValidationError("hf upload requires the hf CLI. Install and authenticate `hf` or `huggingface-cli` first.");
+}
+function validateRepo(repo) {
+    if (!/^[A-Za-z0-9][A-Za-z0-9._-]*\/[A-Za-z0-9][A-Za-z0-9._-]*$/.test(repo)) {
+        throw new ValidationError("hf.repo must be in owner/name format.");
+    }
+}
+function validateRemotePath(path) {
+    if (ASCII_CONTROL_RE.test(path)) {
+        throw new ValidationError("hf remote path cannot contain ASCII control characters.");
+    }
+    for (const segment of path.split("/")) {
+        if (segment === "..") {
+            throw new ValidationError("hf remote path cannot contain '..' segments.");
+        }
+        if (segment.startsWith("-")) {
+            throw new ValidationError("hf remote path segments cannot start with '-'.");
+        }
+    }
+}
+export async function uploadHfFile(source, destination) {
+    validateRepo(destination.repo);
+    const repoType = destination.repo_type ?? "model";
+    const remotePath = withPrefix(destination.path, source.filename);
+    validateRemotePath(remotePath);
+    const command = await findHfCli();
+    let tempDir;
+    let localPath = source.path;
+    try {
+        if (!localPath) {
+            tempDir = await mkdtemp(join(tmpdir(), "comfyui-mcp-hf-upload-"));
+            localPath = join(tempDir, basename(source.filename));
+            await writeFile(localPath, source.data ?? Buffer.alloc(0));
+        }
+        const args = ["upload"];
+        if (repoType !== "model")
+            args.push("--repo-type", repoType);
+        args.push("--", destination.repo, localPath, remotePath);
+        await execFileAsync(command, args);
+        const prefix = repoType === "dataset" ? "datasets/" : repoType === "space" ? "spaces/" : "";
+        return {
+            provider: "hf",
+            url: `https://huggingface.co/${prefix}${destination.repo}/blob/main/${remotePath}`,
+        };
+    }
+    catch (err) {
+        if (err instanceof ValidationError)
+            throw err;
+        throw new ModelError("hf upload failed", {
+            name: err instanceof Error ? err.name : undefined,
+            code: typeof err?.code === "string"
+                ? err.code
+                : undefined,
+        });
+    }
+    finally {
+        if (tempDir)
+            await rm(tempDir, { recursive: true, force: true }).catch(() => undefined);
+    }
+}
+//# sourceMappingURL=hf.js.map

package/dist/services/storage/hf.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"hf.js","sourceRoot":"","sources":["../../../src/services/storage/hf.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC1D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAC3C,OAAO,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACtC,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAEpE,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAExC,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,CAAC;AAC1C,MAAM,gBAAgB,GAAG,iBAAiB,CAAC;AAE3C,KAAK,UAAU,SAAS;IACtB,KAAK,MAAM,OAAO,IAAI,CAAC,IAAI,EAAE,iBAAiB,CAAU,EAAE,CAAC;QACzD,IAAI,CAAC;YACH,MAAM,aAAa,CAAC,OAAO,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC;YAC5C,OAAO,OAAO,CAAC;QACjB,CAAC;QAAC,MAAM,CAAC;YACP,+BAA+B;QACjC,CAAC;IACH,CAAC;IACD,MAAM,IAAI,eAAe,CACvB,0FAA0F,CAC3F,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,IAAY;IAChC,IAAI,CAAC,0DAA0D,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAC3E,MAAM,IAAI,eAAe,CAAC,uCAAuC,CAAC,CAAC;IACrE,CAAC;AACH,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY;IACtC,IAAI,gBAAgB,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAChC,MAAM,IAAI,eAAe,CAAC,yDAAyD,CAAC,CAAC;IACvF,CAAC;IAED,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE,CAAC;QACtC,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;YACrB,MAAM,IAAI,eAAe,CAAC,8CAA8C,CAAC,CAAC;QAC5E,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,eAAe,CAAC,gDAAgD,CAAC,CAAC;QAC9E,CAAC;IACH,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,MAA2B,EAC3B,WAAuF;IAEvF,YAAY,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IAC/B,MAAM,QAAQ,GAAG,WAAW,CAAC,SAAS,IAAI,OAAO,CAAC;IAClD,MAAM,UAAU,GAAG,UAAU,CAAC,WAAW,CAAC,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;IACjE,kBAAkB,CAAC,UAAU,CAAC,CAAC;IAC/B,MAAM,OAAO,GAAG,MAAM,SAAS,EAAE,CAAC;IAClC,IAAI,OAA2B,CAAC;IAChC,IAAI,SAAS,GAAG,MAAM,CAAC,IAAI,CAAC;IAC5B,IAAI,CAAC;QACH,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,wBAAwB,CAAC,CAAC,CAAC;YAClE,SAAS,GAAG,IAAI,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;YACrD,MAAM,SAAS,CAAC,SAAS,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;QAC7D,CAAC;QAED,MAAM,IAAI,GAAG,CAAC,QAAQ,CAAC,CAAC;QACxB,IAAI,QAAQ,KAAK,OAAO;YAAE,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;QAC7D,IAAI,CAAC,IAAI,CAAC,IAAI,EAAE,WAAW,CAAC,IAAI,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QACzD,MAAM,aAAa,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QAEnC,MAAM,MAAM,GAAG,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,KAAK,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC;QAC5F,OAAO;YACL,QAAQ,EAAE,IAAI;YACd,GAAG,EAAE,0BAA0B,MAAM,GAAG,WAAW,CAAC,IAAI,cAAc,UAAU,EAAE;SACnF,CAAC;IACJ,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,eAAe;YAAE,MAAM,GAAG,CAAC;QAC9C,MAAM,IAAI,UAAU,CAAC,kBAAkB,EAAE;YACvC,IAAI,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;YACjD,IAAI,EAAE,OAAQ,GAA0B,EAAE,IAAI,KAAK,QAAQ;gBACzD,CAAC,CAAE,GAAwB,CAAC,IAAI;gBAChC,CAAC,CAAC,SAAS;SACd,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,IAAI,OAAO;YAAE,MAAM,EAAE,CAAC,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC1F,CAAC;AACH,CAAC"}

package/dist/services/storage/http.d.ts

@@ -0,0 +1,6 @@
+import type { StorageUploadResult, StorageUploadSource } from "./types.js";
+export declare function isHttpUrl(url: string): boolean;
+export declare function uploadHttpFile(source: StorageUploadSource, destination: {
+    url: string;
+}): Promise<StorageUploadResult>;
+//# sourceMappingURL=http.d.ts.map

package/dist/services/storage/http.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.d.ts","sourceRoot":"","sources":["../../../src/services/storage/http.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAE3E,wBAAgB,SAAS,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAE9C;AAUD,wBAAsB,cAAc,CAClC,MAAM,EAAE,mBAAmB,EAC3B,WAAW,EAAE;IAAE,GAAG,EAAE,MAAM,CAAA;CAAE,GAC3B,OAAO,CAAC,mBAAmB,CAAC,CAyB9B"}

package/dist/services/storage/http.js

@@ -0,0 +1,41 @@
+import { createReadStream } from "node:fs";
+import { ModelError } from "../../utils/errors.js";
+import { redactUrlForLogs } from "../download-auth.js";
+export function isHttpUrl(url) {
+    return url.startsWith("http://") || url.startsWith("https://");
+}
+function redactedRedirectLocation(location, baseUrl) {
+    try {
+        return redactUrlForLogs(new URL(location, baseUrl).toString());
+    }
+    catch {
+        return "[INVALID_REDIRECT_LOCATION]";
+    }
+}
+export async function uploadHttpFile(source, destination) {
+    const init = {
+        method: "PUT",
+        redirect: "manual",
+        headers: source.contentType ? { "Content-Type": source.contentType } : undefined,
+        body: (source.path ? createReadStream(source.path) : source.data),
+    };
+    if (source.path)
+        init.duplex = "half";
+    const response = await fetch(destination.url, init);
+    if (response.status >= 300 && response.status < 400) {
+        const location = response.headers.get("location");
+        throw new ModelError(`HTTP upload redirect rejected: ${response.status} ${response.statusText}`, {
+            url: redactUrlForLogs(destination.url),
+            status: response.status,
+            location: location ? redactedRedirectLocation(location, destination.url) : undefined,
+        });
+    }
+    if (!response.ok) {
+        throw new ModelError(`HTTP upload failed: ${response.status} ${response.statusText}`, {
+            url: redactUrlForLogs(destination.url),
+            status: response.status,
+        });
+    }
+    return { provider: "http", url: redactUrlForLogs(destination.url) };
+}
+//# sourceMappingURL=http.js.map

package/dist/services/storage/http.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"http.js","sourceRoot":"","sources":["../../../src/services/storage/http.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,OAAO,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnD,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAGvD,MAAM,UAAU,SAAS,CAAC,GAAW;IACnC,OAAO,GAAG,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,GAAG,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;AACjE,CAAC;AAED,SAAS,wBAAwB,CAAC,QAAgB,EAAE,OAAe;IACjE,IAAI,CAAC;QACH,OAAO,gBAAgB,CAAC,IAAI,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,QAAQ,EAAE,CAAC,CAAC;IACjE,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,6BAA6B,CAAC;IACvC,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,MAA2B,EAC3B,WAA4B;IAE5B,MAAM,IAAI,GAAsC;QAC9C,MAAM,EAAE,KAAK;QACb,QAAQ,EAAE,QAAQ;QAClB,OAAO,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,SAAS;QAChF,IAAI,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAmC;KACpG,CAAC;IACF,IAAI,MAAM,CAAC,IAAI;QAAE,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IAEtC,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACpD,IAAI,QAAQ,CAAC,MAAM,IAAI,GAAG,IAAI,QAAQ,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;QACpD,MAAM,QAAQ,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;QAClD,MAAM,IAAI,UAAU,CAAC,kCAAkC,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE;YAC/F,GAAG,EAAE,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC;YACtC,MAAM,EAAE,QAAQ,CAAC,MAAM;YACvB,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,wBAAwB,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS;SACrF,CAAC,CAAC;IACL,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,UAAU,CAAC,uBAAuB,QAAQ,CAAC,MAAM,IAAI,QAAQ,CAAC,UAAU,EAAE,EAAE;YACpF,GAAG,EAAE,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC;YACtC,MAAM,EAAE,QAAQ,CAAC,MAAM;SACxB,CAAC,CAAC;IACL,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,GAAG,EAAE,gBAAgB,CAAC,WAAW,CAAC,GAAG,CAAC,EAAE,CAAC;AACtE,CAAC"}

package/dist/services/storage/index.d.ts

@@ -0,0 +1,28 @@
+import type { CloudStorageAuth, StorageUploadResult, StorageUploadSource } from "./types.js";
+export type UploadDestination = {
+    s3: {
+        bucket: string;
+        prefix?: string;
+        async?: boolean;
+    };
+} | {
+    azure: {
+        container: string;
+        blob_prefix?: string;
+    };
+} | {
+    http: {
+        url: string;
+    };
+} | {
+    hf: {
+        repo: string;
+        repo_type?: "model" | "dataset" | "space";
+        path?: string;
+    };
+};
+export declare function supportsCloudDownload(url: string): boolean;
+export declare function downloadCloudUrlToFile(url: string, targetPath: string, auth?: CloudStorageAuth): Promise<void>;
+export declare function uploadToStorage(source: StorageUploadSource, destination: UploadDestination, auth?: CloudStorageAuth): Promise<StorageUploadResult>;
+export type { CloudStorageAuth, S3Auth, StorageUploadResult, StorageUploadSource, } from "./types.js";
+//# sourceMappingURL=index.d.ts.map

package/dist/services/storage/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/services/storage/index.ts"],"names":[],"mappings":"AAKA,OAAO,KAAK,EACV,gBAAgB,EAChB,mBAAmB,EACnB,mBAAmB,EACpB,MAAM,YAAY,CAAC;AAEpB,MAAM,MAAM,iBAAiB,GACzB;IAAE,EAAE,EAAE;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,CAAA;KAAE,CAAA;CAAE,GAC5D;IAAE,KAAK,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,WAAW,CAAC,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GACtD;IAAE,IAAI,EAAE;QAAE,GAAG,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,GACzB;IAAE,EAAE,EAAE;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,SAAS,CAAC,EAAE,OAAO,GAAG,SAAS,GAAG,OAAO,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,CAAA;CAAE,CAAC;AAEvF,wBAAgB,qBAAqB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAE1D;AAED,wBAAsB,sBAAsB,CAC1C,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,MAAM,EAClB,IAAI,GAAE,gBAAqB,GAC1B,OAAO,CAAC,IAAI,CAAC,CAYf;AAED,wBAAsB,eAAe,CACnC,MAAM,EAAE,mBAAmB,EAC3B,WAAW,EAAE,iBAAiB,EAC9B,IAAI,GAAE,gBAAqB,GAC1B,OAAO,CAAC,mBAAmB,CAAC,CAuB9B;AAED,YAAY,EACV,gBAAgB,EAChB,MAAM,EACN,mBAAmB,EACnB,mBAAmB,GACpB,MAAM,YAAY,CAAC"}

package/dist/services/storage/index.js

@@ -0,0 +1,42 @@
+import { ValidationError } from "../../utils/errors.js";
+import { isAzureBlobUrl, downloadAzureBlobToFile, uploadAzureBlobFile } from "./azure-blob.js";
+import { uploadHfFile } from "./hf.js";
+import { isHttpUrl, uploadHttpFile } from "./http.js";
+import { downloadS3ToFile, isS3Url, uploadS3File } from "./s3.js";
+export function supportsCloudDownload(url) {
+    return isS3Url(url) || isAzureBlobUrl(url);
+}
+export async function downloadCloudUrlToFile(url, targetPath, auth = {}) {
+    if (isS3Url(url)) {
+        await downloadS3ToFile(url, targetPath, auth.s3);
+        return;
+    }
+    if (isAzureBlobUrl(url)) {
+        await downloadAzureBlobToFile(url, targetPath);
+        return;
+    }
+    throw new ValidationError("Unsupported cloud storage download URL. Expected s3://bucket/key or an Azure Blob URL.");
+}
+export async function uploadToStorage(source, destination, auth = {}) {
+    const keys = Object.keys(destination);
+    if (keys.length !== 1) {
+        throw new ValidationError("Provide exactly one upload destination: s3, azure, http, or hf.");
+    }
+    if ("s3" in destination) {
+        return uploadS3File(source, destination.s3, auth.s3);
+    }
+    if ("azure" in destination) {
+        return uploadAzureBlobFile(source, destination.azure);
+    }
+    if ("http" in destination) {
+        if (!isHttpUrl(destination.http.url)) {
+            throw new ValidationError("http.url must start with http:// or https://.");
+        }
+        return uploadHttpFile(source, destination.http);
+    }
+    if ("hf" in destination) {
+        return uploadHfFile(source, destination.hf);
+    }
+    throw new ValidationError("Unsupported upload destination. Expected s3, azure, http, or hf.");
+}
+//# sourceMappingURL=index.js.map

package/dist/services/storage/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/services/storage/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,uBAAuB,EAAE,mBAAmB,EAAE,MAAM,iBAAiB,CAAC;AAC/F,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AACvC,OAAO,EAAE,SAAS,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AACtD,OAAO,EAAE,gBAAgB,EAAE,OAAO,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAalE,MAAM,UAAU,qBAAqB,CAAC,GAAW;IAC/C,OAAO,OAAO,CAAC,GAAG,CAAC,IAAI,cAAc,CAAC,GAAG,CAAC,CAAC;AAC7C,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,sBAAsB,CAC1C,GAAW,EACX,UAAkB,EAClB,OAAyB,EAAE;IAE3B,IAAI,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACjB,MAAM,gBAAgB,CAAC,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;QACjD,OAAO;IACT,CAAC;IACD,IAAI,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,MAAM,uBAAuB,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;QAC/C,OAAO;IACT,CAAC;IACD,MAAM,IAAI,eAAe,CACvB,wFAAwF,CACzF,CAAC;AACJ,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,MAA2B,EAC3B,WAA8B,EAC9B,OAAyB,EAAE;IAE3B,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACtC,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACtB,MAAM,IAAI,eAAe,CAAC,iEAAiE,CAAC,CAAC;IAC/F,CAAC;IAED,IAAI,IAAI,IAAI,WAAW,EAAE,CAAC;QACxB,OAAO,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;IACvD,CAAC;IACD,IAAI,OAAO,IAAI,WAAW,EAAE,CAAC;QAC3B,OAAO,mBAAmB,CAAC,MAAM,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,MAAM,IAAI,WAAW,EAAE,CAAC;QAC1B,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YACrC,MAAM,IAAI,eAAe,CAAC,+CAA+C,CAAC,CAAC;QAC7E,CAAC;QACD,OAAO,cAAc,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,CAAC,CAAC;IAClD,CAAC;IACD,IAAI,IAAI,IAAI,WAAW,EAAE,CAAC;QACxB,OAAO,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,IAAI,eAAe,CAAC,kEAAkE,CAAC,CAAC;AAChG,CAAC"}

package/dist/services/storage/s3.d.ts

@@ -0,0 +1,12 @@
+import type { S3Auth, StorageUploadResult, StorageUploadSource } from "./types.js";
+export declare function isS3Url(url: string): boolean;
+export declare function parseS3Url(url: string): {
+    bucket: string;
+    key: string;
+};
+export declare function downloadS3ToFile(url: string, targetPath: string, auth?: S3Auth): Promise<void>;
+export declare function uploadS3File(source: StorageUploadSource, destination: {
+    bucket: string;
+    prefix?: string;
+}, auth?: S3Auth): Promise<StorageUploadResult>;
+//# sourceMappingURL=s3.d.ts.map

package/dist/services/storage/s3.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"s3.d.ts","sourceRoot":"","sources":["../../../src/services/storage/s3.ts"],"names":[],"mappings":"AAWA,OAAO,KAAK,EAAE,MAAM,EAAE,mBAAmB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAGnF,wBAAgB,OAAO,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAE5C;AAED,wBAAgB,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,GAAG,EAAE,MAAM,CAAA;CAAE,CAWvE;AAqBD,wBAAsB,gBAAgB,CACpC,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,MAAM,EAClB,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,IAAI,CAAC,CAkBf;AAED,wBAAsB,YAAY,CAChC,MAAM,EAAE,mBAAmB,EAC3B,WAAW,EAAE;IAAE,MAAM,EAAE,MAAM,CAAC;IAAC,MAAM,CAAC,EAAE,MAAM,CAAA;CAAE,EAChD,IAAI,CAAC,EAAE,MAAM,GACZ,OAAO,CAAC,mBAAmB,CAAC,CAqB9B"}

package/dist/services/storage/s3.js

@@ -0,0 +1,84 @@
+import { createReadStream } from "node:fs";
+import { createWriteStream } from "node:fs";
+import { pipeline } from "node:stream/promises";
+import { GetObjectCommand, PutObjectCommand, S3Client, } from "@aws-sdk/client-s3";
+import { ModelError, ValidationError } from "../../utils/errors.js";
+import { redactUrlForLogs } from "../download-auth.js";
+import { bodyToReadable, safeErrorDetails, withPrefix } from "./utils.js";
+export function isS3Url(url) {
+    return url.startsWith("s3://");
+}
+export function parseS3Url(url) {
+    let parsed;
+    try {
+        parsed = new URL(url);
+    }
+    catch {
+        throw new ValidationError("Invalid S3 URL. Expected s3://bucket/key.");
+    }
+    if (parsed.protocol !== "s3:" || !parsed.hostname || parsed.pathname.length <= 1) {
+        throw new ValidationError("Invalid S3 URL. Expected s3://bucket/key.");
+    }
+    return { bucket: parsed.hostname, key: decodeURIComponent(parsed.pathname.slice(1)) };
+}
+function makeS3Client(auth) {
+    const endpoint = auth?.endpoint ?? process.env.AWS_S3_ENDPOINT;
+    const region = auth?.region ?? process.env.AWS_REGION ?? (endpoint ? "auto" : undefined);
+    const config = {
+        region,
+        endpoint,
+        forcePathStyle: Boolean(endpoint),
+    };
+    const accessKeyId = auth?.access_key_id ?? process.env.AWS_ACCESS_KEY_ID;
+    const secretAccessKey = auth?.secret_access_key ?? process.env.AWS_SECRET_ACCESS_KEY;
+    const sessionToken = auth?.session_token ?? process.env.AWS_SESSION_TOKEN;
+    if (accessKeyId && secretAccessKey) {
+        config.credentials = { accessKeyId, secretAccessKey, sessionToken };
+    }
+    return new S3Client(config);
+}
+export async function downloadS3ToFile(url, targetPath, auth) {
+    const { bucket, key } = parseS3Url(url);
+    const client = makeS3Client(auth);
+    try {
+        const response = await client.send(new GetObjectCommand({ Bucket: bucket, Key: key }));
+        if (!response.Body) {
+            throw new ModelError("S3 download response has no body", { url: redactUrlForLogs(url) });
+        }
+        await pipeline(bodyToReadable(response.Body), createWriteStream(targetPath));
+    }
+    catch (err) {
+        if (err instanceof ModelError || err instanceof ValidationError)
+            throw err;
+        throw new ModelError("S3 download failed", {
+            url: redactUrlForLogs(url),
+            ...safeErrorDetails(err),
+        });
+    }
+    finally {
+        client.destroy();
+    }
+}
+export async function uploadS3File(source, destination, auth) {
+    const key = withPrefix(destination.prefix, source.filename);
+    const client = makeS3Client(auth);
+    try {
+        await client.send(new PutObjectCommand({
+            Bucket: destination.bucket,
+            Key: key,
+            Body: source.path ? createReadStream(source.path) : source.data,
+            ContentType: source.contentType,
+        }));
+        return { provider: "s3", url: `s3://${destination.bucket}/${key}` };
+    }
+    catch (err) {
+        throw new ModelError("S3 upload failed", {
+            url: `s3://${destination.bucket}/${key}`,
+            ...safeErrorDetails(err),
+        });
+    }
+    finally {
+        client.destroy();
+    }
+}
+//# sourceMappingURL=s3.js.map

package/dist/services/storage/s3.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"s3.js","sourceRoot":"","sources":["../../../src/services/storage/s3.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,SAAS,CAAC;AAC3C,OAAO,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,EAAE,QAAQ,EAAE,MAAM,sBAAsB,CAAC;AAChD,OAAO,EACL,gBAAgB,EAChB,gBAAgB,EAChB,QAAQ,GAET,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,UAAU,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AACpE,OAAO,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC;AAEvD,OAAO,EAAE,cAAc,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE1E,MAAM,UAAU,OAAO,CAAC,GAAW;IACjC,OAAO,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;AACjC,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,GAAW;IACpC,IAAI,MAAW,CAAC;IAChB,IAAI,CAAC;QACH,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC;IAAC,MAAM,CAAC;QACP,MAAM,IAAI,eAAe,CAAC,2CAA2C,CAAC,CAAC;IACzE,CAAC;IACD,IAAI,MAAM,CAAC,QAAQ,KAAK,KAAK,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;QACjF,MAAM,IAAI,eAAe,CAAC,2CAA2C,CAAC,CAAC;IACzE,CAAC;IACD,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,QAAQ,EAAE,GAAG,EAAE,kBAAkB,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;AACxF,CAAC;AAED,SAAS,YAAY,CAAC,IAAa;IACjC,MAAM,QAAQ,GAAG,IAAI,EAAE,QAAQ,IAAI,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;IAC/D,MAAM,MAAM,GAAG,IAAI,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACzF,MAAM,MAAM,GAAmB;QAC7B,MAAM;QACN,QAAQ;QACR,cAAc,EAAE,OAAO,CAAC,QAAQ,CAAC;KAClC,CAAC;IAEF,MAAM,WAAW,GAAG,IAAI,EAAE,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IACzE,MAAM,eAAe,GAAG,IAAI,EAAE,iBAAiB,IAAI,OAAO,CAAC,GAAG,CAAC,qBAAqB,CAAC;IACrF,MAAM,YAAY,GAAG,IAAI,EAAE,aAAa,IAAI,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC;IAC1E,IAAI,WAAW,IAAI,eAAe,EAAE,CAAC;QACnC,MAAM,CAAC,WAAW,GAAG,EAAE,WAAW,EAAE,eAAe,EAAE,YAAY,EAAE,CAAC;IACtE,CAAC;IAED,OAAO,IAAI,QAAQ,CAAC,MAAM,CAAC,CAAC;AAC9B,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,gBAAgB,CACpC,GAAW,EACX,UAAkB,EAClB,IAAa;IAEb,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IACxC,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,gBAAgB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC,CAAC;QACvF,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnB,MAAM,IAAI,UAAU,CAAC,kCAAkC,EAAE,EAAE,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC3F,CAAC;QACD,MAAM,QAAQ,CAAC,cAAc,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,iBAAiB,CAAC,UAAU,CAAC,CAAC,CAAC;IAC/E,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,IAAI,GAAG,YAAY,UAAU,IAAI,GAAG,YAAY,eAAe;YAAE,MAAM,GAAG,CAAC;QAC3E,MAAM,IAAI,UAAU,CAAC,oBAAoB,EAAE;YACzC,GAAG,EAAE,gBAAgB,CAAC,GAAG,CAAC;YAC1B,GAAG,gBAAgB,CAAC,GAAG,CAAC;SACzB,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAChC,MAA2B,EAC3B,WAAgD,EAChD,IAAa;IAEb,MAAM,GAAG,GAAG,UAAU,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC5D,MAAM,MAAM,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAClC,IAAI,CAAC;QACH,MAAM,MAAM,CAAC,IAAI,CACf,IAAI,gBAAgB,CAAC;YACnB,MAAM,EAAE,WAAW,CAAC,MAAM;YAC1B,GAAG,EAAE,GAAG;YACR,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI;YAC/D,WAAW,EAAE,MAAM,CAAC,WAAW;SAChC,CAAC,CACH,CAAC;QACF,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,QAAQ,WAAW,CAAC,MAAM,IAAI,GAAG,EAAE,EAAE,CAAC;IACtE,CAAC;IAAC,OAAO,GAAG,EAAE,CAAC;QACb,MAAM,IAAI,UAAU,CAAC,kBAAkB,EAAE;YACvC,GAAG,EAAE,QAAQ,WAAW,CAAC,MAAM,IAAI,GAAG,EAAE;YACxC,GAAG,gBAAgB,CAAC,GAAG,CAAC;SACzB,CAAC,CAAC;IACL,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;AACH,CAAC"}

package/dist/services/storage/types.d.ts

@@ -0,0 +1,22 @@
+export interface S3Auth {
+    type: "s3";
+    access_key_id: string;
+    secret_access_key: string;
+    session_token?: string;
+    region?: string;
+    endpoint?: string;
+}
+export interface CloudStorageAuth {
+    s3?: S3Auth;
+}
+export interface StorageUploadSource {
+    path?: string;
+    data?: Buffer;
+    filename: string;
+    contentType?: string;
+}
+export interface StorageUploadResult {
+    provider: "s3" | "azure" | "http" | "hf";
+    url: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/services/storage/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/services/storage/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,MAAM;IACrB,IAAI,EAAE,IAAI,CAAC;IACX,aAAa,EAAE,MAAM,CAAC;IACtB,iBAAiB,EAAE,MAAM,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,EAAE,CAAC,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,mBAAmB;IAClC,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,QAAQ,EAAE,MAAM,CAAC;IACjB,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,IAAI,GAAG,OAAO,GAAG,MAAM,GAAG,IAAI,CAAC;IACzC,GAAG,EAAE,MAAM,CAAC;CACb"}

package/dist/services/storage/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/services/storage/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/services/storage/types.ts"],"names":[],"mappings":""}

package/dist/services/storage/utils.d.ts

@@ -0,0 +1,6 @@
+import { Readable } from "node:stream";
+export declare function safeErrorDetails(err: unknown): Record<string, unknown>;
+export declare function bodyToReadable(body: unknown): Readable;
+export declare function normalizePrefix(prefix: string | undefined): string;
+export declare function withPrefix(prefix: string | undefined, filename: string): string;
+//# sourceMappingURL=utils.d.ts.map

package/dist/services/storage/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/services/storage/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAGvC,wBAAgB,gBAAgB,CAAC,GAAG,EAAE,OAAO,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAatE;AAED,wBAAgB,cAAc,CAAC,IAAI,EAAE,OAAO,GAAG,QAAQ,CAUtD;AAED,wBAAgB,eAAe,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,MAAM,CAGlE;AAED,wBAAgB,UAAU,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAG/E"}

package/dist/services/storage/utils.js

@@ -0,0 +1,39 @@
+import { Readable } from "node:stream";
+export function safeErrorDetails(err) {
+    if (!err || typeof err !== "object")
+        return {};
+    const record = err;
+    const metadata = record.$metadata;
+    return {
+        name: typeof record.name === "string" ? record.name : undefined,
+        code: typeof record.Code === "string"
+            ? record.Code
+            : typeof record.code === "string"
+                ? record.code
+                : undefined,
+        status: metadata?.httpStatusCode,
+    };
+}
+export function bodyToReadable(body) {
+    if (body instanceof Readable)
+        return body;
+    if (body instanceof Uint8Array)
+        return Readable.from(body);
+    if (body && typeof body.transformToByteArray === "function") {
+        return Readable.from((async function* () {
+            yield await body
+                .transformToByteArray();
+        })());
+    }
+    return Readable.fromWeb(body);
+}
+export function normalizePrefix(prefix) {
+    if (!prefix)
+        return "";
+    return prefix.replace(/^\/+/, "").replace(/\/+$/, "");
+}
+export function withPrefix(prefix, filename) {
+    const clean = normalizePrefix(prefix);
+    return clean ? `${clean}/${filename}` : filename;
+}
+//# sourceMappingURL=utils.js.map

package/dist/services/storage/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/services/storage/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAGvC,MAAM,UAAU,gBAAgB,CAAC,GAAY;IAC3C,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAC;IAC/C,MAAM,MAAM,GAAG,GAA8B,CAAC;IAC9C,MAAM,QAAQ,GAAG,MAAM,CAAC,SAAgD,CAAC;IACzE,OAAO;QACL,IAAI,EAAE,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;QAC/D,IAAI,EAAE,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ;YACnC,CAAC,CAAC,MAAM,CAAC,IAAI;YACb,CAAC,CAAC,OAAO,MAAM,CAAC,IAAI,KAAK,QAAQ;gBAC/B,CAAC,CAAC,MAAM,CAAC,IAAI;gBACb,CAAC,CAAC,SAAS;QACf,MAAM,EAAE,QAAQ,EAAE,cAAc;KACjC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,cAAc,CAAC,IAAa;IAC1C,IAAI,IAAI,YAAY,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC1C,IAAI,IAAI,YAAY,UAAU;QAAE,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC3D,IAAI,IAAI,IAAI,OAAQ,IAA2C,CAAC,oBAAoB,KAAK,UAAU,EAAE,CAAC;QACpG,OAAO,QAAQ,CAAC,IAAI,CAAC,CAAC,KAAK,SAAS,CAAC;YACnC,MAAM,MAAO,IAA4D;iBACtE,oBAAoB,EAAE,CAAC;QAC5B,CAAC,CAAC,EAAE,CAAC,CAAC;IACR,CAAC;IACD,OAAO,QAAQ,CAAC,OAAO,CAAC,IAAyB,CAAC,CAAC;AACrD,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,MAA0B;IACxD,IAAI,CAAC,MAAM;QAAE,OAAO,EAAE,CAAC;IACvB,OAAO,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;AACxD,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,MAA0B,EAAE,QAAgB;IACrE,MAAM,KAAK,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;IACtC,OAAO,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,QAAQ,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC;AACnD,CAAC"}

package/dist/services/storage-upload.d.ts

@@ -0,0 +1,18 @@
+import { type UploadDestination, type StorageUploadResult } from "./storage/index.js";
+export interface UploadOutputOptions {
+    asset_id?: string;
+    path?: string;
+    destination: UploadDestination;
+}
+export interface UploadOutputResult {
+    source: {
+        filename: string;
+        path?: string;
+        asset_id?: string;
+        mime_type?: string;
+        bytes?: number;
+    };
+    uploads: StorageUploadResult[];
+}
+export declare function uploadOutput(opts: UploadOutputOptions): Promise<UploadOutputResult>;
+//# sourceMappingURL=storage-upload.d.ts.map

package/dist/services/storage-upload.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage-upload.d.ts","sourceRoot":"","sources":["../../src/services/storage-upload.ts"],"names":[],"mappings":"AAKA,OAAO,EAEL,KAAK,iBAAiB,EACtB,KAAK,mBAAmB,EAEzB,MAAM,oBAAoB,CAAC;AAK5B,MAAM,WAAW,mBAAmB;IAClC,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,iBAAiB,CAAC;CAChC;AAED,MAAM,WAAW,kBAAkB;IACjC,MAAM,EAAE;QACN,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IACF,OAAO,EAAE,mBAAmB,EAAE,CAAC;CAChC;AAwGD,wBAAsB,YAAY,CAAC,IAAI,EAAE,mBAAmB,GAAG,OAAO,CAAC,kBAAkB,CAAC,CAyBzF"}

package/dist/services/storage-upload.js

@@ -0,0 +1,121 @@
+import { realpath, stat } from "node:fs/promises";
+import { basename, extname, isAbsolute, relative, resolve, sep } from "node:path";
+import { config } from "../config.js";
+import { AssetRegistry } from "./asset-registry.js";
+import { getOutputImage } from "./image-management.js";
+import { uploadToStorage, } from "./storage/index.js";
+import { ValidationError } from "../utils/errors.js";
+import { logger } from "../utils/logger.js";
+import { redactUrlForLogs } from "./download-auth.js";
+const MIME_BY_EXTENSION = {
+    ".png": "image/png",
+    ".jpg": "image/jpeg",
+    ".jpeg": "image/jpeg",
+    ".webp": "image/webp",
+    ".gif": "image/gif",
+    ".bmp": "image/bmp",
+    ".tiff": "image/tiff",
+    ".tif": "image/tiff",
+    ".mp4": "video/mp4",
+    ".mov": "video/quicktime",
+    ".webm": "video/webm",
+    ".wav": "audio/wav",
+    ".mp3": "audio/mpeg",
+    ".flac": "audio/flac",
+};
+function getOutputDir() {
+    if (!config.comfyuiPath) {
+        throw new ValidationError("COMFYUI_PATH is not configured. Set the COMFYUI_PATH environment variable.");
+    }
+    return resolve(config.comfyuiPath, "output");
+}
+function isInsideOrEqual(root, candidate) {
+    const rel = relative(root, candidate);
+    return rel === "" || (!rel.startsWith("..") && !isAbsolute(rel));
+}
+function assertInsideOutput(root, candidate, label) {
+    if (!isInsideOrEqual(root, candidate) || candidate === root) {
+        throw new ValidationError(`${label} must stay within the ComfyUI output directory.`);
+    }
+}
+function resolveOutputPath(path) {
+    if (path.trim().length === 0) {
+        throw new ValidationError("path must be a non-empty path.");
+    }
+    const outputDir = getOutputDir();
+    const resolved = isAbsolute(path) ? resolve(path) : resolve(outputDir, path);
+    if (resolved === outputDir || !resolved.startsWith(outputDir + sep)) {
+        throw new ValidationError("path must stay within the ComfyUI output directory.");
+    }
+    return resolved;
+}
+function inferMimeType(filename) {
+    return MIME_BY_EXTENSION[extname(filename).toLowerCase()] ?? "application/octet-stream";
+}
+async function sourceFromPath(path) {
+    const lexicalPath = resolveOutputPath(path);
+    const root = await realpath(getOutputDir());
+    const sourcePath = await realpath(lexicalPath).catch(() => undefined);
+    if (!sourcePath) {
+        throw new ValidationError(`Upload source not found: ${lexicalPath}`);
+    }
+    assertInsideOutput(root, sourcePath, "path");
+    const info = await stat(sourcePath).catch(() => undefined);
+    if (!info?.isFile()) {
+        throw new ValidationError(`Upload source not found: ${sourcePath}`);
+    }
+    const filename = basename(sourcePath);
+    return {
+        path: sourcePath,
+        filename,
+        contentType: inferMimeType(filename),
+        bytes: info.size,
+    };
+}
+async function sourceFromAsset(assetId) {
+    const record = AssetRegistry.get(assetId);
+    if (!record) {
+        throw new ValidationError(`No asset found for id "${assetId}". It may have expired or never been registered.`);
+    }
+    const validType = record.type === "output" || record.type === "input" || record.type === "temp";
+    const fetchType = validType
+        ? record.type
+        : "output";
+    const image = await getOutputImage(record.filename, fetchType, record.subfolder);
+    const data = Buffer.from(image.base64, "base64");
+    return {
+        data,
+        filename: image.filename,
+        contentType: image.mimeType,
+        bytes: data.length,
+    };
+}
+function redactDestinationForLogs(destination) {
+    if ("http" in destination)
+        return { http: { url: redactUrlForLogs(destination.http.url) } };
+    return destination;
+}
+export async function uploadOutput(opts) {
+    if (Boolean(opts.asset_id) === Boolean(opts.path)) {
+        throw new ValidationError("Provide exactly one upload source: asset_id or path.");
+    }
+    const source = opts.asset_id
+        ? await sourceFromAsset(opts.asset_id)
+        : await sourceFromPath(opts.path);
+    logger.info("Uploading ComfyUI output to cloud storage", {
+        source: source.path ?? `asset:${opts.asset_id}`,
+        destination: redactDestinationForLogs(opts.destination),
+    });
+    const upload = await uploadToStorage(source, opts.destination);
+    return {
+        source: {
+            filename: source.filename,
+            path: source.path,
+            asset_id: opts.asset_id,
+            mime_type: source.contentType,
+            bytes: source.bytes,
+        },
+        uploads: [upload],
+    };
+}
+//# sourceMappingURL=storage-upload.js.map

package/dist/services/storage-upload.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"storage-upload.js","sourceRoot":"","sources":["../../src/services/storage-upload.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,UAAU,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,EAAE,MAAM,WAAW,CAAC;AAClF,OAAO,EAAE,MAAM,EAAE,MAAM,cAAc,CAAC;AACtC,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AACvD,OAAO,EACL,eAAe,GAIhB,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,EAAE,MAAM,EAAE,MAAM,oBAAoB,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAmBtD,MAAM,iBAAiB,GAA2B;IAChD,MAAM,EAAE,WAAW;IACnB,MAAM,EAAE,YAAY;IACpB,OAAO,EAAE,YAAY;IACrB,OAAO,EAAE,YAAY;IACrB,MAAM,EAAE,WAAW;IACnB,MAAM,EAAE,WAAW;IACnB,OAAO,EAAE,YAAY;IACrB,MAAM,EAAE,YAAY;IACpB,MAAM,EAAE,WAAW;IACnB,MAAM,EAAE,iBAAiB;IACzB,OAAO,EAAE,YAAY;IACrB,MAAM,EAAE,WAAW;IACnB,MAAM,EAAE,YAAY;IACpB,OAAO,EAAE,YAAY;CACtB,CAAC;AAEF,SAAS,YAAY;IACnB,IAAI,CAAC,MAAM,CAAC,WAAW,EAAE,CAAC;QACxB,MAAM,IAAI,eAAe,CACvB,4EAA4E,CAC7E,CAAC;IACJ,CAAC;IACD,OAAO,OAAO,CAAC,MAAM,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,eAAe,CAAC,IAAY,EAAE,SAAiB;IACtD,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACtC,OAAO,GAAG,KAAK,EAAE,IAAI,CAAC,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;AACnE,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAY,EAAE,SAAiB,EAAE,KAAa;IACxE,IAAI,CAAC,eAAe,CAAC,IAAI,EAAE,SAAS,CAAC,IAAI,SAAS,KAAK,IAAI,EAAE,CAAC;QAC5D,MAAM,IAAI,eAAe,CAAC,GAAG,KAAK,iDAAiD,CAAC,CAAC;IACvF,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAY;IACrC,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,MAAM,IAAI,eAAe,CAAC,gCAAgC,CAAC,CAAC;IAC9D,CAAC;IACD,MAAM,SAAS,GAAG,YAAY,EAAE,CAAC;IACjC,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC7E,IAAI,QAAQ,KAAK,SAAS,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,SAAS,GAAG,GAAG,CAAC,EAAE,CAAC;QACpE,MAAM,IAAI,eAAe,CAAC,qDAAqD,CAAC,CAAC;IACnF,CAAC;IACD,OAAO,QAAQ,CAAC;AAClB,CAAC;AAED,SAAS,aAAa,CAAC,QAAgB;IACrC,OAAO,iBAAiB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC,IAAI,0BAA0B,CAAC;AAC1F,CAAC;AAED,KAAK,UAAU,cAAc,CAAC,IAAY;IACxC,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC;IAC5C,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,YAAY,EAAE,CAAC,CAAC;IAC5C,MAAM,UAAU,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IACtE,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,eAAe,CAAC,4BAA4B,WAAW,EAAE,CAAC,CAAC;IACvE,CAAC;IACD,kBAAkB,CAAC,IAAI,EAAE,UAAU,EAAE,MAAM,CAAC,CAAC;IAE7C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;IAC3D,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,CAAC;QACpB,MAAM,IAAI,eAAe,CAAC,4BAA4B,UAAU,EAAE,CAAC,CAAC;IACtE,CAAC;IAED,MAAM,QAAQ,GAAG,QAAQ,CAAC,UAAU,CAAC,CAAC;IACtC,OAAO;QACL,IAAI,EAAE,UAAU;QAChB,QAAQ;QACR,WAAW,EAAE,aAAa,CAAC,QAAQ,CAAC;QACpC,KAAK,EAAE,IAAI,CAAC,IAAI;KACjB,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,eAAe,CAAC,OAAe;IAC5C,MAAM,MAAM,GAAG,aAAa,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC1C,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,eAAe,CACvB,0BAA0B,OAAO,kDAAkD,CACpF,CAAC;IACJ,CAAC;IACD,MAAM,SAAS,GAAG,MAAM,CAAC,IAAI,KAAK,QAAQ,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,CAAC;IAChG,MAAM,SAAS,GAAgC,SAAS;QACtD,CAAC,CAAE,MAAM,CAAC,IAAoC;QAC9C,CAAC,CAAC,QAAQ,CAAC;IACb,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC,MAAM,CAAC,QAAQ,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,CAAC,CAAC;IACjF,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACjD,OAAO;QACL,IAAI;QACJ,QAAQ,EAAE,KAAK,CAAC,QAAQ;QACxB,WAAW,EAAE,KAAK,CAAC,QAAQ;QAC3B,KAAK,EAAE,IAAI,CAAC,MAAM;KACnB,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAAC,WAA8B;IAC9D,IAAI,MAAM,IAAI,WAAW;QAAE,OAAO,EAAE,IAAI,EAAE,EAAE,GAAG,EAAE,gBAAgB,CAAC,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,CAAC;IAC5F,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,IAAyB;IAC1D,IAAI,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,KAAK,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;QAClD,MAAM,IAAI,eAAe,CAAC,sDAAsD,CAAC,CAAC;IACpF,CAAC;IAED,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ;QAC1B,CAAC,CAAC,MAAM,eAAe,CAAC,IAAI,CAAC,QAAQ,CAAC;QACtC,CAAC,CAAC,MAAM,cAAc,CAAC,IAAI,CAAC,IAAK,CAAC,CAAC;IAErC,MAAM,CAAC,IAAI,CAAC,2CAA2C,EAAE;QACvD,MAAM,EAAE,MAAM,CAAC,IAAI,IAAI,SAAS,IAAI,CAAC,QAAQ,EAAE;QAC/C,WAAW,EAAE,wBAAwB,CAAC,IAAI,CAAC,WAAW,CAAC;KACxD,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC/D,OAAO;QACL,MAAM,EAAE;YACN,QAAQ,EAAE,MAAM,CAAC,QAAQ;YACzB,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,SAAS,EAAE,MAAM,CAAC,WAAW;YAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;SACpB;QACD,OAAO,EAAE,CAAC,MAAM,CAAC;KAClB,CAAC;AACJ,CAAC"}

package/dist/tools/image-convert.d.ts

@@ -0,0 +1,3 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export declare function registerImageConvertTools(server: McpServer): void;
+//# sourceMappingURL=image-convert.d.ts.map

package/dist/tools/image-convert.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"image-convert.d.ts","sourceRoot":"","sources":["../../src/tools/image-convert.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,yCAAyC,CAAC;AA4CzE,wBAAgB,yBAAyB,CAAC,MAAM,EAAE,SAAS,GAAG,IAAI,CAoBjE"}