@primuslabs/fhetransform-sdk 0.4.0

package/dist/wasm/worker-client.browser.js

@@ -0,0 +1,21 @@
+import { BrowserWorkerSideTransport, BrowserWorkerTransport } from "./worker-transport.browser.js";
+import { FheWasmClientDirect, FheWasmClientViaWorker } from "./fhe-wasm-client.js";
+import { WasmWorkerClient } from "./worker-client.js";
+export function getWorkerTransport() {
+    return new BrowserWorkerSideTransport();
+}
+function getWasmClient() {
+    const worker = new Worker(new URL("./worker.js", import.meta.url), {
+        type: "module"
+    });
+    const client = new WasmWorkerClient(new BrowserWorkerTransport(worker));
+    return client;
+}
+export function createFheWasmClient(options) {
+    const useWorker = options?.useWorker ?? true;
+    if (useWorker) {
+        return new FheWasmClientViaWorker(getWasmClient());
+    }
+    return new FheWasmClientDirect();
+}
+//# sourceMappingURL=worker-client.browser.js.map

package/dist/wasm/worker-client.browser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-client.browser.js","sourceRoot":"","sources":["../../src/wasm/worker-client.browser.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,0BAA0B,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACnG,OAAO,EAAiB,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAClG,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAItD,MAAM,UAAU,kBAAkB;IAChC,OAAO,IAAI,0BAA0B,EAAE,CAAC;AAC1C,CAAC;AAED,SAAS,aAAa;IACpB,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE;QACjE,IAAI,EAAE,QAAQ;KACf,CAAC,CAAC;IAEH,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,IAAI,sBAAsB,CAAC,MAAM,CAAC,CAAC,CAAC;IACxE,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,OAAiC;IACnE,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,IAAI,CAAC;IAC7C,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,IAAI,sBAAsB,CAAC,aAAa,EAAE,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,IAAI,mBAAmB,EAAE,CAAC;AACnC,CAAC"}

package/dist/wasm/worker-client.d.ts

@@ -0,0 +1,34 @@
+import type { WorkerTransport } from "./worker-transport.js";
+import { EncryptWithZkProofInputU8 } from "./fhe-wasm-client.js";
+/**
+ * Client for managing WASM worker communication
+ */
+export declare class WasmWorkerClient {
+    private transport;
+    private pendingRequests;
+    private ready;
+    private readyPromise;
+    private resolveReady;
+    private readonly timeoutMs;
+    constructor(transport: WorkerTransport, timeoutMs?: number);
+    /**
+     * Handle incoming worker messages
+     */
+    private handleMessage;
+    /**
+     * Handle worker transport errors
+     */
+    private handleError;
+    /**
+     * Encrypt integers with zero-knowledge proof via worker
+     */
+    encryptIntegerWithZkProof(args: EncryptWithZkProofInputU8): Promise<Uint8Array>;
+    /**
+     * Check if worker is ready
+     */
+    isReady(): boolean;
+    /**
+     * Clean up pending requests
+     */
+    cleanup(): void;
+}

package/dist/wasm/worker-client.js

@@ -0,0 +1,160 @@
+import { logger } from '../core/logger.js';
+import { FheErrorCode } from '../core/errors/error-codes.js';
+import { FheError } from '../core/errors/fhe-error.js';
+/**
+ * Generate unique request ID
+ */
+function getUUID() {
+    return `${Date.now()}_${Math.random().toString(36).substring(2, 11)}`;
+}
+/**
+ * Client for managing WASM worker communication
+ */
+export class WasmWorkerClient {
+    transport;
+    pendingRequests = new Map();
+    ready = false;
+    readyPromise;
+    resolveReady;
+    timeoutMs;
+    constructor(transport, timeoutMs = 30000) {
+        this.transport = transport;
+        this.timeoutMs = timeoutMs;
+        this.transport.onMessage(this.handleMessage.bind(this));
+        this.transport.onError(this.handleError.bind(this));
+        this.readyPromise = new Promise(resolve => {
+            this.resolveReady = resolve;
+        });
+        logger.debug('WasmWorkerClient created', { timeoutMs: this.timeoutMs });
+    }
+    /**
+     * Handle incoming worker messages
+     */
+    handleMessage(msg) {
+        try {
+            logger.debug('Worker message received', { cmd: msg.cmd, id: msg.id });
+            // Handle ready event
+            if (msg.cmd === "ready" && !this.ready) {
+                this.ready = true;
+                this.resolveReady();
+                logger.info('Worker ready');
+                return;
+            }
+            // Handle encrypt response
+            if (msg.cmd === "encrypt") {
+                const pending = this.pendingRequests.get(msg.id);
+                if (!pending) {
+                    logger.warn('No pending request found for ID', { id: msg.id });
+                    return;
+                }
+                this.pendingRequests.delete(msg.id);
+                clearTimeout(pending.timeout);
+                if (msg.code === "OK") {
+                    logger.debug('Encrypt request succeeded', { id: msg.id });
+                    pending.resolve(new Uint8Array(msg.result));
+                }
+                else {
+                    logger.warn('Encrypt request failed', { id: msg.id, message: msg.message });
+                    pending.reject(new FheError({ code: FheErrorCode.EncryptionFailed, message: msg.message ?? 'WASM worker encryption failed' }));
+                }
+            }
+        }
+        catch (error) {
+            logger.error('Error handling worker message', { error });
+        }
+    }
+    /**
+     * Handle worker transport errors
+     */
+    handleError(error) {
+        logger.error('Worker transport error', { error });
+        // Reject all pending requests
+        for (const [id, pending] of this.pendingRequests) {
+            clearTimeout(pending.timeout);
+            pending.reject(new FheError({ code: FheErrorCode.Network, message: `Worker transport error: ${error.message}`, cause: error }));
+            this.pendingRequests.delete(id);
+        }
+    }
+    /**
+     * Encrypt integers with zero-knowledge proof via worker
+     */
+    async encryptIntegerWithZkProof(args) {
+        try {
+            logger.debug('Starting encryption with ZK proof via worker', {
+                inputSize: args.input.length,
+                pkSize: args.pk.length,
+                crsSize: args.crs.length,
+                metadataSize: args.metadata.length,
+                refresh: args.refresh,
+            });
+            // Wait for worker to be ready
+            await this.readyPromise;
+            const id = getUUID();
+            return new Promise((resolve, reject) => {
+                // Set timeout for request
+                const timeout = setTimeout(() => {
+                    this.pendingRequests.delete(id);
+                    const errorMsg = `Encrypt request timeout after ${this.timeoutMs}ms (ID: ${id})`;
+                    logger.error(errorMsg);
+                    reject(new FheError({ code: FheErrorCode.Timeout, message: errorMsg }));
+                }, this.timeoutMs);
+                // Store pending request
+                this.pendingRequests.set(id, { resolve, reject, timeout });
+                // Prepare transferable buffers
+                const buffers = [
+                    args.pk.buffer,
+                    args.crs.buffer,
+                    args.input.buffer,
+                    args.metadata.buffer,
+                ];
+                // Send message to worker
+                try {
+                    this.transport.postMessage({
+                        cmd: "encrypt",
+                        id,
+                        args: {
+                            pk: args.pk.buffer,
+                            crs: args.crs.buffer,
+                            input: args.input.buffer,
+                            metadata: args.metadata.buffer,
+                            refresh: args.refresh,
+                        },
+                    }, buffers);
+                    logger.debug('Encrypt request sent to worker', { id });
+                }
+                catch (error) {
+                    clearTimeout(timeout);
+                    this.pendingRequests.delete(id);
+                    logger.error('Failed to post encrypt message to worker', { id, error });
+                    reject(new FheError({
+                        code: FheErrorCode.Network,
+                        message: `Failed to send encrypt request: ${error instanceof Error ? error.message : String(error)}`,
+                        cause: error,
+                    }));
+                }
+            });
+        }
+        catch (error) {
+            logger.error('encryptIntegerWithZkProof failed', { error });
+            throw error;
+        }
+    }
+    /**
+     * Check if worker is ready
+     */
+    isReady() {
+        return this.ready;
+    }
+    /**
+     * Clean up pending requests
+     */
+    cleanup() {
+        logger.info('Cleaning up WasmWorkerClient');
+        for (const [id, pending] of this.pendingRequests) {
+            clearTimeout(pending.timeout);
+            pending.reject(new FheError({ code: FheErrorCode.Aborted, message: 'Client cleaned up' }));
+            this.pendingRequests.delete(id);
+        }
+    }
+}
+//# sourceMappingURL=worker-client.js.map

package/dist/wasm/worker-client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-client.js","sourceRoot":"","sources":["../../src/wasm/worker-client.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAQvD;;GAEG;AACH,SAAS,OAAO;IACd,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC;AACxE,CAAC;AAED;;GAEG;AACH,MAAM,OAAO,gBAAgB;IAQjB;IAPF,eAAe,GAAG,IAAI,GAAG,EAA0B,CAAC;IACpD,KAAK,GAAG,KAAK,CAAC;IACd,YAAY,CAAgB;IAC5B,YAAY,CAAc;IACjB,SAAS,CAAS;IAEnC,YACU,SAA0B,EAClC,YAAoB,KAAK;QADjB,cAAS,GAAT,SAAS,CAAiB;QAGlC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAE3B,IAAI,CAAC,SAAS,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACxD,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAEpD,IAAI,CAAC,YAAY,GAAG,IAAI,OAAO,CAAO,OAAO,CAAC,EAAE;YAC9C,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;QAC9B,CAAC,CAAC,CAAC;QAEH,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC;IAC1E,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,GAA0B;QAC9C,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,EAAE,EAAG,GAAW,CAAC,EAAE,EAAE,CAAC,CAAC;YAE/E,qBAAqB;YACrB,IAAI,GAAG,CAAC,GAAG,KAAK,OAAO,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;gBACvC,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;gBAClB,IAAI,CAAC,YAAY,EAAE,CAAC;gBACpB,MAAM,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;gBAC5B,OAAO;YACT,CAAC;YAED,0BAA0B;YAC1B,IAAI,GAAG,CAAC,GAAG,KAAK,SAAS,EAAE,CAAC;gBAC1B,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACjD,IAAI,CAAC,OAAO,EAAE,CAAC;oBACb,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC/D,OAAO;gBACT,CAAC;gBAED,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACpC,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBAE9B,IAAI,GAAG,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;oBACtB,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;oBAC1D,OAAO,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC;gBAC9C,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,IAAI,CAAC,wBAAwB,EAAE,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;oBAC5E,OAAO,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,gBAAgB,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,IAAI,+BAA+B,EAAE,CAAC,CAAC,CAAC;gBACjI,CAAC;YACH,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,+BAA+B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3D,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,KAAY;QAC9B,MAAM,CAAC,KAAK,CAAC,wBAAwB,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;QAElD,8BAA8B;QAC9B,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACjD,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,OAAO,EAAE,OAAO,EAAE,2BAA2B,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YAChI,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,yBAAyB,CAAC,IAA+B;QAC7D,IAAI,CAAC;YACH,MAAM,CAAC,KAAK,CAAC,8CAA8C,EAAE;gBAC3D,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;gBAC5B,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,MAAM;gBACtB,OAAO,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM;gBACxB,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM;gBAClC,OAAO,EAAE,IAAI,CAAC,OAAO;aACtB,CAAC,CAAC;YAEH,8BAA8B;YAC9B,MAAM,IAAI,CAAC,YAAY,CAAC;YAExB,MAAM,EAAE,GAAG,OAAO,EAAE,CAAC;YAErB,OAAO,IAAI,OAAO,CAAa,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACjD,0BAA0B;gBAC1B,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;oBAC9B,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAChC,MAAM,QAAQ,GAAG,iCAAiC,IAAI,CAAC,SAAS,WAAW,EAAE,GAAG,CAAC;oBACjF,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;oBACvB,MAAM,CAAC,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;gBAC1E,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;gBAEnB,wBAAwB;gBACxB,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,CAAC;gBAE3D,+BAA+B;gBAC/B,MAAM,OAAO,GAAG;oBACd,IAAI,CAAC,EAAE,CAAC,MAAM;oBACd,IAAI,CAAC,GAAG,CAAC,MAAM;oBACf,IAAI,CAAC,KAAK,CAAC,MAAM;oBACjB,IAAI,CAAC,QAAQ,CAAC,MAAM;iBACrB,CAAC;gBAEF,yBAAyB;gBACzB,IAAI,CAAC;oBACH,IAAI,CAAC,SAAS,CAAC,WAAW,CACxB;wBACE,GAAG,EAAE,SAAS;wBACd,EAAE;wBACF,IAAI,EAAE;4BACJ,EAAE,EAAE,IAAI,CAAC,EAAE,CAAC,MAAM;4BAClB,GAAG,EAAE,IAAI,CAAC,GAAG,CAAC,MAAM;4BACpB,KAAK,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM;4BACxB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,MAAM;4BAC9B,OAAO,EAAE,IAAI,CAAC,OAAO;yBACtB;qBACF,EACD,OAAO,CACR,CAAC;oBACF,MAAM,CAAC,KAAK,CAAC,gCAAgC,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;gBACzD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,YAAY,CAAC,OAAO,CAAC,CAAC;oBACtB,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;oBAChC,MAAM,CAAC,KAAK,CAAC,0CAA0C,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;oBACxE,MAAM,CAAC,IAAI,QAAQ,CAAC;wBAClB,IAAI,EAAE,YAAY,CAAC,OAAO;wBAC1B,OAAO,EAAE,mCAAmC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;wBACpG,KAAK,EAAE,KAAK;qBACb,CAAC,CAAC,CAAC;gBACN,CAAC;YACH,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;YAC5D,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,OAAO;QACL,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,OAAO;QACL,MAAM,CAAC,IAAI,CAAC,8BAA8B,CAAC,CAAC;QAE5C,KAAK,MAAM,CAAC,EAAE,EAAE,OAAO,CAAC,IAAI,IAAI,CAAC,eAAe,EAAE,CAAC;YACjD,YAAY,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAC9B,OAAO,CAAC,MAAM,CAAC,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,OAAO,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC,CAAC;YAC3F,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAClC,CAAC;IACH,CAAC;CACF"}

package/dist/wasm/worker-client.node.d.ts

@@ -0,0 +1,6 @@
+import { NodeWorkerSideTransport } from "./worker-transport.node.js";
+import { FheWasmClient } from "./fhe-wasm-client.js";
+export declare function getWorkerTransport(): NodeWorkerSideTransport;
+export declare function createFheWasmClient(options?: {
+    useWorker?: boolean;
+}): FheWasmClient;

package/dist/wasm/worker-client.node.js

@@ -0,0 +1,20 @@
+import { Worker } from "node:worker_threads";
+import { WasmWorkerClient } from "./worker-client.js";
+import { NodeWorkerTransport, NodeWorkerSideTransport } from "./worker-transport.node.js";
+import { FheWasmClientDirect, FheWasmClientViaWorker } from "./fhe-wasm-client.js";
+export function getWorkerTransport() {
+    return new NodeWorkerSideTransport();
+}
+function getWasmClient() {
+    const worker = new Worker(new URL("./worker.js", import.meta.url));
+    const client = new WasmWorkerClient(new NodeWorkerTransport(worker));
+    return client;
+}
+export function createFheWasmClient(options) {
+    const useWorker = options?.useWorker ?? false;
+    if (useWorker) {
+        return new FheWasmClientViaWorker(getWasmClient());
+    }
+    return new FheWasmClientDirect();
+}
+//# sourceMappingURL=worker-client.node.js.map

package/dist/wasm/worker-client.node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-client.node.js","sourceRoot":"","sources":["../../src/wasm/worker-client.node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,mBAAmB,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AAC1F,OAAO,EAAiB,mBAAmB,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAElG,MAAM,UAAU,kBAAkB;IAChC,OAAO,IAAI,uBAAuB,EAAE,CAAC;AACvC,CAAC;AAED,SAAS,aAAa;IACpB,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,GAAG,CAAC,aAAa,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IACnE,MAAM,MAAM,GAAG,IAAI,gBAAgB,CAAC,IAAI,mBAAmB,CAAC,MAAM,CAAC,CAAC,CAAC;IACrE,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,OAAiC;IACnE,MAAM,SAAS,GAAG,OAAO,EAAE,SAAS,IAAI,KAAK,CAAC;IAC9C,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,IAAI,sBAAsB,CAAC,aAAa,EAAE,CAAC,CAAC;IACrD,CAAC;IACD,OAAO,IAAI,mBAAmB,EAAE,CAAC;AACnC,CAAC"}

package/dist/wasm/worker-message.d.ts

@@ -0,0 +1,61 @@
+import { EncryptWithZkProofInput } from "./fhe-wasm-module.js";
+/**
+ * Base interface for all worker messages
+ */
+export interface WorkerMessageBase {
+    cmd: string;
+}
+/**
+ * Base interface for worker requests with request ID
+ */
+export interface WorkerRequestBase extends WorkerMessageBase {
+    id: string;
+}
+/**
+ * Base interface for worker responses with status code
+ */
+export interface WorkerResponseBase extends WorkerMessageBase {
+    id: string;
+    code: "OK" | "ERROR";
+}
+/**
+ * Worker ready event sent during initialization
+ */
+export interface WorkerReadyEvent extends WorkerMessageBase {
+    cmd: "ready";
+    wasmReady: boolean;
+}
+export type WorkerEvent = WorkerReadyEvent;
+/**
+ * Encryption request with zero-knowledge proof parameters
+ */
+export interface WorkerEncryptRequest extends WorkerRequestBase {
+    cmd: "encrypt";
+    args: EncryptWithZkProofInput;
+}
+export type WorkerRequestMessage = WorkerEncryptRequest;
+/**
+ * Successful encryption response containing result buffer
+ */
+export interface EncryptSuccess extends WorkerResponseBase {
+    cmd: "encrypt";
+    code: "OK";
+    result: ArrayBuffer;
+}
+/**
+ * Failed encryption response with error message
+ */
+export interface EncryptError extends WorkerResponseBase {
+    cmd: "encrypt";
+    code: "ERROR";
+    message: string;
+}
+export type WorkerResponse = EncryptSuccess | EncryptError;
+export type WorkerResponseMessage = WorkerResponse | WorkerEvent;
+/**
+ * Configuration for WASM worker client
+ */
+export interface WasmWorkerClientConfig {
+    workerUrl?: string;
+    timeoutMs?: number;
+}

package/dist/wasm/worker-message.js

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

package/dist/wasm/worker-message.js.map

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

package/dist/wasm/worker-transport.browser.d.ts

@@ -0,0 +1,15 @@
+import type { WorkerTransport } from "./worker-transport.js";
+export declare class BrowserWorkerTransport implements WorkerTransport {
+    private worker;
+    constructor(worker: Worker);
+    postMessage(message: any, transfer?: Transferable[]): void;
+    onMessage(callback: (message: any) => void): void;
+    onError(callback: (error: Error) => void): void;
+    close(): void;
+}
+export declare class BrowserWorkerSideTransport implements WorkerTransport {
+    postMessage(message: any, transfer?: Transferable[]): void;
+    onMessage(callback: (message: any) => void): void;
+    onError(_callback: (error: Error) => void): void;
+    close(): void;
+}

package/dist/wasm/worker-transport.browser.js

@@ -0,0 +1,35 @@
+import { FheErrorCode } from "../core/errors/error-codes.js";
+import { FheError } from "../core/errors/fhe-error.js";
+export class BrowserWorkerTransport {
+    worker;
+    constructor(worker) {
+        this.worker = worker;
+    }
+    postMessage(message, transfer) {
+        this.worker.postMessage(message, transfer ?? []);
+    }
+    onMessage(callback) {
+        this.worker.addEventListener("message", e => callback(e.data));
+    }
+    onError(callback) {
+        this.worker.addEventListener("error", (e) => {
+            callback(e.error ?? new FheError({ code: FheErrorCode.Network, message: e.message ?? "Worker error" }));
+        });
+    }
+    close() {
+        this.worker.terminate();
+    }
+}
+export class BrowserWorkerSideTransport {
+    postMessage(message, transfer) {
+        self.postMessage(message, transfer ?? []);
+    }
+    onMessage(callback) {
+        self.addEventListener("message", (e) => callback(e.data));
+    }
+    onError(_callback) { }
+    close() {
+        self.close();
+    }
+}
+//# sourceMappingURL=worker-transport.browser.js.map

package/dist/wasm/worker-transport.browser.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-transport.browser.js","sourceRoot":"","sources":["../../src/wasm/worker-transport.browser.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAEvD,MAAM,OAAO,sBAAsB;IACb;IAApB,YAAoB,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAI,CAAC;IAEvC,WAAW,CAAC,OAAY,EAAE,QAAyB;QACjD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,SAAS,CAAC,QAAgC;QACxC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACjE,CAAC;IAED,OAAO,CAAC,QAAgC;QACtC,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE;YAC1C,QAAQ,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,cAAc,EAAE,CAAC,CAAC,CAAC;QAC1G,CAAC,CAAC,CAAC;IACL,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IAC1B,CAAC;CACF;AAED,MAAM,OAAO,0BAA0B;IACrC,WAAW,CAAC,OAAY,EAAE,QAAyB;QACjD,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,QAAQ,IAAI,EAAE,CAAC,CAAC;IAC5C,CAAC;IAED,SAAS,CAAC,QAAgC;QACxC,IAAI,CAAC,gBAAgB,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IAC5D,CAAC;IAED,OAAO,CAAC,SAAiC,IAAwB,CAAC;IAElE,KAAK;QACH,IAAI,CAAC,KAAK,EAAE,CAAC;IACf,CAAC;CACF"}

package/dist/wasm/worker-transport.d.ts

@@ -0,0 +1,6 @@
+export interface WorkerTransport {
+    postMessage(message: any, transfer?: Transferable[]): void;
+    onMessage(callback: (message: any) => void): void;
+    onError(callback: (error: Error) => void): void;
+    close(): void;
+}

package/dist/wasm/worker-transport.js

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

package/dist/wasm/worker-transport.js.map

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

package/dist/wasm/worker-transport.node.d.ts

@@ -0,0 +1,17 @@
+import { type Worker } from "node:worker_threads";
+import type { WorkerTransport } from "./worker-transport.js";
+export declare class NodeWorkerTransport implements WorkerTransport {
+    private worker;
+    constructor(worker: Worker);
+    postMessage(message: any, transfer?: Transferable[]): void;
+    onMessage(callback: (message: any) => void): void;
+    onError(callback: (error: Error) => void): void;
+    close(): void;
+}
+export declare class NodeWorkerSideTransport implements WorkerTransport {
+    constructor();
+    postMessage(message: any, transfer?: Transferable[]): void;
+    onMessage(callback: (message: any) => void): void;
+    onError(_callback: (error: Error) => void): void;
+    close(): void;
+}

package/dist/wasm/worker-transport.node.js

@@ -0,0 +1,39 @@
+import { parentPort } from "node:worker_threads";
+import { FheErrorCode } from "../core/errors/error-codes.js";
+import { FheError } from "../core/errors/fhe-error.js";
+export class NodeWorkerTransport {
+    worker;
+    constructor(worker) {
+        this.worker = worker;
+    }
+    postMessage(message, transfer) {
+        this.worker.postMessage(message, transfer);
+    }
+    onMessage(callback) {
+        this.worker.on("message", callback);
+    }
+    onError(callback) {
+        this.worker.on("error", callback);
+    }
+    close() {
+        this.worker.terminate();
+    }
+}
+export class NodeWorkerSideTransport {
+    constructor() {
+        if (!parentPort) {
+            throw new FheError({ code: FheErrorCode.Unsupported, message: "parentPort is null" });
+        }
+    }
+    postMessage(message, transfer) {
+        parentPort.postMessage(message, transfer);
+    }
+    onMessage(callback) {
+        parentPort.on("message", callback);
+    }
+    onError(_callback) { }
+    close() {
+        parentPort.close();
+    }
+}
+//# sourceMappingURL=worker-transport.node.js.map

package/dist/wasm/worker-transport.node.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker-transport.node.js","sourceRoot":"","sources":["../../src/wasm/worker-transport.node.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAe,MAAM,qBAAqB,CAAC;AAE9D,OAAO,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,6BAA6B,CAAC;AAEvD,MAAM,OAAO,mBAAmB;IACV;IAApB,YAAoB,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAI,CAAC;IAEvC,WAAW,CAAC,OAAY,EAAE,QAAyB;QACjD,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,OAAO,EAAE,QAAe,CAAC,CAAC;IACpD,CAAC;IAED,SAAS,CAAC,QAAgC;QACxC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED,OAAO,CAAC,QAAgC;QACtC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;IAC1B,CAAC;CACF;AAGD,MAAM,OAAO,uBAAuB;IAClC;QACE,IAAI,CAAC,UAAU,EAAE,CAAC;YAChB,MAAM,IAAI,QAAQ,CAAC,EAAE,IAAI,EAAE,YAAY,CAAC,WAAW,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC,CAAC;QACxF,CAAC;IACH,CAAC;IAED,WAAW,CAAC,OAAY,EAAE,QAAyB;QACjD,UAAW,CAAC,WAAW,CAAC,OAAO,EAAE,QAAe,CAAC,CAAC;IACpD,CAAC;IAED,SAAS,CAAC,QAAgC;QACxC,UAAW,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;IACtC,CAAC;IAED,OAAO,CAAC,SAAiC,IAAwB,CAAC;IAElE,KAAK;QACH,UAAW,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;CACF"}

package/dist/wasm/worker.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/wasm/worker.js

@@ -0,0 +1,69 @@
+/// <reference lib="webworker" />
+import { getWorkerTransport } from '#worker-client';
+import { logger } from '../core/logger.js';
+import { encryptIntegerWithZkProof, loadFheWasmModule } from './fhe-wasm-module.js';
+const transport = getWorkerTransport();
+/**
+ * Handle encrypt request with zero-knowledge proof
+ */
+async function handleEncryptIntegerWithZkProof(data) {
+    const { cmd, id, args } = data;
+    try {
+        logger.debug('[Worker] Processing encrypt request', { id, refresh: args.refresh });
+        // Execute encryption
+        const res = await encryptIntegerWithZkProof(args);
+        // Send success response with transferable buffer
+        transport.postMessage({
+            cmd,
+            id,
+            code: "OK",
+            result: res.result.buffer,
+        }, [res.result.buffer]);
+        logger.debug('[Worker] Encrypt request completed successfully', { id });
+    }
+    catch (error) {
+        // Send error response
+        const errorMessage = error instanceof Error ? error.message : String(error);
+        logger.error('[Worker] Encrypt request failed', { id, error: errorMessage });
+        transport.postMessage({
+            cmd,
+            id,
+            code: "ERROR",
+            message: errorMessage,
+        });
+    }
+}
+/**
+ * Set up message handler
+ */
+transport.onMessage(async (message) => {
+    try {
+        const msg = message;
+        logger.debug('[Worker] Received message', { cmd: msg.cmd, id: msg.id });
+        switch (msg.cmd) {
+            case 'encrypt':
+                await handleEncryptIntegerWithZkProof(msg);
+                break;
+            default:
+                logger.warn('[Worker] Unknown command received', { cmd: msg.cmd, id: msg.id });
+        }
+    }
+    catch (error) {
+        logger.error('[Worker] Failed to handle message', { error });
+    }
+});
+/**
+ * Initialize WASM module and signal ready
+ */
+(async () => {
+    logger.info('[Worker] Initializing WASM module...');
+    await loadFheWasmModule();
+    logger.info('[Worker] WASM module loaded');
+    // Signal worker is ready
+    transport.postMessage({
+        cmd: "ready",
+        wasmReady: true,
+    });
+    logger.info('[Worker] Ready signal sent');
+})();
+//# sourceMappingURL=worker.js.map

package/dist/wasm/worker.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"worker.js","sourceRoot":"","sources":["../../src/wasm/worker.ts"],"names":[],"mappings":"AAAA,iCAAiC;AAEjC,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACpD,OAAO,EAAE,MAAM,EAAE,MAAM,mBAAmB,CAAC;AAC3C,OAAO,EAAE,yBAAyB,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AASpF,MAAM,SAAS,GAAG,kBAAkB,EAAE,CAAC;AAEvC;;GAEG;AACH,KAAK,UAAU,+BAA+B,CAAC,IAA0B;IACvE,MAAM,EAAE,GAAG,EAAE,EAAE,EAAE,IAAI,EAAE,GAAG,IAAI,CAAC;IAE/B,IAAI,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAEnF,qBAAqB;QACrB,MAAM,GAAG,GAAG,MAAM,yBAAyB,CAAC,IAAI,CAAC,CAAC;QAElD,iDAAiD;QACjD,SAAS,CAAC,WAAW,CACnB;YACE,GAAG;YACH,EAAE;YACF,IAAI,EAAE,IAAI;YACV,MAAM,EAAE,GAAG,CAAC,MAAM,CAAC,MAAqB;SAChB,EAC1B,CAAC,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CACpB,CAAC;QAEF,MAAM,CAAC,KAAK,CAAC,iDAAiD,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;IAC1E,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,sBAAsB;QACtB,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC5E,MAAM,CAAC,KAAK,CAAC,iCAAiC,EAAE,EAAE,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;QAE7E,SAAS,CAAC,WAAW,CAAC;YACpB,GAAG;YACH,EAAE;YACF,IAAI,EAAE,OAAO;YACb,OAAO,EAAE,YAAY;SACC,CAAC,CAAC;IAC5B,CAAC;AACH,CAAC;AAED;;GAEG;AACH,SAAS,CAAC,SAAS,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE;IACpC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,OAA+B,CAAC;QAE5C,MAAM,CAAC,KAAK,CAAC,2BAA2B,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;QAExE,QAAQ,GAAG,CAAC,GAAG,EAAE,CAAC;YAChB,KAAK,SAAS;gBACZ,MAAM,+BAA+B,CAAC,GAAG,CAAC,CAAC;gBAC3C,MAAM;YACR;gBACE,MAAM,CAAC,IAAI,CAAC,mCAAmC,EAAE,EAAE,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;QACnF,CAAC;IACH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC;IAC/D,CAAC;AACH,CAAC,CAAC,CAAC;AAEH;;GAEG;AACH,CAAC,KAAK,IAAI,EAAE;IACV,MAAM,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;IACpD,MAAM,iBAAiB,EAAE,CAAC;IAC1B,MAAM,CAAC,IAAI,CAAC,6BAA6B,CAAC,CAAC;IAE3C,yBAAyB;IACzB,SAAS,CAAC,WAAW,CAAC;QACpB,GAAG,EAAE,OAAO;QACZ,SAAS,EAAE,IAAI;KACW,CAAC,CAAC;IAE9B,MAAM,CAAC,IAAI,CAAC,4BAA4B,CAAC,CAAC;AAC5C,CAAC,CAAC,EAAE,CAAC"}

package/package.json

@@ -0,0 +1,81 @@
+{
+  "name": "@primuslabs/fhetransform-sdk",
+  "version": "0.4.0",
+  "description": "SDK for FHETransform.",
+  "author": "",
+  "keywords": [
+    "FHE"
+  ],
+  "license": "MIT",
+  "type": "module",
+  "main": "./dist/index.js",
+  "types": "./dist/index.d.ts",
+  "exports": {
+    ".": {
+      "types": "./dist/index.d.ts",
+      "import": "./dist/index.js"
+    }
+  },
+  "imports": {
+    "#runtime": {
+      "browser": "./dist/runtime/browser.js",
+      "node": "./dist/runtime/node.js",
+      "default": "./dist/runtime/node.js"
+    },
+    "#worker-client": {
+      "browser": "./dist/wasm/worker-client.browser.js",
+      "node": "./dist/wasm/worker-client.node.js",
+      "default": "./dist/wasm/worker-client.node.js"
+    },
+    "#wasm-loader": {
+      "browser": "./dist/wasm/wasm-loader.browser.js",
+      "node": "./dist/wasm/wasm-loader.node.js",
+      "default": "./dist/wasm/wasm-loader.node.js"
+    }
+  },
+  "bin": {
+    "eusdc-cli": "./dist/cli/eusdc.js"
+  },
+  "scripts": {
+    "build": "tsc && chmod +x dist/cli/*.js && npm run copy:native && npm run docs:errors",
+    "copy:native": "mkdir -p dist/native && cp -rf src/native/* dist/native/",
+    "docs:errors": "node --import tsx scripts/generate-error-docs.ts",
+    "test": "npx vitest run --coverage",
+    "usage:browser": "npx vite",
+    "usage:node": "npx tsx examples/full_usage.ts",
+    "lint": "eslint src --ext .ts",
+    "format": "prettier --write \"src/**/*.ts\"",
+    "clean": "rm -rf dist"
+  },
+  "dependencies": {
+    "axios": "^1.18.1",
+    "commander": "^15.0.0",
+    "ethers": "^6.15.0",
+    "idb-keyval": "^6.2.5",
+    "loglevel": "^1.9.2",
+    "tslog": "^5.1.0",
+    "unstorage": "^1.17.5"
+  },
+  "devDependencies": {
+    "@types/node": "^25.9.1",
+    "@typescript-eslint/eslint-plugin": "^8.60.0",
+    "@typescript-eslint/parser": "^8.60.0",
+    "@vitest/browser": "^4.1.10",
+    "@vitest/coverage-v8": "^4.1.10",
+    "dotenv": "^17.4.2",
+    "eslint": "^10.4.1",
+    "playwright": "^1.61.1",
+    "prettier": "^3.8.3",
+    "tsx": "^4.22.3",
+    "typescript": "^5.8.3",
+    "vite": "^8.0.14",
+    "vitest": "^4.1.10"
+  },
+  "engines": {
+    "node": ">=20",
+    "npm": ">=7.0.0"
+  },
+  "files": [
+    "dist"
+  ]
+}