@cofhe/hardhat-plugin 0.1.1 → 0.2.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (13) hide show
  1. package/CHANGELOG.md +26 -0
  2. package/README.md +8 -8
  3. package/dist/index.d.mts +40 -30
  4. package/dist/index.d.ts +40 -30
  5. package/dist/index.js +57 -59
  6. package/dist/index.js.map +1 -1
  7. package/dist/index.mjs +63 -53
  8. package/dist/index.mjs.map +1 -1
  9. package/package.json +6 -6
  10. package/src/consts.ts +0 -2
  11. package/src/deploy.ts +13 -9
  12. package/src/index.ts +102 -53
  13. package/src/expectResultUtils.ts +0 -24
package/dist/index.js.map CHANGED
@@ -1 +1 @@
1
- {"version":3,"sources":["../src/index.ts","../src/fund.ts","../src/consts.ts","../src/deploy.ts","../src/logging.ts","../src/utils.ts","../src/expectResultUtils.ts"],"sourcesContent":["/* eslint-disable no-empty-pattern */\n/* eslint-disable turbo/no-undeclared-env-vars */\n/* eslint-disable no-unused-vars */\nimport chalk from 'chalk';\nimport { type PublicClient, type WalletClient } from 'viem';\nimport { extendConfig, extendEnvironment, task, types } from 'hardhat/config';\nimport { TASK_TEST, TASK_NODE } from 'hardhat/builtin-tasks/task-names';\nimport { HardhatEthersSigner } from '@nomicfoundation/hardhat-ethers/signers';\nimport { type CofhesdkClient, type CofhesdkConfig, type CofhesdkInputConfig, type Result } from '@cofhe/sdk';\nimport { createCofhesdkClient, createCofhesdkConfig } from '@cofhe/sdk/node';\nimport { HardhatSignerAdapter } from '@cofhe/sdk/adapters';\n\nimport { localcofheFundAccount } from './fund.js';\nimport {\n MOCKS_ZK_VERIFIER_SIGNER_ADDRESS,\n TASK_COFHE_MOCKS_DEPLOY,\n TASK_COFHE_MOCKS_SET_LOG_OPS,\n TASK_COFHE_USE_FAUCET,\n} from './consts.js';\nimport { deployMocks, type DeployMocksArgs } from './deploy.js';\nimport { mock_setLoggingEnabled, mock_withLogs } from './logging.js';\nimport { mock_expectPlaintext } from './utils.js';\nimport { mock_getPlaintext } from './utils.js';\nimport {\n expectResultError,\n expectResultPartialValue,\n expectResultSuccess,\n expectResultValue,\n} from './expectResultUtils.js';\nexport {\n MockACLArtifact,\n MockQueryDecrypterArtifact,\n MockTaskManagerArtifact,\n MockZkVerifierArtifact,\n TestBedArtifact,\n} from '@cofhe/mock-contracts';\n\n/**\n * Configuration interface for the CoFHE Hardhat plugin.\n * Allows users to configure mock logging and gas warning settings.\n */\ndeclare module 'hardhat/types/config' {\n interface HardhatUserConfig {\n cofhesdk?: {\n /** Whether to log mock operations (default: true) */\n logMocks?: boolean;\n /** Whether to show gas usage warnings for mock operations (default: true) */\n gasWarning?: boolean;\n };\n }\n\n interface HardhatConfig {\n cofhesdk: {\n /** Whether to log mock operations (default: true) */\n logMocks: boolean;\n /** Whether to show gas usage warnings for mock operations (default: true) */\n gasWarning: boolean;\n };\n }\n}\n\nextendConfig((config, userConfig) => {\n // Allow users to override the localcofhe network config\n if (userConfig.networks && userConfig.networks.localcofhe) {\n return;\n }\n\n // Default config\n config.networks.localcofhe = {\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 10_000,\n httpHeaders: {},\n url: 'http://127.0.0.1:42069',\n accounts: [\n '0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80',\n '0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d',\n '0x5de4111afa1a4b94908f83103eb1f1706367c2e68ca870fc3fb9a804cdab365a',\n '0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6',\n ],\n };\n\n // Only add Sepolia config if user hasn't defined it\n if (!userConfig.networks?.['eth-sepolia']) {\n config.networks['eth-sepolia'] = {\n url: process.env.SEPOLIA_RPC_URL ?? 'https://ethereum-sepolia.publicnode.com',\n accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],\n chainId: 11155111,\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 60_000,\n httpHeaders: {},\n };\n }\n\n // Only add Arbitrum Sepolia config if user hasn't defined it\n if (!userConfig.networks?.['arb-sepolia']) {\n config.networks['arb-sepolia'] = {\n url: process.env.ARBITRUM_SEPOLIA_RPC_URL ?? 'https://sepolia-rollup.arbitrum.io/rpc',\n accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],\n chainId: 421614,\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 60_000,\n httpHeaders: {},\n };\n }\n\n // Add cofhe config\n config.cofhesdk = {\n logMocks: userConfig.cofhesdk?.logMocks ?? true,\n gasWarning: userConfig.cofhesdk?.gasWarning ?? true,\n };\n});\n\ntype UseFaucetArgs = {\n address?: string;\n};\n\ntask(TASK_COFHE_USE_FAUCET, 'Fund an account from the funder')\n .addOptionalParam('address', 'Address to fund', undefined, types.string)\n .setAction(async ({ address }: UseFaucetArgs, hre) => {\n const { network } = hre;\n const { name: networkName } = network;\n\n if (networkName !== 'localcofhe') {\n console.info(chalk.yellow(`Programmatic faucet only supported for localcofhe`));\n return;\n }\n\n if (!address) {\n console.info(chalk.red(`Failed to get address to fund`));\n return;\n }\n\n console.info(chalk.green(`Getting funds from faucet for ${address}`));\n\n try {\n await localcofheFundAccount(hre, address);\n } catch (e) {\n console.info(chalk.red(`failed to get funds from localcofhe for ${address}: ${e}`));\n }\n });\n\n// DEPLOY TASKS\n\ntask(TASK_COFHE_MOCKS_DEPLOY, 'Deploys the mock contracts on the Hardhat network')\n .addOptionalParam('deployTestBed', 'Whether to deploy the test bed', true, types.boolean)\n .addOptionalParam('silent', 'Whether to suppress output', false, types.boolean)\n .setAction(async ({ deployTestBed, silent }: DeployMocksArgs, hre) => {\n await deployMocks(hre, {\n deployTestBed: deployTestBed ?? true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n silent: silent ?? false,\n });\n });\n\ntask(TASK_TEST, 'Deploy mock contracts on hardhat').setAction(async ({}, hre, runSuper) => {\n await deployMocks(hre, {\n deployTestBed: true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n });\n return runSuper();\n});\n\ntask(TASK_NODE, 'Deploy mock contracts on hardhat').setAction(async ({}, hre, runSuper) => {\n await deployMocks(hre, {\n deployTestBed: true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n });\n return runSuper();\n});\n\n// SET LOG OPS\n\ntask(TASK_COFHE_MOCKS_SET_LOG_OPS, 'Set logging for the Mock CoFHE contracts')\n .addParam('enable', 'Whether to enable logging', false, types.boolean)\n .setAction(async ({ enable }, hre) => {\n await mock_setLoggingEnabled(hre, enable);\n });\n\n// MOCK UTILS\n\nexport * from './utils.js';\nexport * from './expectResultUtils.js';\nexport * from './fund.js';\nexport * from './logging.js';\nexport * from './deploy.js';\n\n/**\n * Runtime environment extensions for the CoFHE Hardhat plugin.\n * Provides access to CoFHE initialization, environment checks, and mock utilities.\n */\ndeclare module 'hardhat/types/runtime' {\n export interface HardhatRuntimeEnvironment {\n cofhesdk: {\n /**\n * Create a CoFHE SDK configuration for use with cofhesdk.createCofhesdkClient(...)\n * @param {CofhesdkInputConfig} config - The CoFHE SDK input configuration\n * @returns {CofhesdkConfig} The CoFHE SDK configuration\n */\n createCofhesdkConfig: (config: CofhesdkInputConfig) => Promise<CofhesdkConfig>;\n /**\n * Create a CoFHE SDK client instance\n * @param {CofhesdkConfig} config - The CoFHE SDK configuration (use createCofhesdkConfig to create with Node.js defaults)\n * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance\n */\n createCofhesdkClient: (config: CofhesdkConfig) => CofhesdkClient;\n /**\n * Create viem clients from a Hardhat ethers signer, to be used with `cofhesdkClient.connect(...)`\n * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use\n * @returns {Promise<{ publicClient: PublicClient; walletClient: WalletClient }>} The viem clients\n */\n hardhatSignerAdapter: (\n signer: HardhatEthersSigner\n ) => Promise<{ publicClient: PublicClient; walletClient: WalletClient }>;\n\n /**\n * Assert that a Result type returned from a function is successful and return its value (result.success === true)\n * @param {Result<T>} result - The Result to check\n * @returns {T} The inner data of the Result (non null)\n */\n expectResultSuccess: <T>(result: Result<T> | Promise<Result<T>>) => Promise<T>;\n\n /**\n * Assert that a Result type contains an error matching the partial string (result.success === false && result.error.includes(errorPartial))\n * @param {Result<T>} result - The Result to check\n * @param {string} errorPartial - The partial error string to match\n */\n expectResultError: <T>(result: Result<T> | Promise<Result<T>>, errorPartial: string) => Promise<void>;\n\n /**\n * Assert that a Result type contains a specific value (result.success === true && result.data === value)\n * @param {Result<T>} result - The Result to check\n * @param {T} value - The inner data of the Result (non null)\n */\n expectResultValue: <T>(result: Result<T> | Promise<Result<T>>, value: T) => Promise<T>;\n\n /**\n * Assert that a Result type contains a value matching the partial object (result.success === true && result.data.includes(partial))\n * @param {Result<T>} result - The Result to check\n * @param {Partial<T>} partial - The partial object to match against\n * @returns {T} The inner data of the Result (non null)\n */\n expectResultPartialValue: <T>(result: Result<T> | Promise<Result<T>>, partial: Partial<T>) => Promise<T>;\n\n mocks: {\n /**\n * **[MOCKS ONLY]**\n *\n * Execute a block of code with cofhe mock contracts logging enabled.\n *\n * _(If logging only a function, we recommend passing the function name as the closureName (ex \"counter.increment()\"))_\n *\n * Example usage:\n *\n * ```ts\n * await hre.cofhesdk.mocks.withLogs(\"counter.increment()\", async () => {\n * await counter.increment();\n * });\n * ```\n *\n * Expected output:\n * ```\n * ┌──────────────────┬──────────────────────────────────────────────────\n * │ [COFHE-MOCKS] │ \"counter.increment()\" logs:\n * ├──────────────────┴──────────────────────────────────────────────────\n * ├ FHE.add | euint32(4473..3424)[0] + euint32(1157..3648)[1] => euint32(1106..1872)[1]\n * ├ FHE.allowThis | euint32(1106..1872)[1] -> 0x663f..6602\n * ├ FHE.allow | euint32(1106..1872)[1] -> 0x3c44..93bc\n * └─────────────────────────────────────────────────────────────────────\n * ```\n * @param {string} closureName - Name of the code block to log within\n * @param {() => Promise<void>} closure - The async function to execute\n */\n withLogs: (closureName: string, closure: () => Promise<void>) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Enable logging from cofhe mock contracts\n * @param {string} closureName - Optional name of the code block to enable logging for\n */\n enableLogs: (closureName?: string) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Disable logging from cofhe mock contracts\n */\n disableLogs: () => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Deploy the cofhe mock contracts (normally this is done automatically)\n * @param {DeployMocksArgs} options - Deployment options\n */\n deployMocks: (options: DeployMocksArgs) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Get the plaintext value for a ciphertext hash\n * @param {bigint} ctHash - The ciphertext hash to look up\n * @returns {Promise<bigint>} The plaintext value\n */\n getPlaintext: (ctHash: bigint) => Promise<bigint>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Assert that a ciphertext hash represents an expected plaintext value\n * @param {bigint} ctHash - The ciphertext hash to check\n * @param {bigint} expectedValue - The expected plaintext value\n */\n expectPlaintext: (ctHash: bigint, expectedValue: bigint) => Promise<void>;\n };\n };\n }\n}\n\nextendEnvironment((hre) => {\n hre.cofhesdk = {\n createCofhesdkConfig: async (config: CofhesdkInputConfig) => {\n // Create zkv wallet client\n // This wallet interacts with the MockZkVerifier contract so that the user's connected wallet doesn't have to\n const zkvHhSigner = await hre.ethers.getImpersonatedSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n const { walletClient: zkvWalletClient } = await HardhatSignerAdapter(zkvHhSigner);\n\n // Inject zkv wallet client into config\n const configWithZkvWalletClient = {\n ...config,\n _internal: {\n ...config._internal,\n zkvWalletClient,\n },\n };\n\n return createCofhesdkConfig(configWithZkvWalletClient);\n },\n createCofhesdkClient: (config: CofhesdkConfig) => {\n return createCofhesdkClient(config);\n },\n hardhatSignerAdapter: async (signer: HardhatEthersSigner) => {\n return HardhatSignerAdapter(signer);\n },\n expectResultSuccess: async <T>(result: Result<T> | Promise<Result<T>>) => {\n const awaitedResult = await result;\n return expectResultSuccess(awaitedResult);\n },\n expectResultError: async <T>(result: Result<T> | Promise<Result<T>>, errorPartial: string) => {\n const awaitedResult = await result;\n return expectResultError(awaitedResult, errorPartial);\n },\n expectResultValue: async <T>(result: Result<T> | Promise<Result<T>>, value: T) => {\n const awaitedResult = await result;\n return expectResultValue(awaitedResult, value);\n },\n expectResultPartialValue: async <T>(result: Result<T> | Promise<Result<T>>, partial: Partial<T>) => {\n const awaitedResult = await result;\n return expectResultPartialValue(awaitedResult, partial);\n },\n mocks: {\n withLogs: async (closureName: string, closure: () => Promise<void>) => {\n return mock_withLogs(hre, closureName, closure);\n },\n enableLogs: async (closureName?: string) => {\n return mock_setLoggingEnabled(hre, true, closureName);\n },\n disableLogs: async () => {\n return mock_setLoggingEnabled(hre, false);\n },\n deployMocks: async (options: DeployMocksArgs) => {\n return deployMocks(hre, options);\n },\n getPlaintext: async (ctHash: bigint) => {\n const [signer] = await hre.ethers.getSigners();\n return mock_getPlaintext(signer.provider, ctHash);\n },\n expectPlaintext: async (ctHash: bigint, expectedValue: bigint) => {\n const [signer] = await hre.ethers.getSigners();\n return mock_expectPlaintext(signer.provider, ctHash, expectedValue);\n },\n },\n };\n});\n","/* eslint-disable turbo/no-undeclared-env-vars */\nimport { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport '@nomicfoundation/hardhat-ethers';\n\n/**\n * Sends funds to the specified address\n * @param hre Hardhat Runtime Environment\n * @param toAddress Address to send funds to\n * @param amount Amount to send in ETH (default: 10)\n * @returns Transaction receipt or null if failed\n */\nexport async function localcofheFundAccount(hre: HardhatRuntimeEnvironment, toAddress: string, amount: string = '10') {\n // Load private key from environment\n const privateKey =\n process.env.FUNDER_PRIVATE_KEY ?? '0xb6b15c8cb491557369f3c7d2c287b053eb229daa9c22138887752191c9520659';\n if (!privateKey) {\n console.error('Error: FUNDER_PRIVATE_KEY environment variable not set');\n return null;\n }\n\n try {\n // Create wallet from private key\n const wallet = new hre.ethers.Wallet(privateKey, hre.ethers.provider);\n\n // Get wallet balance\n const balance = await hre.ethers.provider.getBalance(wallet.address);\n console.log(`Funder wallet address: ${wallet.address}`);\n console.log(`Funder wallet balance: ${hre.ethers.formatEther(balance)} ETH`);\n\n // Check if wallet has enough funds\n const amountToSend = hre.ethers.parseEther(amount);\n if (balance < amountToSend) {\n console.error(\n `Error: Funder wallet doesn't have enough funds. Current balance: ${hre.ethers.formatEther(balance)} ETH`\n );\n return null;\n }\n\n // Send transaction\n console.log(`Sending ${amount} ETH to ${toAddress}...`);\n const tx = await wallet.sendTransaction({\n to: toAddress,\n value: amountToSend,\n });\n\n console.log(`Transaction sent! Hash: ${tx.hash}`);\n console.log('Waiting for confirmation...');\n\n // Wait for transaction to be mined\n const receipt = await tx.wait();\n console.log(`Transaction confirmed in block ${receipt?.blockNumber}`);\n console.log(`Successfully sent ${amount} ETH to ${toAddress}`);\n\n return receipt;\n } catch (error) {\n console.error('Error sending funds:', error);\n return null;\n }\n}\n\n/**\n * Checks a wallet's balance and funds it if below 1 ETH\n * @param hre Hardhat Runtime Environment\n * @param walletAddress Address of the wallet to check and potentially fund\n * @returns Promise that resolves when the funding operation completes (if needed)\n */\nexport async function localcofheFundWalletIfNeeded(hre: HardhatRuntimeEnvironment, walletAddress: string) {\n // Check wallet balance and fund if needed\n const walletBalance = await hre.ethers.provider.getBalance(walletAddress);\n console.log(`Wallet balance: ${hre.ethers.formatEther(walletBalance)} ETH`);\n\n if (walletBalance < hre.ethers.parseEther('1')) {\n console.log(`Wallet balance is less than 1 ETH. Funding ${walletAddress}...`);\n const receipt = await localcofheFundAccount(hre, walletAddress);\n if (receipt) {\n const newBalance = await hre.ethers.provider.getBalance(walletAddress);\n console.log(`Wallet new balance: ${hre.ethers.formatEther(newBalance)} ETH`);\n } else {\n console.error(`Failed to fund ${walletAddress}`);\n }\n }\n}\n","export const TASK_COFHE_USE_FAUCET = 'task:cofhe:usefaucet';\nexport const TASK_COFHE_MOCKS_SET_LOG_OPS = 'task:cofhe-mocks:setlogops';\nexport const TASK_COFHE_MOCKS_DEPLOY = 'task:cofhe-mocks:deploy';\n\nexport const TASK_MANAGER_ADDRESS = '0xeA30c4B8b44078Bbf8a6ef5b9f1eC1626C7848D9';\nexport const MOCKS_ACL_ADDRESS = '0x0000000000000000000000000000000000000400';\nexport const MOCKS_ZK_VERIFIER_ADDRESS = '0x0000000000000000000000000000000000000100';\nexport const MOCKS_QUERY_DECRYPTER_ADDRESS = '0x0000000000000000000000000000000000000200';\nexport const TEST_BED_ADDRESS = '0x0000000000000000000000000000000000000300';\nexport const MOCKS_ZK_VERIFIER_SIGNER_ADDRESS = '0x6E12D8C87503D4287c294f2Fdef96ACd9DFf6bd2';\n","import { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport chalk from 'chalk';\nimport { Contract } from 'ethers';\n\nimport {\n TASK_MANAGER_ADDRESS,\n MOCKS_ZK_VERIFIER_ADDRESS,\n MOCKS_QUERY_DECRYPTER_ADDRESS,\n TEST_BED_ADDRESS,\n MOCKS_ZK_VERIFIER_SIGNER_ADDRESS,\n MOCKS_ACL_ADDRESS,\n} from './consts.js';\n\nimport {\n MockTaskManagerArtifact,\n MockACLArtifact,\n MockZkVerifierArtifact,\n MockQueryDecrypterArtifact,\n TestBedArtifact,\n} from '@cofhe/mock-contracts';\n\n// Deploy\n\nconst deployMockTaskManager = async (hre: HardhatRuntimeEnvironment) => {\n const [signer] = await hre.ethers.getSigners();\n\n // Deploy MockTaskManager\n await hardhatSetCode(hre, TASK_MANAGER_ADDRESS, MockTaskManagerArtifact.deployedBytecode);\n const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);\n\n // Initialize MockTaskManager\n const initTx = await taskManager.initialize(signer.address);\n await initTx.wait();\n\n // Check if MockTaskManager exists\n const tmExists = await taskManager.exists();\n if (!tmExists) {\n throw new Error('MockTaskManager does not exist');\n }\n\n return taskManager;\n};\n\nconst deployMockACL = async (hre: HardhatRuntimeEnvironment): Promise<Contract> => {\n // Deploy MockACL\n await hardhatSetCode(hre, MOCKS_ACL_ADDRESS, MockACLArtifact.deployedBytecode);\n const acl = await hre.ethers.getContractAt(MockACLArtifact.abi, MOCKS_ACL_ADDRESS);\n\n // Check if ACL exists\n const exists = await acl.exists();\n if (!exists) {\n logError('MockACL does not exist', 2);\n throw new Error('MockACL does not exist');\n }\n\n return acl;\n};\n\nconst deployMockZkVerifier = async (hre: HardhatRuntimeEnvironment) => {\n await hardhatSetCode(hre, MOCKS_ZK_VERIFIER_ADDRESS, MockZkVerifierArtifact.deployedBytecode);\n const zkVerifier = await hre.ethers.getContractAt(MockZkVerifierArtifact.abi, MOCKS_ZK_VERIFIER_ADDRESS);\n\n const zkVerifierExists = await zkVerifier.exists();\n if (!zkVerifierExists) {\n logError('MockZkVerifier does not exist', 2);\n throw new Error('MockZkVerifier does not exist');\n }\n\n return zkVerifier;\n};\n\nconst deployMockQueryDecrypter = async (hre: HardhatRuntimeEnvironment, acl: Contract) => {\n await hardhatSetCode(hre, MOCKS_QUERY_DECRYPTER_ADDRESS, MockQueryDecrypterArtifact.deployedBytecode);\n const queryDecrypter = await hre.ethers.getContractAt(MockQueryDecrypterArtifact.abi, MOCKS_QUERY_DECRYPTER_ADDRESS);\n\n // Initialize MockQueryDecrypter\n const initTx = await queryDecrypter.initialize(TASK_MANAGER_ADDRESS, await acl.getAddress());\n await initTx.wait();\n\n // Check if MockQueryDecrypter exists\n const queryDecrypterExists = await queryDecrypter.exists();\n if (!queryDecrypterExists) {\n logError('MockQueryDecrypter does not exist', 2);\n throw new Error('MockQueryDecrypter does not exist');\n }\n\n return queryDecrypter;\n};\n\nconst deployTestBedContract = async (hre: HardhatRuntimeEnvironment) => {\n await hardhatSetCode(hre, TEST_BED_ADDRESS, TestBedArtifact.deployedBytecode);\n const testBed = await hre.ethers.getContractAt(TestBedArtifact.abi, TEST_BED_ADDRESS);\n await testBed.waitForDeployment();\n return testBed;\n};\n\n// Funding\n\nconst fundZkVerifierSigner = async (hre: HardhatRuntimeEnvironment) => {\n const zkVerifierSigner = await hre.ethers.getSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n await hre.network.provider.send('hardhat_setBalance', [\n zkVerifierSigner.address,\n '0x' + hre.ethers.parseEther('10').toString(16),\n ]);\n};\n\n// Initializations\n\nconst setTaskManagerACL = async (taskManager: Contract, acl: Contract) => {\n const setAclTx = await taskManager.setACLContract(await acl.getAddress());\n await setAclTx.wait();\n};\n\nexport type DeployMocksArgs = {\n deployTestBed?: boolean;\n gasWarning?: boolean;\n silent?: boolean;\n};\n\nexport const deployMocks = async (\n hre: HardhatRuntimeEnvironment,\n options: DeployMocksArgs = {\n deployTestBed: true,\n gasWarning: true,\n silent: false,\n }\n) => {\n // Check if network is Hardhat, if not log skip message and return\n const isHardhat = await checkNetworkAndSkip(hre);\n if (!isHardhat) return;\n\n const logEmptyIfNoisy = () => {\n if (!options.silent) {\n logEmpty();\n }\n };\n const logSuccessIfNoisy = (message: string, indent = 0) => {\n if (!options.silent) {\n logSuccess(message, indent);\n }\n };\n const logDeploymentIfNoisy = (contractName: string, address: string) => {\n if (!options.silent) {\n logDeployment(contractName, address);\n }\n };\n const logWarningIfNoisy = (message: string, indent = 0) => {\n if (!options.silent) {\n logWarning(message, indent);\n }\n };\n\n // Log start message\n logEmptyIfNoisy();\n logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: deploy mocks'), 0);\n logEmptyIfNoisy();\n\n // Compile mock contracts\n logEmptyIfNoisy();\n logSuccessIfNoisy('Mock contracts compiled', 1);\n\n // Deploy mock contracts\n const taskManager = await deployMockTaskManager(hre);\n logDeploymentIfNoisy('MockTaskManager', await taskManager.getAddress());\n\n const acl = await deployMockACL(hre);\n logDeploymentIfNoisy('MockACL', await acl.getAddress());\n\n await setTaskManagerACL(taskManager, acl);\n logSuccessIfNoisy('ACL address set in TaskManager', 2);\n\n await fundZkVerifierSigner(hre);\n logSuccessIfNoisy(`ZkVerifier signer (${MOCKS_ZK_VERIFIER_SIGNER_ADDRESS}) funded`, 1);\n\n const zkVerifierSignerBalance = await hre.ethers.provider.getBalance(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n logSuccessIfNoisy(`ETH balance: ${zkVerifierSignerBalance.toString()}`, 2);\n\n const zkVerifier = await deployMockZkVerifier(hre);\n logDeploymentIfNoisy('MockZkVerifier', await zkVerifier.getAddress());\n\n const queryDecrypter = await deployMockQueryDecrypter(hre, acl);\n logDeploymentIfNoisy('MockQueryDecrypter', await queryDecrypter.getAddress());\n\n if (options.deployTestBed) {\n logSuccessIfNoisy('TestBed deployment enabled', 2);\n const testBed = await deployTestBedContract(hre);\n logDeploymentIfNoisy('TestBed', await testBed.getAddress());\n }\n\n // Log success message\n logEmptyIfNoisy();\n logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: mocks deployed successfully'), 0);\n\n // Log warning about mocks increased gas costs\n if (options.gasWarning) {\n logEmptyIfNoisy();\n logWarningIfNoisy(\n \"When using mocks, FHE operations (eg FHE.add / FHE.mul) report a higher gas price due to additional on-chain mocking logic. Deploy your contracts on a testnet chain to check the true gas costs.\\n(Disable this warning by setting '@cofhe/sdk.gasWarning' to false in your hardhat config\",\n 0\n );\n }\n\n logEmptyIfNoisy();\n};\n\n// Utils\n\nconst hardhatSetCode = async (hre: HardhatRuntimeEnvironment, address: string, bytecode: string) => {\n await hre.network.provider.send('hardhat_setCode', [address, bytecode]);\n};\n\n// Network\n\nconst checkNetworkAndSkip = async (hre: HardhatRuntimeEnvironment) => {\n const network = hre.network.name;\n const isHardhat = network === 'hardhat';\n if (!isHardhat) logSuccess(`cofhe-hardhat-plugin - deploy mocks - skipped on non-hardhat network ${network}`, 0);\n return isHardhat;\n};\n\n// Logging\n\nconst logEmpty = () => {\n console.log('');\n};\n\nconst logSuccess = (message: string, indent = 1) => {\n console.log(chalk.green(`${' '.repeat(indent)}✓ ${message}`));\n};\n\nconst logWarning = (message: string, indent = 1) => {\n console.log(chalk.bold(chalk.yellow(`${' '.repeat(indent)}⚠ NOTE:`)), message);\n};\n\nconst logError = (message: string, indent = 1) => {\n console.log(chalk.red(`${' '.repeat(indent)}✗ ${message}`));\n};\n\nconst logDeployment = (contractName: string, address: string) => {\n const paddedName = `${contractName} deployed`.padEnd(36);\n logSuccess(`${paddedName} ${chalk.bold(address)}`);\n};\n","import chalk from 'chalk';\nimport { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport { TASK_MANAGER_ADDRESS } from './consts';\nimport { MockTaskManagerArtifact } from '@cofhe/mock-contracts';\n\nconst getDeployedMockTaskManager = async (hre: HardhatRuntimeEnvironment) => {\n // Fetch the deployed MockTaskManager\n const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);\n\n return taskManager;\n};\n\nconst getLoggingEnabled = async (hre: HardhatRuntimeEnvironment) => {\n const taskManager = await getDeployedMockTaskManager(hre);\n return await taskManager.logOps();\n};\n\nconst setLoggingEnabled = async (hre: HardhatRuntimeEnvironment, enabled: boolean) => {\n const taskManager = await getDeployedMockTaskManager(hre);\n await taskManager.setLogOps(enabled);\n};\n\n// prettier-ignore\nconst printLogsEnabledMessage = (closureMessage: string) => {\n console.log(\"┌──────────────────┬──────────────────────────────────────────────────\");\n console.log(`│ [COFHE-MOCKS] │ ${closureMessage}`);\n console.log(\"├──────────────────┴──────────────────────────────────────────────────\");\n};\n\n// prettier-ignore\nconst printLogsBlockEnd = () => {\n console.log(\"└─────────────────────────────────────────────────────────────────────\");\n};\n\nexport const mock_setLoggingEnabled = async (\n hre: HardhatRuntimeEnvironment,\n enabled: boolean,\n closureName?: string\n) => {\n try {\n const initiallyEnabled = await getLoggingEnabled(hre);\n\n await setLoggingEnabled(hre, enabled);\n\n // Only print if enabling logs\n if (enabled) {\n printLogsEnabledMessage(`${closureName ? `\"${chalk.bold(closureName)}\" logs:` : 'Logs:'}`);\n }\n\n // Only print if disabling logs AND logs currently enabled\n if (!enabled && initiallyEnabled) {\n printLogsBlockEnd();\n }\n } catch (error) {\n console.log(chalk.red('mock_setLoggingEnabled error'), error);\n }\n};\n\nexport const mock_withLogs = async (\n hre: HardhatRuntimeEnvironment,\n closureName: string,\n closure: () => Promise<void>\n) => {\n const initiallyEnabled = await getLoggingEnabled(hre);\n\n await setLoggingEnabled(hre, true);\n printLogsEnabledMessage(`\"${chalk.bold(closureName)}\" logs:`);\n await closure();\n printLogsBlockEnd();\n\n // If logs were disabled, disable them again\n if (!initiallyEnabled) {\n await setLoggingEnabled(hre, false);\n }\n};\n","import { TASK_MANAGER_ADDRESS, MOCKS_ZK_VERIFIER_ADDRESS } from './consts.js';\nimport { expect } from 'chai';\nimport { ethers } from 'ethers';\nimport { type HardhatEthersProvider } from '@nomicfoundation/hardhat-ethers/internal/hardhat-ethers-provider';\n\nconst mock_checkIsTestnet = async (fnName: string, provider: HardhatEthersProvider | ethers.JsonRpcProvider) => {\n // Testnet is checked by testing if MockZkVerifier is deployed\n\n // Get bytecode at ZK_VERIFIER_ADDRESS\n const bytecode = await provider.getCode(MOCKS_ZK_VERIFIER_ADDRESS);\n\n // If bytecode is empty, we are on a testnet\n const isTestnet = bytecode.length === 0;\n\n // Log if we are on a testnet\n if (isTestnet) {\n console.log(`${fnName} - skipped on non-testnet chain`);\n }\n\n return isTestnet;\n};\n\nexport const mock_getPlaintext = async (provider: HardhatEthersProvider | ethers.JsonRpcProvider, ctHash: bigint) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_getPlaintext.name, provider)) return;\n\n // Connect to MockTaskManager\n const taskManager = new ethers.Contract(\n TASK_MANAGER_ADDRESS,\n ['function mockStorage(uint256) view returns (uint256)'],\n provider\n );\n\n // Fetch the plaintext\n const plaintext = await taskManager.mockStorage(ctHash);\n\n return plaintext;\n};\n\nexport const mock_getPlaintextExists = async (\n provider: HardhatEthersProvider | ethers.JsonRpcProvider,\n ctHash: bigint\n) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_getPlaintextExists.name, provider)) return;\n\n // Connect to MockTaskManager\n const taskManager = new ethers.Contract(\n TASK_MANAGER_ADDRESS,\n ['function inMockStorage(uint256) view returns (bool)'],\n provider\n );\n\n // Fetch the plaintext exists\n const plaintextExists = await taskManager.inMockStorage(ctHash);\n\n return plaintextExists;\n};\n\nexport const mock_expectPlaintext = async (\n provider: HardhatEthersProvider | ethers.JsonRpcProvider,\n ctHash: bigint,\n expectedValue: bigint\n) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_expectPlaintext.name, provider)) return;\n\n // Expect the plaintext to exist\n const plaintextExists = await mock_getPlaintextExists(provider, ctHash);\n expect(plaintextExists).equal(true, 'Plaintext does not exist');\n\n // Expect the plaintext to have the expected value\n const plaintext = await mock_getPlaintext(provider, ctHash);\n expect(plaintext).equal(expectedValue, 'Plaintext value is incorrect');\n};\n","import { type Result } from '@cofhe/sdk';\nimport { expect } from 'chai';\n\nexport const expectResultError = <T>(result: Result<T>, errorPartial: string) => {\n expect(result.success).to.eq(false, 'Result should be an error');\n expect(result.error).to.include(errorPartial, `Error should contain error partial: ${errorPartial}`);\n};\n\nexport const expectResultSuccess = <T>(result: Result<T>): T => {\n expect(result.success).to.eq(true, 'Result should be a success');\n return result.data!;\n};\n\nexport const expectResultValue = <T>(result: Result<T>, value: T): T => {\n expect(result.success).to.eq(true, 'Result should be a success');\n expect(result.data).to.eq(value, `Result should have the expected value ${value}`);\n return result.data!;\n};\n\nexport const expectResultPartialValue = <T>(result: Result<T>, partial: Partial<T>): T => {\n expect(result.success).to.eq(true, 'Result should be a success');\n expect(result.data).to.include(partial, `Result should have the expected partial ${partial}`);\n return result.data!;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAGA,IAAAA,gBAAkB;AAClB,kBAAqD;AACrD,oBAA6D;AAC7D,wBAAqC;AACrC,qBAAoC;AACpC,IAAAC,cAAgG;AAChG,kBAA2D;AAC3D,sBAAqC;;;ACTrC,mBAA+C;AAC/C,4BAAO;AASP,eAAsB,sBAAsB,KAAgC,WAAmB,SAAiB,MAAM;AAEpH,QAAM,aACJ,QAAQ,IAAI,sBAAsB;AACpC,MAAI,CAAC,YAAY;AACf,YAAQ,MAAM,wDAAwD;AACtE,WAAO;AAAA,EACT;AAEA,MAAI;AAEF,UAAM,SAAS,IAAI,IAAI,OAAO,OAAO,YAAY,IAAI,OAAO,QAAQ;AAGpE,UAAM,UAAU,MAAM,IAAI,OAAO,SAAS,WAAW,OAAO,OAAO;AACnE,YAAQ,IAAI,0BAA0B,OAAO,OAAO,EAAE;AACtD,YAAQ,IAAI,0BAA0B,IAAI,OAAO,YAAY,OAAO,CAAC,MAAM;AAG3E,UAAM,eAAe,IAAI,OAAO,WAAW,MAAM;AACjD,QAAI,UAAU,cAAc;AAC1B,cAAQ;AAAA,QACN,oEAAoE,IAAI,OAAO,YAAY,OAAO,CAAC;AAAA,MACrG;AACA,aAAO;AAAA,IACT;AAGA,YAAQ,IAAI,WAAW,MAAM,WAAW,SAAS,KAAK;AACtD,UAAM,KAAK,MAAM,OAAO,gBAAgB;AAAA,MACtC,IAAI;AAAA,MACJ,OAAO;AAAA,IACT,CAAC;AAED,YAAQ,IAAI,2BAA2B,GAAG,IAAI,EAAE;AAChD,YAAQ,IAAI,6BAA6B;AAGzC,UAAM,UAAU,MAAM,GAAG,KAAK;AAC9B,YAAQ,IAAI,kCAAkC,SAAS,WAAW,EAAE;AACpE,YAAQ,IAAI,qBAAqB,MAAM,WAAW,SAAS,EAAE;AAE7D,WAAO;AAAA,EACT,SAAS,OAAO;AACd,YAAQ,MAAM,wBAAwB,KAAK;AAC3C,WAAO;AAAA,EACT;AACF;AAQA,eAAsB,6BAA6B,KAAgC,eAAuB;AAExG,QAAM,gBAAgB,MAAM,IAAI,OAAO,SAAS,WAAW,aAAa;AACxE,UAAQ,IAAI,mBAAmB,IAAI,OAAO,YAAY,aAAa,CAAC,MAAM;AAE1E,MAAI,gBAAgB,IAAI,OAAO,WAAW,GAAG,GAAG;AAC9C,YAAQ,IAAI,8CAA8C,aAAa,KAAK;AAC5E,UAAM,UAAU,MAAM,sBAAsB,KAAK,aAAa;AAC9D,QAAI,SAAS;AACX,YAAM,aAAa,MAAM,IAAI,OAAO,SAAS,WAAW,aAAa;AACrE,cAAQ,IAAI,uBAAuB,IAAI,OAAO,YAAY,UAAU,CAAC,MAAM;AAAA,IAC7E,OAAO;AACL,cAAQ,MAAM,kBAAkB,aAAa,EAAE;AAAA,IACjD;AAAA,EACF;AACF;;;ACjFO,IAAM,wBAAwB;AAC9B,IAAM,+BAA+B;AACrC,IAAM,0BAA0B;AAEhC,IAAM,uBAAuB;AAC7B,IAAM,oBAAoB;AAC1B,IAAM,4BAA4B;AAClC,IAAM,gCAAgC;AACtC,IAAM,mBAAmB;AACzB,IAAM,mCAAmC;;;ACThD,IAAAC,gBAA+C;AAC/C,mBAAkB;AAClB,oBAAyB;AAWzB,4BAMO;AAIP,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAG7C,QAAM,eAAe,KAAK,sBAAsB,8CAAwB,gBAAgB;AACxF,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,8CAAwB,KAAK,oBAAoB;AAGpG,QAAM,SAAS,MAAM,YAAY,WAAW,OAAO,OAAO;AAC1D,QAAM,OAAO,KAAK;AAGlB,QAAM,WAAW,MAAM,YAAY,OAAO;AAC1C,MAAI,CAAC,UAAU;AACb,UAAM,IAAI,MAAM,gCAAgC;AAAA,EAClD;AAEA,SAAO;AACT;AAEA,IAAM,gBAAgB,OAAO,QAAsD;AAEjF,QAAM,eAAe,KAAK,mBAAmB,sCAAgB,gBAAgB;AAC7E,QAAM,MAAM,MAAM,IAAI,OAAO,cAAc,sCAAgB,KAAK,iBAAiB;AAGjF,QAAM,SAAS,MAAM,IAAI,OAAO;AAChC,MAAI,CAAC,QAAQ;AACX,aAAS,0BAA0B,CAAC;AACpC,UAAM,IAAI,MAAM,wBAAwB;AAAA,EAC1C;AAEA,SAAO;AACT;AAEA,IAAM,uBAAuB,OAAO,QAAmC;AACrE,QAAM,eAAe,KAAK,2BAA2B,6CAAuB,gBAAgB;AAC5F,QAAM,aAAa,MAAM,IAAI,OAAO,cAAc,6CAAuB,KAAK,yBAAyB;AAEvG,QAAM,mBAAmB,MAAM,WAAW,OAAO;AACjD,MAAI,CAAC,kBAAkB;AACrB,aAAS,iCAAiC,CAAC;AAC3C,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,SAAO;AACT;AAEA,IAAM,2BAA2B,OAAO,KAAgC,QAAkB;AACxF,QAAM,eAAe,KAAK,+BAA+B,iDAA2B,gBAAgB;AACpG,QAAM,iBAAiB,MAAM,IAAI,OAAO,cAAc,iDAA2B,KAAK,6BAA6B;AAGnH,QAAM,SAAS,MAAM,eAAe,WAAW,sBAAsB,MAAM,IAAI,WAAW,CAAC;AAC3F,QAAM,OAAO,KAAK;AAGlB,QAAM,uBAAuB,MAAM,eAAe,OAAO;AACzD,MAAI,CAAC,sBAAsB;AACzB,aAAS,qCAAqC,CAAC;AAC/C,UAAM,IAAI,MAAM,mCAAmC;AAAA,EACrD;AAEA,SAAO;AACT;AAEA,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,eAAe,KAAK,kBAAkB,sCAAgB,gBAAgB;AAC5E,QAAM,UAAU,MAAM,IAAI,OAAO,cAAc,sCAAgB,KAAK,gBAAgB;AACpF,QAAM,QAAQ,kBAAkB;AAChC,SAAO;AACT;AAIA,IAAM,uBAAuB,OAAO,QAAmC;AACrE,QAAM,mBAAmB,MAAM,IAAI,OAAO,UAAU,gCAAgC;AACpF,QAAM,IAAI,QAAQ,SAAS,KAAK,sBAAsB;AAAA,IACpD,iBAAiB;AAAA,IACjB,OAAO,IAAI,OAAO,WAAW,IAAI,EAAE,SAAS,EAAE;AAAA,EAChD,CAAC;AACH;AAIA,IAAM,oBAAoB,OAAO,aAAuB,QAAkB;AACxE,QAAM,WAAW,MAAM,YAAY,eAAe,MAAM,IAAI,WAAW,CAAC;AACxE,QAAM,SAAS,KAAK;AACtB;AAQO,IAAM,cAAc,OACzB,KACA,UAA2B;AAAA,EACzB,eAAe;AAAA,EACf,YAAY;AAAA,EACZ,QAAQ;AACV,MACG;AAEH,QAAM,YAAY,MAAM,oBAAoB,GAAG;AAC/C,MAAI,CAAC;AAAW;AAEhB,QAAM,kBAAkB,MAAM;AAC5B,QAAI,CAAC,QAAQ,QAAQ;AACnB,eAAS;AAAA,IACX;AAAA,EACF;AACA,QAAM,oBAAoB,CAAC,SAAiB,SAAS,MAAM;AACzD,QAAI,CAAC,QAAQ,QAAQ;AACnB,iBAAW,SAAS,MAAM;AAAA,IAC5B;AAAA,EACF;AACA,QAAM,uBAAuB,CAAC,cAAsB,YAAoB;AACtE,QAAI,CAAC,QAAQ,QAAQ;AACnB,oBAAc,cAAc,OAAO;AAAA,IACrC;AAAA,EACF;AACA,QAAM,oBAAoB,CAAC,SAAiB,SAAS,MAAM;AACzD,QAAI,CAAC,QAAQ,QAAQ;AACnB,iBAAW,SAAS,MAAM;AAAA,IAC5B;AAAA,EACF;AAGA,kBAAgB;AAChB,oBAAkB,aAAAC,QAAM,KAAK,sCAAsC,GAAG,CAAC;AACvE,kBAAgB;AAGhB,kBAAgB;AAChB,oBAAkB,2BAA2B,CAAC;AAG9C,QAAM,cAAc,MAAM,sBAAsB,GAAG;AACnD,uBAAqB,mBAAmB,MAAM,YAAY,WAAW,CAAC;AAEtE,QAAM,MAAM,MAAM,cAAc,GAAG;AACnC,uBAAqB,WAAW,MAAM,IAAI,WAAW,CAAC;AAEtD,QAAM,kBAAkB,aAAa,GAAG;AACxC,oBAAkB,kCAAkC,CAAC;AAErD,QAAM,qBAAqB,GAAG;AAC9B,oBAAkB,sBAAsB,gCAAgC,YAAY,CAAC;AAErF,QAAM,0BAA0B,MAAM,IAAI,OAAO,SAAS,WAAW,gCAAgC;AACrG,oBAAkB,gBAAgB,wBAAwB,SAAS,CAAC,IAAI,CAAC;AAEzE,QAAM,aAAa,MAAM,qBAAqB,GAAG;AACjD,uBAAqB,kBAAkB,MAAM,WAAW,WAAW,CAAC;AAEpE,QAAM,iBAAiB,MAAM,yBAAyB,KAAK,GAAG;AAC9D,uBAAqB,sBAAsB,MAAM,eAAe,WAAW,CAAC;AAE5E,MAAI,QAAQ,eAAe;AACzB,sBAAkB,8BAA8B,CAAC;AACjD,UAAM,UAAU,MAAM,sBAAsB,GAAG;AAC/C,yBAAqB,WAAW,MAAM,QAAQ,WAAW,CAAC;AAAA,EAC5D;AAGA,kBAAgB;AAChB,oBAAkB,aAAAA,QAAM,KAAK,qDAAqD,GAAG,CAAC;AAGtF,MAAI,QAAQ,YAAY;AACtB,oBAAgB;AAChB;AAAA,MACE;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,kBAAgB;AAClB;AAIA,IAAM,iBAAiB,OAAO,KAAgC,SAAiB,aAAqB;AAClG,QAAM,IAAI,QAAQ,SAAS,KAAK,mBAAmB,CAAC,SAAS,QAAQ,CAAC;AACxE;AAIA,IAAM,sBAAsB,OAAO,QAAmC;AACpE,QAAM,UAAU,IAAI,QAAQ;AAC5B,QAAM,YAAY,YAAY;AAC9B,MAAI,CAAC;AAAW,eAAW,wEAAwE,OAAO,IAAI,CAAC;AAC/G,SAAO;AACT;AAIA,IAAM,WAAW,MAAM;AACrB,UAAQ,IAAI,EAAE;AAChB;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,aAAAA,QAAM,MAAM,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC/D;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,aAAAA,QAAM,KAAK,aAAAA,QAAM,OAAO,GAAG,KAAK,OAAO,MAAM,CAAC,cAAS,CAAC,GAAG,OAAO;AAChF;AAEA,IAAM,WAAW,CAAC,SAAiB,SAAS,MAAM;AAChD,UAAQ,IAAI,aAAAA,QAAM,IAAI,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC7D;AAEA,IAAM,gBAAgB,CAAC,cAAsB,YAAoB;AAC/D,QAAM,aAAa,GAAG,YAAY,YAAY,OAAO,EAAE;AACvD,aAAW,GAAG,UAAU,IAAI,aAAAA,QAAM,KAAK,OAAO,CAAC,EAAE;AACnD;;;ACjPA,IAAAC,gBAAkB;AAClB,IAAAC,gBAA+C;AAE/C,IAAAC,yBAAwC;AAExC,IAAM,6BAA6B,OAAO,QAAmC;AAE3E,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,+CAAwB,KAAK,oBAAoB;AAEpG,SAAO;AACT;AAEA,IAAM,oBAAoB,OAAO,QAAmC;AAClE,QAAM,cAAc,MAAM,2BAA2B,GAAG;AACxD,SAAO,MAAM,YAAY,OAAO;AAClC;AAEA,IAAM,oBAAoB,OAAO,KAAgC,YAAqB;AACpF,QAAM,cAAc,MAAM,2BAA2B,GAAG;AACxD,QAAM,YAAY,UAAU,OAAO;AACrC;AAGA,IAAM,0BAA0B,CAAC,mBAA2B;AAC1D,UAAQ,IAAI,saAAwE;AACpF,UAAQ,IAAI,kCAAwB,cAAc,EAAE;AACpD,UAAQ,IAAI,saAAwE;AACtF;AAGA,IAAM,oBAAoB,MAAM;AAC9B,UAAQ,IAAI,saAAwE;AACtF;AAEO,IAAM,yBAAyB,OACpC,KACA,SACA,gBACG;AACH,MAAI;AACF,UAAM,mBAAmB,MAAM,kBAAkB,GAAG;AAEpD,UAAM,kBAAkB,KAAK,OAAO;AAGpC,QAAI,SAAS;AACX,8BAAwB,GAAG,cAAc,IAAI,cAAAC,QAAM,KAAK,WAAW,CAAC,YAAY,OAAO,EAAE;AAAA,IAC3F;AAGA,QAAI,CAAC,WAAW,kBAAkB;AAChC,wBAAkB;AAAA,IACpB;AAAA,EACF,SAAS,OAAO;AACd,YAAQ,IAAI,cAAAA,QAAM,IAAI,8BAA8B,GAAG,KAAK;AAAA,EAC9D;AACF;AAEO,IAAM,gBAAgB,OAC3B,KACA,aACA,YACG;AACH,QAAM,mBAAmB,MAAM,kBAAkB,GAAG;AAEpD,QAAM,kBAAkB,KAAK,IAAI;AACjC,0BAAwB,IAAI,cAAAA,QAAM,KAAK,WAAW,CAAC,SAAS;AAC5D,QAAM,QAAQ;AACd,oBAAkB;AAGlB,MAAI,CAAC,kBAAkB;AACrB,UAAM,kBAAkB,KAAK,KAAK;AAAA,EACpC;AACF;;;ACzEA,kBAAuB;AACvB,IAAAC,iBAAuB;AACvB,qCAA2C;AAE3C,IAAM,sBAAsB,OAAO,QAAgB,aAA6D;AAI9G,QAAM,WAAW,MAAM,SAAS,QAAQ,yBAAyB;AAGjE,QAAM,YAAY,SAAS,WAAW;AAGtC,MAAI,WAAW;AACb,YAAQ,IAAI,GAAG,MAAM,iCAAiC;AAAA,EACxD;AAEA,SAAO;AACT;AAEO,IAAM,oBAAoB,OAAO,UAA0D,WAAmB;AAEnH,MAAI,MAAM,oBAAoB,kBAAkB,MAAM,QAAQ;AAAG;AAGjE,QAAM,cAAc,IAAI,sBAAO;AAAA,IAC7B;AAAA,IACA,CAAC,sDAAsD;AAAA,IACvD;AAAA,EACF;AAGA,QAAM,YAAY,MAAM,YAAY,YAAY,MAAM;AAEtD,SAAO;AACT;AAEO,IAAM,0BAA0B,OACrC,UACA,WACG;AAEH,MAAI,MAAM,oBAAoB,wBAAwB,MAAM,QAAQ;AAAG;AAGvE,QAAM,cAAc,IAAI,sBAAO;AAAA,IAC7B;AAAA,IACA,CAAC,qDAAqD;AAAA,IACtD;AAAA,EACF;AAGA,QAAM,kBAAkB,MAAM,YAAY,cAAc,MAAM;AAE9D,SAAO;AACT;AAEO,IAAM,uBAAuB,OAClC,UACA,QACA,kBACG;AAEH,MAAI,MAAM,oBAAoB,qBAAqB,MAAM,QAAQ;AAAG;AAGpE,QAAM,kBAAkB,MAAM,wBAAwB,UAAU,MAAM;AACtE,0BAAO,eAAe,EAAE,MAAM,MAAM,0BAA0B;AAG9D,QAAM,YAAY,MAAM,kBAAkB,UAAU,MAAM;AAC1D,0BAAO,SAAS,EAAE,MAAM,eAAe,8BAA8B;AACvE;;;AC1EA,iBAA4B;AAC5B,IAAAC,eAAuB;AAEhB,IAAM,oBAAoB,CAAI,QAAmB,iBAAyB;AAC/E,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,OAAO,2BAA2B;AAC/D,2BAAO,OAAO,KAAK,EAAE,GAAG,QAAQ,cAAc,uCAAuC,YAAY,EAAE;AACrG;AAEO,IAAM,sBAAsB,CAAI,WAAyB;AAC9D,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,SAAO,OAAO;AAChB;AAEO,IAAM,oBAAoB,CAAI,QAAmB,UAAgB;AACtE,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,2BAAO,OAAO,IAAI,EAAE,GAAG,GAAG,OAAO,yCAAyC,KAAK,EAAE;AACjF,SAAO,OAAO;AAChB;AAEO,IAAM,2BAA2B,CAAI,QAAmB,YAA2B;AACxF,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,2BAAO,OAAO,IAAI,EAAE,GAAG,QAAQ,SAAS,2CAA2C,OAAO,EAAE;AAC5F,SAAO,OAAO;AAChB;;;ANMA,IAAAC,yBAMO;AAAA,IA0BP,4BAAa,CAAC,QAAQ,eAAe;AAEnC,MAAI,WAAW,YAAY,WAAW,SAAS,YAAY;AACzD;AAAA,EACF;AAGA,SAAO,SAAS,aAAa;AAAA,IAC3B,KAAK;AAAA,IACL,eAAe;AAAA,IACf,UAAU;AAAA,IACV,SAAS;AAAA,IACT,aAAa,CAAC;AAAA,IACd,KAAK;AAAA,IACL,UAAU;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAGA,MAAI,CAAC,WAAW,WAAW,aAAa,GAAG;AACzC,WAAO,SAAS,aAAa,IAAI;AAAA,MAC/B,KAAK,QAAQ,IAAI,mBAAmB;AAAA,MACpC,UAAU,QAAQ,IAAI,cAAc,CAAC,QAAQ,IAAI,WAAW,IAAI,CAAC;AAAA,MACjE,SAAS;AAAA,MACT,KAAK;AAAA,MACL,eAAe;AAAA,MACf,UAAU;AAAA,MACV,SAAS;AAAA,MACT,aAAa,CAAC;AAAA,IAChB;AAAA,EACF;AAGA,MAAI,CAAC,WAAW,WAAW,aAAa,GAAG;AACzC,WAAO,SAAS,aAAa,IAAI;AAAA,MAC/B,KAAK,QAAQ,IAAI,4BAA4B;AAAA,MAC7C,UAAU,QAAQ,IAAI,cAAc,CAAC,QAAQ,IAAI,WAAW,IAAI,CAAC;AAAA,MACjE,SAAS;AAAA,MACT,KAAK;AAAA,MACL,eAAe;AAAA,MACf,UAAU;AAAA,MACV,SAAS;AAAA,MACT,aAAa,CAAC;AAAA,IAChB;AAAA,EACF;AAGA,SAAO,WAAW;AAAA,IAChB,UAAU,WAAW,UAAU,YAAY;AAAA,IAC3C,YAAY,WAAW,UAAU,cAAc;AAAA,EACjD;AACF,CAAC;AAAA,IAMD,oBAAK,uBAAuB,iCAAiC,EAC1D,iBAAiB,WAAW,mBAAmB,QAAW,oBAAM,MAAM,EACtE,UAAU,OAAO,EAAE,QAAQ,GAAkB,QAAQ;AACpD,QAAM,EAAE,QAAQ,IAAI;AACpB,QAAM,EAAE,MAAM,YAAY,IAAI;AAE9B,MAAI,gBAAgB,cAAc;AAChC,YAAQ,KAAK,cAAAC,QAAM,OAAO,mDAAmD,CAAC;AAC9E;AAAA,EACF;AAEA,MAAI,CAAC,SAAS;AACZ,YAAQ,KAAK,cAAAA,QAAM,IAAI,+BAA+B,CAAC;AACvD;AAAA,EACF;AAEA,UAAQ,KAAK,cAAAA,QAAM,MAAM,iCAAiC,OAAO,EAAE,CAAC;AAEpE,MAAI;AACF,UAAM,sBAAsB,KAAK,OAAO;AAAA,EAC1C,SAAS,GAAG;AACV,YAAQ,KAAK,cAAAA,QAAM,IAAI,2CAA2C,OAAO,KAAK,CAAC,EAAE,CAAC;AAAA,EACpF;AACF,CAAC;AAAA,IAIH,oBAAK,yBAAyB,mDAAmD,EAC9E,iBAAiB,iBAAiB,kCAAkC,MAAM,oBAAM,OAAO,EACvF,iBAAiB,UAAU,8BAA8B,OAAO,oBAAM,OAAO,EAC7E,UAAU,OAAO,EAAE,eAAe,OAAO,GAAoB,QAAQ;AACpE,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe,iBAAiB;AAAA,IAChC,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,IAC9C,QAAQ,UAAU;AAAA,EACpB,CAAC;AACH,CAAC;AAAA,IAEH,oBAAK,6BAAW,kCAAkC,EAAE,UAAU,OAAO,CAAC,GAAG,KAAK,aAAa;AACzF,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe;AAAA,IACf,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,EAChD,CAAC;AACD,SAAO,SAAS;AAClB,CAAC;AAAA,IAED,oBAAK,6BAAW,kCAAkC,EAAE,UAAU,OAAO,CAAC,GAAG,KAAK,aAAa;AACzF,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe;AAAA,IACf,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,EAChD,CAAC;AACD,SAAO,SAAS;AAClB,CAAC;AAAA,IAID,oBAAK,8BAA8B,0CAA0C,EAC1E,SAAS,UAAU,6BAA6B,OAAO,oBAAM,OAAO,EACpE,UAAU,OAAO,EAAE,OAAO,GAAG,QAAQ;AACpC,QAAM,uBAAuB,KAAK,MAAM;AAC1C,CAAC;AAAA,IA+IH,iCAAkB,CAAC,QAAQ;AACzB,MAAI,WAAW;AAAA,IACb,sBAAsB,OAAO,WAAgC;AAG3D,YAAM,cAAc,MAAM,IAAI,OAAO,sBAAsB,gCAAgC;AAC3F,YAAM,EAAE,cAAc,gBAAgB,IAAI,UAAM,sCAAqB,WAAW;AAGhF,YAAM,4BAA4B;AAAA,QAChC,GAAG;AAAA,QACH,WAAW;AAAA,UACT,GAAG,OAAO;AAAA,UACV;AAAA,QACF;AAAA,MACF;AAEA,iBAAO,kCAAqB,yBAAyB;AAAA,IACvD;AAAA,IACA,sBAAsB,CAAC,WAA2B;AAChD,iBAAO,kCAAqB,MAAM;AAAA,IACpC;AAAA,IACA,sBAAsB,OAAO,WAAgC;AAC3D,iBAAO,sCAAqB,MAAM;AAAA,IACpC;AAAA,IACA,qBAAqB,OAAU,WAA2C;AACxE,YAAM,gBAAgB,MAAM;AAC5B,aAAO,oBAAoB,aAAa;AAAA,IAC1C;AAAA,IACA,mBAAmB,OAAU,QAAwC,iBAAyB;AAC5F,YAAM,gBAAgB,MAAM;AAC5B,aAAO,kBAAkB,eAAe,YAAY;AAAA,IACtD;AAAA,IACA,mBAAmB,OAAU,QAAwC,UAAa;AAChF,YAAM,gBAAgB,MAAM;AAC5B,aAAO,kBAAkB,eAAe,KAAK;AAAA,IAC/C;AAAA,IACA,0BAA0B,OAAU,QAAwC,YAAwB;AAClG,YAAM,gBAAgB,MAAM;AAC5B,aAAO,yBAAyB,eAAe,OAAO;AAAA,IACxD;AAAA,IACA,OAAO;AAAA,MACL,UAAU,OAAO,aAAqB,YAAiC;AACrE,eAAO,cAAc,KAAK,aAAa,OAAO;AAAA,MAChD;AAAA,MACA,YAAY,OAAO,gBAAyB;AAC1C,eAAO,uBAAuB,KAAK,MAAM,WAAW;AAAA,MACtD;AAAA,MACA,aAAa,YAAY;AACvB,eAAO,uBAAuB,KAAK,KAAK;AAAA,MAC1C;AAAA,MACA,aAAa,OAAO,YAA6B;AAC/C,eAAO,YAAY,KAAK,OAAO;AAAA,MACjC;AAAA,MACA,cAAc,OAAO,WAAmB;AACtC,cAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAC7C,eAAO,kBAAkB,OAAO,UAAU,MAAM;AAAA,MAClD;AAAA,MACA,iBAAiB,OAAO,QAAgB,kBAA0B;AAChE,cAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAC7C,eAAO,qBAAqB,OAAO,UAAU,QAAQ,aAAa;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AACF,CAAC;","names":["import_chalk","import_sdk","import_types","chalk","import_chalk","import_types","import_mock_contracts","chalk","import_ethers","import_chai","import_mock_contracts","chalk"]}
1
+ {"version":3,"sources":["../src/index.ts","../src/fund.ts","../src/consts.ts","../src/deploy.ts","../src/logging.ts","../src/utils.ts"],"sourcesContent":["/* eslint-disable no-empty-pattern */\n/* eslint-disable turbo/no-undeclared-env-vars */\nimport chalk from 'chalk';\nimport { type PublicClient, type WalletClient } from 'viem';\nimport { extendConfig, extendEnvironment, task, types } from 'hardhat/config';\nimport { TASK_TEST, TASK_NODE } from 'hardhat/builtin-tasks/task-names';\nimport { HardhatEthersSigner } from '@nomicfoundation/hardhat-ethers/signers';\nimport {\n MOCKS_ZK_VERIFIER_SIGNER_ADDRESS,\n type CofhesdkClient,\n type CofhesdkConfig,\n type CofhesdkInputConfig,\n} from '@cofhe/sdk';\nimport { createCofhesdkClient, createCofhesdkConfig } from '@cofhe/sdk/node';\nimport { HardhatSignerAdapter } from '@cofhe/sdk/adapters';\n\nimport { localcofheFundAccount } from './fund.js';\nimport { TASK_COFHE_MOCKS_DEPLOY, TASK_COFHE_MOCKS_SET_LOG_OPS, TASK_COFHE_USE_FAUCET } from './consts.js';\nimport { deployMocks, type DeployMocksArgs } from './deploy.js';\nimport { mock_setLoggingEnabled, mock_withLogs } from './logging.js';\nimport { mock_expectPlaintext } from './utils.js';\nimport { mock_getPlaintext } from './utils.js';\nimport type { Contract } from 'ethers';\nimport {\n MockACLArtifact,\n MockQueryDecrypterArtifact,\n MockTaskManagerArtifact,\n MockZkVerifierArtifact,\n TestBedArtifact,\n} from '@cofhe/mock-contracts';\nimport { hardhat } from '@cofhe/sdk/chains';\nexport {\n MockACLArtifact,\n MockQueryDecrypterArtifact,\n MockTaskManagerArtifact,\n MockZkVerifierArtifact,\n TestBedArtifact,\n} from '@cofhe/mock-contracts';\n\n/**\n * Configuration interface for the CoFHE Hardhat plugin.\n * Allows users to configure mock logging and gas warning settings.\n */\ndeclare module 'hardhat/types/config' {\n interface HardhatUserConfig {\n cofhesdk?: {\n /** Whether to log mock operations (default: true) */\n logMocks?: boolean;\n /** Whether to show gas usage warnings for mock operations (default: true) */\n gasWarning?: boolean;\n };\n }\n\n interface HardhatConfig {\n cofhesdk: {\n /** Whether to log mock operations (default: true) */\n logMocks: boolean;\n /** Whether to show gas usage warnings for mock operations (default: true) */\n gasWarning: boolean;\n };\n }\n}\n\nextendConfig((config, userConfig) => {\n // Allow users to override the localcofhe network config\n if (userConfig.networks && userConfig.networks.localcofhe) {\n return;\n }\n\n // Default config\n config.networks.localcofhe = {\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 10_000,\n httpHeaders: {},\n url: 'http://127.0.0.1:42069',\n accounts: [\n '0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80',\n '0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d',\n '0x5de4111afa1a4b94908f83103eb1f1706367c2e68ca870fc3fb9a804cdab365a',\n '0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6',\n ],\n };\n\n // Only add Sepolia config if user hasn't defined it\n if (!userConfig.networks?.['eth-sepolia']) {\n config.networks['eth-sepolia'] = {\n url: process.env.SEPOLIA_RPC_URL ?? 'https://ethereum-sepolia.publicnode.com',\n accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],\n chainId: 11155111,\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 60_000,\n httpHeaders: {},\n };\n }\n\n // Only add Arbitrum Sepolia config if user hasn't defined it\n if (!userConfig.networks?.['arb-sepolia']) {\n config.networks['arb-sepolia'] = {\n url: process.env.ARBITRUM_SEPOLIA_RPC_URL ?? 'https://sepolia-rollup.arbitrum.io/rpc',\n accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],\n chainId: 421614,\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 60_000,\n httpHeaders: {},\n };\n }\n\n // Add cofhe config\n config.cofhesdk = {\n logMocks: userConfig.cofhesdk?.logMocks ?? true,\n gasWarning: userConfig.cofhesdk?.gasWarning ?? true,\n };\n});\n\ntype UseFaucetArgs = {\n address?: string;\n};\n\ntask(TASK_COFHE_USE_FAUCET, 'Fund an account from the funder')\n .addOptionalParam('address', 'Address to fund', undefined, types.string)\n .setAction(async ({ address }: UseFaucetArgs, hre) => {\n const { network } = hre;\n const { name: networkName } = network;\n\n if (networkName !== 'localcofhe') {\n console.info(chalk.yellow(`Programmatic faucet only supported for localcofhe`));\n return;\n }\n\n if (!address) {\n console.info(chalk.red(`Failed to get address to fund`));\n return;\n }\n\n console.info(chalk.green(`Getting funds from faucet for ${address}`));\n\n try {\n await localcofheFundAccount(hre, address);\n } catch (e) {\n console.info(chalk.red(`failed to get funds from localcofhe for ${address}: ${e}`));\n }\n });\n\n// DEPLOY TASKS\n\ntask(TASK_COFHE_MOCKS_DEPLOY, 'Deploys the mock contracts on the Hardhat network')\n .addOptionalParam('deployTestBed', 'Whether to deploy the test bed', true, types.boolean)\n .addOptionalParam('silent', 'Whether to suppress output', false, types.boolean)\n .setAction(async ({ deployTestBed, silent }: DeployMocksArgs, hre) => {\n await deployMocks(hre, {\n deployTestBed: deployTestBed ?? true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n silent: silent ?? false,\n });\n });\n\ntask(TASK_TEST, 'Deploy mock contracts on hardhat').setAction(async ({}, hre, runSuper) => {\n await deployMocks(hre, {\n deployTestBed: true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n });\n return runSuper();\n});\n\ntask(TASK_NODE, 'Deploy mock contracts on hardhat').setAction(async ({}, hre, runSuper) => {\n await deployMocks(hre, {\n deployTestBed: true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n });\n return runSuper();\n});\n\n// SET LOG OPS\n\ntask(TASK_COFHE_MOCKS_SET_LOG_OPS, 'Set logging for the Mock CoFHE contracts')\n .addParam('enable', 'Whether to enable logging', false, types.boolean)\n .setAction(async ({ enable }, hre) => {\n await mock_setLoggingEnabled(hre, enable);\n });\n\n// MOCK UTILS\n\nexport * from './utils.js';\nexport * from './fund.js';\nexport * from './logging.js';\nexport * from './deploy.js';\n\n/**\n * Runtime environment extensions for the CoFHE Hardhat plugin.\n * Provides access to CoFHE initialization, environment checks, and mock utilities.\n */\ndeclare module 'hardhat/types/runtime' {\n export interface HardhatRuntimeEnvironment {\n cofhesdk: {\n /**\n * Create a CoFHE SDK configuration for use with cofhesdk.createCofhesdkClient(...)\n * @param {CofhesdkInputConfig} config - The CoFHE SDK input configuration\n * @returns {CofhesdkConfig} The CoFHE SDK configuration\n */\n createCofhesdkConfig: (config: CofhesdkInputConfig) => Promise<CofhesdkConfig>;\n /**\n * Create a CoFHE SDK client instance\n * @param {CofhesdkConfig} config - The CoFHE SDK configuration (use createCofhesdkConfig to create with Node.js defaults)\n * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance\n */\n createCofhesdkClient: (config: CofhesdkConfig) => CofhesdkClient;\n /**\n * Create viem clients from a Hardhat ethers signer, to be used with `cofhesdkClient.connect(...)`\n * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use\n * @returns {Promise<{ publicClient: PublicClient; walletClient: WalletClient }>} The viem clients\n */\n hardhatSignerAdapter: (\n signer: HardhatEthersSigner\n ) => Promise<{ publicClient: PublicClient; walletClient: WalletClient }>;\n /**\n * Connect a CoFHE SDK client with a Hardhat ethers signer\n * @param {CofhesdkClient} client - The CoFHE SDK client to connect\n * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use\n * @returns {Promise<void>}\n */\n connectWithHardhatSigner: (client: CofhesdkClient, signer: HardhatEthersSigner) => Promise<void>;\n /**\n * Create and connect to a batteries included client.\n * Also generates a self-usage a permit for the signer.\n * If customization is needed, use createCofhesdkClient and connectWithHardhatSigner.\n * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use (optional - defaults to first signer)\n * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance\n */\n createBatteriesIncludedCofhesdkClient: (signer?: HardhatEthersSigner) => Promise<CofhesdkClient>;\n\n mocks: {\n /**\n * **[MOCKS ONLY]**\n *\n * Execute a block of code with cofhe mock contracts logging enabled.\n *\n * _(If logging only a function, we recommend passing the function name as the closureName (ex \"counter.increment()\"))_\n *\n * Example usage:\n *\n * ```ts\n * await hre.cofhesdk.mocks.withLogs(\"counter.increment()\", async () => {\n * await counter.increment();\n * });\n * ```\n *\n * Expected output:\n * ```\n * ┌──────────────────┬──────────────────────────────────────────────────\n * │ [COFHE-MOCKS] │ \"counter.increment()\" logs:\n * ├──────────────────┴──────────────────────────────────────────────────\n * ├ FHE.add | euint32(4473..3424)[0] + euint32(1157..3648)[1] => euint32(1106..1872)[1]\n * ├ FHE.allowThis | euint32(1106..1872)[1] -> 0x663f..6602\n * ├ FHE.allow | euint32(1106..1872)[1] -> 0x3c44..93bc\n * └─────────────────────────────────────────────────────────────────────\n * ```\n * @param {string} closureName - Name of the code block to log within\n * @param {() => Promise<void>} closure - The async function to execute\n */\n withLogs: (closureName: string, closure: () => Promise<void>) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Enable logging from cofhe mock contracts\n * @param {string} closureName - Optional name of the code block to enable logging for\n */\n enableLogs: (closureName?: string) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Disable logging from cofhe mock contracts\n */\n disableLogs: () => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Deploy the cofhe mock contracts (normally this is done automatically)\n * @param {DeployMocksArgs} options - Deployment options\n */\n deployMocks: (options: DeployMocksArgs) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Get the plaintext value for a ciphertext hash\n * @param {bigint} ctHash - The ciphertext hash to look up\n * @returns {Promise<bigint>} The plaintext value\n */\n getPlaintext: (ctHash: bigint) => Promise<bigint>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Assert that a ciphertext hash represents an expected plaintext value\n * @param {bigint} ctHash - The ciphertext hash to check\n * @param {bigint} expectedValue - The expected plaintext value\n */\n expectPlaintext: (ctHash: bigint, expectedValue: bigint) => Promise<void>;\n\n /**\n * Get the MockTaskManager contract\n * @returns {Promise<Contract>} The MockTaskManager contract\n */\n getMockTaskManager: () => Promise<Contract>;\n\n /**\n * Get the MockACL contract\n * @returns {Promise<Contract>} The MockACL contract\n */\n getMockACL: () => Promise<Contract>;\n\n /**\n * Get the MockQueryDecrypter contract\n * @returns {Promise<Contract>} The MockQueryDecrypter contract\n */\n getMockQueryDecrypter: () => Promise<Contract>;\n\n /**\n * Get the MockZkVerifier contract\n * @returns {Promise<Contract>} The MockZkVerifier contract\n */\n getMockZkVerifier: () => Promise<Contract>;\n\n /**\n * Get the TestBed contract\n * @returns {Promise<Contract>} The TestBed contract\n */\n getTestBed: () => Promise<Contract>;\n };\n };\n }\n}\n\nextendEnvironment((hre) => {\n hre.cofhesdk = {\n createCofhesdkConfig: async (config: CofhesdkInputConfig) => {\n // Create zkv wallet client\n // This wallet interacts with the MockZkVerifier contract so that the user's connected wallet doesn't have to\n const zkvHhSigner = await hre.ethers.getImpersonatedSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n const { walletClient: zkvWalletClient } = await HardhatSignerAdapter(zkvHhSigner);\n\n // Inject zkv wallet client into config\n const configWithZkvWalletClient = {\n environment: 'hardhat' as const,\n ...config,\n _internal: {\n ...config._internal,\n zkvWalletClient,\n },\n };\n\n return createCofhesdkConfig(configWithZkvWalletClient);\n },\n createCofhesdkClient: (config: CofhesdkConfig) => {\n return createCofhesdkClient(config);\n },\n hardhatSignerAdapter: async (signer: HardhatEthersSigner) => {\n return HardhatSignerAdapter(signer);\n },\n connectWithHardhatSigner: async (client: CofhesdkClient, signer: HardhatEthersSigner) => {\n const { publicClient, walletClient } = await HardhatSignerAdapter(signer);\n return client.connect(publicClient, walletClient);\n },\n createBatteriesIncludedCofhesdkClient: async (signer?: HardhatEthersSigner) => {\n // Get signer if not provided\n if (!signer) {\n [signer] = await hre.ethers.getSigners();\n }\n\n // Create config\n const config = await hre.cofhesdk.createCofhesdkConfig({\n environment: 'hardhat',\n supportedChains: [hardhat],\n });\n\n // Create client\n const client = hre.cofhesdk.createCofhesdkClient(config);\n\n // Connect client\n await hre.cofhesdk.connectWithHardhatSigner(client, signer);\n\n // Create self-usage permit\n await client.permits.createSelf({\n issuer: signer.address,\n });\n\n // Return client\n return client;\n },\n mocks: {\n withLogs: async (closureName: string, closure: () => Promise<void>) => {\n return mock_withLogs(hre, closureName, closure);\n },\n enableLogs: async (closureName?: string) => {\n return mock_setLoggingEnabled(hre, true, closureName);\n },\n disableLogs: async () => {\n return mock_setLoggingEnabled(hre, false);\n },\n deployMocks: async (options: DeployMocksArgs) => {\n return deployMocks(hre, options);\n },\n getPlaintext: async (ctHash: bigint) => {\n const [signer] = await hre.ethers.getSigners();\n return mock_getPlaintext(signer.provider, ctHash);\n },\n expectPlaintext: async (ctHash: bigint, expectedValue: bigint) => {\n const [signer] = await hre.ethers.getSigners();\n return mock_expectPlaintext(signer.provider, ctHash, expectedValue);\n },\n getMockTaskManager: async () => {\n return await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, MockTaskManagerArtifact.fixedAddress);\n },\n getMockACL: async () => {\n const tm = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, MockTaskManagerArtifact.fixedAddress);\n const aclAddress = await tm.acl();\n return await hre.ethers.getContractAt(MockACLArtifact.abi, aclAddress);\n },\n getMockQueryDecrypter: async () => {\n return await hre.ethers.getContractAt(MockQueryDecrypterArtifact.abi, MockQueryDecrypterArtifact.fixedAddress);\n },\n getMockZkVerifier: async () => {\n return await hre.ethers.getContractAt(MockZkVerifierArtifact.abi, MockZkVerifierArtifact.fixedAddress);\n },\n getTestBed: async () => {\n return await hre.ethers.getContractAt(TestBedArtifact.abi, TestBedArtifact.fixedAddress);\n },\n },\n };\n});\n","/* eslint-disable turbo/no-undeclared-env-vars */\nimport { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport '@nomicfoundation/hardhat-ethers';\n\n/**\n * Sends funds to the specified address\n * @param hre Hardhat Runtime Environment\n * @param toAddress Address to send funds to\n * @param amount Amount to send in ETH (default: 10)\n * @returns Transaction receipt or null if failed\n */\nexport async function localcofheFundAccount(hre: HardhatRuntimeEnvironment, toAddress: string, amount: string = '10') {\n // Load private key from environment\n const privateKey =\n process.env.FUNDER_PRIVATE_KEY ?? '0xb6b15c8cb491557369f3c7d2c287b053eb229daa9c22138887752191c9520659';\n if (!privateKey) {\n console.error('Error: FUNDER_PRIVATE_KEY environment variable not set');\n return null;\n }\n\n try {\n // Create wallet from private key\n const wallet = new hre.ethers.Wallet(privateKey, hre.ethers.provider);\n\n // Get wallet balance\n const balance = await hre.ethers.provider.getBalance(wallet.address);\n console.log(`Funder wallet address: ${wallet.address}`);\n console.log(`Funder wallet balance: ${hre.ethers.formatEther(balance)} ETH`);\n\n // Check if wallet has enough funds\n const amountToSend = hre.ethers.parseEther(amount);\n if (balance < amountToSend) {\n console.error(\n `Error: Funder wallet doesn't have enough funds. Current balance: ${hre.ethers.formatEther(balance)} ETH`\n );\n return null;\n }\n\n // Send transaction\n console.log(`Sending ${amount} ETH to ${toAddress}...`);\n const tx = await wallet.sendTransaction({\n to: toAddress,\n value: amountToSend,\n });\n\n console.log(`Transaction sent! Hash: ${tx.hash}`);\n console.log('Waiting for confirmation...');\n\n // Wait for transaction to be mined\n const receipt = await tx.wait();\n console.log(`Transaction confirmed in block ${receipt?.blockNumber}`);\n console.log(`Successfully sent ${amount} ETH to ${toAddress}`);\n\n return receipt;\n } catch (error) {\n console.error('Error sending funds:', error);\n return null;\n }\n}\n\n/**\n * Checks a wallet's balance and funds it if below 1 ETH\n * @param hre Hardhat Runtime Environment\n * @param walletAddress Address of the wallet to check and potentially fund\n * @returns Promise that resolves when the funding operation completes (if needed)\n */\nexport async function localcofheFundWalletIfNeeded(hre: HardhatRuntimeEnvironment, walletAddress: string) {\n // Check wallet balance and fund if needed\n const walletBalance = await hre.ethers.provider.getBalance(walletAddress);\n console.log(`Wallet balance: ${hre.ethers.formatEther(walletBalance)} ETH`);\n\n if (walletBalance < hre.ethers.parseEther('1')) {\n console.log(`Wallet balance is less than 1 ETH. Funding ${walletAddress}...`);\n const receipt = await localcofheFundAccount(hre, walletAddress);\n if (receipt) {\n const newBalance = await hre.ethers.provider.getBalance(walletAddress);\n console.log(`Wallet new balance: ${hre.ethers.formatEther(newBalance)} ETH`);\n } else {\n console.error(`Failed to fund ${walletAddress}`);\n }\n }\n}\n","export const TASK_COFHE_USE_FAUCET = 'task:cofhe:usefaucet';\nexport const TASK_COFHE_MOCKS_SET_LOG_OPS = 'task:cofhe-mocks:setlogops';\nexport const TASK_COFHE_MOCKS_DEPLOY = 'task:cofhe-mocks:deploy';\n\nexport const TASK_MANAGER_ADDRESS = '0xeA30c4B8b44078Bbf8a6ef5b9f1eC1626C7848D9';\nexport const MOCKS_ZK_VERIFIER_ADDRESS = '0x0000000000000000000000000000000000000100';\nexport const MOCKS_QUERY_DECRYPTER_ADDRESS = '0x0000000000000000000000000000000000000200';\nexport const TEST_BED_ADDRESS = '0x0000000000000000000000000000000000000300';\n","import { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport chalk from 'chalk';\nimport { Contract } from 'ethers';\n\nimport {\n TASK_MANAGER_ADDRESS,\n MOCKS_ZK_VERIFIER_ADDRESS,\n MOCKS_QUERY_DECRYPTER_ADDRESS,\n TEST_BED_ADDRESS,\n} from './consts.js';\n\nimport {\n MockTaskManagerArtifact,\n MockACLArtifact,\n MockZkVerifierArtifact,\n MockQueryDecrypterArtifact,\n TestBedArtifact,\n} from '@cofhe/mock-contracts';\nimport { MOCKS_ZK_VERIFIER_SIGNER_ADDRESS } from '@cofhe/sdk';\n\n// Deploy\n\nconst deployMockTaskManager = async (hre: HardhatRuntimeEnvironment) => {\n const [signer] = await hre.ethers.getSigners();\n\n // Deploy MockTaskManager\n await hardhatSetCode(hre, TASK_MANAGER_ADDRESS, MockTaskManagerArtifact.deployedBytecode);\n const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);\n\n // Initialize MockTaskManager\n const initTx = await taskManager.initialize(signer.address);\n await initTx.wait();\n\n // Check if MockTaskManager exists\n const tmExists = await taskManager.exists();\n if (!tmExists) {\n throw new Error('MockTaskManager does not exist');\n }\n\n return taskManager;\n};\n\nconst deployMockACL = async (hre: HardhatRuntimeEnvironment): Promise<Contract> => {\n // Deploy MockACL (uses ethers to deploy to ensure constructor called and EIP712 domain set)\n const acl = await ethersDeployContract(hre, MockACLArtifact.abi, MockACLArtifact.bytecode);\n\n // Check if ACL exists\n const exists = await acl.exists();\n if (!exists) {\n logError('MockACL does not exist', 2);\n throw new Error('MockACL does not exist');\n }\n\n return acl;\n};\n\nconst deployMockZkVerifier = async (hre: HardhatRuntimeEnvironment) => {\n await hardhatSetCode(hre, MOCKS_ZK_VERIFIER_ADDRESS, MockZkVerifierArtifact.deployedBytecode);\n const zkVerifier = await hre.ethers.getContractAt(MockZkVerifierArtifact.abi, MOCKS_ZK_VERIFIER_ADDRESS);\n\n const zkVerifierExists = await zkVerifier.exists();\n if (!zkVerifierExists) {\n logError('MockZkVerifier does not exist', 2);\n throw new Error('MockZkVerifier does not exist');\n }\n\n return zkVerifier;\n};\n\nconst deployMockQueryDecrypter = async (hre: HardhatRuntimeEnvironment, acl: Contract) => {\n await hardhatSetCode(hre, MOCKS_QUERY_DECRYPTER_ADDRESS, MockQueryDecrypterArtifact.deployedBytecode);\n const queryDecrypter = await hre.ethers.getContractAt(MockQueryDecrypterArtifact.abi, MOCKS_QUERY_DECRYPTER_ADDRESS);\n\n // Initialize MockQueryDecrypter\n const initTx = await queryDecrypter.initialize(TASK_MANAGER_ADDRESS, await acl.getAddress());\n await initTx.wait();\n\n // Check if MockQueryDecrypter exists\n const queryDecrypterExists = await queryDecrypter.exists();\n if (!queryDecrypterExists) {\n logError('MockQueryDecrypter does not exist', 2);\n throw new Error('MockQueryDecrypter does not exist');\n }\n\n return queryDecrypter;\n};\n\nconst deployTestBedContract = async (hre: HardhatRuntimeEnvironment) => {\n await hardhatSetCode(hre, TEST_BED_ADDRESS, TestBedArtifact.deployedBytecode);\n const testBed = await hre.ethers.getContractAt(TestBedArtifact.abi, TEST_BED_ADDRESS);\n await testBed.waitForDeployment();\n return testBed;\n};\n\n// Funding\n\nconst fundZkVerifierSigner = async (hre: HardhatRuntimeEnvironment) => {\n const zkVerifierSigner = await hre.ethers.getSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n await hre.network.provider.send('hardhat_setBalance', [\n zkVerifierSigner.address,\n '0x' + hre.ethers.parseEther('10').toString(16),\n ]);\n};\n\n// Initializations\n\nconst setTaskManagerACL = async (taskManager: Contract, acl: Contract) => {\n const setAclTx = await taskManager.setACLContract(await acl.getAddress());\n await setAclTx.wait();\n};\n\nexport type DeployMocksArgs = {\n deployTestBed?: boolean;\n gasWarning?: boolean;\n silent?: boolean;\n};\n\nexport const deployMocks = async (\n hre: HardhatRuntimeEnvironment,\n options: DeployMocksArgs = {\n deployTestBed: true,\n gasWarning: true,\n silent: false,\n }\n) => {\n // Check if network is Hardhat, if not log skip message and return\n const isHardhat = await checkNetworkAndSkip(hre);\n if (!isHardhat) return;\n\n const logEmptyIfNoisy = () => {\n if (!options.silent) {\n logEmpty();\n }\n };\n const logSuccessIfNoisy = (message: string, indent = 0) => {\n if (!options.silent) {\n logSuccess(message, indent);\n }\n };\n const logDeploymentIfNoisy = (contractName: string, address: string) => {\n if (!options.silent) {\n logDeployment(contractName, address);\n }\n };\n const logWarningIfNoisy = (message: string, indent = 0) => {\n if (!options.silent) {\n logWarning(message, indent);\n }\n };\n\n // Log start message\n logEmptyIfNoisy();\n logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: deploy mocks'), 0);\n logEmptyIfNoisy();\n\n // Deploy mock contracts\n const taskManager = await deployMockTaskManager(hre);\n logDeploymentIfNoisy('MockTaskManager', await taskManager.getAddress());\n\n const acl = await deployMockACL(hre);\n logDeploymentIfNoisy('MockACL', await acl.getAddress());\n\n await setTaskManagerACL(taskManager, acl);\n logSuccessIfNoisy('ACL address set in TaskManager', 2);\n\n await fundZkVerifierSigner(hre);\n logSuccessIfNoisy(`ZkVerifier signer (${MOCKS_ZK_VERIFIER_SIGNER_ADDRESS}) funded`, 1);\n\n const zkVerifierSignerBalance = await hre.ethers.provider.getBalance(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n logSuccessIfNoisy(`ETH balance: ${zkVerifierSignerBalance.toString()}`, 2);\n\n const zkVerifier = await deployMockZkVerifier(hre);\n logDeploymentIfNoisy('MockZkVerifier', await zkVerifier.getAddress());\n\n const queryDecrypter = await deployMockQueryDecrypter(hre, acl);\n logDeploymentIfNoisy('MockQueryDecrypter', await queryDecrypter.getAddress());\n\n if (options.deployTestBed) {\n logSuccessIfNoisy('TestBed deployment enabled', 2);\n const testBed = await deployTestBedContract(hre);\n logDeploymentIfNoisy('TestBed', await testBed.getAddress());\n }\n\n // Log success message\n logEmptyIfNoisy();\n logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: mocks deployed successfully'), 0);\n\n // Log warning about mocks increased gas costs\n if (options.gasWarning) {\n logEmptyIfNoisy();\n logWarningIfNoisy(\n \"When using mocks, FHE operations (eg FHE.add / FHE.mul) report a higher gas price due to additional on-chain mocking logic. Deploy your contracts on a testnet chain to check the true gas costs.\\n(Disable this warning by setting '@cofhe/sdk.gasWarning' to false in your hardhat config\",\n 0\n );\n }\n\n logEmptyIfNoisy();\n};\n\n// Utils\n\nconst hardhatSetCode = async (hre: HardhatRuntimeEnvironment, address: string, bytecode: string) => {\n await hre.network.provider.send('hardhat_setCode', [address, bytecode]);\n};\n\nconst ethersDeployContract = async (hre: HardhatRuntimeEnvironment, abi: any, bytecode: string) => {\n const [signer] = await hre.ethers.getSigners();\n\n const factory = new hre.ethers.ContractFactory(abi, bytecode, signer);\n const contract = await factory.deploy(/* constructor args */);\n await contract.waitForDeployment();\n\n return contract as Contract;\n};\n\n// Network\n\nconst checkNetworkAndSkip = async (hre: HardhatRuntimeEnvironment) => {\n const network = hre.network.name;\n const isHardhat = network === 'hardhat';\n if (!isHardhat) logSuccess(`cofhe-hardhat-plugin - deploy mocks - skipped on non-hardhat network ${network}`, 0);\n return isHardhat;\n};\n\n// Logging\n\nconst logEmpty = () => {\n console.log('');\n};\n\nconst logSuccess = (message: string, indent = 1) => {\n console.log(chalk.green(`${' '.repeat(indent)}✓ ${message}`));\n};\n\nconst logWarning = (message: string, indent = 1) => {\n console.log(chalk.bold(chalk.yellow(`${' '.repeat(indent)}⚠ NOTE:`)), message);\n};\n\nconst logError = (message: string, indent = 1) => {\n console.log(chalk.red(`${' '.repeat(indent)}✗ ${message}`));\n};\n\nconst logDeployment = (contractName: string, address: string) => {\n const paddedName = `${contractName} deployed`.padEnd(36);\n logSuccess(`${paddedName} ${chalk.bold(address)}`);\n};\n","import chalk from 'chalk';\nimport { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport { TASK_MANAGER_ADDRESS } from './consts';\nimport { MockTaskManagerArtifact } from '@cofhe/mock-contracts';\n\nconst getDeployedMockTaskManager = async (hre: HardhatRuntimeEnvironment) => {\n // Fetch the deployed MockTaskManager\n const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);\n\n return taskManager;\n};\n\nconst getLoggingEnabled = async (hre: HardhatRuntimeEnvironment) => {\n const taskManager = await getDeployedMockTaskManager(hre);\n return await taskManager.logOps();\n};\n\nconst setLoggingEnabled = async (hre: HardhatRuntimeEnvironment, enabled: boolean) => {\n const taskManager = await getDeployedMockTaskManager(hre);\n await taskManager.setLogOps(enabled);\n};\n\n// prettier-ignore\nconst printLogsEnabledMessage = (closureMessage: string) => {\n console.log(\"┌──────────────────┬──────────────────────────────────────────────────\");\n console.log(`│ [COFHE-MOCKS] │ ${closureMessage}`);\n console.log(\"├──────────────────┴──────────────────────────────────────────────────\");\n};\n\n// prettier-ignore\nconst printLogsBlockEnd = () => {\n console.log(\"└─────────────────────────────────────────────────────────────────────\");\n};\n\nexport const mock_setLoggingEnabled = async (\n hre: HardhatRuntimeEnvironment,\n enabled: boolean,\n closureName?: string\n) => {\n try {\n const initiallyEnabled = await getLoggingEnabled(hre);\n\n await setLoggingEnabled(hre, enabled);\n\n // Only print if enabling logs\n if (enabled) {\n printLogsEnabledMessage(`${closureName ? `\"${chalk.bold(closureName)}\" logs:` : 'Logs:'}`);\n }\n\n // Only print if disabling logs AND logs currently enabled\n if (!enabled && initiallyEnabled) {\n printLogsBlockEnd();\n }\n } catch (error) {\n console.log(chalk.red('mock_setLoggingEnabled error'), error);\n }\n};\n\nexport const mock_withLogs = async (\n hre: HardhatRuntimeEnvironment,\n closureName: string,\n closure: () => Promise<void>\n) => {\n const initiallyEnabled = await getLoggingEnabled(hre);\n\n await setLoggingEnabled(hre, true);\n printLogsEnabledMessage(`\"${chalk.bold(closureName)}\" logs:`);\n await closure();\n printLogsBlockEnd();\n\n // If logs were disabled, disable them again\n if (!initiallyEnabled) {\n await setLoggingEnabled(hre, false);\n }\n};\n","import { TASK_MANAGER_ADDRESS, MOCKS_ZK_VERIFIER_ADDRESS } from './consts.js';\nimport { expect } from 'chai';\nimport { ethers } from 'ethers';\nimport { type HardhatEthersProvider } from '@nomicfoundation/hardhat-ethers/internal/hardhat-ethers-provider';\n\nconst mock_checkIsTestnet = async (fnName: string, provider: HardhatEthersProvider | ethers.JsonRpcProvider) => {\n // Testnet is checked by testing if MockZkVerifier is deployed\n\n // Get bytecode at ZK_VERIFIER_ADDRESS\n const bytecode = await provider.getCode(MOCKS_ZK_VERIFIER_ADDRESS);\n\n // If bytecode is empty, we are on a testnet\n const isTestnet = bytecode.length === 0;\n\n // Log if we are on a testnet\n if (isTestnet) {\n console.log(`${fnName} - skipped on non-testnet chain`);\n }\n\n return isTestnet;\n};\n\nexport const mock_getPlaintext = async (provider: HardhatEthersProvider | ethers.JsonRpcProvider, ctHash: bigint) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_getPlaintext.name, provider)) return;\n\n // Connect to MockTaskManager\n const taskManager = new ethers.Contract(\n TASK_MANAGER_ADDRESS,\n ['function mockStorage(uint256) view returns (uint256)'],\n provider\n );\n\n // Fetch the plaintext\n const plaintext = await taskManager.mockStorage(ctHash);\n\n return plaintext;\n};\n\nexport const mock_getPlaintextExists = async (\n provider: HardhatEthersProvider | ethers.JsonRpcProvider,\n ctHash: bigint\n) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_getPlaintextExists.name, provider)) return;\n\n // Connect to MockTaskManager\n const taskManager = new ethers.Contract(\n TASK_MANAGER_ADDRESS,\n ['function inMockStorage(uint256) view returns (bool)'],\n provider\n );\n\n // Fetch the plaintext exists\n const plaintextExists = await taskManager.inMockStorage(ctHash);\n\n return plaintextExists;\n};\n\nexport const mock_expectPlaintext = async (\n provider: HardhatEthersProvider | ethers.JsonRpcProvider,\n ctHash: bigint,\n expectedValue: bigint\n) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_expectPlaintext.name, provider)) return;\n\n // Expect the plaintext to exist\n const plaintextExists = await mock_getPlaintextExists(provider, ctHash);\n expect(plaintextExists).equal(true, 'Plaintext does not exist');\n\n // Expect the plaintext to have the expected value\n const plaintext = await mock_getPlaintext(provider, ctHash);\n expect(plaintext).equal(expectedValue, 'Plaintext value is incorrect');\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAEA,IAAAA,gBAAkB;AAClB,kBAAqD;AACrD,oBAA6D;AAC7D,wBAAqC;AACrC,qBAAoC;AACpC,IAAAC,cAKO;AACP,kBAA2D;AAC3D,sBAAqC;;;ACbrC,mBAA+C;AAC/C,4BAAO;AASP,eAAsB,sBAAsB,KAAgC,WAAmB,SAAiB,MAAM;AAEpH,QAAM,aACJ,QAAQ,IAAI,sBAAsB;AACpC,MAAI,CAAC,YAAY;AACf,YAAQ,MAAM,wDAAwD;AACtE,WAAO;AAAA,EACT;AAEA,MAAI;AAEF,UAAM,SAAS,IAAI,IAAI,OAAO,OAAO,YAAY,IAAI,OAAO,QAAQ;AAGpE,UAAM,UAAU,MAAM,IAAI,OAAO,SAAS,WAAW,OAAO,OAAO;AACnE,YAAQ,IAAI,0BAA0B,OAAO,OAAO,EAAE;AACtD,YAAQ,IAAI,0BAA0B,IAAI,OAAO,YAAY,OAAO,CAAC,MAAM;AAG3E,UAAM,eAAe,IAAI,OAAO,WAAW,MAAM;AACjD,QAAI,UAAU,cAAc;AAC1B,cAAQ;AAAA,QACN,oEAAoE,IAAI,OAAO,YAAY,OAAO,CAAC;AAAA,MACrG;AACA,aAAO;AAAA,IACT;AAGA,YAAQ,IAAI,WAAW,MAAM,WAAW,SAAS,KAAK;AACtD,UAAM,KAAK,MAAM,OAAO,gBAAgB;AAAA,MACtC,IAAI;AAAA,MACJ,OAAO;AAAA,IACT,CAAC;AAED,YAAQ,IAAI,2BAA2B,GAAG,IAAI,EAAE;AAChD,YAAQ,IAAI,6BAA6B;AAGzC,UAAM,UAAU,MAAM,GAAG,KAAK;AAC9B,YAAQ,IAAI,kCAAkC,SAAS,WAAW,EAAE;AACpE,YAAQ,IAAI,qBAAqB,MAAM,WAAW,SAAS,EAAE;AAE7D,WAAO;AAAA,EACT,SAAS,OAAO;AACd,YAAQ,MAAM,wBAAwB,KAAK;AAC3C,WAAO;AAAA,EACT;AACF;AAQA,eAAsB,6BAA6B,KAAgC,eAAuB;AAExG,QAAM,gBAAgB,MAAM,IAAI,OAAO,SAAS,WAAW,aAAa;AACxE,UAAQ,IAAI,mBAAmB,IAAI,OAAO,YAAY,aAAa,CAAC,MAAM;AAE1E,MAAI,gBAAgB,IAAI,OAAO,WAAW,GAAG,GAAG;AAC9C,YAAQ,IAAI,8CAA8C,aAAa,KAAK;AAC5E,UAAM,UAAU,MAAM,sBAAsB,KAAK,aAAa;AAC9D,QAAI,SAAS;AACX,YAAM,aAAa,MAAM,IAAI,OAAO,SAAS,WAAW,aAAa;AACrE,cAAQ,IAAI,uBAAuB,IAAI,OAAO,YAAY,UAAU,CAAC,MAAM;AAAA,IAC7E,OAAO;AACL,cAAQ,MAAM,kBAAkB,aAAa,EAAE;AAAA,IACjD;AAAA,EACF;AACF;;;ACjFO,IAAM,wBAAwB;AAC9B,IAAM,+BAA+B;AACrC,IAAM,0BAA0B;AAEhC,IAAM,uBAAuB;AAC7B,IAAM,4BAA4B;AAClC,IAAM,gCAAgC;AACtC,IAAM,mBAAmB;;;ACPhC,IAAAC,gBAA+C;AAC/C,mBAAkB;AAClB,oBAAyB;AASzB,4BAMO;AACP,iBAAiD;AAIjD,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAG7C,QAAM,eAAe,KAAK,sBAAsB,8CAAwB,gBAAgB;AACxF,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,8CAAwB,KAAK,oBAAoB;AAGpG,QAAM,SAAS,MAAM,YAAY,WAAW,OAAO,OAAO;AAC1D,QAAM,OAAO,KAAK;AAGlB,QAAM,WAAW,MAAM,YAAY,OAAO;AAC1C,MAAI,CAAC,UAAU;AACb,UAAM,IAAI,MAAM,gCAAgC;AAAA,EAClD;AAEA,SAAO;AACT;AAEA,IAAM,gBAAgB,OAAO,QAAsD;AAEjF,QAAM,MAAM,MAAM,qBAAqB,KAAK,sCAAgB,KAAK,sCAAgB,QAAQ;AAGzF,QAAM,SAAS,MAAM,IAAI,OAAO;AAChC,MAAI,CAAC,QAAQ;AACX,aAAS,0BAA0B,CAAC;AACpC,UAAM,IAAI,MAAM,wBAAwB;AAAA,EAC1C;AAEA,SAAO;AACT;AAEA,IAAM,uBAAuB,OAAO,QAAmC;AACrE,QAAM,eAAe,KAAK,2BAA2B,6CAAuB,gBAAgB;AAC5F,QAAM,aAAa,MAAM,IAAI,OAAO,cAAc,6CAAuB,KAAK,yBAAyB;AAEvG,QAAM,mBAAmB,MAAM,WAAW,OAAO;AACjD,MAAI,CAAC,kBAAkB;AACrB,aAAS,iCAAiC,CAAC;AAC3C,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,SAAO;AACT;AAEA,IAAM,2BAA2B,OAAO,KAAgC,QAAkB;AACxF,QAAM,eAAe,KAAK,+BAA+B,iDAA2B,gBAAgB;AACpG,QAAM,iBAAiB,MAAM,IAAI,OAAO,cAAc,iDAA2B,KAAK,6BAA6B;AAGnH,QAAM,SAAS,MAAM,eAAe,WAAW,sBAAsB,MAAM,IAAI,WAAW,CAAC;AAC3F,QAAM,OAAO,KAAK;AAGlB,QAAM,uBAAuB,MAAM,eAAe,OAAO;AACzD,MAAI,CAAC,sBAAsB;AACzB,aAAS,qCAAqC,CAAC;AAC/C,UAAM,IAAI,MAAM,mCAAmC;AAAA,EACrD;AAEA,SAAO;AACT;AAEA,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,eAAe,KAAK,kBAAkB,sCAAgB,gBAAgB;AAC5E,QAAM,UAAU,MAAM,IAAI,OAAO,cAAc,sCAAgB,KAAK,gBAAgB;AACpF,QAAM,QAAQ,kBAAkB;AAChC,SAAO;AACT;AAIA,IAAM,uBAAuB,OAAO,QAAmC;AACrE,QAAM,mBAAmB,MAAM,IAAI,OAAO,UAAU,2CAAgC;AACpF,QAAM,IAAI,QAAQ,SAAS,KAAK,sBAAsB;AAAA,IACpD,iBAAiB;AAAA,IACjB,OAAO,IAAI,OAAO,WAAW,IAAI,EAAE,SAAS,EAAE;AAAA,EAChD,CAAC;AACH;AAIA,IAAM,oBAAoB,OAAO,aAAuB,QAAkB;AACxE,QAAM,WAAW,MAAM,YAAY,eAAe,MAAM,IAAI,WAAW,CAAC;AACxE,QAAM,SAAS,KAAK;AACtB;AAQO,IAAM,cAAc,OACzB,KACA,UAA2B;AAAA,EACzB,eAAe;AAAA,EACf,YAAY;AAAA,EACZ,QAAQ;AACV,MACG;AAEH,QAAM,YAAY,MAAM,oBAAoB,GAAG;AAC/C,MAAI,CAAC;AAAW;AAEhB,QAAM,kBAAkB,MAAM;AAC5B,QAAI,CAAC,QAAQ,QAAQ;AACnB,eAAS;AAAA,IACX;AAAA,EACF;AACA,QAAM,oBAAoB,CAAC,SAAiB,SAAS,MAAM;AACzD,QAAI,CAAC,QAAQ,QAAQ;AACnB,iBAAW,SAAS,MAAM;AAAA,IAC5B;AAAA,EACF;AACA,QAAM,uBAAuB,CAAC,cAAsB,YAAoB;AACtE,QAAI,CAAC,QAAQ,QAAQ;AACnB,oBAAc,cAAc,OAAO;AAAA,IACrC;AAAA,EACF;AACA,QAAM,oBAAoB,CAAC,SAAiB,SAAS,MAAM;AACzD,QAAI,CAAC,QAAQ,QAAQ;AACnB,iBAAW,SAAS,MAAM;AAAA,IAC5B;AAAA,EACF;AAGA,kBAAgB;AAChB,oBAAkB,aAAAC,QAAM,KAAK,sCAAsC,GAAG,CAAC;AACvE,kBAAgB;AAGhB,QAAM,cAAc,MAAM,sBAAsB,GAAG;AACnD,uBAAqB,mBAAmB,MAAM,YAAY,WAAW,CAAC;AAEtE,QAAM,MAAM,MAAM,cAAc,GAAG;AACnC,uBAAqB,WAAW,MAAM,IAAI,WAAW,CAAC;AAEtD,QAAM,kBAAkB,aAAa,GAAG;AACxC,oBAAkB,kCAAkC,CAAC;AAErD,QAAM,qBAAqB,GAAG;AAC9B,oBAAkB,sBAAsB,2CAAgC,YAAY,CAAC;AAErF,QAAM,0BAA0B,MAAM,IAAI,OAAO,SAAS,WAAW,2CAAgC;AACrG,oBAAkB,gBAAgB,wBAAwB,SAAS,CAAC,IAAI,CAAC;AAEzE,QAAM,aAAa,MAAM,qBAAqB,GAAG;AACjD,uBAAqB,kBAAkB,MAAM,WAAW,WAAW,CAAC;AAEpE,QAAM,iBAAiB,MAAM,yBAAyB,KAAK,GAAG;AAC9D,uBAAqB,sBAAsB,MAAM,eAAe,WAAW,CAAC;AAE5E,MAAI,QAAQ,eAAe;AACzB,sBAAkB,8BAA8B,CAAC;AACjD,UAAM,UAAU,MAAM,sBAAsB,GAAG;AAC/C,yBAAqB,WAAW,MAAM,QAAQ,WAAW,CAAC;AAAA,EAC5D;AAGA,kBAAgB;AAChB,oBAAkB,aAAAA,QAAM,KAAK,qDAAqD,GAAG,CAAC;AAGtF,MAAI,QAAQ,YAAY;AACtB,oBAAgB;AAChB;AAAA,MACE;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,kBAAgB;AAClB;AAIA,IAAM,iBAAiB,OAAO,KAAgC,SAAiB,aAAqB;AAClG,QAAM,IAAI,QAAQ,SAAS,KAAK,mBAAmB,CAAC,SAAS,QAAQ,CAAC;AACxE;AAEA,IAAM,uBAAuB,OAAO,KAAgC,KAAU,aAAqB;AACjG,QAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAE7C,QAAM,UAAU,IAAI,IAAI,OAAO,gBAAgB,KAAK,UAAU,MAAM;AACpE,QAAM,WAAW,MAAM,QAAQ;AAAA;AAAA,EAA6B;AAC5D,QAAM,SAAS,kBAAkB;AAEjC,SAAO;AACT;AAIA,IAAM,sBAAsB,OAAO,QAAmC;AACpE,QAAM,UAAU,IAAI,QAAQ;AAC5B,QAAM,YAAY,YAAY;AAC9B,MAAI,CAAC;AAAW,eAAW,wEAAwE,OAAO,IAAI,CAAC;AAC/G,SAAO;AACT;AAIA,IAAM,WAAW,MAAM;AACrB,UAAQ,IAAI,EAAE;AAChB;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,aAAAA,QAAM,MAAM,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC/D;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,aAAAA,QAAM,KAAK,aAAAA,QAAM,OAAO,GAAG,KAAK,OAAO,MAAM,CAAC,cAAS,CAAC,GAAG,OAAO;AAChF;AAEA,IAAM,WAAW,CAAC,SAAiB,SAAS,MAAM;AAChD,UAAQ,IAAI,aAAAA,QAAM,IAAI,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC7D;AAEA,IAAM,gBAAgB,CAAC,cAAsB,YAAoB;AAC/D,QAAM,aAAa,GAAG,YAAY,YAAY,OAAO,EAAE;AACvD,aAAW,GAAG,UAAU,IAAI,aAAAA,QAAM,KAAK,OAAO,CAAC,EAAE;AACnD;;;ACrPA,IAAAC,gBAAkB;AAClB,IAAAC,gBAA+C;AAE/C,IAAAC,yBAAwC;AAExC,IAAM,6BAA6B,OAAO,QAAmC;AAE3E,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,+CAAwB,KAAK,oBAAoB;AAEpG,SAAO;AACT;AAEA,IAAM,oBAAoB,OAAO,QAAmC;AAClE,QAAM,cAAc,MAAM,2BAA2B,GAAG;AACxD,SAAO,MAAM,YAAY,OAAO;AAClC;AAEA,IAAM,oBAAoB,OAAO,KAAgC,YAAqB;AACpF,QAAM,cAAc,MAAM,2BAA2B,GAAG;AACxD,QAAM,YAAY,UAAU,OAAO;AACrC;AAGA,IAAM,0BAA0B,CAAC,mBAA2B;AAC1D,UAAQ,IAAI,saAAwE;AACpF,UAAQ,IAAI,kCAAwB,cAAc,EAAE;AACpD,UAAQ,IAAI,saAAwE;AACtF;AAGA,IAAM,oBAAoB,MAAM;AAC9B,UAAQ,IAAI,saAAwE;AACtF;AAEO,IAAM,yBAAyB,OACpC,KACA,SACA,gBACG;AACH,MAAI;AACF,UAAM,mBAAmB,MAAM,kBAAkB,GAAG;AAEpD,UAAM,kBAAkB,KAAK,OAAO;AAGpC,QAAI,SAAS;AACX,8BAAwB,GAAG,cAAc,IAAI,cAAAC,QAAM,KAAK,WAAW,CAAC,YAAY,OAAO,EAAE;AAAA,IAC3F;AAGA,QAAI,CAAC,WAAW,kBAAkB;AAChC,wBAAkB;AAAA,IACpB;AAAA,EACF,SAAS,OAAO;AACd,YAAQ,IAAI,cAAAA,QAAM,IAAI,8BAA8B,GAAG,KAAK;AAAA,EAC9D;AACF;AAEO,IAAM,gBAAgB,OAC3B,KACA,aACA,YACG;AACH,QAAM,mBAAmB,MAAM,kBAAkB,GAAG;AAEpD,QAAM,kBAAkB,KAAK,IAAI;AACjC,0BAAwB,IAAI,cAAAA,QAAM,KAAK,WAAW,CAAC,SAAS;AAC5D,QAAM,QAAQ;AACd,oBAAkB;AAGlB,MAAI,CAAC,kBAAkB;AACrB,UAAM,kBAAkB,KAAK,KAAK;AAAA,EACpC;AACF;;;ACzEA,kBAAuB;AACvB,IAAAC,iBAAuB;AACvB,qCAA2C;AAE3C,IAAM,sBAAsB,OAAO,QAAgB,aAA6D;AAI9G,QAAM,WAAW,MAAM,SAAS,QAAQ,yBAAyB;AAGjE,QAAM,YAAY,SAAS,WAAW;AAGtC,MAAI,WAAW;AACb,YAAQ,IAAI,GAAG,MAAM,iCAAiC;AAAA,EACxD;AAEA,SAAO;AACT;AAEO,IAAM,oBAAoB,OAAO,UAA0D,WAAmB;AAEnH,MAAI,MAAM,oBAAoB,kBAAkB,MAAM,QAAQ;AAAG;AAGjE,QAAM,cAAc,IAAI,sBAAO;AAAA,IAC7B;AAAA,IACA,CAAC,sDAAsD;AAAA,IACvD;AAAA,EACF;AAGA,QAAM,YAAY,MAAM,YAAY,YAAY,MAAM;AAEtD,SAAO;AACT;AAEO,IAAM,0BAA0B,OACrC,UACA,WACG;AAEH,MAAI,MAAM,oBAAoB,wBAAwB,MAAM,QAAQ;AAAG;AAGvE,QAAM,cAAc,IAAI,sBAAO;AAAA,IAC7B;AAAA,IACA,CAAC,qDAAqD;AAAA,IACtD;AAAA,EACF;AAGA,QAAM,kBAAkB,MAAM,YAAY,cAAc,MAAM;AAE9D,SAAO;AACT;AAEO,IAAM,uBAAuB,OAClC,UACA,QACA,kBACG;AAEH,MAAI,MAAM,oBAAoB,qBAAqB,MAAM,QAAQ;AAAG;AAGpE,QAAM,kBAAkB,MAAM,wBAAwB,UAAU,MAAM;AACtE,0BAAO,eAAe,EAAE,MAAM,MAAM,0BAA0B;AAG9D,QAAM,YAAY,MAAM,kBAAkB,UAAU,MAAM;AAC1D,0BAAO,SAAS,EAAE,MAAM,eAAe,8BAA8B;AACvE;;;ALnDA,IAAAC,yBAMO;AACP,oBAAwB;AACxB,IAAAA,yBAMO;AAAA,IA0BP,4BAAa,CAAC,QAAQ,eAAe;AAEnC,MAAI,WAAW,YAAY,WAAW,SAAS,YAAY;AACzD;AAAA,EACF;AAGA,SAAO,SAAS,aAAa;AAAA,IAC3B,KAAK;AAAA,IACL,eAAe;AAAA,IACf,UAAU;AAAA,IACV,SAAS;AAAA,IACT,aAAa,CAAC;AAAA,IACd,KAAK;AAAA,IACL,UAAU;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAGA,MAAI,CAAC,WAAW,WAAW,aAAa,GAAG;AACzC,WAAO,SAAS,aAAa,IAAI;AAAA,MAC/B,KAAK,QAAQ,IAAI,mBAAmB;AAAA,MACpC,UAAU,QAAQ,IAAI,cAAc,CAAC,QAAQ,IAAI,WAAW,IAAI,CAAC;AAAA,MACjE,SAAS;AAAA,MACT,KAAK;AAAA,MACL,eAAe;AAAA,MACf,UAAU;AAAA,MACV,SAAS;AAAA,MACT,aAAa,CAAC;AAAA,IAChB;AAAA,EACF;AAGA,MAAI,CAAC,WAAW,WAAW,aAAa,GAAG;AACzC,WAAO,SAAS,aAAa,IAAI;AAAA,MAC/B,KAAK,QAAQ,IAAI,4BAA4B;AAAA,MAC7C,UAAU,QAAQ,IAAI,cAAc,CAAC,QAAQ,IAAI,WAAW,IAAI,CAAC;AAAA,MACjE,SAAS;AAAA,MACT,KAAK;AAAA,MACL,eAAe;AAAA,MACf,UAAU;AAAA,MACV,SAAS;AAAA,MACT,aAAa,CAAC;AAAA,IAChB;AAAA,EACF;AAGA,SAAO,WAAW;AAAA,IAChB,UAAU,WAAW,UAAU,YAAY;AAAA,IAC3C,YAAY,WAAW,UAAU,cAAc;AAAA,EACjD;AACF,CAAC;AAAA,IAMD,oBAAK,uBAAuB,iCAAiC,EAC1D,iBAAiB,WAAW,mBAAmB,QAAW,oBAAM,MAAM,EACtE,UAAU,OAAO,EAAE,QAAQ,GAAkB,QAAQ;AACpD,QAAM,EAAE,QAAQ,IAAI;AACpB,QAAM,EAAE,MAAM,YAAY,IAAI;AAE9B,MAAI,gBAAgB,cAAc;AAChC,YAAQ,KAAK,cAAAC,QAAM,OAAO,mDAAmD,CAAC;AAC9E;AAAA,EACF;AAEA,MAAI,CAAC,SAAS;AACZ,YAAQ,KAAK,cAAAA,QAAM,IAAI,+BAA+B,CAAC;AACvD;AAAA,EACF;AAEA,UAAQ,KAAK,cAAAA,QAAM,MAAM,iCAAiC,OAAO,EAAE,CAAC;AAEpE,MAAI;AACF,UAAM,sBAAsB,KAAK,OAAO;AAAA,EAC1C,SAAS,GAAG;AACV,YAAQ,KAAK,cAAAA,QAAM,IAAI,2CAA2C,OAAO,KAAK,CAAC,EAAE,CAAC;AAAA,EACpF;AACF,CAAC;AAAA,IAIH,oBAAK,yBAAyB,mDAAmD,EAC9E,iBAAiB,iBAAiB,kCAAkC,MAAM,oBAAM,OAAO,EACvF,iBAAiB,UAAU,8BAA8B,OAAO,oBAAM,OAAO,EAC7E,UAAU,OAAO,EAAE,eAAe,OAAO,GAAoB,QAAQ;AACpE,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe,iBAAiB;AAAA,IAChC,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,IAC9C,QAAQ,UAAU;AAAA,EACpB,CAAC;AACH,CAAC;AAAA,IAEH,oBAAK,6BAAW,kCAAkC,EAAE,UAAU,OAAO,CAAC,GAAG,KAAK,aAAa;AACzF,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe;AAAA,IACf,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,EAChD,CAAC;AACD,SAAO,SAAS;AAClB,CAAC;AAAA,IAED,oBAAK,6BAAW,kCAAkC,EAAE,UAAU,OAAO,CAAC,GAAG,KAAK,aAAa;AACzF,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe;AAAA,IACf,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,EAChD,CAAC;AACD,SAAO,SAAS;AAClB,CAAC;AAAA,IAID,oBAAK,8BAA8B,0CAA0C,EAC1E,SAAS,UAAU,6BAA6B,OAAO,oBAAM,OAAO,EACpE,UAAU,OAAO,EAAE,OAAO,GAAG,QAAQ;AACpC,QAAM,uBAAuB,KAAK,MAAM;AAC1C,CAAC;AAAA,IA8JH,iCAAkB,CAAC,QAAQ;AACzB,MAAI,WAAW;AAAA,IACb,sBAAsB,OAAO,WAAgC;AAG3D,YAAM,cAAc,MAAM,IAAI,OAAO,sBAAsB,4CAAgC;AAC3F,YAAM,EAAE,cAAc,gBAAgB,IAAI,UAAM,sCAAqB,WAAW;AAGhF,YAAM,4BAA4B;AAAA,QAChC,aAAa;AAAA,QACb,GAAG;AAAA,QACH,WAAW;AAAA,UACT,GAAG,OAAO;AAAA,UACV;AAAA,QACF;AAAA,MACF;AAEA,iBAAO,kCAAqB,yBAAyB;AAAA,IACvD;AAAA,IACA,sBAAsB,CAAC,WAA2B;AAChD,iBAAO,kCAAqB,MAAM;AAAA,IACpC;AAAA,IACA,sBAAsB,OAAO,WAAgC;AAC3D,iBAAO,sCAAqB,MAAM;AAAA,IACpC;AAAA,IACA,0BAA0B,OAAO,QAAwB,WAAgC;AACvF,YAAM,EAAE,cAAc,aAAa,IAAI,UAAM,sCAAqB,MAAM;AACxE,aAAO,OAAO,QAAQ,cAAc,YAAY;AAAA,IAClD;AAAA,IACA,uCAAuC,OAAO,WAAiC;AAE7E,UAAI,CAAC,QAAQ;AACX,SAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAAA,MACzC;AAGA,YAAM,SAAS,MAAM,IAAI,SAAS,qBAAqB;AAAA,QACrD,aAAa;AAAA,QACb,iBAAiB,CAAC,qBAAO;AAAA,MAC3B,CAAC;AAGD,YAAM,SAAS,IAAI,SAAS,qBAAqB,MAAM;AAGvD,YAAM,IAAI,SAAS,yBAAyB,QAAQ,MAAM;AAG1D,YAAM,OAAO,QAAQ,WAAW;AAAA,QAC9B,QAAQ,OAAO;AAAA,MACjB,CAAC;AAGD,aAAO;AAAA,IACT;AAAA,IACA,OAAO;AAAA,MACL,UAAU,OAAO,aAAqB,YAAiC;AACrE,eAAO,cAAc,KAAK,aAAa,OAAO;AAAA,MAChD;AAAA,MACA,YAAY,OAAO,gBAAyB;AAC1C,eAAO,uBAAuB,KAAK,MAAM,WAAW;AAAA,MACtD;AAAA,MACA,aAAa,YAAY;AACvB,eAAO,uBAAuB,KAAK,KAAK;AAAA,MAC1C;AAAA,MACA,aAAa,OAAO,YAA6B;AAC/C,eAAO,YAAY,KAAK,OAAO;AAAA,MACjC;AAAA,MACA,cAAc,OAAO,WAAmB;AACtC,cAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAC7C,eAAO,kBAAkB,OAAO,UAAU,MAAM;AAAA,MAClD;AAAA,MACA,iBAAiB,OAAO,QAAgB,kBAA0B;AAChE,cAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAC7C,eAAO,qBAAqB,OAAO,UAAU,QAAQ,aAAa;AAAA,MACpE;AAAA,MACA,oBAAoB,YAAY;AAC9B,eAAO,MAAM,IAAI,OAAO,cAAc,+CAAwB,KAAK,+CAAwB,YAAY;AAAA,MACzG;AAAA,MACA,YAAY,YAAY;AACtB,cAAM,KAAK,MAAM,IAAI,OAAO,cAAc,+CAAwB,KAAK,+CAAwB,YAAY;AAC3G,cAAM,aAAa,MAAM,GAAG,IAAI;AAChC,eAAO,MAAM,IAAI,OAAO,cAAc,uCAAgB,KAAK,UAAU;AAAA,MACvE;AAAA,MACA,uBAAuB,YAAY;AACjC,eAAO,MAAM,IAAI,OAAO,cAAc,kDAA2B,KAAK,kDAA2B,YAAY;AAAA,MAC/G;AAAA,MACA,mBAAmB,YAAY;AAC7B,eAAO,MAAM,IAAI,OAAO,cAAc,8CAAuB,KAAK,8CAAuB,YAAY;AAAA,MACvG;AAAA,MACA,YAAY,YAAY;AACtB,eAAO,MAAM,IAAI,OAAO,cAAc,uCAAgB,KAAK,uCAAgB,YAAY;AAAA,MACzF;AAAA,IACF;AAAA,EACF;AACF,CAAC;","names":["import_chalk","import_sdk","import_types","chalk","import_chalk","import_types","import_mock_contracts","chalk","import_ethers","import_mock_contracts","chalk"]}
package/dist/index.mjs CHANGED
@@ -4,7 +4,9 @@ import "viem";
4
4
  import { extendConfig, extendEnvironment, task, types } from "hardhat/config";
5
5
  import { TASK_TEST, TASK_NODE } from "hardhat/builtin-tasks/task-names";
6
6
  import "@nomicfoundation/hardhat-ethers/signers";
7
- import "@cofhe/sdk";
7
+ import {
8
+ MOCKS_ZK_VERIFIER_SIGNER_ADDRESS as MOCKS_ZK_VERIFIER_SIGNER_ADDRESS2
9
+ } from "@cofhe/sdk";
8
10
  import { createCofhesdkClient, createCofhesdkConfig } from "@cofhe/sdk/node";
9
11
  import { HardhatSignerAdapter } from "@cofhe/sdk/adapters";
10
12
 
@@ -65,11 +67,9 @@ var TASK_COFHE_USE_FAUCET = "task:cofhe:usefaucet";
65
67
  var TASK_COFHE_MOCKS_SET_LOG_OPS = "task:cofhe-mocks:setlogops";
66
68
  var TASK_COFHE_MOCKS_DEPLOY = "task:cofhe-mocks:deploy";
67
69
  var TASK_MANAGER_ADDRESS = "0xeA30c4B8b44078Bbf8a6ef5b9f1eC1626C7848D9";
68
- var MOCKS_ACL_ADDRESS = "0x0000000000000000000000000000000000000400";
69
70
  var MOCKS_ZK_VERIFIER_ADDRESS = "0x0000000000000000000000000000000000000100";
70
71
  var MOCKS_QUERY_DECRYPTER_ADDRESS = "0x0000000000000000000000000000000000000200";
71
72
  var TEST_BED_ADDRESS = "0x0000000000000000000000000000000000000300";
72
- var MOCKS_ZK_VERIFIER_SIGNER_ADDRESS = "0x6E12D8C87503D4287c294f2Fdef96ACd9DFf6bd2";
73
73
 
74
74
  // src/deploy.ts
75
75
  import "hardhat/types";
@@ -82,6 +82,7 @@ import {
82
82
  MockQueryDecrypterArtifact,
83
83
  TestBedArtifact
84
84
  } from "@cofhe/mock-contracts";
85
+ import { MOCKS_ZK_VERIFIER_SIGNER_ADDRESS } from "@cofhe/sdk";
85
86
  var deployMockTaskManager = async (hre) => {
86
87
  const [signer] = await hre.ethers.getSigners();
87
88
  await hardhatSetCode(hre, TASK_MANAGER_ADDRESS, MockTaskManagerArtifact.deployedBytecode);
@@ -95,8 +96,7 @@ var deployMockTaskManager = async (hre) => {
95
96
  return taskManager;
96
97
  };
97
98
  var deployMockACL = async (hre) => {
98
- await hardhatSetCode(hre, MOCKS_ACL_ADDRESS, MockACLArtifact.deployedBytecode);
99
- const acl = await hre.ethers.getContractAt(MockACLArtifact.abi, MOCKS_ACL_ADDRESS);
99
+ const acl = await ethersDeployContract(hre, MockACLArtifact.abi, MockACLArtifact.bytecode);
100
100
  const exists = await acl.exists();
101
101
  if (!exists) {
102
102
  logError("MockACL does not exist", 2);
@@ -174,8 +174,6 @@ var deployMocks = async (hre, options = {
174
174
  logEmptyIfNoisy();
175
175
  logSuccessIfNoisy(chalk.bold("cofhe-hardhat-plugin :: deploy mocks"), 0);
176
176
  logEmptyIfNoisy();
177
- logEmptyIfNoisy();
178
- logSuccessIfNoisy("Mock contracts compiled", 1);
179
177
  const taskManager = await deployMockTaskManager(hre);
180
178
  logDeploymentIfNoisy("MockTaskManager", await taskManager.getAddress());
181
179
  const acl = await deployMockACL(hre);
@@ -209,6 +207,15 @@ var deployMocks = async (hre, options = {
209
207
  var hardhatSetCode = async (hre, address, bytecode) => {
210
208
  await hre.network.provider.send("hardhat_setCode", [address, bytecode]);
211
209
  };
210
+ var ethersDeployContract = async (hre, abi, bytecode) => {
211
+ const [signer] = await hre.ethers.getSigners();
212
+ const factory = new hre.ethers.ContractFactory(abi, bytecode, signer);
213
+ const contract = await factory.deploy(
214
+ /* constructor args */
215
+ );
216
+ await contract.waitForDeployment();
217
+ return contract;
218
+ };
212
219
  var checkNetworkAndSkip = async (hre) => {
213
220
  const network = hre.network.name;
214
221
  const isHardhat = network === "hardhat";
@@ -325,28 +332,6 @@ var mock_expectPlaintext = async (provider, ctHash, expectedValue) => {
325
332
  expect(plaintext).equal(expectedValue, "Plaintext value is incorrect");
326
333
  };
327
334
 
328
- // src/expectResultUtils.ts
329
- import "@cofhe/sdk";
330
- import { expect as expect2 } from "chai";
331
- var expectResultError = (result, errorPartial) => {
332
- expect2(result.success).to.eq(false, "Result should be an error");
333
- expect2(result.error).to.include(errorPartial, `Error should contain error partial: ${errorPartial}`);
334
- };
335
- var expectResultSuccess = (result) => {
336
- expect2(result.success).to.eq(true, "Result should be a success");
337
- return result.data;
338
- };
339
- var expectResultValue = (result, value) => {
340
- expect2(result.success).to.eq(true, "Result should be a success");
341
- expect2(result.data).to.eq(value, `Result should have the expected value ${value}`);
342
- return result.data;
343
- };
344
- var expectResultPartialValue = (result, partial) => {
345
- expect2(result.success).to.eq(true, "Result should be a success");
346
- expect2(result.data).to.include(partial, `Result should have the expected partial ${partial}`);
347
- return result.data;
348
- };
349
-
350
335
  // src/index.ts
351
336
  import {
352
337
  MockACLArtifact as MockACLArtifact2,
@@ -355,6 +340,14 @@ import {
355
340
  MockZkVerifierArtifact as MockZkVerifierArtifact2,
356
341
  TestBedArtifact as TestBedArtifact2
357
342
  } from "@cofhe/mock-contracts";
343
+ import { hardhat } from "@cofhe/sdk/chains";
344
+ import {
345
+ MockACLArtifact as MockACLArtifact3,
346
+ MockQueryDecrypterArtifact as MockQueryDecrypterArtifact3,
347
+ MockTaskManagerArtifact as MockTaskManagerArtifact4,
348
+ MockZkVerifierArtifact as MockZkVerifierArtifact3,
349
+ TestBedArtifact as TestBedArtifact3
350
+ } from "@cofhe/mock-contracts";
358
351
  extendConfig((config, userConfig) => {
359
352
  if (userConfig.networks && userConfig.networks.localcofhe) {
360
353
  return;
@@ -447,9 +440,10 @@ task(TASK_COFHE_MOCKS_SET_LOG_OPS, "Set logging for the Mock CoFHE contracts").a
447
440
  extendEnvironment((hre) => {
448
441
  hre.cofhesdk = {
449
442
  createCofhesdkConfig: async (config) => {
450
- const zkvHhSigner = await hre.ethers.getImpersonatedSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
443
+ const zkvHhSigner = await hre.ethers.getImpersonatedSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS2);
451
444
  const { walletClient: zkvWalletClient } = await HardhatSignerAdapter(zkvHhSigner);
452
445
  const configWithZkvWalletClient = {
446
+ environment: "hardhat",
453
447
  ...config,
454
448
  _internal: {
455
449
  ...config._internal,
@@ -464,21 +458,24 @@ extendEnvironment((hre) => {
464
458
  hardhatSignerAdapter: async (signer) => {
465
459
  return HardhatSignerAdapter(signer);
466
460
  },
467
- expectResultSuccess: async (result) => {
468
- const awaitedResult = await result;
469
- return expectResultSuccess(awaitedResult);
470
- },
471
- expectResultError: async (result, errorPartial) => {
472
- const awaitedResult = await result;
473
- return expectResultError(awaitedResult, errorPartial);
461
+ connectWithHardhatSigner: async (client, signer) => {
462
+ const { publicClient, walletClient } = await HardhatSignerAdapter(signer);
463
+ return client.connect(publicClient, walletClient);
474
464
  },
475
- expectResultValue: async (result, value) => {
476
- const awaitedResult = await result;
477
- return expectResultValue(awaitedResult, value);
478
- },
479
- expectResultPartialValue: async (result, partial) => {
480
- const awaitedResult = await result;
481
- return expectResultPartialValue(awaitedResult, partial);
465
+ createBatteriesIncludedCofhesdkClient: async (signer) => {
466
+ if (!signer) {
467
+ [signer] = await hre.ethers.getSigners();
468
+ }
469
+ const config = await hre.cofhesdk.createCofhesdkConfig({
470
+ environment: "hardhat",
471
+ supportedChains: [hardhat]
472
+ });
473
+ const client = hre.cofhesdk.createCofhesdkClient(config);
474
+ await hre.cofhesdk.connectWithHardhatSigner(client, signer);
475
+ await client.permits.createSelf({
476
+ issuer: signer.address
477
+ });
478
+ return client;
482
479
  },
483
480
  mocks: {
484
481
  withLogs: async (closureName, closure) => {
@@ -500,21 +497,34 @@ extendEnvironment((hre) => {
500
497
  expectPlaintext: async (ctHash, expectedValue) => {
501
498
  const [signer] = await hre.ethers.getSigners();
502
499
  return mock_expectPlaintext(signer.provider, ctHash, expectedValue);
500
+ },
501
+ getMockTaskManager: async () => {
502
+ return await hre.ethers.getContractAt(MockTaskManagerArtifact3.abi, MockTaskManagerArtifact3.fixedAddress);
503
+ },
504
+ getMockACL: async () => {
505
+ const tm = await hre.ethers.getContractAt(MockTaskManagerArtifact3.abi, MockTaskManagerArtifact3.fixedAddress);
506
+ const aclAddress = await tm.acl();
507
+ return await hre.ethers.getContractAt(MockACLArtifact2.abi, aclAddress);
508
+ },
509
+ getMockQueryDecrypter: async () => {
510
+ return await hre.ethers.getContractAt(MockQueryDecrypterArtifact2.abi, MockQueryDecrypterArtifact2.fixedAddress);
511
+ },
512
+ getMockZkVerifier: async () => {
513
+ return await hre.ethers.getContractAt(MockZkVerifierArtifact2.abi, MockZkVerifierArtifact2.fixedAddress);
514
+ },
515
+ getTestBed: async () => {
516
+ return await hre.ethers.getContractAt(TestBedArtifact2.abi, TestBedArtifact2.fixedAddress);
503
517
  }
504
518
  }
505
519
  };
506
520
  });
507
521
  export {
508
- MockACLArtifact2 as MockACLArtifact,
509
- MockQueryDecrypterArtifact2 as MockQueryDecrypterArtifact,
510
- MockTaskManagerArtifact3 as MockTaskManagerArtifact,
511
- MockZkVerifierArtifact2 as MockZkVerifierArtifact,
512
- TestBedArtifact2 as TestBedArtifact,
522
+ MockACLArtifact3 as MockACLArtifact,
523
+ MockQueryDecrypterArtifact3 as MockQueryDecrypterArtifact,
524
+ MockTaskManagerArtifact4 as MockTaskManagerArtifact,
525
+ MockZkVerifierArtifact3 as MockZkVerifierArtifact,
526
+ TestBedArtifact3 as TestBedArtifact,
513
527
  deployMocks,
514
- expectResultError,
515
- expectResultPartialValue,
516
- expectResultSuccess,
517
- expectResultValue,
518
528
  localcofheFundAccount,
519
529
  localcofheFundWalletIfNeeded,
520
530
  mock_expectPlaintext,