@cofhe/hardhat-plugin 0.2.1 → 0.3.0

package/CHANGELOG.md

@@ -1,5 +1,35 @@
 # @cofhe/hardhat-plugin Changelog
 
+## 0.3.0
+
+### Minor Changes
+
+- 35024b6: Remove `sdk` from function names and exported types. Rename:
+
+  - `createCofhesdkConfig` -> `createCofheConfig`
+  - `createCofhesdkClient` -> `createCofheClient`
+  - `hre.cofhesdk.*` -> `hre.cofhe.*`
+  - `hre.cofhesdk.createCofheConfig()` → `hre.cofhe.createConfig()`
+  - `hre.cofhesdk.createCofheClient()` → `hre.cofhe.createClient()`
+  - `hre.cofhesdk.createBatteriesIncludedCofheClient()` → `hre.cofhe.createClientWithBatteries()`
+
+- 29c2401: implement decrypt-with-proof flows and related tests:
+
+  - Implement production `decryptForTx` backed by Threshold Network `POST /decrypt`, with explicit permit vs global-allowance selection.
+  - Rename mocks “Query Decrypter” -> “Threshold Network” and update SDK constants/contracts/artifacts accordingly.
+  - Extend mock contracts + hardhat plugin to publish & verify decryption results on-chain, and add end-to-end integration tests.
+
+### Patch Changes
+
+- 5467d77: Adds `config.mocks.encryptDelay: number | [number, number, number, number, number]` to allow configurable mock encryption delay. Defaults to 0 delay on hardhat to keep tests quick.
+- Updated dependencies [35024b6]
+- Updated dependencies [5467d77]
+- Updated dependencies [73b1502]
+- Updated dependencies [29c2401]
+- Updated dependencies [650ea48]
+  - @cofhe/mock-contracts@0.3.0
+  - @cofhe/sdk@0.3.0
+
 ## 0.2.1
 
 ### Patch Changes

package/README.md

@@ -46,7 +46,7 @@ This plugin uses [cofhe-mock-contracts](https://github.com/FhenixProtocol/cofhe-
 The mock contracts include:
 
 - MockTaskManager: Manages FHE operations and stores plaintext values
-- MockQueryDecrypter: Handles decryption requests
+- MockThresholdNetwork: Handles decryption requests
 - MockZkVerifier: Simulates verification of encrypted inputs
 - ACL: Handles access control
 
@@ -64,6 +64,12 @@ Mock contracts are automatically deployed when using the Hardhat network:
 - Running tests: `npx hardhat test`
 - Starting a local node: `npx hardhat node`
 
+To disable this automatic deployment (for example, when your tests only use external RPCs and don't need the in-process Hardhat network), set:
+
+```bash
+COFHE_SKIP_MOCKS_DEPLOY=1
+```
+
 You can also manually deploy mock contracts:
 
 ```bash

package/dist/index.d.mts

@@ -1,6 +1,6 @@
 import { PublicClient, WalletClient } from 'viem';
 import { HardhatEthersSigner } from '@nomicfoundation/hardhat-ethers/signers';
-import { CofhesdkInputConfig, CofhesdkConfig, CofhesdkClient } from '@cofhe/sdk';
+import { CofheInputConfig, CofheConfig, CofheClient } from '@cofhe/sdk';
 import { HardhatRuntimeEnvironment } from 'hardhat/types';
 import * as ethers from 'ethers';
 import { Contract, ethers as ethers$1 } from 'ethers';
@@ -49,7 +49,7 @@ declare const mock_withLogs: (hre: HardhatRuntimeEnvironment, closureName: strin
  */
 declare module 'hardhat/types/config' {
     interface HardhatUserConfig {
-        cofhesdk?: {
+        cofhe?: {
             /** Whether to log mock operations (default: true) */
             logMocks?: boolean;
             /** Whether to show gas usage warnings for mock operations (default: true) */
@@ -57,7 +57,7 @@ declare module 'hardhat/types/config' {
         };
     }
     interface HardhatConfig {
-        cofhesdk: {
+        cofhe: {
             /** Whether to log mock operations (default: true) */
             logMocks: boolean;
             /** Whether to show gas usage warnings for mock operations (default: true) */
@@ -72,21 +72,21 @@ declare module 'hardhat/types/config' {
  */
 declare module 'hardhat/types/runtime' {
     interface HardhatRuntimeEnvironment {
-        cofhesdk: {
+        cofhe: {
             /**
-             * Create a CoFHE SDK configuration for use with cofhesdk.createCofhesdkClient(...)
-             * @param {CofhesdkInputConfig} config - The CoFHE SDK input configuration
-             * @returns {CofhesdkConfig} The CoFHE SDK configuration
+             * Create a CoFHE configuration for use with hre.cofhe.createClient(...)
+             * @param {CofheInputConfig} config - The CoFHE input configuration
+             * @returns {CofheConfig} The CoFHE configuration
              */
-            createCofhesdkConfig: (config: CofhesdkInputConfig) => Promise<CofhesdkConfig>;
+            createConfig: (config: CofheInputConfig) => Promise<CofheConfig>;
             /**
-             * Create a CoFHE SDK client instance
-             * @param {CofhesdkConfig} config - The CoFHE SDK configuration (use createCofhesdkConfig to create with Node.js defaults)
-             * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance
+             * Create a CoFHE client instance
+             * @param {CofheConfig} config - The CoFHE configuration (use createCofheConfig to create with Node.js defaults)
+             * @returns {Promise<CofheClient>} The CoFHE client instance
              */
-            createCofhesdkClient: (config: CofhesdkConfig) => CofhesdkClient;
+            createClient: (config: CofheConfig) => CofheClient;
             /**
-             * Create viem clients from a Hardhat ethers signer, to be used with `cofhesdkClient.connect(...)`
+             * Create viem clients from a Hardhat ethers signer, to be used with `cofheClient.connect(...)`
              * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use
              * @returns {Promise<{ publicClient: PublicClient; walletClient: WalletClient }>} The viem clients
              */
@@ -95,20 +95,20 @@ declare module 'hardhat/types/runtime' {
                 walletClient: WalletClient;
             }>;
             /**
-             * Connect a CoFHE SDK client with a Hardhat ethers signer
-             * @param {CofhesdkClient} client - The CoFHE SDK client to connect
+             * Connect a CoFHE client with a Hardhat ethers signer
+             * @param {CofheClient} client - The CoFHE client to connect
              * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use
              * @returns {Promise<void>}
              */
-            connectWithHardhatSigner: (client: CofhesdkClient, signer: HardhatEthersSigner) => Promise<void>;
+            connectWithHardhatSigner: (client: CofheClient, signer: HardhatEthersSigner) => Promise<void>;
             /**
              * Create and connect to a batteries included client.
              * Also generates a self-usage a permit for the signer.
-             * If customization is needed, use createCofhesdkClient and connectWithHardhatSigner.
+             * If customization is needed, use createCofheClient and connectWithHardhatSigner.
              * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use (optional - defaults to first signer)
-             * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance
+             * @returns {Promise<CofheClient>} The CoFHE client instance
              */
-            createBatteriesIncludedCofhesdkClient: (signer?: HardhatEthersSigner) => Promise<CofhesdkClient>;
+            createClientWithBatteries: (signer?: HardhatEthersSigner) => Promise<CofheClient>;
             mocks: {
                 /**
                  * **[MOCKS ONLY]**
@@ -120,7 +120,7 @@ declare module 'hardhat/types/runtime' {
                  * Example usage:
                  *
                  * ```ts
-                 * await hre.cofhesdk.mocks.withLogs("counter.increment()", async () => {
+                 * await hre.cofhe.mocks.withLogs("counter.increment()", async () => {
                  *   await counter.increment();
                  * });
                  * ```
@@ -186,10 +186,10 @@ declare module 'hardhat/types/runtime' {
                  */
                 getMockACL: () => Promise<Contract>;
                 /**
-                 * Get the MockQueryDecrypter contract
-                 * @returns {Promise<Contract>} The MockQueryDecrypter contract
+                 * Get the MockThresholdNetwork contract
+                 * @returns {Promise<Contract>} The MockThresholdNetwork contract
                  */
-                getMockQueryDecrypter: () => Promise<Contract>;
+                getMockThresholdNetwork: () => Promise<Contract>;
                 /**
                  * Get the MockZkVerifier contract
                  * @returns {Promise<Contract>} The MockZkVerifier contract
@@ -204,5 +204,11 @@ declare module 'hardhat/types/runtime' {
         };
     }
 }
+/**
+ * Builds the mocks config for the hardhat plugin.
+ * Defaults `encryptDelay` to `0` so tests run without artificial wait times,
+ * unless the user has explicitly provided a value.
+ */
+declare function buildHardhatPluginMocksConfig(mocksConfig: CofheInputConfig['mocks']): NonNullable<CofheInputConfig['mocks']>;
 
-export { type DeployMocksArgs, TASK_COFHE_MOCKS_DEPLOY, TASK_COFHE_MOCKS_SET_LOG_OPS, TASK_COFHE_USE_FAUCET, deployMockContractFromArtifact, deployMocks, getFixedMockContract, localcofheFundAccount, localcofheFundWalletIfNeeded, mock_expectPlaintext, mock_getPlaintext, mock_getPlaintextExists, mock_setLoggingEnabled, mock_withLogs };
+export { type DeployMocksArgs, TASK_COFHE_MOCKS_DEPLOY, TASK_COFHE_MOCKS_SET_LOG_OPS, TASK_COFHE_USE_FAUCET, buildHardhatPluginMocksConfig, deployMockContractFromArtifact, deployMocks, getFixedMockContract, localcofheFundAccount, localcofheFundWalletIfNeeded, mock_expectPlaintext, mock_getPlaintext, mock_getPlaintextExists, mock_setLoggingEnabled, mock_withLogs };

package/dist/index.d.ts

@@ -1,6 +1,6 @@
 import { PublicClient, WalletClient } from 'viem';
 import { HardhatEthersSigner } from '@nomicfoundation/hardhat-ethers/signers';
-import { CofhesdkInputConfig, CofhesdkConfig, CofhesdkClient } from '@cofhe/sdk';
+import { CofheInputConfig, CofheConfig, CofheClient } from '@cofhe/sdk';
 import { HardhatRuntimeEnvironment } from 'hardhat/types';
 import * as ethers from 'ethers';
 import { Contract, ethers as ethers$1 } from 'ethers';
@@ -49,7 +49,7 @@ declare const mock_withLogs: (hre: HardhatRuntimeEnvironment, closureName: strin
  */
 declare module 'hardhat/types/config' {
     interface HardhatUserConfig {
-        cofhesdk?: {
+        cofhe?: {
             /** Whether to log mock operations (default: true) */
             logMocks?: boolean;
             /** Whether to show gas usage warnings for mock operations (default: true) */
@@ -57,7 +57,7 @@ declare module 'hardhat/types/config' {
         };
     }
     interface HardhatConfig {
-        cofhesdk: {
+        cofhe: {
             /** Whether to log mock operations (default: true) */
             logMocks: boolean;
             /** Whether to show gas usage warnings for mock operations (default: true) */
@@ -72,21 +72,21 @@ declare module 'hardhat/types/config' {
  */
 declare module 'hardhat/types/runtime' {
     interface HardhatRuntimeEnvironment {
-        cofhesdk: {
+        cofhe: {
             /**
-             * Create a CoFHE SDK configuration for use with cofhesdk.createCofhesdkClient(...)
-             * @param {CofhesdkInputConfig} config - The CoFHE SDK input configuration
-             * @returns {CofhesdkConfig} The CoFHE SDK configuration
+             * Create a CoFHE configuration for use with hre.cofhe.createClient(...)
+             * @param {CofheInputConfig} config - The CoFHE input configuration
+             * @returns {CofheConfig} The CoFHE configuration
              */
-            createCofhesdkConfig: (config: CofhesdkInputConfig) => Promise<CofhesdkConfig>;
+            createConfig: (config: CofheInputConfig) => Promise<CofheConfig>;
             /**
-             * Create a CoFHE SDK client instance
-             * @param {CofhesdkConfig} config - The CoFHE SDK configuration (use createCofhesdkConfig to create with Node.js defaults)
-             * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance
+             * Create a CoFHE client instance
+             * @param {CofheConfig} config - The CoFHE configuration (use createCofheConfig to create with Node.js defaults)
+             * @returns {Promise<CofheClient>} The CoFHE client instance
              */
-            createCofhesdkClient: (config: CofhesdkConfig) => CofhesdkClient;
+            createClient: (config: CofheConfig) => CofheClient;
             /**
-             * Create viem clients from a Hardhat ethers signer, to be used with `cofhesdkClient.connect(...)`
+             * Create viem clients from a Hardhat ethers signer, to be used with `cofheClient.connect(...)`
              * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use
              * @returns {Promise<{ publicClient: PublicClient; walletClient: WalletClient }>} The viem clients
              */
@@ -95,20 +95,20 @@ declare module 'hardhat/types/runtime' {
                 walletClient: WalletClient;
             }>;
             /**
-             * Connect a CoFHE SDK client with a Hardhat ethers signer
-             * @param {CofhesdkClient} client - The CoFHE SDK client to connect
+             * Connect a CoFHE client with a Hardhat ethers signer
+             * @param {CofheClient} client - The CoFHE client to connect
              * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use
              * @returns {Promise<void>}
              */
-            connectWithHardhatSigner: (client: CofhesdkClient, signer: HardhatEthersSigner) => Promise<void>;
+            connectWithHardhatSigner: (client: CofheClient, signer: HardhatEthersSigner) => Promise<void>;
             /**
              * Create and connect to a batteries included client.
              * Also generates a self-usage a permit for the signer.
-             * If customization is needed, use createCofhesdkClient and connectWithHardhatSigner.
+             * If customization is needed, use createCofheClient and connectWithHardhatSigner.
              * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use (optional - defaults to first signer)
-             * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance
+             * @returns {Promise<CofheClient>} The CoFHE client instance
              */
-            createBatteriesIncludedCofhesdkClient: (signer?: HardhatEthersSigner) => Promise<CofhesdkClient>;
+            createClientWithBatteries: (signer?: HardhatEthersSigner) => Promise<CofheClient>;
             mocks: {
                 /**
                  * **[MOCKS ONLY]**
@@ -120,7 +120,7 @@ declare module 'hardhat/types/runtime' {
                  * Example usage:
                  *
                  * ```ts
-                 * await hre.cofhesdk.mocks.withLogs("counter.increment()", async () => {
+                 * await hre.cofhe.mocks.withLogs("counter.increment()", async () => {
                  *   await counter.increment();
                  * });
                  * ```
@@ -186,10 +186,10 @@ declare module 'hardhat/types/runtime' {
                  */
                 getMockACL: () => Promise<Contract>;
                 /**
-                 * Get the MockQueryDecrypter contract
-                 * @returns {Promise<Contract>} The MockQueryDecrypter contract
+                 * Get the MockThresholdNetwork contract
+                 * @returns {Promise<Contract>} The MockThresholdNetwork contract
                  */
-                getMockQueryDecrypter: () => Promise<Contract>;
+                getMockThresholdNetwork: () => Promise<Contract>;
                 /**
                  * Get the MockZkVerifier contract
                  * @returns {Promise<Contract>} The MockZkVerifier contract
@@ -204,5 +204,11 @@ declare module 'hardhat/types/runtime' {
         };
     }
 }
+/**
+ * Builds the mocks config for the hardhat plugin.
+ * Defaults `encryptDelay` to `0` so tests run without artificial wait times,
+ * unless the user has explicitly provided a value.
+ */
+declare function buildHardhatPluginMocksConfig(mocksConfig: CofheInputConfig['mocks']): NonNullable<CofheInputConfig['mocks']>;
 
-export { type DeployMocksArgs, TASK_COFHE_MOCKS_DEPLOY, TASK_COFHE_MOCKS_SET_LOG_OPS, TASK_COFHE_USE_FAUCET, deployMockContractFromArtifact, deployMocks, getFixedMockContract, localcofheFundAccount, localcofheFundWalletIfNeeded, mock_expectPlaintext, mock_getPlaintext, mock_getPlaintextExists, mock_setLoggingEnabled, mock_withLogs };
+export { type DeployMocksArgs, TASK_COFHE_MOCKS_DEPLOY, TASK_COFHE_MOCKS_SET_LOG_OPS, TASK_COFHE_USE_FAUCET, buildHardhatPluginMocksConfig, deployMockContractFromArtifact, deployMocks, getFixedMockContract, localcofheFundAccount, localcofheFundWalletIfNeeded, mock_expectPlaintext, mock_getPlaintext, mock_getPlaintextExists, mock_setLoggingEnabled, mock_withLogs };

package/dist/index.js

@@ -33,6 +33,7 @@ __export(src_exports, {
   TASK_COFHE_MOCKS_DEPLOY: () => TASK_COFHE_MOCKS_DEPLOY,
   TASK_COFHE_MOCKS_SET_LOG_OPS: () => TASK_COFHE_MOCKS_SET_LOG_OPS,
   TASK_COFHE_USE_FAUCET: () => TASK_COFHE_USE_FAUCET,
+  buildHardhatPluginMocksConfig: () => buildHardhatPluginMocksConfig,
   deployMockContractFromArtifact: () => deployMockContractFromArtifact,
   deployMocks: () => deployMocks,
   getFixedMockContract: () => getFixedMockContract,
@@ -208,8 +209,8 @@ var deployMocks = async (hre, options = {
   logSuccess(`ETH balance: ${zkVerifierSignerBalance.toString()}`, 2);
   const zkVerifier = await deployMockZkVerifier(hre);
   logDeployment("MockZkVerifier", await zkVerifier.getAddress());
-  const queryDecrypter = await deployMockQueryDecrypter(hre, acl);
-  logDeployment("MockQueryDecrypter", await queryDecrypter.getAddress());
+  const thresholdNetwork = await deployMockThresholdNetwork(hre, acl);
+  logDeployment("MockThresholdNetwork", await thresholdNetwork.getAddress());
   if (options.deployTestBed) {
     logSuccess("TestBed deployment enabled", 2);
     const testBed = await deployTestBedContract(hre);
@@ -271,15 +272,15 @@ var deployMockZkVerifier = async (hre) => {
   }
   return zkVerifier;
 };
-var deployMockQueryDecrypter = async (hre, acl) => {
-  const queryDecrypter = await deployMockContractFromArtifact(hre, import_mock_contracts.MockQueryDecrypterArtifact);
-  const initTx = await queryDecrypter.initialize(import_sdk2.TASK_MANAGER_ADDRESS, await acl.getAddress());
+var deployMockThresholdNetwork = async (hre, acl) => {
+  const thresholdNetwork = await deployMockContractFromArtifact(hre, import_mock_contracts.MockThresholdNetworkArtifact);
+  const initTx = await thresholdNetwork.initialize(import_sdk2.TASK_MANAGER_ADDRESS, await acl.getAddress());
   await initTx.wait();
-  const queryDecrypterExists = await queryDecrypter.exists();
-  if (!queryDecrypterExists) {
-    throw new Error("MockQueryDecrypter does not exist");
+  const exists = await thresholdNetwork.exists();
+  if (!exists) {
+    throw new Error("MockThresholdNetwork does not exist");
   }
-  return queryDecrypter;
+  return thresholdNetwork;
 };
 var deployTestBedContract = async (hre) => {
   return deployMockContractFromArtifact(hre, import_mock_contracts.TestBedArtifact);
@@ -398,9 +399,9 @@ var import_mock_contracts3 = require("@cofhe/mock-contracts");
       httpHeaders: {}
     };
   }
-  config.cofhesdk = {
-    logMocks: userConfig.cofhesdk?.logMocks ?? true,
-    gasWarning: userConfig.cofhesdk?.gasWarning ?? true
+  config.cofhe = {
+    logMocks: userConfig.cofhe?.logMocks ?? true,
+    gasWarning: userConfig.cofhe?.gasWarning ?? true
   };
 });
 (0, import_config.task)(TASK_COFHE_USE_FAUCET, "Fund an account from the funder").addOptionalParam("address", "Address to fund", void 0, import_config.types.string).setAction(async ({ address }, hre) => {
@@ -424,44 +425,65 @@ var import_mock_contracts3 = require("@cofhe/mock-contracts");
 (0, import_config.task)(TASK_COFHE_MOCKS_DEPLOY, "Deploys the mock contracts on the Hardhat network").addOptionalParam("deployTestBed", "Whether to deploy the test bed", true, import_config.types.boolean).addOptionalParam("silent", "Whether to suppress output", false, import_config.types.boolean).setAction(async ({ deployTestBed, silent }, hre) => {
   await deployMocks(hre, {
     deployTestBed: deployTestBed ?? true,
-    gasWarning: hre.config.cofhesdk.gasWarning ?? true,
+    gasWarning: hre.config.cofhe.gasWarning ?? true,
     silent: silent ?? false
   });
 });
 (0, import_config.task)(import_task_names.TASK_TEST, "Deploy mock contracts on hardhat").setAction(async ({}, hre, runSuper) => {
-  await deployMocks(hre, {
-    deployTestBed: true,
-    gasWarning: hre.config.cofhesdk.gasWarning ?? true
-  });
+  const skipAutoDeploy = (() => {
+    const raw = process.env.COFHE_SKIP_MOCKS_DEPLOY ?? "";
+    const normalized = raw.trim().toLowerCase();
+    return normalized === "1" || normalized === "true" || normalized === "yes";
+  })();
+  if (!skipAutoDeploy) {
+    await deployMocks(hre, {
+      deployTestBed: true,
+      gasWarning: hre.config.cofhe.gasWarning ?? true
+    });
+  }
   return runSuper();
 });
 (0, import_config.task)(import_task_names.TASK_NODE, "Deploy mock contracts on hardhat").setAction(async ({}, hre, runSuper) => {
-  await deployMocks(hre, {
-    deployTestBed: true,
-    gasWarning: hre.config.cofhesdk.gasWarning ?? true
-  });
+  const skipAutoDeploy = (() => {
+    const raw = process.env.COFHE_SKIP_MOCKS_DEPLOY ?? "";
+    const normalized = raw.trim().toLowerCase();
+    return normalized === "1" || normalized === "true" || normalized === "yes";
+  })();
+  if (!skipAutoDeploy) {
+    await deployMocks(hre, {
+      deployTestBed: true,
+      gasWarning: hre.config.cofhe.gasWarning ?? true
+    });
+  }
   return runSuper();
 });
 (0, import_config.task)(TASK_COFHE_MOCKS_SET_LOG_OPS, "Set logging for the Mock CoFHE contracts").addParam("enable", "Whether to enable logging", false, import_config.types.boolean).setAction(async ({ enable }, hre) => {
   await mock_setLoggingEnabled(hre, enable);
 });
+function buildHardhatPluginMocksConfig(mocksConfig) {
+  return {
+    ...mocksConfig,
+    encryptDelay: mocksConfig?.encryptDelay ?? 0
+  };
+}
 (0, import_config.extendEnvironment)((hre) => {
-  hre.cofhesdk = {
-    createCofhesdkConfig: async (config) => {
+  hre.cofhe = {
+    createConfig: async (config) => {
       const zkvHhSigner = await hre.ethers.getImpersonatedSigner(import_sdk3.MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
       const { walletClient: zkvWalletClient } = await (0, import_adapters.HardhatSignerAdapter)(zkvHhSigner);
       const configWithZkvWalletClient = {
         environment: "hardhat",
         ...config,
+        mocks: buildHardhatPluginMocksConfig(config.mocks),
         _internal: {
           ...config._internal,
           zkvWalletClient
         }
       };
-      return (0, import_node.createCofhesdkConfig)(configWithZkvWalletClient);
+      return (0, import_node.createCofheConfig)(configWithZkvWalletClient);
     },
-    createCofhesdkClient: (config) => {
-      return (0, import_node.createCofhesdkClient)(config);
+    createClient: (config) => {
+      return (0, import_node.createCofheClient)(config);
     },
     hardhatSignerAdapter: async (signer) => {
       return (0, import_adapters.HardhatSignerAdapter)(signer);
@@ -470,16 +492,16 @@ var import_mock_contracts3 = require("@cofhe/mock-contracts");
       const { publicClient, walletClient } = await (0, import_adapters.HardhatSignerAdapter)(signer);
       return client.connect(publicClient, walletClient);
     },
-    createBatteriesIncludedCofhesdkClient: async (signer) => {
+    createClientWithBatteries: async (signer) => {
       if (!signer) {
         [signer] = await hre.ethers.getSigners();
       }
-      const config = await hre.cofhesdk.createCofhesdkConfig({
+      const config = await hre.cofhe.createConfig({
         environment: "hardhat",
         supportedChains: [import_chains.hardhat]
       });
-      const client = hre.cofhesdk.createCofhesdkClient(config);
-      await hre.cofhesdk.connectWithHardhatSigner(client, signer);
+      const client = hre.cofhe.createClient(config);
+      await hre.cofhe.connectWithHardhatSigner(client, signer);
       await client.permits.createSelf({
         issuer: signer.address
       });
@@ -512,7 +534,7 @@ var import_mock_contracts3 = require("@cofhe/mock-contracts");
         const aclAddress = await taskManager.acl();
         return hre.ethers.getContractAt(import_mock_contracts3.MockACLArtifact.abi, aclAddress);
       },
-      getMockQueryDecrypter: async () => getFixedMockContract(hre, import_mock_contracts3.MockQueryDecrypterArtifact),
+      getMockThresholdNetwork: async () => getFixedMockContract(hre, import_mock_contracts3.MockThresholdNetworkArtifact),
       getMockZkVerifier: async () => getFixedMockContract(hre, import_mock_contracts3.MockZkVerifierArtifact),
       getTestBed: async () => getFixedMockContract(hre, import_mock_contracts3.TestBedArtifact)
     }
@@ -523,6 +545,7 @@ var import_mock_contracts3 = require("@cofhe/mock-contracts");
   TASK_COFHE_MOCKS_DEPLOY,
   TASK_COFHE_MOCKS_SET_LOG_OPS,
   TASK_COFHE_USE_FAUCET,
+  buildHardhatPluginMocksConfig,
   deployMockContractFromArtifact,
   deployMocks,
   getFixedMockContract,