@enclave-e3/sdk 0.1.6 → 0.1.8

package/README.md

@@ -217,7 +217,7 @@ await sdk.requestE3({
 });
 
 // Activate an E3 computation
-await sdk.activateE3(e3Id: bigint, publicKey: `0x${string}`, gasLimit?: bigint);
+await sdk.activateE3(e3Id: bigint, gasLimit?: bigint);
 
 // Publish input data
 await sdk.publishInput(e3Id: bigint, data: `0x${string}`, gasLimit?: bigint);

package/dist/index.cjs

@@ -30,17 +30,15 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
-  BFV_PARAMS_SET: () => BFV_PARAMS_SET,
-  BfvProtocolParams: () => BfvProtocolParams,
   ContractClient: () => ContractClient,
   DEFAULT_COMPUTE_PROVIDER_PARAMS: () => DEFAULT_COMPUTE_PROVIDER_PARAMS,
   DEFAULT_E3_CONFIG: () => DEFAULT_E3_CONFIG,
   EnclaveEventType: () => EnclaveEventType,
   EnclaveSDK: () => EnclaveSDK,
   EventListener: () => EventListener,
-  FheProtocol: () => FheProtocol,
   RegistryEventType: () => RegistryEventType,
   SDKError: () => SDKError,
+  ThresholdBfvParamsPresetNames: () => ThresholdBfvParamsPresetNames,
   calculateStartWindow: () => calculateStartWindow,
   convertToPolynomial: () => convertToPolynomial,
   convertToPolynomialArray: () => convertToPolynomialArray,
@@ -107,14 +105,6 @@ function generateEventId(log) {
 function getCurrentTimestamp() {
   return Math.floor(Date.now() / 1e3);
 }
-var INSECURE_SET_2048_1032193_1 = {
-  degree: 2048,
-  plaintext_modulus: 1032193,
-  moduli: [0x3fffffff000001n],
-  // BigInt for the modulus
-  error1_variance: "10"
-};
-var BFV_PARAMS_SET = INSECURE_SET_2048_1032193_1;
 var DEFAULT_COMPUTE_PROVIDER_PARAMS = {
   name: "risc0",
   parallel: false,
@@ -130,7 +120,14 @@ var DEFAULT_E3_CONFIG = {
   payment_amount: "0"
   // 0 ETH in wei
 };
-function encodeBfvParams(degree = BFV_PARAMS_SET.degree, plaintext_modulus = BFV_PARAMS_SET.plaintext_modulus, moduli = BFV_PARAMS_SET.moduli, error1_variance = BFV_PARAMS_SET.error1_variance) {
+function encodeBfvParams(params) {
+  const { degree, plaintextModulus, moduli, error1Variance } = params;
+  if (error1Variance === void 0) {
+    throw new SDKError(
+      "error1Variance is required in ProtocolParams. All BFV parameter sets must specify error1_variance.",
+      "MISSING_ERROR1_VARIANCE"
+    );
+  }
   return (0, import_viem.encodeAbiParameters)(
     [
       {
@@ -147,9 +144,9 @@ function encodeBfvParams(degree = BFV_PARAMS_SET.degree, plaintext_modulus = BFV
     [
       {
         degree: BigInt(degree),
-        plaintext_modulus: BigInt(plaintext_modulus),
+        plaintext_modulus: BigInt(plaintextModulus),
         moduli: [...moduli],
-        error1_variance
+        error1_variance: error1Variance
       }
     ]
   );
@@ -305,9 +302,9 @@ var ContractClient = class {
   }
   /**
    * Activate an E3 computation
-   * activate(uint256 e3Id, bytes memory publicKey)
+   * activate(uint256 e3Id)
    */
-  async activateE3(e3Id, publicKey, gasLimit) {
+  async activateE3(e3Id, gasLimit) {
     if (!this.walletClient) {
       throw new SDKError("Wallet client required for write operations", "NO_WALLET");
     }
@@ -323,7 +320,7 @@ var ContractClient = class {
         address: this.addresses.enclave,
         abi: import_types.Enclave__factory.abi,
         functionName: "activate",
-        args: [e3Id, publicKey],
+        args: [e3Id],
         account,
         gas: gasLimit
       });
@@ -694,37 +691,10 @@ var RegistryEventType = /* @__PURE__ */ ((RegistryEventType2) => {
   RegistryEventType2["INITIALIZED"] = "Initialized";
   return RegistryEventType2;
 })(RegistryEventType || {});
-var FheProtocol = /* @__PURE__ */ ((FheProtocol2) => {
-  FheProtocol2["BFV"] = "BFV";
-  FheProtocol2["TRBFV"] = "TRBFV";
-  return FheProtocol2;
-})(FheProtocol || {});
-var BfvProtocolParams = {
-  /**
-   * Recommended parameters for BFV protocol
-   * - Degree: 2048
-   * - Plaintext modulus: 1032193
-   * - Moduli:0x3FFFFFFF000001
-   */
-  BFV_NORMAL: {
-    degree: 2048,
-    plaintextModulus: 1032193n,
-    moduli: [0x3fffffff000001n],
-    error1Variance: "10"
-  },
-  /**
-   * Recommended parameters for TrBFV protocol
-   * - Degree: 8192
-   * - Plaintext modulus: 1000
-   * - Moduli: [0x00800000022a0001, 0x00800000021a0001, 0x0080000002120001, 0x0080000001f60001]
-   */
-  BFV_THRESHOLD: {
-    degree: 8192,
-    plaintextModulus: 1000n,
-    moduli: [0x00800000022a0001n, 0x00800000021a0001n, 0x0080000002120001n, 0x0080000001f60001n],
-    error1Variance: "10"
-  }
-};
+var ThresholdBfvParamsPresetNames = [
+  "INSECURE_THRESHOLD_BFV_512",
+  "SECURE_THRESHOLD_BFV_8192"
+];
 
 // src/enclave-sdk.ts
 var import_wasm = require("@enclave-e3/wasm");
@@ -812,15 +782,18 @@ var EnclaveSDK = class _EnclaveSDK {
     if (!isValidAddress(config.contracts.feeToken)) {
       throw new SDKError("Invalid FeeToken contract address", "INVALID_ADDRESS");
     }
-    this.eventListener = new EventListener(config.publicClient);
-    this.contractClient = new ContractClient(config.publicClient, config.walletClient, config.contracts);
-    if (!Object.values(FheProtocol).includes(config.protocol)) {
-      throw new SDKError(`Invalid protocol: ${config.protocol}`, "INVALID_PROTOCOL");
+    if (!config.thresholdBfvParamsPresetName) {
+      throw new SDKError("Threshold BFV parameters preset name is required", "MISSING_THRESHOLD_BFV_PARAMS_PRESET_NAME");
     }
-    this.protocol = config.protocol;
-    if (config.protocolParams) {
-      this.protocolParams = config.protocolParams;
+    if (!Object.values(ThresholdBfvParamsPresetNames).includes(config.thresholdBfvParamsPresetName)) {
+      throw new SDKError(
+        `Invalid threshold BFV parameters preset name: ${config.thresholdBfvParamsPresetName}`,
+        "INVALID_THRESHOLD_BFV_PARAMS_PRESET_NAME"
+      );
     }
+    this.thresholdBfvParamsPresetName = config.thresholdBfvParamsPresetName;
+    this.eventListener = new EventListener(config.publicClient);
+    this.contractClient = new ContractClient(config.publicClient, config.walletClient, config.contracts);
     this.publicClient = config.publicClient;
   }
   static chains = {
@@ -832,8 +805,7 @@ var EnclaveSDK = class _EnclaveSDK {
   eventListener;
   contractClient;
   initialized = false;
-  protocol;
-  protocolParams;
+  thresholdBfvParamsPresetName;
   publicClient;
   /**
    * Initialize the SDK
@@ -855,9 +827,9 @@ var EnclaveSDK = class _EnclaveSDK {
   getPublicClient = () => {
     return this.publicClient;
   };
-  async getBfvParamsSet(name) {
+  async getThresholdBfvParamsSet() {
     await (0, import_init.default)();
-    let params = (0, import_wasm.get_bfv_params)(name);
+    let params = (0, import_wasm.get_bfv_params)(this.thresholdBfvParamsPresetName);
     return {
       degree: Number(params.degree),
       // degree is returned as a bigint from wasm
@@ -866,17 +838,15 @@ var EnclaveSDK = class _EnclaveSDK {
       error1Variance: params.error1_variance
     };
   }
-  async getProtocolParams() {
+  async computePublicKeyCommitment(publicKey) {
     await (0, import_init.default)();
-    if (this.protocolParams) {
-      return this.protocolParams;
-    }
-    switch (this.protocol) {
-      case "BFV" /* BFV */:
-        return await this.getBfvParamsSet("INSECURE_SET_2048_1032193_1");
-      case "TRBFV" /* TRBFV */:
-        return await this.getBfvParamsSet("INSECURE_SET_512_10_1");
-    }
+    const protocolParams = await this.getThresholdBfvParamsSet();
+    return (0, import_wasm.compute_pk_commitment)(
+      publicKey,
+      protocolParams.degree,
+      protocolParams.plaintextModulus,
+      BigUint64Array.from(protocolParams.moduli)
+    );
   }
   /**
    * Encrypt a number using the configured protocol
@@ -886,7 +856,7 @@ var EnclaveSDK = class _EnclaveSDK {
    */
   async encryptNumber(data, publicKey) {
     await (0, import_init.default)();
-    const protocolParams = await this.getProtocolParams();
+    const protocolParams = await this.getThresholdBfvParamsSet();
     return (0, import_wasm.bfv_encrypt_number)(
       data,
       publicKey,
@@ -903,7 +873,7 @@ var EnclaveSDK = class _EnclaveSDK {
    */
   async encryptVector(data, publicKey) {
     await (0, import_init.default)();
-    const protocolParams = await this.getProtocolParams();
+    const protocolParams = await this.getThresholdBfvParamsSet();
     return (0, import_wasm.bfv_encrypt_vector)(
       data,
       publicKey,
@@ -921,7 +891,7 @@ var EnclaveSDK = class _EnclaveSDK {
    */
   async encryptNumberAndGenInputs(data, publicKey) {
     await (0, import_init.default)();
-    const protocolParams = await this.getProtocolParams();
+    const protocolParams = await this.getThresholdBfvParamsSet();
     const [encryptedData, circuitInputs] = (0, import_wasm.bfv_verifiable_encrypt_number)(
       data,
       publicKey,
@@ -958,7 +928,7 @@ var EnclaveSDK = class _EnclaveSDK {
    */
   async encryptVectorAndGenInputs(data, publicKey) {
     await (0, import_init.default)();
-    const protocolParams = await this.getProtocolParams();
+    const protocolParams = await this.getThresholdBfvParamsSet();
     const [encryptedData, circuitInputs] = (0, import_wasm.bfv_verifiable_encrypt_vector)(
       data,
       publicKey,
@@ -1032,11 +1002,11 @@ var EnclaveSDK = class _EnclaveSDK {
   /**
    * Activate an E3 computation
    */
-  async activateE3(e3Id, publicKey, gasLimit) {
+  async activateE3(e3Id, gasLimit) {
     if (!this.initialized) {
       await this.initialize();
     }
-    return this.contractClient.activateE3(e3Id, publicKey, gasLimit);
+    return this.contractClient.activateE3(e3Id, gasLimit);
   }
   /**
    * Publish input for an E3 computation
@@ -1184,24 +1154,21 @@ var EnclaveSDK = class _EnclaveSDK {
       walletClient,
       contracts: options.contracts,
       chainId: options.chainId,
-      protocol: options.protocol,
-      protocolParams: options.protocolParams
+      thresholdBfvParamsPresetName: options.thresholdBfvParamsPresetName
     });
   }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
-  BFV_PARAMS_SET,
-  BfvProtocolParams,
   ContractClient,
   DEFAULT_COMPUTE_PROVIDER_PARAMS,
   DEFAULT_E3_CONFIG,
   EnclaveEventType,
   EnclaveSDK,
   EventListener,
-  FheProtocol,
   RegistryEventType,
   SDKError,
+  ThresholdBfvParamsPresetNames,
   calculateStartWindow,
   convertToPolynomial,
   convertToPolynomialArray,