@cofhe/hardhat-plugin 0.1.0 → 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 (15) hide show
  1. package/CHANGELOG.md +71 -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 +14 -7
  10. package/src/consts.ts +8 -0
  11. package/src/deploy.ts +246 -0
  12. package/src/fund.ts +82 -0
  13. package/src/index.ts +439 -0
  14. package/src/logging.ts +75 -0
  15. package/src/utils.ts +75 -0
package/dist/index.mjs.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":";AAGA,OAAOA,YAAW;AAClB,OAAqD;AACrD,SAAS,cAAc,mBAAmB,MAAM,aAAa;AAC7D,SAAS,WAAW,iBAAiB;AACrC,OAAoC;AACpC,OAAgG;AAChG,SAAS,sBAAsB,4BAA4B;AAC3D,SAAS,4BAA4B;;;ACTrC,OAA+C;AAC/C,OAAO;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,OAA+C;AAC/C,OAAO,WAAW;AAClB,OAAyB;AAWzB;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AAIP,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAG7C,QAAM,eAAe,KAAK,sBAAsB,wBAAwB,gBAAgB;AACxF,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,wBAAwB,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,gBAAgB,gBAAgB;AAC7E,QAAM,MAAM,MAAM,IAAI,OAAO,cAAc,gBAAgB,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,uBAAuB,gBAAgB;AAC5F,QAAM,aAAa,MAAM,IAAI,OAAO,cAAc,uBAAuB,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,2BAA2B,gBAAgB;AACpG,QAAM,iBAAiB,MAAM,IAAI,OAAO,cAAc,2BAA2B,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,gBAAgB,gBAAgB;AAC5E,QAAM,UAAU,MAAM,IAAI,OAAO,cAAc,gBAAgB,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,MAAM,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,MAAM,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,MAAM,MAAM,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC/D;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,MAAM,KAAK,MAAM,OAAO,GAAG,KAAK,OAAO,MAAM,CAAC,cAAS,CAAC,GAAG,OAAO;AAChF;AAEA,IAAM,WAAW,CAAC,SAAiB,SAAS,MAAM;AAChD,UAAQ,IAAI,MAAM,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,MAAM,KAAK,OAAO,CAAC,EAAE;AACnD;;;ACjPA,OAAOC,YAAW;AAClB,OAA+C;AAE/C,SAAS,2BAAAC,gCAA+B;AAExC,IAAM,6BAA6B,OAAO,QAAmC;AAE3E,QAAM,cAAc,MAAM,IAAI,OAAO,cAAcA,yBAAwB,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,IAAIC,OAAM,KAAK,WAAW,CAAC,YAAY,OAAO,EAAE;AAAA,IAC3F;AAGA,QAAI,CAAC,WAAW,kBAAkB;AAChC,wBAAkB;AAAA,IACpB;AAAA,EACF,SAAS,OAAO;AACd,YAAQ,IAAIA,OAAM,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,IAAIA,OAAM,KAAK,WAAW,CAAC,SAAS;AAC5D,QAAM,QAAQ;AACd,oBAAkB;AAGlB,MAAI,CAAC,kBAAkB;AACrB,UAAM,kBAAkB,KAAK,KAAK;AAAA,EACpC;AACF;;;ACzEA,SAAS,cAAc;AACvB,SAAS,cAAc;AACvB,OAA2C;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,OAAO;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,OAAO;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,SAAO,eAAe,EAAE,MAAM,MAAM,0BAA0B;AAG9D,QAAM,YAAY,MAAM,kBAAkB,UAAU,MAAM;AAC1D,SAAO,SAAS,EAAE,MAAM,eAAe,8BAA8B;AACvE;;;AC1EA,OAA4B;AAC5B,SAAS,UAAAC,eAAc;AAEhB,IAAM,oBAAoB,CAAI,QAAmB,iBAAyB;AAC/E,EAAAA,QAAO,OAAO,OAAO,EAAE,GAAG,GAAG,OAAO,2BAA2B;AAC/D,EAAAA,QAAO,OAAO,KAAK,EAAE,GAAG,QAAQ,cAAc,uCAAuC,YAAY,EAAE;AACrG;AAEO,IAAM,sBAAsB,CAAI,WAAyB;AAC9D,EAAAA,QAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,SAAO,OAAO;AAChB;AAEO,IAAM,oBAAoB,CAAI,QAAmB,UAAgB;AACtE,EAAAA,QAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,EAAAA,QAAO,OAAO,IAAI,EAAE,GAAG,GAAG,OAAO,yCAAyC,KAAK,EAAE;AACjF,SAAO,OAAO;AAChB;AAEO,IAAM,2BAA2B,CAAI,QAAmB,YAA2B;AACxF,EAAAA,QAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,EAAAA,QAAO,OAAO,IAAI,EAAE,GAAG,QAAQ,SAAS,2CAA2C,OAAO,EAAE;AAC5F,SAAO,OAAO;AAChB;;;ANMA;AAAA,EACE,mBAAAC;AAAA,EACA,8BAAAC;AAAA,EACA,2BAAAC;AAAA,EACA,0BAAAC;AAAA,EACA,mBAAAC;AAAA,OACK;AA0BP,aAAa,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;AAMD,KAAK,uBAAuB,iCAAiC,EAC1D,iBAAiB,WAAW,mBAAmB,QAAW,MAAM,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,KAAKC,OAAM,OAAO,mDAAmD,CAAC;AAC9E;AAAA,EACF;AAEA,MAAI,CAAC,SAAS;AACZ,YAAQ,KAAKA,OAAM,IAAI,+BAA+B,CAAC;AACvD;AAAA,EACF;AAEA,UAAQ,KAAKA,OAAM,MAAM,iCAAiC,OAAO,EAAE,CAAC;AAEpE,MAAI;AACF,UAAM,sBAAsB,KAAK,OAAO;AAAA,EAC1C,SAAS,GAAG;AACV,YAAQ,KAAKA,OAAM,IAAI,2CAA2C,OAAO,KAAK,CAAC,EAAE,CAAC;AAAA,EACpF;AACF,CAAC;AAIH,KAAK,yBAAyB,mDAAmD,EAC9E,iBAAiB,iBAAiB,kCAAkC,MAAM,MAAM,OAAO,EACvF,iBAAiB,UAAU,8BAA8B,OAAO,MAAM,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;AAEH,KAAK,WAAW,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;AAED,KAAK,WAAW,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;AAID,KAAK,8BAA8B,0CAA0C,EAC1E,SAAS,UAAU,6BAA6B,OAAO,MAAM,OAAO,EACpE,UAAU,OAAO,EAAE,OAAO,GAAG,QAAQ;AACpC,QAAM,uBAAuB,KAAK,MAAM;AAC1C,CAAC;AA+IH,kBAAkB,CAAC,QAAQ;AACzB,MAAI,WAAW;AAAA,IACb,sBAAsB,OAAO,WAAgC;AAG3D,YAAM,cAAc,MAAM,IAAI,OAAO,sBAAsB,gCAAgC;AAC3F,YAAM,EAAE,cAAc,gBAAgB,IAAI,MAAM,qBAAqB,WAAW;AAGhF,YAAM,4BAA4B;AAAA,QAChC,GAAG;AAAA,QACH,WAAW;AAAA,UACT,GAAG,OAAO;AAAA,UACV;AAAA,QACF;AAAA,MACF;AAEA,aAAO,qBAAqB,yBAAyB;AAAA,IACvD;AAAA,IACA,sBAAsB,CAAC,WAA2B;AAChD,aAAO,qBAAqB,MAAM;AAAA,IACpC;AAAA,IACA,sBAAsB,OAAO,WAAgC;AAC3D,aAAO,qBAAqB,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":["chalk","chalk","MockTaskManagerArtifact","chalk","expect","MockACLArtifact","MockQueryDecrypterArtifact","MockTaskManagerArtifact","MockZkVerifierArtifact","TestBedArtifact","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":";AAEA,OAAOA,YAAW;AAClB,OAAqD;AACrD,SAAS,cAAc,mBAAmB,MAAM,aAAa;AAC7D,SAAS,WAAW,iBAAiB;AACrC,OAAoC;AACpC;AAAA,EACE,oCAAAC;AAAA,OAIK;AACP,SAAS,sBAAsB,4BAA4B;AAC3D,SAAS,4BAA4B;;;ACbrC,OAA+C;AAC/C,OAAO;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,OAA+C;AAC/C,OAAO,WAAW;AAClB,OAAyB;AASzB;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AACP,SAAS,wCAAwC;AAIjD,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAG7C,QAAM,eAAe,KAAK,sBAAsB,wBAAwB,gBAAgB;AACxF,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,wBAAwB,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,gBAAgB,KAAK,gBAAgB,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,uBAAuB,gBAAgB;AAC5F,QAAM,aAAa,MAAM,IAAI,OAAO,cAAc,uBAAuB,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,2BAA2B,gBAAgB;AACpG,QAAM,iBAAiB,MAAM,IAAI,OAAO,cAAc,2BAA2B,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,gBAAgB,gBAAgB;AAC5E,QAAM,UAAU,MAAM,IAAI,OAAO,cAAc,gBAAgB,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,MAAM,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,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,MAAM,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,MAAM,MAAM,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC/D;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,MAAM,KAAK,MAAM,OAAO,GAAG,KAAK,OAAO,MAAM,CAAC,cAAS,CAAC,GAAG,OAAO;AAChF;AAEA,IAAM,WAAW,CAAC,SAAiB,SAAS,MAAM;AAChD,UAAQ,IAAI,MAAM,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,MAAM,KAAK,OAAO,CAAC,EAAE;AACnD;;;ACrPA,OAAOC,YAAW;AAClB,OAA+C;AAE/C,SAAS,2BAAAC,gCAA+B;AAExC,IAAM,6BAA6B,OAAO,QAAmC;AAE3E,QAAM,cAAc,MAAM,IAAI,OAAO,cAAcA,yBAAwB,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,IAAIC,OAAM,KAAK,WAAW,CAAC,YAAY,OAAO,EAAE;AAAA,IAC3F;AAGA,QAAI,CAAC,WAAW,kBAAkB;AAChC,wBAAkB;AAAA,IACpB;AAAA,EACF,SAAS,OAAO;AACd,YAAQ,IAAIA,OAAM,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,IAAIA,OAAM,KAAK,WAAW,CAAC,SAAS;AAC5D,QAAM,QAAQ;AACd,oBAAkB;AAGlB,MAAI,CAAC,kBAAkB;AACrB,UAAM,kBAAkB,KAAK,KAAK;AAAA,EACpC;AACF;;;ACzEA,SAAS,cAAc;AACvB,SAAS,cAAc;AACvB,OAA2C;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,OAAO;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,OAAO;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,SAAO,eAAe,EAAE,MAAM,MAAM,0BAA0B;AAG9D,QAAM,YAAY,MAAM,kBAAkB,UAAU,MAAM;AAC1D,SAAO,SAAS,EAAE,MAAM,eAAe,8BAA8B;AACvE;;;ALnDA;AAAA,EACE,mBAAAC;AAAA,EACA,8BAAAC;AAAA,EACA,2BAAAC;AAAA,EACA,0BAAAC;AAAA,EACA,mBAAAC;AAAA,OACK;AACP,SAAS,eAAe;AACxB;AAAA,EACE,mBAAAJ;AAAA,EACA,8BAAAC;AAAA,EACA,2BAAAC;AAAA,EACA,0BAAAC;AAAA,EACA,mBAAAC;AAAA,OACK;AA0BP,aAAa,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;AAMD,KAAK,uBAAuB,iCAAiC,EAC1D,iBAAiB,WAAW,mBAAmB,QAAW,MAAM,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,KAAKC,OAAM,OAAO,mDAAmD,CAAC;AAC9E;AAAA,EACF;AAEA,MAAI,CAAC,SAAS;AACZ,YAAQ,KAAKA,OAAM,IAAI,+BAA+B,CAAC;AACvD;AAAA,EACF;AAEA,UAAQ,KAAKA,OAAM,MAAM,iCAAiC,OAAO,EAAE,CAAC;AAEpE,MAAI;AACF,UAAM,sBAAsB,KAAK,OAAO;AAAA,EAC1C,SAAS,GAAG;AACV,YAAQ,KAAKA,OAAM,IAAI,2CAA2C,OAAO,KAAK,CAAC,EAAE,CAAC;AAAA,EACpF;AACF,CAAC;AAIH,KAAK,yBAAyB,mDAAmD,EAC9E,iBAAiB,iBAAiB,kCAAkC,MAAM,MAAM,OAAO,EACvF,iBAAiB,UAAU,8BAA8B,OAAO,MAAM,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;AAEH,KAAK,WAAW,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;AAED,KAAK,WAAW,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;AAID,KAAK,8BAA8B,0CAA0C,EAC1E,SAAS,UAAU,6BAA6B,OAAO,MAAM,OAAO,EACpE,UAAU,OAAO,EAAE,OAAO,GAAG,QAAQ;AACpC,QAAM,uBAAuB,KAAK,MAAM;AAC1C,CAAC;AA8JH,kBAAkB,CAAC,QAAQ;AACzB,MAAI,WAAW;AAAA,IACb,sBAAsB,OAAO,WAAgC;AAG3D,YAAM,cAAc,MAAM,IAAI,OAAO,sBAAsBC,iCAAgC;AAC3F,YAAM,EAAE,cAAc,gBAAgB,IAAI,MAAM,qBAAqB,WAAW;AAGhF,YAAM,4BAA4B;AAAA,QAChC,aAAa;AAAA,QACb,GAAG;AAAA,QACH,WAAW;AAAA,UACT,GAAG,OAAO;AAAA,UACV;AAAA,QACF;AAAA,MACF;AAEA,aAAO,qBAAqB,yBAAyB;AAAA,IACvD;AAAA,IACA,sBAAsB,CAAC,WAA2B;AAChD,aAAO,qBAAqB,MAAM;AAAA,IACpC;AAAA,IACA,sBAAsB,OAAO,WAAgC;AAC3D,aAAO,qBAAqB,MAAM;AAAA,IACpC;AAAA,IACA,0BAA0B,OAAO,QAAwB,WAAgC;AACvF,YAAM,EAAE,cAAc,aAAa,IAAI,MAAM,qBAAqB,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,OAAO;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,cAAcJ,yBAAwB,KAAKA,yBAAwB,YAAY;AAAA,MACzG;AAAA,MACA,YAAY,YAAY;AACtB,cAAM,KAAK,MAAM,IAAI,OAAO,cAAcA,yBAAwB,KAAKA,yBAAwB,YAAY;AAC3G,cAAM,aAAa,MAAM,GAAG,IAAI;AAChC,eAAO,MAAM,IAAI,OAAO,cAAcF,iBAAgB,KAAK,UAAU;AAAA,MACvE;AAAA,MACA,uBAAuB,YAAY;AACjC,eAAO,MAAM,IAAI,OAAO,cAAcC,4BAA2B,KAAKA,4BAA2B,YAAY;AAAA,MAC/G;AAAA,MACA,mBAAmB,YAAY;AAC7B,eAAO,MAAM,IAAI,OAAO,cAAcE,wBAAuB,KAAKA,wBAAuB,YAAY;AAAA,MACvG;AAAA,MACA,YAAY,YAAY;AACtB,eAAO,MAAM,IAAI,OAAO,cAAcC,iBAAgB,KAAKA,iBAAgB,YAAY;AAAA,MACzF;AAAA,IACF;AAAA,EACF;AACF,CAAC;","names":["chalk","MOCKS_ZK_VERIFIER_SIGNER_ADDRESS","chalk","MockTaskManagerArtifact","chalk","MockACLArtifact","MockQueryDecrypterArtifact","MockTaskManagerArtifact","MockZkVerifierArtifact","TestBedArtifact","chalk","MOCKS_ZK_VERIFIER_SIGNER_ADDRESS"]}
package/package.json CHANGED
@@ -1,13 +1,20 @@
1
1
  {
2
2
  "name": "@cofhe/hardhat-plugin",
3
- "version": "0.1.0",
3
+ "version": "0.2.0",
4
4
  "main": "./dist/index.js",
5
5
  "module": "./dist/index.mjs",
6
6
  "types": "./dist/index.d.ts",
7
7
  "sideEffects": false,
8
8
  "license": "MIT",
9
+ "repository": {
10
+ "type": "git",
11
+ "url": "https://github.com/FhenixProtocol/cofhesdk.git",
12
+ "directory": "packages/hardhat-plugin"
13
+ },
9
14
  "files": [
10
- "dist/**"
15
+ "dist/**",
16
+ "src/**",
17
+ "CHANGELOG.md"
11
18
  ],
12
19
  "dependencies": {
13
20
  "@fhenixprotocol/cofhe-contracts": "0.0.13",
@@ -15,10 +22,10 @@
15
22
  "@openzeppelin/contracts-upgradeable": "^5.0.0",
16
23
  "chalk": "^4.1.2",
17
24
  "fast-glob": "^3.3.2",
18
- "viem": "^2.0.0",
25
+ "viem": "^2.38.6",
19
26
  "chai": "^4.2.0",
20
- "@cofhe/sdk": "0.1.0",
21
- "@cofhe/mock-contracts": "0.1.0"
27
+ "@cofhe/sdk": "0.2.0",
28
+ "@cofhe/mock-contracts": "0.2.0"
22
29
  },
23
30
  "devDependencies": {
24
31
  "@nomicfoundation/hardhat-ethers": "^3.0.5",
@@ -35,8 +42,8 @@
35
42
  "prettier-plugin-solidity": "^1.3.0",
36
43
  "tsup": "^8.0.2",
37
44
  "typescript": "5.5.4",
38
- "@cofhe/tsconfig": "0.1.0",
39
- "@cofhe/eslint-config": "0.1.0"
45
+ "@cofhe/tsconfig": "0.1.1",
46
+ "@cofhe/eslint-config": "0.2.0"
40
47
  },
41
48
  "peerDependencies": {
42
49
  "hardhat": "^2.0.0",
package/src/consts.ts ADDED
@@ -0,0 +1,8 @@
1
+ export const TASK_COFHE_USE_FAUCET = 'task:cofhe:usefaucet';
2
+ export const TASK_COFHE_MOCKS_SET_LOG_OPS = 'task:cofhe-mocks:setlogops';
3
+ export const TASK_COFHE_MOCKS_DEPLOY = 'task:cofhe-mocks:deploy';
4
+
5
+ export const TASK_MANAGER_ADDRESS = '0xeA30c4B8b44078Bbf8a6ef5b9f1eC1626C7848D9';
6
+ export const MOCKS_ZK_VERIFIER_ADDRESS = '0x0000000000000000000000000000000000000100';
7
+ export const MOCKS_QUERY_DECRYPTER_ADDRESS = '0x0000000000000000000000000000000000000200';
8
+ export const TEST_BED_ADDRESS = '0x0000000000000000000000000000000000000300';
package/src/deploy.ts ADDED
@@ -0,0 +1,246 @@
1
+ import { type HardhatRuntimeEnvironment } from 'hardhat/types';
2
+ import chalk from 'chalk';
3
+ import { Contract } from 'ethers';
4
+
5
+ import {
6
+ TASK_MANAGER_ADDRESS,
7
+ MOCKS_ZK_VERIFIER_ADDRESS,
8
+ MOCKS_QUERY_DECRYPTER_ADDRESS,
9
+ TEST_BED_ADDRESS,
10
+ } from './consts.js';
11
+
12
+ import {
13
+ MockTaskManagerArtifact,
14
+ MockACLArtifact,
15
+ MockZkVerifierArtifact,
16
+ MockQueryDecrypterArtifact,
17
+ TestBedArtifact,
18
+ } from '@cofhe/mock-contracts';
19
+ import { MOCKS_ZK_VERIFIER_SIGNER_ADDRESS } from '@cofhe/sdk';
20
+
21
+ // Deploy
22
+
23
+ const deployMockTaskManager = async (hre: HardhatRuntimeEnvironment) => {
24
+ const [signer] = await hre.ethers.getSigners();
25
+
26
+ // Deploy MockTaskManager
27
+ await hardhatSetCode(hre, TASK_MANAGER_ADDRESS, MockTaskManagerArtifact.deployedBytecode);
28
+ const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);
29
+
30
+ // Initialize MockTaskManager
31
+ const initTx = await taskManager.initialize(signer.address);
32
+ await initTx.wait();
33
+
34
+ // Check if MockTaskManager exists
35
+ const tmExists = await taskManager.exists();
36
+ if (!tmExists) {
37
+ throw new Error('MockTaskManager does not exist');
38
+ }
39
+
40
+ return taskManager;
41
+ };
42
+
43
+ const deployMockACL = async (hre: HardhatRuntimeEnvironment): Promise<Contract> => {
44
+ // Deploy MockACL (uses ethers to deploy to ensure constructor called and EIP712 domain set)
45
+ const acl = await ethersDeployContract(hre, MockACLArtifact.abi, MockACLArtifact.bytecode);
46
+
47
+ // Check if ACL exists
48
+ const exists = await acl.exists();
49
+ if (!exists) {
50
+ logError('MockACL does not exist', 2);
51
+ throw new Error('MockACL does not exist');
52
+ }
53
+
54
+ return acl;
55
+ };
56
+
57
+ const deployMockZkVerifier = async (hre: HardhatRuntimeEnvironment) => {
58
+ await hardhatSetCode(hre, MOCKS_ZK_VERIFIER_ADDRESS, MockZkVerifierArtifact.deployedBytecode);
59
+ const zkVerifier = await hre.ethers.getContractAt(MockZkVerifierArtifact.abi, MOCKS_ZK_VERIFIER_ADDRESS);
60
+
61
+ const zkVerifierExists = await zkVerifier.exists();
62
+ if (!zkVerifierExists) {
63
+ logError('MockZkVerifier does not exist', 2);
64
+ throw new Error('MockZkVerifier does not exist');
65
+ }
66
+
67
+ return zkVerifier;
68
+ };
69
+
70
+ const deployMockQueryDecrypter = async (hre: HardhatRuntimeEnvironment, acl: Contract) => {
71
+ await hardhatSetCode(hre, MOCKS_QUERY_DECRYPTER_ADDRESS, MockQueryDecrypterArtifact.deployedBytecode);
72
+ const queryDecrypter = await hre.ethers.getContractAt(MockQueryDecrypterArtifact.abi, MOCKS_QUERY_DECRYPTER_ADDRESS);
73
+
74
+ // Initialize MockQueryDecrypter
75
+ const initTx = await queryDecrypter.initialize(TASK_MANAGER_ADDRESS, await acl.getAddress());
76
+ await initTx.wait();
77
+
78
+ // Check if MockQueryDecrypter exists
79
+ const queryDecrypterExists = await queryDecrypter.exists();
80
+ if (!queryDecrypterExists) {
81
+ logError('MockQueryDecrypter does not exist', 2);
82
+ throw new Error('MockQueryDecrypter does not exist');
83
+ }
84
+
85
+ return queryDecrypter;
86
+ };
87
+
88
+ const deployTestBedContract = async (hre: HardhatRuntimeEnvironment) => {
89
+ await hardhatSetCode(hre, TEST_BED_ADDRESS, TestBedArtifact.deployedBytecode);
90
+ const testBed = await hre.ethers.getContractAt(TestBedArtifact.abi, TEST_BED_ADDRESS);
91
+ await testBed.waitForDeployment();
92
+ return testBed;
93
+ };
94
+
95
+ // Funding
96
+
97
+ const fundZkVerifierSigner = async (hre: HardhatRuntimeEnvironment) => {
98
+ const zkVerifierSigner = await hre.ethers.getSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
99
+ await hre.network.provider.send('hardhat_setBalance', [
100
+ zkVerifierSigner.address,
101
+ '0x' + hre.ethers.parseEther('10').toString(16),
102
+ ]);
103
+ };
104
+
105
+ // Initializations
106
+
107
+ const setTaskManagerACL = async (taskManager: Contract, acl: Contract) => {
108
+ const setAclTx = await taskManager.setACLContract(await acl.getAddress());
109
+ await setAclTx.wait();
110
+ };
111
+
112
+ export type DeployMocksArgs = {
113
+ deployTestBed?: boolean;
114
+ gasWarning?: boolean;
115
+ silent?: boolean;
116
+ };
117
+
118
+ export const deployMocks = async (
119
+ hre: HardhatRuntimeEnvironment,
120
+ options: DeployMocksArgs = {
121
+ deployTestBed: true,
122
+ gasWarning: true,
123
+ silent: false,
124
+ }
125
+ ) => {
126
+ // Check if network is Hardhat, if not log skip message and return
127
+ const isHardhat = await checkNetworkAndSkip(hre);
128
+ if (!isHardhat) return;
129
+
130
+ const logEmptyIfNoisy = () => {
131
+ if (!options.silent) {
132
+ logEmpty();
133
+ }
134
+ };
135
+ const logSuccessIfNoisy = (message: string, indent = 0) => {
136
+ if (!options.silent) {
137
+ logSuccess(message, indent);
138
+ }
139
+ };
140
+ const logDeploymentIfNoisy = (contractName: string, address: string) => {
141
+ if (!options.silent) {
142
+ logDeployment(contractName, address);
143
+ }
144
+ };
145
+ const logWarningIfNoisy = (message: string, indent = 0) => {
146
+ if (!options.silent) {
147
+ logWarning(message, indent);
148
+ }
149
+ };
150
+
151
+ // Log start message
152
+ logEmptyIfNoisy();
153
+ logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: deploy mocks'), 0);
154
+ logEmptyIfNoisy();
155
+
156
+ // Deploy mock contracts
157
+ const taskManager = await deployMockTaskManager(hre);
158
+ logDeploymentIfNoisy('MockTaskManager', await taskManager.getAddress());
159
+
160
+ const acl = await deployMockACL(hre);
161
+ logDeploymentIfNoisy('MockACL', await acl.getAddress());
162
+
163
+ await setTaskManagerACL(taskManager, acl);
164
+ logSuccessIfNoisy('ACL address set in TaskManager', 2);
165
+
166
+ await fundZkVerifierSigner(hre);
167
+ logSuccessIfNoisy(`ZkVerifier signer (${MOCKS_ZK_VERIFIER_SIGNER_ADDRESS}) funded`, 1);
168
+
169
+ const zkVerifierSignerBalance = await hre.ethers.provider.getBalance(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
170
+ logSuccessIfNoisy(`ETH balance: ${zkVerifierSignerBalance.toString()}`, 2);
171
+
172
+ const zkVerifier = await deployMockZkVerifier(hre);
173
+ logDeploymentIfNoisy('MockZkVerifier', await zkVerifier.getAddress());
174
+
175
+ const queryDecrypter = await deployMockQueryDecrypter(hre, acl);
176
+ logDeploymentIfNoisy('MockQueryDecrypter', await queryDecrypter.getAddress());
177
+
178
+ if (options.deployTestBed) {
179
+ logSuccessIfNoisy('TestBed deployment enabled', 2);
180
+ const testBed = await deployTestBedContract(hre);
181
+ logDeploymentIfNoisy('TestBed', await testBed.getAddress());
182
+ }
183
+
184
+ // Log success message
185
+ logEmptyIfNoisy();
186
+ logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: mocks deployed successfully'), 0);
187
+
188
+ // Log warning about mocks increased gas costs
189
+ if (options.gasWarning) {
190
+ logEmptyIfNoisy();
191
+ logWarningIfNoisy(
192
+ "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",
193
+ 0
194
+ );
195
+ }
196
+
197
+ logEmptyIfNoisy();
198
+ };
199
+
200
+ // Utils
201
+
202
+ const hardhatSetCode = async (hre: HardhatRuntimeEnvironment, address: string, bytecode: string) => {
203
+ await hre.network.provider.send('hardhat_setCode', [address, bytecode]);
204
+ };
205
+
206
+ const ethersDeployContract = async (hre: HardhatRuntimeEnvironment, abi: any, bytecode: string) => {
207
+ const [signer] = await hre.ethers.getSigners();
208
+
209
+ const factory = new hre.ethers.ContractFactory(abi, bytecode, signer);
210
+ const contract = await factory.deploy(/* constructor args */);
211
+ await contract.waitForDeployment();
212
+
213
+ return contract as Contract;
214
+ };
215
+
216
+ // Network
217
+
218
+ const checkNetworkAndSkip = async (hre: HardhatRuntimeEnvironment) => {
219
+ const network = hre.network.name;
220
+ const isHardhat = network === 'hardhat';
221
+ if (!isHardhat) logSuccess(`cofhe-hardhat-plugin - deploy mocks - skipped on non-hardhat network ${network}`, 0);
222
+ return isHardhat;
223
+ };
224
+
225
+ // Logging
226
+
227
+ const logEmpty = () => {
228
+ console.log('');
229
+ };
230
+
231
+ const logSuccess = (message: string, indent = 1) => {
232
+ console.log(chalk.green(`${' '.repeat(indent)}✓ ${message}`));
233
+ };
234
+
235
+ const logWarning = (message: string, indent = 1) => {
236
+ console.log(chalk.bold(chalk.yellow(`${' '.repeat(indent)}⚠ NOTE:`)), message);
237
+ };
238
+
239
+ const logError = (message: string, indent = 1) => {
240
+ console.log(chalk.red(`${' '.repeat(indent)}✗ ${message}`));
241
+ };
242
+
243
+ const logDeployment = (contractName: string, address: string) => {
244
+ const paddedName = `${contractName} deployed`.padEnd(36);
245
+ logSuccess(`${paddedName} ${chalk.bold(address)}`);
246
+ };
package/src/fund.ts ADDED
@@ -0,0 +1,82 @@
1
+ /* eslint-disable turbo/no-undeclared-env-vars */
2
+ import { type HardhatRuntimeEnvironment } from 'hardhat/types';
3
+ import '@nomicfoundation/hardhat-ethers';
4
+
5
+ /**
6
+ * Sends funds to the specified address
7
+ * @param hre Hardhat Runtime Environment
8
+ * @param toAddress Address to send funds to
9
+ * @param amount Amount to send in ETH (default: 10)
10
+ * @returns Transaction receipt or null if failed
11
+ */
12
+ export async function localcofheFundAccount(hre: HardhatRuntimeEnvironment, toAddress: string, amount: string = '10') {
13
+ // Load private key from environment
14
+ const privateKey =
15
+ process.env.FUNDER_PRIVATE_KEY ?? '0xb6b15c8cb491557369f3c7d2c287b053eb229daa9c22138887752191c9520659';
16
+ if (!privateKey) {
17
+ console.error('Error: FUNDER_PRIVATE_KEY environment variable not set');
18
+ return null;
19
+ }
20
+
21
+ try {
22
+ // Create wallet from private key
23
+ const wallet = new hre.ethers.Wallet(privateKey, hre.ethers.provider);
24
+
25
+ // Get wallet balance
26
+ const balance = await hre.ethers.provider.getBalance(wallet.address);
27
+ console.log(`Funder wallet address: ${wallet.address}`);
28
+ console.log(`Funder wallet balance: ${hre.ethers.formatEther(balance)} ETH`);
29
+
30
+ // Check if wallet has enough funds
31
+ const amountToSend = hre.ethers.parseEther(amount);
32
+ if (balance < amountToSend) {
33
+ console.error(
34
+ `Error: Funder wallet doesn't have enough funds. Current balance: ${hre.ethers.formatEther(balance)} ETH`
35
+ );
36
+ return null;
37
+ }
38
+
39
+ // Send transaction
40
+ console.log(`Sending ${amount} ETH to ${toAddress}...`);
41
+ const tx = await wallet.sendTransaction({
42
+ to: toAddress,
43
+ value: amountToSend,
44
+ });
45
+
46
+ console.log(`Transaction sent! Hash: ${tx.hash}`);
47
+ console.log('Waiting for confirmation...');
48
+
49
+ // Wait for transaction to be mined
50
+ const receipt = await tx.wait();
51
+ console.log(`Transaction confirmed in block ${receipt?.blockNumber}`);
52
+ console.log(`Successfully sent ${amount} ETH to ${toAddress}`);
53
+
54
+ return receipt;
55
+ } catch (error) {
56
+ console.error('Error sending funds:', error);
57
+ return null;
58
+ }
59
+ }
60
+
61
+ /**
62
+ * Checks a wallet's balance and funds it if below 1 ETH
63
+ * @param hre Hardhat Runtime Environment
64
+ * @param walletAddress Address of the wallet to check and potentially fund
65
+ * @returns Promise that resolves when the funding operation completes (if needed)
66
+ */
67
+ export async function localcofheFundWalletIfNeeded(hre: HardhatRuntimeEnvironment, walletAddress: string) {
68
+ // Check wallet balance and fund if needed
69
+ const walletBalance = await hre.ethers.provider.getBalance(walletAddress);
70
+ console.log(`Wallet balance: ${hre.ethers.formatEther(walletBalance)} ETH`);
71
+
72
+ if (walletBalance < hre.ethers.parseEther('1')) {
73
+ console.log(`Wallet balance is less than 1 ETH. Funding ${walletAddress}...`);
74
+ const receipt = await localcofheFundAccount(hre, walletAddress);
75
+ if (receipt) {
76
+ const newBalance = await hre.ethers.provider.getBalance(walletAddress);
77
+ console.log(`Wallet new balance: ${hre.ethers.formatEther(newBalance)} ETH`);
78
+ } else {
79
+ console.error(`Failed to fund ${walletAddress}`);
80
+ }
81
+ }
82
+ }