@cofhe/sdk 1.1.0-alpha.3 → 1.1.0-alpha.4

package/dist/node.cjs

@@ -2332,6 +2332,8 @@ var CofheConfigSchema = zod.z.object({
   supportedChains: zod.z.array(zod.z.custom()),
   /** Default acp expiration in seconds, default is 30 days */
   defaultACPExpiration: zod.z.number().optional().default(60 * 60 * 24 * 30),
+  /** Decrypts sent to the threshold network at a time per client; `Infinity` disables the cap (default 8) */
+  decryptConcurrency: zod.z.union([zod.z.number().min(1), zod.z.literal(Infinity)]).optional().default(8),
   /** Storage method for fhe keys (defaults to indexedDB on web, filesystem on node) */
   fheKeyStorage: zod.z.object({
     getItem: zod.z.custom((val) => typeof val === "function", {
@@ -3874,6 +3876,32 @@ function throwIfSubmitRetryTimedOut(params) {
     });
   }
 }
+function decryptAbortedError(operationLabel, signal) {
+  return new CofheError({
+    code: "ABORTED" /* Aborted */,
+    message: `${operationLabel} aborted by the caller`,
+    cause: signal.reason instanceof Error ? signal.reason : void 0
+  });
+}
+function throwIfAborted(signal, operationLabel) {
+  if (signal?.aborted)
+    throw decryptAbortedError(operationLabel, signal);
+}
+function sleepUnlessAborted(ms, signal, operationLabel) {
+  return new Promise((resolve, reject) => {
+    if (signal?.aborted)
+      return reject(decryptAbortedError(operationLabel, signal));
+    const onAbort = () => {
+      clearTimeout(timer);
+      reject(decryptAbortedError(operationLabel, signal));
+    };
+    const timer = setTimeout(() => {
+      signal?.removeEventListener("abort", onAbort);
+      resolve();
+    }, ms);
+    signal?.addEventListener("abort", onAbort, { once: true });
+  });
+}
 function normalizeTnSignature(signature) {
   if (typeof signature !== "string") {
     throw new CofheError({
@@ -3991,7 +4019,7 @@ function parseCompletedSealOutputResponse(params) {
   }
   return convertSealedData(sealed);
 }
-async function submitSealOutputRequest(thresholdNetworkUrl, ctHash, chainId, acp, overallStartTime, retry404TimeoutMs, timeoutMs, onPoll) {
+async function submitSealOutputRequest(thresholdNetworkUrl, ctHash, chainId, acp, overallStartTime, retry404TimeoutMs, timeoutMs, onPoll, signal) {
   const body = {
     ct_tempkey: BigInt(ctHash).toString(16).padStart(64, "0"),
     host_chain_id: chainId,
@@ -4000,6 +4028,7 @@ async function submitSealOutputRequest(thresholdNetworkUrl, ctHash, chainId, acp
   let attemptIndex = 0;
   let last404ApiErrorMessage;
   for (; ; ) {
+    throwIfAborted(signal, "sealOutput");
     let response;
     try {
       response = await cofheFetch(
@@ -4010,10 +4039,12 @@ async function submitSealOutputRequest(thresholdNetworkUrl, ctHash, chainId, acp
           headers: {
             "Content-Type": "application/json"
           },
-          body: JSON.stringify(body)
+          body: JSON.stringify(body),
+          signal
         }
       );
     } catch (e) {
+      throwIfAborted(signal, "sealOutput");
       throw new CofheError({
         code: "SEAL_OUTPUT_FAILED" /* SealOutputFailed */,
         message: `sealOutput request failed`,
@@ -4108,13 +4139,14 @@ async function submitSealOutputRequest(thresholdNetworkUrl, ctHash, chainId, acp
       intervalMs: SUBMIT_RETRY_INTERVAL_MS,
       timeoutMs
     });
-    await new Promise((resolve) => setTimeout(resolve, SUBMIT_RETRY_INTERVAL_MS));
+    await sleepUnlessAborted(SUBMIT_RETRY_INTERVAL_MS, signal, "sealOutput");
     attemptIndex += 1;
   }
 }
-async function pollSealOutputStatus(thresholdNetworkUrl, requestId, overallStartTime, timeoutMs, onPoll) {
+async function pollSealOutputStatus(thresholdNetworkUrl, requestId, overallStartTime, timeoutMs, onPoll, signal) {
   let attemptIndex = 0;
   while (true) {
+    throwIfAborted(signal, "sealOutput");
     const elapsedMs = Date.now() - overallStartTime;
     const intervalMs = computeMinuteRampPollIntervalMs(elapsedMs, {
       minIntervalMs: POLL_INTERVAL_MS,
@@ -4149,7 +4181,8 @@ async function pollSealOutputStatus(thresholdNetworkUrl, requestId, overallStart
           method: "GET",
           headers: {
             "Content-Type": "application/json"
-          }
+          },
+          signal
         }
       );
     } catch (e) {
@@ -4210,7 +4243,7 @@ async function pollSealOutputStatus(thresholdNetworkUrl, requestId, overallStart
         requestId
       });
     }
-    await new Promise((resolve) => setTimeout(resolve, intervalMs));
+    await sleepUnlessAborted(intervalMs, signal, "sealOutput");
     attemptIndex += 1;
   }
   throw new CofheError({
@@ -4223,7 +4256,7 @@ async function pollSealOutputStatus(thresholdNetworkUrl, requestId, overallStart
   });
 }
 async function tnSealOutputV2(params) {
-  const { thresholdNetworkUrl, ctHash, chainId, acp, retry404TimeoutMs, onPoll } = params;
+  const { thresholdNetworkUrl, ctHash, chainId, acp, retry404TimeoutMs, onPoll, signal } = params;
   const timeoutMs = params.timeoutMs ?? DEFAULT_DECRYPT_TIMEOUT_MS;
   const normalized404RetryTimeoutMs = normalize404RetryTimeoutMs({
     timeoutMs: retry404TimeoutMs,
@@ -4239,12 +4272,20 @@ async function tnSealOutputV2(params) {
     overallStartTime,
     normalized404RetryTimeoutMs,
     timeoutMs,
-    onPoll
+    onPoll,
+    signal
   );
   if (submitResult.kind === "completed") {
     return submitResult.sealed;
   }
-  return await pollSealOutputStatus(thresholdNetworkUrl, submitResult.requestId, overallStartTime, timeoutMs, onPoll);
+  return await pollSealOutputStatus(
+    thresholdNetworkUrl,
+    submitResult.requestId,
+    overallStartTime,
+    timeoutMs,
+    onPoll,
+    signal
+  );
 }
 
 // core/decrypt/decryptForViewBuilder.ts
@@ -4257,6 +4298,8 @@ var DecryptForViewBuilder = class extends BaseBuilder {
   pollCallback;
   retry404TimeoutMs = DEFAULT_404_RETRY_TIMEOUT_MS2;
   timeoutMs;
+  signal;
+  limit;
   constructor(params) {
     super({
       config: params.config,
@@ -4270,6 +4313,7 @@ var DecryptForViewBuilder = class extends BaseBuilder {
     this.utype = params.utype;
     this.acpHash = params.acpHash;
     this.acp = params.acp;
+    this.limit = params.limit ?? ((task) => task());
   }
   /**
    * @param chainId - Chain to decrypt values from. Used to fetch the threshold network URL and use the correct acp.
@@ -4340,6 +4384,15 @@ var DecryptForViewBuilder = class extends BaseBuilder {
     this.timeoutMs = timeoutMs;
     return this;
   }
+  /**
+   * Stops the decrypt when `signal` fires: the pending request is aborted, no further attempt is
+   * sent, and `execute()` rejects with `Aborted`. A request already processing on the threshold
+   * network is not cancelled.
+   */
+  setSignal(signal) {
+    this.signal = signal;
+    return this;
+  }
   withACP(acpOrACPHash) {
     if (typeof acpOrACPHash === "string") {
       this.acpHash = acpOrACPHash;
@@ -4463,15 +4516,20 @@ var DecryptForViewBuilder = class extends BaseBuilder {
     this.assertPublicClient();
     const thresholdNetworkUrl = await this.getThresholdNetworkUrl();
     const wireAcp = ACPUtils.getPublic(acp, true);
-    const sealed = await tnSealOutputV2({
-      ctHash: this.ctHash,
-      chainId: this.chainId,
-      acp: wireAcp,
-      thresholdNetworkUrl,
-      retry404TimeoutMs: this.retry404TimeoutMs,
-      timeoutMs: this.timeoutMs,
-      onPoll: this.pollCallback
-    });
+    const sealed = await this.limit(
+      () => tnSealOutputV2({
+        ctHash: this.ctHash,
+        chainId: this.chainId,
+        acp: wireAcp,
+        thresholdNetworkUrl,
+        retry404TimeoutMs: this.retry404TimeoutMs,
+        timeoutMs: this.timeoutMs,
+        onPoll: this.pollCallback,
+        signal: this.signal
+      }),
+      "sealOutput",
+      this.signal
+    );
     return ACPUtils.unseal(acp, sealed);
   }
   /**
@@ -4688,7 +4746,7 @@ function assertDecryptStatusResponseV2(value) {
   }
   return value;
 }
-async function submitDecryptRequestV2(thresholdNetworkUrl, ctHash, chainId, acp, overallStartTime, retry404TimeoutMs, timeoutMs, onPoll) {
+async function submitDecryptRequestV2(thresholdNetworkUrl, ctHash, chainId, acp, overallStartTime, retry404TimeoutMs, timeoutMs, onPoll, signal) {
   const body = {
     ct_tempkey: BigInt(ctHash).toString(16).padStart(64, "0"),
     host_chain_id: chainId
@@ -4699,6 +4757,7 @@ async function submitDecryptRequestV2(thresholdNetworkUrl, ctHash, chainId, acp,
   let attemptIndex = 0;
   let last404ApiErrorMessage;
   for (; ; ) {
+    throwIfAborted(signal, "decrypt");
     let response;
     try {
       response = await cofheFetch(
@@ -4709,10 +4768,12 @@ async function submitDecryptRequestV2(thresholdNetworkUrl, ctHash, chainId, acp,
           headers: {
             "Content-Type": "application/json"
           },
-          body: JSON.stringify(body)
+          body: JSON.stringify(body),
+          signal
         }
       );
     } catch (e) {
+      throwIfAborted(signal, "decrypt");
       throw new CofheError({
         code: "DECRYPT_FAILED" /* DecryptFailed */,
         message: `decrypt request failed`,
@@ -4809,13 +4870,14 @@ async function submitDecryptRequestV2(thresholdNetworkUrl, ctHash, chainId, acp,
       intervalMs: SUBMIT_RETRY_INTERVAL_MS2,
       timeoutMs
     });
-    await new Promise((resolve) => setTimeout(resolve, SUBMIT_RETRY_INTERVAL_MS2));
+    await sleepUnlessAborted(SUBMIT_RETRY_INTERVAL_MS2, signal, "decrypt");
     attemptIndex += 1;
   }
 }
-async function pollDecryptStatusV2(thresholdNetworkUrl, requestId, overallStartTime, timeoutMs, onPoll) {
+async function pollDecryptStatusV2(thresholdNetworkUrl, requestId, overallStartTime, timeoutMs, onPoll, signal) {
   let attemptIndex = 0;
   while (true) {
+    throwIfAborted(signal, "decrypt");
     const elapsedMs = Date.now() - overallStartTime;
     const intervalMs = computeMinuteRampPollIntervalMs(elapsedMs, {
       minIntervalMs: POLL_INTERVAL_MS2,
@@ -4850,7 +4912,8 @@ async function pollDecryptStatusV2(thresholdNetworkUrl, requestId, overallStartT
           method: "GET",
           headers: {
             "Content-Type": "application/json"
-          }
+          },
+          signal
         }
       );
     } catch (e) {
@@ -4912,7 +4975,7 @@ async function pollDecryptStatusV2(thresholdNetworkUrl, requestId, overallStartT
         requestId
       });
     }
-    await new Promise((resolve) => setTimeout(resolve, intervalMs));
+    await sleepUnlessAborted(intervalMs, signal, "decrypt");
     attemptIndex += 1;
   }
   throw new CofheError({
@@ -4925,7 +4988,7 @@ async function pollDecryptStatusV2(thresholdNetworkUrl, requestId, overallStartT
   });
 }
 async function tnDecryptV2(params) {
-  const { thresholdNetworkUrl, ctHash, chainId, acp, retry404TimeoutMs, onPoll } = params;
+  const { thresholdNetworkUrl, ctHash, chainId, acp, retry404TimeoutMs, onPoll, signal } = params;
   const timeoutMs = params.timeoutMs ?? DEFAULT_DECRYPT_TIMEOUT_MS;
   const normalized404RetryTimeoutMs = normalize404RetryTimeoutMs({
     timeoutMs: retry404TimeoutMs,
@@ -4941,12 +5004,20 @@ async function tnDecryptV2(params) {
     overallStartTime,
     normalized404RetryTimeoutMs,
     timeoutMs,
-    onPoll
+    onPoll,
+    signal
   );
   if (submitResult.kind === "completed") {
     return submitResult;
   }
-  return await pollDecryptStatusV2(thresholdNetworkUrl, submitResult.requestId, overallStartTime, timeoutMs, onPoll);
+  return await pollDecryptStatusV2(
+    thresholdNetworkUrl,
+    submitResult.requestId,
+    overallStartTime,
+    timeoutMs,
+    onPoll,
+    signal
+  );
 }
 
 // core/decrypt/decryptForTxBuilder.ts
@@ -4959,6 +5030,8 @@ var DecryptForTxBuilder = class extends BaseBuilder {
   pollCallback;
   retry404TimeoutMs = DEFAULT_404_RETRY_TIMEOUT_MS3;
   timeoutMs;
+  signal;
+  limit;
   constructor(params) {
     super({
       config: params.config,
@@ -4969,6 +5042,7 @@ var DecryptForTxBuilder = class extends BaseBuilder {
       requireConnected: params.requireConnected
     });
     this.ctHash = params.ctHash;
+    this.limit = params.limit ?? ((task) => task());
   }
   setChainId(chainId) {
     this.chainId = chainId;
@@ -5006,6 +5080,10 @@ var DecryptForTxBuilder = class extends BaseBuilder {
     this.timeoutMs = timeoutMs;
     return this;
   }
+  setSignal(signal) {
+    this.signal = signal;
+    return this;
+  }
   withACP(acpOrACPHash) {
     if (this.acpSelection === "with-acp") {
       throw new CofheError({
@@ -5133,15 +5211,20 @@ var DecryptForTxBuilder = class extends BaseBuilder {
     this.assertPublicClient();
     const thresholdNetworkUrl = await this.getThresholdNetworkUrl();
     const wireAcp = acp ? ACPUtils.getPublic(acp, true) : null;
-    const { decryptedValue, signature } = await tnDecryptV2({
-      ctHash: this.ctHash,
-      chainId: this.chainId,
-      acp: wireAcp,
-      thresholdNetworkUrl,
-      retry404TimeoutMs: this.retry404TimeoutMs,
-      timeoutMs: this.timeoutMs,
-      onPoll: this.pollCallback
-    });
+    const { decryptedValue, signature } = await this.limit(
+      () => tnDecryptV2({
+        ctHash: this.ctHash,
+        chainId: this.chainId,
+        acp: wireAcp,
+        thresholdNetworkUrl,
+        retry404TimeoutMs: this.retry404TimeoutMs,
+        timeoutMs: this.timeoutMs,
+        onPoll: this.pollCallback,
+        signal: this.signal
+      }),
+      "decrypt",
+      this.signal
+    );
     return {
       ctHash: this.ctHash,
       decryptedValue,
@@ -5179,6 +5262,39 @@ var DecryptForTxBuilder = class extends BaseBuilder {
     }
   }
 };
+
+// core/decrypt/decryptLimiter.ts
+function createDecryptLimiter(max) {
+  let active = 0;
+  const waiting = [];
+  return async (task, operationLabel, signal) => {
+    if (signal?.aborted)
+      throw decryptAbortedError(operationLabel, signal);
+    if (active >= max) {
+      await new Promise((resume, reject) => {
+        const onAbort = () => {
+          const index = waiting.indexOf(start);
+          if (index >= 0)
+            waiting.splice(index, 1);
+          reject(decryptAbortedError(operationLabel, signal));
+        };
+        const start = () => {
+          signal?.removeEventListener("abort", onAbort);
+          resume();
+        };
+        signal?.addEventListener("abort", onAbort, { once: true });
+        waiting.push(start);
+      });
+    }
+    active += 1;
+    try {
+      return await task();
+    } finally {
+      active -= 1;
+      waiting.shift()?.();
+    }
+  };
+}
 var decryptResultSignerAbi = viem.parseAbi(["function decryptResultSigner() view returns (address)"]);
 var UINT_TYPE_MASK2 = 0x7fn;
 var TYPE_BYTE_OFFSET2 = 8n;
@@ -5324,11 +5440,13 @@ function createCofheClientBase(opts) {
       requireConnected: _requireConnected
     });
   }
+  const decryptLimiter = createDecryptLimiter(opts.config.decryptConcurrency);
   function decryptForView(ctHash, utype) {
     const state = connectStore.getState();
     return new DecryptForViewBuilder({
       ctHash,
       utype,
+      limit: decryptLimiter,
       chainId: state.chainId,
       account: state.account,
       config: opts.config,
@@ -5341,6 +5459,7 @@ function createCofheClientBase(opts) {
     const state = connectStore.getState();
     return new DecryptForTxBuilder({
       ctHash,
+      limit: decryptLimiter,
       chainId: state.chainId,
       account: state.account,
       config: opts.config,

package/dist/node.d.cts

@@ -1,4 +1,4 @@
-import { C as CofheInputConfig, a as CofheConfig, b as CofheClient } from './clientTypes-BCAMwrqk.cjs';
+import { C as CofheInputConfig, a as CofheConfig, b as CofheClient } from './clientTypes-K6XNB9H2.cjs';
 import 'viem';
 import './types-C07FK-cL.cjs';
 import 'zod';

package/dist/node.d.ts

@@ -1,4 +1,4 @@
-import { C as CofheInputConfig, a as CofheConfig, b as CofheClient } from './clientTypes-LvslG7o0.js';
+import { C as CofheInputConfig, a as CofheConfig, b as CofheClient } from './clientTypes-DFXuL9kz.js';
 import 'viem';
 import './types-C07FK-cL.js';
 import 'zod';

package/dist/node.js

@@ -1,8 +1,8 @@
-import { createCofheConfigBase, createCofheClientBase } from './chunk-KHM6LYLU.js';
+import { createCofheConfigBase, createCofheClientBase } from './chunk-ZYCIUX5C.js';
 import './chunk-UKQAJK4N.js';
-import './chunk-WAA2U7CT.js';
-import { fromHexString } from './chunk-3SD2UCRE.js';
-import { TFHE_RS_SAFE_SERIALIZATION_SIZE_LIMIT } from './chunk-OC5TGQGG.js';
+import './chunk-37SG44CR.js';
+import { fromHexString } from './chunk-PLHQYVND.js';
+import { TFHE_RS_SAFE_SERIALIZATION_SIZE_LIMIT } from './chunk-NJTYZ3CW.js';
 import { promises } from 'fs';
 import { join } from 'path';
 import { init_panic_hook, ProvenCompactCiphertextList, CompactPkeCrs, TfheCompactPublicKey } from 'node-tfhe';