@cofhe/hardhat-plugin 0.1.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.
@@ -0,0 +1 @@
1
+ {"version":3,"sources":["../src/index.ts","../src/fund.ts","../src/consts.ts","../src/deploy.ts","../src/logging.ts","../src/utils.ts","../src/expectResultUtils.ts"],"sourcesContent":["/* eslint-disable no-empty-pattern */\n/* eslint-disable turbo/no-undeclared-env-vars */\n/* eslint-disable no-unused-vars */\nimport chalk from 'chalk';\nimport { type PublicClient, type WalletClient } from 'viem';\nimport { extendConfig, extendEnvironment, task, types } from 'hardhat/config';\nimport { TASK_TEST, TASK_NODE } from 'hardhat/builtin-tasks/task-names';\nimport { HardhatEthersSigner } from '@nomicfoundation/hardhat-ethers/signers';\nimport { type CofhesdkClient, type CofhesdkConfig, type CofhesdkInputConfig, type Result } from '@cofhe/sdk';\nimport { createCofhesdkClient, createCofhesdkConfig } from '@cofhe/sdk/node';\nimport { HardhatSignerAdapter } from '@cofhe/sdk/adapters';\n\nimport { localcofheFundAccount } from './fund.js';\nimport {\n MOCKS_ZK_VERIFIER_SIGNER_ADDRESS,\n TASK_COFHE_MOCKS_DEPLOY,\n TASK_COFHE_MOCKS_SET_LOG_OPS,\n TASK_COFHE_USE_FAUCET,\n} from './consts.js';\nimport { deployMocks, type DeployMocksArgs } from './deploy.js';\nimport { mock_setLoggingEnabled, mock_withLogs } from './logging.js';\nimport { mock_expectPlaintext } from './utils.js';\nimport { mock_getPlaintext } from './utils.js';\nimport {\n expectResultError,\n expectResultPartialValue,\n expectResultSuccess,\n expectResultValue,\n} from './expectResultUtils.js';\nexport {\n MockACLArtifact,\n MockQueryDecrypterArtifact,\n MockTaskManagerArtifact,\n MockZkVerifierArtifact,\n TestBedArtifact,\n} from '@cofhe/mock-contracts';\n\n/**\n * Configuration interface for the CoFHE Hardhat plugin.\n * Allows users to configure mock logging and gas warning settings.\n */\ndeclare module 'hardhat/types/config' {\n interface HardhatUserConfig {\n cofhesdk?: {\n /** Whether to log mock operations (default: true) */\n logMocks?: boolean;\n /** Whether to show gas usage warnings for mock operations (default: true) */\n gasWarning?: boolean;\n };\n }\n\n interface HardhatConfig {\n cofhesdk: {\n /** Whether to log mock operations (default: true) */\n logMocks: boolean;\n /** Whether to show gas usage warnings for mock operations (default: true) */\n gasWarning: boolean;\n };\n }\n}\n\nextendConfig((config, userConfig) => {\n // Allow users to override the localcofhe network config\n if (userConfig.networks && userConfig.networks.localcofhe) {\n return;\n }\n\n // Default config\n config.networks.localcofhe = {\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 10_000,\n httpHeaders: {},\n url: 'http://127.0.0.1:42069',\n accounts: [\n '0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80',\n '0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d',\n '0x5de4111afa1a4b94908f83103eb1f1706367c2e68ca870fc3fb9a804cdab365a',\n '0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6',\n ],\n };\n\n // Only add Sepolia config if user hasn't defined it\n if (!userConfig.networks?.['eth-sepolia']) {\n config.networks['eth-sepolia'] = {\n url: process.env.SEPOLIA_RPC_URL ?? 'https://ethereum-sepolia.publicnode.com',\n accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],\n chainId: 11155111,\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 60_000,\n httpHeaders: {},\n };\n }\n\n // Only add Arbitrum Sepolia config if user hasn't defined it\n if (!userConfig.networks?.['arb-sepolia']) {\n config.networks['arb-sepolia'] = {\n url: process.env.ARBITRUM_SEPOLIA_RPC_URL ?? 'https://sepolia-rollup.arbitrum.io/rpc',\n accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],\n chainId: 421614,\n gas: 'auto',\n gasMultiplier: 1.2,\n gasPrice: 'auto',\n timeout: 60_000,\n httpHeaders: {},\n };\n }\n\n // Add cofhe config\n config.cofhesdk = {\n logMocks: userConfig.cofhesdk?.logMocks ?? true,\n gasWarning: userConfig.cofhesdk?.gasWarning ?? true,\n };\n});\n\ntype UseFaucetArgs = {\n address?: string;\n};\n\ntask(TASK_COFHE_USE_FAUCET, 'Fund an account from the funder')\n .addOptionalParam('address', 'Address to fund', undefined, types.string)\n .setAction(async ({ address }: UseFaucetArgs, hre) => {\n const { network } = hre;\n const { name: networkName } = network;\n\n if (networkName !== 'localcofhe') {\n console.info(chalk.yellow(`Programmatic faucet only supported for localcofhe`));\n return;\n }\n\n if (!address) {\n console.info(chalk.red(`Failed to get address to fund`));\n return;\n }\n\n console.info(chalk.green(`Getting funds from faucet for ${address}`));\n\n try {\n await localcofheFundAccount(hre, address);\n } catch (e) {\n console.info(chalk.red(`failed to get funds from localcofhe for ${address}: ${e}`));\n }\n });\n\n// DEPLOY TASKS\n\ntask(TASK_COFHE_MOCKS_DEPLOY, 'Deploys the mock contracts on the Hardhat network')\n .addOptionalParam('deployTestBed', 'Whether to deploy the test bed', true, types.boolean)\n .addOptionalParam('silent', 'Whether to suppress output', false, types.boolean)\n .setAction(async ({ deployTestBed, silent }: DeployMocksArgs, hre) => {\n await deployMocks(hre, {\n deployTestBed: deployTestBed ?? true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n silent: silent ?? false,\n });\n });\n\ntask(TASK_TEST, 'Deploy mock contracts on hardhat').setAction(async ({}, hre, runSuper) => {\n await deployMocks(hre, {\n deployTestBed: true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n });\n return runSuper();\n});\n\ntask(TASK_NODE, 'Deploy mock contracts on hardhat').setAction(async ({}, hre, runSuper) => {\n await deployMocks(hre, {\n deployTestBed: true,\n gasWarning: hre.config.cofhesdk.gasWarning ?? true,\n });\n return runSuper();\n});\n\n// SET LOG OPS\n\ntask(TASK_COFHE_MOCKS_SET_LOG_OPS, 'Set logging for the Mock CoFHE contracts')\n .addParam('enable', 'Whether to enable logging', false, types.boolean)\n .setAction(async ({ enable }, hre) => {\n await mock_setLoggingEnabled(hre, enable);\n });\n\n// MOCK UTILS\n\nexport * from './utils.js';\nexport * from './expectResultUtils.js';\nexport * from './fund.js';\nexport * from './logging.js';\nexport * from './deploy.js';\n\n/**\n * Runtime environment extensions for the CoFHE Hardhat plugin.\n * Provides access to CoFHE initialization, environment checks, and mock utilities.\n */\ndeclare module 'hardhat/types/runtime' {\n export interface HardhatRuntimeEnvironment {\n cofhesdk: {\n /**\n * Create a CoFHE SDK configuration for use with cofhesdk.createCofhesdkClient(...)\n * @param {CofhesdkInputConfig} config - The CoFHE SDK input configuration\n * @returns {CofhesdkConfig} The CoFHE SDK configuration\n */\n createCofhesdkConfig: (config: CofhesdkInputConfig) => Promise<CofhesdkConfig>;\n /**\n * Create a CoFHE SDK client instance\n * @param {CofhesdkConfig} config - The CoFHE SDK configuration (use createCofhesdkConfig to create with Node.js defaults)\n * @returns {Promise<CofhesdkClient>} The CoFHE SDK client instance\n */\n createCofhesdkClient: (config: CofhesdkConfig) => CofhesdkClient;\n /**\n * Create viem clients from a Hardhat ethers signer, to be used with `cofhesdkClient.connect(...)`\n * @param {HardhatEthersSigner} signer - The Hardhat ethers signer to use\n * @returns {Promise<{ publicClient: PublicClient; walletClient: WalletClient }>} The viem clients\n */\n hardhatSignerAdapter: (\n signer: HardhatEthersSigner\n ) => Promise<{ publicClient: PublicClient; walletClient: WalletClient }>;\n\n /**\n * Assert that a Result type returned from a function is successful and return its value (result.success === true)\n * @param {Result<T>} result - The Result to check\n * @returns {T} The inner data of the Result (non null)\n */\n expectResultSuccess: <T>(result: Result<T> | Promise<Result<T>>) => Promise<T>;\n\n /**\n * Assert that a Result type contains an error matching the partial string (result.success === false && result.error.includes(errorPartial))\n * @param {Result<T>} result - The Result to check\n * @param {string} errorPartial - The partial error string to match\n */\n expectResultError: <T>(result: Result<T> | Promise<Result<T>>, errorPartial: string) => Promise<void>;\n\n /**\n * Assert that a Result type contains a specific value (result.success === true && result.data === value)\n * @param {Result<T>} result - The Result to check\n * @param {T} value - The inner data of the Result (non null)\n */\n expectResultValue: <T>(result: Result<T> | Promise<Result<T>>, value: T) => Promise<T>;\n\n /**\n * Assert that a Result type contains a value matching the partial object (result.success === true && result.data.includes(partial))\n * @param {Result<T>} result - The Result to check\n * @param {Partial<T>} partial - The partial object to match against\n * @returns {T} The inner data of the Result (non null)\n */\n expectResultPartialValue: <T>(result: Result<T> | Promise<Result<T>>, partial: Partial<T>) => Promise<T>;\n\n mocks: {\n /**\n * **[MOCKS ONLY]**\n *\n * Execute a block of code with cofhe mock contracts logging enabled.\n *\n * _(If logging only a function, we recommend passing the function name as the closureName (ex \"counter.increment()\"))_\n *\n * Example usage:\n *\n * ```ts\n * await hre.cofhesdk.mocks.withLogs(\"counter.increment()\", async () => {\n * await counter.increment();\n * });\n * ```\n *\n * Expected output:\n * ```\n * ┌──────────────────┬──────────────────────────────────────────────────\n * │ [COFHE-MOCKS] │ \"counter.increment()\" logs:\n * ├──────────────────┴──────────────────────────────────────────────────\n * ├ FHE.add | euint32(4473..3424)[0] + euint32(1157..3648)[1] => euint32(1106..1872)[1]\n * ├ FHE.allowThis | euint32(1106..1872)[1] -> 0x663f..6602\n * ├ FHE.allow | euint32(1106..1872)[1] -> 0x3c44..93bc\n * └─────────────────────────────────────────────────────────────────────\n * ```\n * @param {string} closureName - Name of the code block to log within\n * @param {() => Promise<void>} closure - The async function to execute\n */\n withLogs: (closureName: string, closure: () => Promise<void>) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Enable logging from cofhe mock contracts\n * @param {string} closureName - Optional name of the code block to enable logging for\n */\n enableLogs: (closureName?: string) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Disable logging from cofhe mock contracts\n */\n disableLogs: () => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Deploy the cofhe mock contracts (normally this is done automatically)\n * @param {DeployMocksArgs} options - Deployment options\n */\n deployMocks: (options: DeployMocksArgs) => Promise<void>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Get the plaintext value for a ciphertext hash\n * @param {bigint} ctHash - The ciphertext hash to look up\n * @returns {Promise<bigint>} The plaintext value\n */\n getPlaintext: (ctHash: bigint) => Promise<bigint>;\n\n /**\n * **[MOCKS ONLY]**\n *\n * Assert that a ciphertext hash represents an expected plaintext value\n * @param {bigint} ctHash - The ciphertext hash to check\n * @param {bigint} expectedValue - The expected plaintext value\n */\n expectPlaintext: (ctHash: bigint, expectedValue: bigint) => Promise<void>;\n };\n };\n }\n}\n\nextendEnvironment((hre) => {\n hre.cofhesdk = {\n createCofhesdkConfig: async (config: CofhesdkInputConfig) => {\n // Create zkv wallet client\n // This wallet interacts with the MockZkVerifier contract so that the user's connected wallet doesn't have to\n const zkvHhSigner = await hre.ethers.getImpersonatedSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n const { walletClient: zkvWalletClient } = await HardhatSignerAdapter(zkvHhSigner);\n\n // Inject zkv wallet client into config\n const configWithZkvWalletClient = {\n ...config,\n _internal: {\n ...config._internal,\n zkvWalletClient,\n },\n };\n\n return createCofhesdkConfig(configWithZkvWalletClient);\n },\n createCofhesdkClient: (config: CofhesdkConfig) => {\n return createCofhesdkClient(config);\n },\n hardhatSignerAdapter: async (signer: HardhatEthersSigner) => {\n return HardhatSignerAdapter(signer);\n },\n expectResultSuccess: async <T>(result: Result<T> | Promise<Result<T>>) => {\n const awaitedResult = await result;\n return expectResultSuccess(awaitedResult);\n },\n expectResultError: async <T>(result: Result<T> | Promise<Result<T>>, errorPartial: string) => {\n const awaitedResult = await result;\n return expectResultError(awaitedResult, errorPartial);\n },\n expectResultValue: async <T>(result: Result<T> | Promise<Result<T>>, value: T) => {\n const awaitedResult = await result;\n return expectResultValue(awaitedResult, value);\n },\n expectResultPartialValue: async <T>(result: Result<T> | Promise<Result<T>>, partial: Partial<T>) => {\n const awaitedResult = await result;\n return expectResultPartialValue(awaitedResult, partial);\n },\n mocks: {\n withLogs: async (closureName: string, closure: () => Promise<void>) => {\n return mock_withLogs(hre, closureName, closure);\n },\n enableLogs: async (closureName?: string) => {\n return mock_setLoggingEnabled(hre, true, closureName);\n },\n disableLogs: async () => {\n return mock_setLoggingEnabled(hre, false);\n },\n deployMocks: async (options: DeployMocksArgs) => {\n return deployMocks(hre, options);\n },\n getPlaintext: async (ctHash: bigint) => {\n const [signer] = await hre.ethers.getSigners();\n return mock_getPlaintext(signer.provider, ctHash);\n },\n expectPlaintext: async (ctHash: bigint, expectedValue: bigint) => {\n const [signer] = await hre.ethers.getSigners();\n return mock_expectPlaintext(signer.provider, ctHash, expectedValue);\n },\n },\n };\n});\n","/* eslint-disable turbo/no-undeclared-env-vars */\nimport { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport '@nomicfoundation/hardhat-ethers';\n\n/**\n * Sends funds to the specified address\n * @param hre Hardhat Runtime Environment\n * @param toAddress Address to send funds to\n * @param amount Amount to send in ETH (default: 10)\n * @returns Transaction receipt or null if failed\n */\nexport async function localcofheFundAccount(hre: HardhatRuntimeEnvironment, toAddress: string, amount: string = '10') {\n // Load private key from environment\n const privateKey =\n process.env.FUNDER_PRIVATE_KEY ?? '0xb6b15c8cb491557369f3c7d2c287b053eb229daa9c22138887752191c9520659';\n if (!privateKey) {\n console.error('Error: FUNDER_PRIVATE_KEY environment variable not set');\n return null;\n }\n\n try {\n // Create wallet from private key\n const wallet = new hre.ethers.Wallet(privateKey, hre.ethers.provider);\n\n // Get wallet balance\n const balance = await hre.ethers.provider.getBalance(wallet.address);\n console.log(`Funder wallet address: ${wallet.address}`);\n console.log(`Funder wallet balance: ${hre.ethers.formatEther(balance)} ETH`);\n\n // Check if wallet has enough funds\n const amountToSend = hre.ethers.parseEther(amount);\n if (balance < amountToSend) {\n console.error(\n `Error: Funder wallet doesn't have enough funds. Current balance: ${hre.ethers.formatEther(balance)} ETH`\n );\n return null;\n }\n\n // Send transaction\n console.log(`Sending ${amount} ETH to ${toAddress}...`);\n const tx = await wallet.sendTransaction({\n to: toAddress,\n value: amountToSend,\n });\n\n console.log(`Transaction sent! Hash: ${tx.hash}`);\n console.log('Waiting for confirmation...');\n\n // Wait for transaction to be mined\n const receipt = await tx.wait();\n console.log(`Transaction confirmed in block ${receipt?.blockNumber}`);\n console.log(`Successfully sent ${amount} ETH to ${toAddress}`);\n\n return receipt;\n } catch (error) {\n console.error('Error sending funds:', error);\n return null;\n }\n}\n\n/**\n * Checks a wallet's balance and funds it if below 1 ETH\n * @param hre Hardhat Runtime Environment\n * @param walletAddress Address of the wallet to check and potentially fund\n * @returns Promise that resolves when the funding operation completes (if needed)\n */\nexport async function localcofheFundWalletIfNeeded(hre: HardhatRuntimeEnvironment, walletAddress: string) {\n // Check wallet balance and fund if needed\n const walletBalance = await hre.ethers.provider.getBalance(walletAddress);\n console.log(`Wallet balance: ${hre.ethers.formatEther(walletBalance)} ETH`);\n\n if (walletBalance < hre.ethers.parseEther('1')) {\n console.log(`Wallet balance is less than 1 ETH. Funding ${walletAddress}...`);\n const receipt = await localcofheFundAccount(hre, walletAddress);\n if (receipt) {\n const newBalance = await hre.ethers.provider.getBalance(walletAddress);\n console.log(`Wallet new balance: ${hre.ethers.formatEther(newBalance)} ETH`);\n } else {\n console.error(`Failed to fund ${walletAddress}`);\n }\n }\n}\n","export const TASK_COFHE_USE_FAUCET = 'task:cofhe:usefaucet';\nexport const TASK_COFHE_MOCKS_SET_LOG_OPS = 'task:cofhe-mocks:setlogops';\nexport const TASK_COFHE_MOCKS_DEPLOY = 'task:cofhe-mocks:deploy';\n\nexport const TASK_MANAGER_ADDRESS = '0xeA30c4B8b44078Bbf8a6ef5b9f1eC1626C7848D9';\nexport const MOCKS_ACL_ADDRESS = '0x0000000000000000000000000000000000000400';\nexport const MOCKS_ZK_VERIFIER_ADDRESS = '0x0000000000000000000000000000000000000100';\nexport const MOCKS_QUERY_DECRYPTER_ADDRESS = '0x0000000000000000000000000000000000000200';\nexport const TEST_BED_ADDRESS = '0x0000000000000000000000000000000000000300';\nexport const MOCKS_ZK_VERIFIER_SIGNER_ADDRESS = '0x6E12D8C87503D4287c294f2Fdef96ACd9DFf6bd2';\n","import { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport chalk from 'chalk';\nimport { Contract } from 'ethers';\n\nimport {\n TASK_MANAGER_ADDRESS,\n MOCKS_ZK_VERIFIER_ADDRESS,\n MOCKS_QUERY_DECRYPTER_ADDRESS,\n TEST_BED_ADDRESS,\n MOCKS_ZK_VERIFIER_SIGNER_ADDRESS,\n MOCKS_ACL_ADDRESS,\n} from './consts.js';\n\nimport {\n MockTaskManagerArtifact,\n MockACLArtifact,\n MockZkVerifierArtifact,\n MockQueryDecrypterArtifact,\n TestBedArtifact,\n} from '@cofhe/mock-contracts';\n\n// Deploy\n\nconst deployMockTaskManager = async (hre: HardhatRuntimeEnvironment) => {\n const [signer] = await hre.ethers.getSigners();\n\n // Deploy MockTaskManager\n await hardhatSetCode(hre, TASK_MANAGER_ADDRESS, MockTaskManagerArtifact.deployedBytecode);\n const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);\n\n // Initialize MockTaskManager\n const initTx = await taskManager.initialize(signer.address);\n await initTx.wait();\n\n // Check if MockTaskManager exists\n const tmExists = await taskManager.exists();\n if (!tmExists) {\n throw new Error('MockTaskManager does not exist');\n }\n\n return taskManager;\n};\n\nconst deployMockACL = async (hre: HardhatRuntimeEnvironment): Promise<Contract> => {\n // Deploy MockACL\n await hardhatSetCode(hre, MOCKS_ACL_ADDRESS, MockACLArtifact.deployedBytecode);\n const acl = await hre.ethers.getContractAt(MockACLArtifact.abi, MOCKS_ACL_ADDRESS);\n\n // Check if ACL exists\n const exists = await acl.exists();\n if (!exists) {\n logError('MockACL does not exist', 2);\n throw new Error('MockACL does not exist');\n }\n\n return acl;\n};\n\nconst deployMockZkVerifier = async (hre: HardhatRuntimeEnvironment) => {\n await hardhatSetCode(hre, MOCKS_ZK_VERIFIER_ADDRESS, MockZkVerifierArtifact.deployedBytecode);\n const zkVerifier = await hre.ethers.getContractAt(MockZkVerifierArtifact.abi, MOCKS_ZK_VERIFIER_ADDRESS);\n\n const zkVerifierExists = await zkVerifier.exists();\n if (!zkVerifierExists) {\n logError('MockZkVerifier does not exist', 2);\n throw new Error('MockZkVerifier does not exist');\n }\n\n return zkVerifier;\n};\n\nconst deployMockQueryDecrypter = async (hre: HardhatRuntimeEnvironment, acl: Contract) => {\n await hardhatSetCode(hre, MOCKS_QUERY_DECRYPTER_ADDRESS, MockQueryDecrypterArtifact.deployedBytecode);\n const queryDecrypter = await hre.ethers.getContractAt(MockQueryDecrypterArtifact.abi, MOCKS_QUERY_DECRYPTER_ADDRESS);\n\n // Initialize MockQueryDecrypter\n const initTx = await queryDecrypter.initialize(TASK_MANAGER_ADDRESS, await acl.getAddress());\n await initTx.wait();\n\n // Check if MockQueryDecrypter exists\n const queryDecrypterExists = await queryDecrypter.exists();\n if (!queryDecrypterExists) {\n logError('MockQueryDecrypter does not exist', 2);\n throw new Error('MockQueryDecrypter does not exist');\n }\n\n return queryDecrypter;\n};\n\nconst deployTestBedContract = async (hre: HardhatRuntimeEnvironment) => {\n await hardhatSetCode(hre, TEST_BED_ADDRESS, TestBedArtifact.deployedBytecode);\n const testBed = await hre.ethers.getContractAt(TestBedArtifact.abi, TEST_BED_ADDRESS);\n await testBed.waitForDeployment();\n return testBed;\n};\n\n// Funding\n\nconst fundZkVerifierSigner = async (hre: HardhatRuntimeEnvironment) => {\n const zkVerifierSigner = await hre.ethers.getSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n await hre.network.provider.send('hardhat_setBalance', [\n zkVerifierSigner.address,\n '0x' + hre.ethers.parseEther('10').toString(16),\n ]);\n};\n\n// Initializations\n\nconst setTaskManagerACL = async (taskManager: Contract, acl: Contract) => {\n const setAclTx = await taskManager.setACLContract(await acl.getAddress());\n await setAclTx.wait();\n};\n\nexport type DeployMocksArgs = {\n deployTestBed?: boolean;\n gasWarning?: boolean;\n silent?: boolean;\n};\n\nexport const deployMocks = async (\n hre: HardhatRuntimeEnvironment,\n options: DeployMocksArgs = {\n deployTestBed: true,\n gasWarning: true,\n silent: false,\n }\n) => {\n // Check if network is Hardhat, if not log skip message and return\n const isHardhat = await checkNetworkAndSkip(hre);\n if (!isHardhat) return;\n\n const logEmptyIfNoisy = () => {\n if (!options.silent) {\n logEmpty();\n }\n };\n const logSuccessIfNoisy = (message: string, indent = 0) => {\n if (!options.silent) {\n logSuccess(message, indent);\n }\n };\n const logDeploymentIfNoisy = (contractName: string, address: string) => {\n if (!options.silent) {\n logDeployment(contractName, address);\n }\n };\n const logWarningIfNoisy = (message: string, indent = 0) => {\n if (!options.silent) {\n logWarning(message, indent);\n }\n };\n\n // Log start message\n logEmptyIfNoisy();\n logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: deploy mocks'), 0);\n logEmptyIfNoisy();\n\n // Compile mock contracts\n logEmptyIfNoisy();\n logSuccessIfNoisy('Mock contracts compiled', 1);\n\n // Deploy mock contracts\n const taskManager = await deployMockTaskManager(hre);\n logDeploymentIfNoisy('MockTaskManager', await taskManager.getAddress());\n\n const acl = await deployMockACL(hre);\n logDeploymentIfNoisy('MockACL', await acl.getAddress());\n\n await setTaskManagerACL(taskManager, acl);\n logSuccessIfNoisy('ACL address set in TaskManager', 2);\n\n await fundZkVerifierSigner(hre);\n logSuccessIfNoisy(`ZkVerifier signer (${MOCKS_ZK_VERIFIER_SIGNER_ADDRESS}) funded`, 1);\n\n const zkVerifierSignerBalance = await hre.ethers.provider.getBalance(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);\n logSuccessIfNoisy(`ETH balance: ${zkVerifierSignerBalance.toString()}`, 2);\n\n const zkVerifier = await deployMockZkVerifier(hre);\n logDeploymentIfNoisy('MockZkVerifier', await zkVerifier.getAddress());\n\n const queryDecrypter = await deployMockQueryDecrypter(hre, acl);\n logDeploymentIfNoisy('MockQueryDecrypter', await queryDecrypter.getAddress());\n\n if (options.deployTestBed) {\n logSuccessIfNoisy('TestBed deployment enabled', 2);\n const testBed = await deployTestBedContract(hre);\n logDeploymentIfNoisy('TestBed', await testBed.getAddress());\n }\n\n // Log success message\n logEmptyIfNoisy();\n logSuccessIfNoisy(chalk.bold('cofhe-hardhat-plugin :: mocks deployed successfully'), 0);\n\n // Log warning about mocks increased gas costs\n if (options.gasWarning) {\n logEmptyIfNoisy();\n logWarningIfNoisy(\n \"When using mocks, FHE operations (eg FHE.add / FHE.mul) report a higher gas price due to additional on-chain mocking logic. Deploy your contracts on a testnet chain to check the true gas costs.\\n(Disable this warning by setting '@cofhe/sdk.gasWarning' to false in your hardhat config\",\n 0\n );\n }\n\n logEmptyIfNoisy();\n};\n\n// Utils\n\nconst hardhatSetCode = async (hre: HardhatRuntimeEnvironment, address: string, bytecode: string) => {\n await hre.network.provider.send('hardhat_setCode', [address, bytecode]);\n};\n\n// Network\n\nconst checkNetworkAndSkip = async (hre: HardhatRuntimeEnvironment) => {\n const network = hre.network.name;\n const isHardhat = network === 'hardhat';\n if (!isHardhat) logSuccess(`cofhe-hardhat-plugin - deploy mocks - skipped on non-hardhat network ${network}`, 0);\n return isHardhat;\n};\n\n// Logging\n\nconst logEmpty = () => {\n console.log('');\n};\n\nconst logSuccess = (message: string, indent = 1) => {\n console.log(chalk.green(`${' '.repeat(indent)}✓ ${message}`));\n};\n\nconst logWarning = (message: string, indent = 1) => {\n console.log(chalk.bold(chalk.yellow(`${' '.repeat(indent)}⚠ NOTE:`)), message);\n};\n\nconst logError = (message: string, indent = 1) => {\n console.log(chalk.red(`${' '.repeat(indent)}✗ ${message}`));\n};\n\nconst logDeployment = (contractName: string, address: string) => {\n const paddedName = `${contractName} deployed`.padEnd(36);\n logSuccess(`${paddedName} ${chalk.bold(address)}`);\n};\n","import chalk from 'chalk';\nimport { type HardhatRuntimeEnvironment } from 'hardhat/types';\nimport { TASK_MANAGER_ADDRESS } from './consts';\nimport { MockTaskManagerArtifact } from '@cofhe/mock-contracts';\n\nconst getDeployedMockTaskManager = async (hre: HardhatRuntimeEnvironment) => {\n // Fetch the deployed MockTaskManager\n const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);\n\n return taskManager;\n};\n\nconst getLoggingEnabled = async (hre: HardhatRuntimeEnvironment) => {\n const taskManager = await getDeployedMockTaskManager(hre);\n return await taskManager.logOps();\n};\n\nconst setLoggingEnabled = async (hre: HardhatRuntimeEnvironment, enabled: boolean) => {\n const taskManager = await getDeployedMockTaskManager(hre);\n await taskManager.setLogOps(enabled);\n};\n\n// prettier-ignore\nconst printLogsEnabledMessage = (closureMessage: string) => {\n console.log(\"┌──────────────────┬──────────────────────────────────────────────────\");\n console.log(`│ [COFHE-MOCKS] │ ${closureMessage}`);\n console.log(\"├──────────────────┴──────────────────────────────────────────────────\");\n};\n\n// prettier-ignore\nconst printLogsBlockEnd = () => {\n console.log(\"└─────────────────────────────────────────────────────────────────────\");\n};\n\nexport const mock_setLoggingEnabled = async (\n hre: HardhatRuntimeEnvironment,\n enabled: boolean,\n closureName?: string\n) => {\n try {\n const initiallyEnabled = await getLoggingEnabled(hre);\n\n await setLoggingEnabled(hre, enabled);\n\n // Only print if enabling logs\n if (enabled) {\n printLogsEnabledMessage(`${closureName ? `\"${chalk.bold(closureName)}\" logs:` : 'Logs:'}`);\n }\n\n // Only print if disabling logs AND logs currently enabled\n if (!enabled && initiallyEnabled) {\n printLogsBlockEnd();\n }\n } catch (error) {\n console.log(chalk.red('mock_setLoggingEnabled error'), error);\n }\n};\n\nexport const mock_withLogs = async (\n hre: HardhatRuntimeEnvironment,\n closureName: string,\n closure: () => Promise<void>\n) => {\n const initiallyEnabled = await getLoggingEnabled(hre);\n\n await setLoggingEnabled(hre, true);\n printLogsEnabledMessage(`\"${chalk.bold(closureName)}\" logs:`);\n await closure();\n printLogsBlockEnd();\n\n // If logs were disabled, disable them again\n if (!initiallyEnabled) {\n await setLoggingEnabled(hre, false);\n }\n};\n","import { TASK_MANAGER_ADDRESS, MOCKS_ZK_VERIFIER_ADDRESS } from './consts.js';\nimport { expect } from 'chai';\nimport { ethers } from 'ethers';\nimport { type HardhatEthersProvider } from '@nomicfoundation/hardhat-ethers/internal/hardhat-ethers-provider';\n\nconst mock_checkIsTestnet = async (fnName: string, provider: HardhatEthersProvider | ethers.JsonRpcProvider) => {\n // Testnet is checked by testing if MockZkVerifier is deployed\n\n // Get bytecode at ZK_VERIFIER_ADDRESS\n const bytecode = await provider.getCode(MOCKS_ZK_VERIFIER_ADDRESS);\n\n // If bytecode is empty, we are on a testnet\n const isTestnet = bytecode.length === 0;\n\n // Log if we are on a testnet\n if (isTestnet) {\n console.log(`${fnName} - skipped on non-testnet chain`);\n }\n\n return isTestnet;\n};\n\nexport const mock_getPlaintext = async (provider: HardhatEthersProvider | ethers.JsonRpcProvider, ctHash: bigint) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_getPlaintext.name, provider)) return;\n\n // Connect to MockTaskManager\n const taskManager = new ethers.Contract(\n TASK_MANAGER_ADDRESS,\n ['function mockStorage(uint256) view returns (uint256)'],\n provider\n );\n\n // Fetch the plaintext\n const plaintext = await taskManager.mockStorage(ctHash);\n\n return plaintext;\n};\n\nexport const mock_getPlaintextExists = async (\n provider: HardhatEthersProvider | ethers.JsonRpcProvider,\n ctHash: bigint\n) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_getPlaintextExists.name, provider)) return;\n\n // Connect to MockTaskManager\n const taskManager = new ethers.Contract(\n TASK_MANAGER_ADDRESS,\n ['function inMockStorage(uint256) view returns (bool)'],\n provider\n );\n\n // Fetch the plaintext exists\n const plaintextExists = await taskManager.inMockStorage(ctHash);\n\n return plaintextExists;\n};\n\nexport const mock_expectPlaintext = async (\n provider: HardhatEthersProvider | ethers.JsonRpcProvider,\n ctHash: bigint,\n expectedValue: bigint\n) => {\n // Skip with log if called on a non-testnet chain\n if (await mock_checkIsTestnet(mock_expectPlaintext.name, provider)) return;\n\n // Expect the plaintext to exist\n const plaintextExists = await mock_getPlaintextExists(provider, ctHash);\n expect(plaintextExists).equal(true, 'Plaintext does not exist');\n\n // Expect the plaintext to have the expected value\n const plaintext = await mock_getPlaintext(provider, ctHash);\n expect(plaintext).equal(expectedValue, 'Plaintext value is incorrect');\n};\n","import { type Result } from '@cofhe/sdk';\nimport { expect } from 'chai';\n\nexport const expectResultError = <T>(result: Result<T>, errorPartial: string) => {\n expect(result.success).to.eq(false, 'Result should be an error');\n expect(result.error).to.include(errorPartial, `Error should contain error partial: ${errorPartial}`);\n};\n\nexport const expectResultSuccess = <T>(result: Result<T>): T => {\n expect(result.success).to.eq(true, 'Result should be a success');\n return result.data!;\n};\n\nexport const expectResultValue = <T>(result: Result<T>, value: T): T => {\n expect(result.success).to.eq(true, 'Result should be a success');\n expect(result.data).to.eq(value, `Result should have the expected value ${value}`);\n return result.data!;\n};\n\nexport const expectResultPartialValue = <T>(result: Result<T>, partial: Partial<T>): T => {\n expect(result.success).to.eq(true, 'Result should be a success');\n expect(result.data).to.include(partial, `Result should have the expected partial ${partial}`);\n return result.data!;\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAGA,IAAAA,gBAAkB;AAClB,kBAAqD;AACrD,oBAA6D;AAC7D,wBAAqC;AACrC,qBAAoC;AACpC,IAAAC,cAAgG;AAChG,kBAA2D;AAC3D,sBAAqC;;;ACTrC,mBAA+C;AAC/C,4BAAO;AASP,eAAsB,sBAAsB,KAAgC,WAAmB,SAAiB,MAAM;AAEpH,QAAM,aACJ,QAAQ,IAAI,sBAAsB;AACpC,MAAI,CAAC,YAAY;AACf,YAAQ,MAAM,wDAAwD;AACtE,WAAO;AAAA,EACT;AAEA,MAAI;AAEF,UAAM,SAAS,IAAI,IAAI,OAAO,OAAO,YAAY,IAAI,OAAO,QAAQ;AAGpE,UAAM,UAAU,MAAM,IAAI,OAAO,SAAS,WAAW,OAAO,OAAO;AACnE,YAAQ,IAAI,0BAA0B,OAAO,OAAO,EAAE;AACtD,YAAQ,IAAI,0BAA0B,IAAI,OAAO,YAAY,OAAO,CAAC,MAAM;AAG3E,UAAM,eAAe,IAAI,OAAO,WAAW,MAAM;AACjD,QAAI,UAAU,cAAc;AAC1B,cAAQ;AAAA,QACN,oEAAoE,IAAI,OAAO,YAAY,OAAO,CAAC;AAAA,MACrG;AACA,aAAO;AAAA,IACT;AAGA,YAAQ,IAAI,WAAW,MAAM,WAAW,SAAS,KAAK;AACtD,UAAM,KAAK,MAAM,OAAO,gBAAgB;AAAA,MACtC,IAAI;AAAA,MACJ,OAAO;AAAA,IACT,CAAC;AAED,YAAQ,IAAI,2BAA2B,GAAG,IAAI,EAAE;AAChD,YAAQ,IAAI,6BAA6B;AAGzC,UAAM,UAAU,MAAM,GAAG,KAAK;AAC9B,YAAQ,IAAI,kCAAkC,SAAS,WAAW,EAAE;AACpE,YAAQ,IAAI,qBAAqB,MAAM,WAAW,SAAS,EAAE;AAE7D,WAAO;AAAA,EACT,SAAS,OAAO;AACd,YAAQ,MAAM,wBAAwB,KAAK;AAC3C,WAAO;AAAA,EACT;AACF;AAQA,eAAsB,6BAA6B,KAAgC,eAAuB;AAExG,QAAM,gBAAgB,MAAM,IAAI,OAAO,SAAS,WAAW,aAAa;AACxE,UAAQ,IAAI,mBAAmB,IAAI,OAAO,YAAY,aAAa,CAAC,MAAM;AAE1E,MAAI,gBAAgB,IAAI,OAAO,WAAW,GAAG,GAAG;AAC9C,YAAQ,IAAI,8CAA8C,aAAa,KAAK;AAC5E,UAAM,UAAU,MAAM,sBAAsB,KAAK,aAAa;AAC9D,QAAI,SAAS;AACX,YAAM,aAAa,MAAM,IAAI,OAAO,SAAS,WAAW,aAAa;AACrE,cAAQ,IAAI,uBAAuB,IAAI,OAAO,YAAY,UAAU,CAAC,MAAM;AAAA,IAC7E,OAAO;AACL,cAAQ,MAAM,kBAAkB,aAAa,EAAE;AAAA,IACjD;AAAA,EACF;AACF;;;ACjFO,IAAM,wBAAwB;AAC9B,IAAM,+BAA+B;AACrC,IAAM,0BAA0B;AAEhC,IAAM,uBAAuB;AAC7B,IAAM,oBAAoB;AAC1B,IAAM,4BAA4B;AAClC,IAAM,gCAAgC;AACtC,IAAM,mBAAmB;AACzB,IAAM,mCAAmC;;;ACThD,IAAAC,gBAA+C;AAC/C,mBAAkB;AAClB,oBAAyB;AAWzB,4BAMO;AAIP,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAG7C,QAAM,eAAe,KAAK,sBAAsB,8CAAwB,gBAAgB;AACxF,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,8CAAwB,KAAK,oBAAoB;AAGpG,QAAM,SAAS,MAAM,YAAY,WAAW,OAAO,OAAO;AAC1D,QAAM,OAAO,KAAK;AAGlB,QAAM,WAAW,MAAM,YAAY,OAAO;AAC1C,MAAI,CAAC,UAAU;AACb,UAAM,IAAI,MAAM,gCAAgC;AAAA,EAClD;AAEA,SAAO;AACT;AAEA,IAAM,gBAAgB,OAAO,QAAsD;AAEjF,QAAM,eAAe,KAAK,mBAAmB,sCAAgB,gBAAgB;AAC7E,QAAM,MAAM,MAAM,IAAI,OAAO,cAAc,sCAAgB,KAAK,iBAAiB;AAGjF,QAAM,SAAS,MAAM,IAAI,OAAO;AAChC,MAAI,CAAC,QAAQ;AACX,aAAS,0BAA0B,CAAC;AACpC,UAAM,IAAI,MAAM,wBAAwB;AAAA,EAC1C;AAEA,SAAO;AACT;AAEA,IAAM,uBAAuB,OAAO,QAAmC;AACrE,QAAM,eAAe,KAAK,2BAA2B,6CAAuB,gBAAgB;AAC5F,QAAM,aAAa,MAAM,IAAI,OAAO,cAAc,6CAAuB,KAAK,yBAAyB;AAEvG,QAAM,mBAAmB,MAAM,WAAW,OAAO;AACjD,MAAI,CAAC,kBAAkB;AACrB,aAAS,iCAAiC,CAAC;AAC3C,UAAM,IAAI,MAAM,+BAA+B;AAAA,EACjD;AAEA,SAAO;AACT;AAEA,IAAM,2BAA2B,OAAO,KAAgC,QAAkB;AACxF,QAAM,eAAe,KAAK,+BAA+B,iDAA2B,gBAAgB;AACpG,QAAM,iBAAiB,MAAM,IAAI,OAAO,cAAc,iDAA2B,KAAK,6BAA6B;AAGnH,QAAM,SAAS,MAAM,eAAe,WAAW,sBAAsB,MAAM,IAAI,WAAW,CAAC;AAC3F,QAAM,OAAO,KAAK;AAGlB,QAAM,uBAAuB,MAAM,eAAe,OAAO;AACzD,MAAI,CAAC,sBAAsB;AACzB,aAAS,qCAAqC,CAAC;AAC/C,UAAM,IAAI,MAAM,mCAAmC;AAAA,EACrD;AAEA,SAAO;AACT;AAEA,IAAM,wBAAwB,OAAO,QAAmC;AACtE,QAAM,eAAe,KAAK,kBAAkB,sCAAgB,gBAAgB;AAC5E,QAAM,UAAU,MAAM,IAAI,OAAO,cAAc,sCAAgB,KAAK,gBAAgB;AACpF,QAAM,QAAQ,kBAAkB;AAChC,SAAO;AACT;AAIA,IAAM,uBAAuB,OAAO,QAAmC;AACrE,QAAM,mBAAmB,MAAM,IAAI,OAAO,UAAU,gCAAgC;AACpF,QAAM,IAAI,QAAQ,SAAS,KAAK,sBAAsB;AAAA,IACpD,iBAAiB;AAAA,IACjB,OAAO,IAAI,OAAO,WAAW,IAAI,EAAE,SAAS,EAAE;AAAA,EAChD,CAAC;AACH;AAIA,IAAM,oBAAoB,OAAO,aAAuB,QAAkB;AACxE,QAAM,WAAW,MAAM,YAAY,eAAe,MAAM,IAAI,WAAW,CAAC;AACxE,QAAM,SAAS,KAAK;AACtB;AAQO,IAAM,cAAc,OACzB,KACA,UAA2B;AAAA,EACzB,eAAe;AAAA,EACf,YAAY;AAAA,EACZ,QAAQ;AACV,MACG;AAEH,QAAM,YAAY,MAAM,oBAAoB,GAAG;AAC/C,MAAI,CAAC;AAAW;AAEhB,QAAM,kBAAkB,MAAM;AAC5B,QAAI,CAAC,QAAQ,QAAQ;AACnB,eAAS;AAAA,IACX;AAAA,EACF;AACA,QAAM,oBAAoB,CAAC,SAAiB,SAAS,MAAM;AACzD,QAAI,CAAC,QAAQ,QAAQ;AACnB,iBAAW,SAAS,MAAM;AAAA,IAC5B;AAAA,EACF;AACA,QAAM,uBAAuB,CAAC,cAAsB,YAAoB;AACtE,QAAI,CAAC,QAAQ,QAAQ;AACnB,oBAAc,cAAc,OAAO;AAAA,IACrC;AAAA,EACF;AACA,QAAM,oBAAoB,CAAC,SAAiB,SAAS,MAAM;AACzD,QAAI,CAAC,QAAQ,QAAQ;AACnB,iBAAW,SAAS,MAAM;AAAA,IAC5B;AAAA,EACF;AAGA,kBAAgB;AAChB,oBAAkB,aAAAC,QAAM,KAAK,sCAAsC,GAAG,CAAC;AACvE,kBAAgB;AAGhB,kBAAgB;AAChB,oBAAkB,2BAA2B,CAAC;AAG9C,QAAM,cAAc,MAAM,sBAAsB,GAAG;AACnD,uBAAqB,mBAAmB,MAAM,YAAY,WAAW,CAAC;AAEtE,QAAM,MAAM,MAAM,cAAc,GAAG;AACnC,uBAAqB,WAAW,MAAM,IAAI,WAAW,CAAC;AAEtD,QAAM,kBAAkB,aAAa,GAAG;AACxC,oBAAkB,kCAAkC,CAAC;AAErD,QAAM,qBAAqB,GAAG;AAC9B,oBAAkB,sBAAsB,gCAAgC,YAAY,CAAC;AAErF,QAAM,0BAA0B,MAAM,IAAI,OAAO,SAAS,WAAW,gCAAgC;AACrG,oBAAkB,gBAAgB,wBAAwB,SAAS,CAAC,IAAI,CAAC;AAEzE,QAAM,aAAa,MAAM,qBAAqB,GAAG;AACjD,uBAAqB,kBAAkB,MAAM,WAAW,WAAW,CAAC;AAEpE,QAAM,iBAAiB,MAAM,yBAAyB,KAAK,GAAG;AAC9D,uBAAqB,sBAAsB,MAAM,eAAe,WAAW,CAAC;AAE5E,MAAI,QAAQ,eAAe;AACzB,sBAAkB,8BAA8B,CAAC;AACjD,UAAM,UAAU,MAAM,sBAAsB,GAAG;AAC/C,yBAAqB,WAAW,MAAM,QAAQ,WAAW,CAAC;AAAA,EAC5D;AAGA,kBAAgB;AAChB,oBAAkB,aAAAA,QAAM,KAAK,qDAAqD,GAAG,CAAC;AAGtF,MAAI,QAAQ,YAAY;AACtB,oBAAgB;AAChB;AAAA,MACE;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAEA,kBAAgB;AAClB;AAIA,IAAM,iBAAiB,OAAO,KAAgC,SAAiB,aAAqB;AAClG,QAAM,IAAI,QAAQ,SAAS,KAAK,mBAAmB,CAAC,SAAS,QAAQ,CAAC;AACxE;AAIA,IAAM,sBAAsB,OAAO,QAAmC;AACpE,QAAM,UAAU,IAAI,QAAQ;AAC5B,QAAM,YAAY,YAAY;AAC9B,MAAI,CAAC;AAAW,eAAW,wEAAwE,OAAO,IAAI,CAAC;AAC/G,SAAO;AACT;AAIA,IAAM,WAAW,MAAM;AACrB,UAAQ,IAAI,EAAE;AAChB;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,aAAAA,QAAM,MAAM,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC/D;AAEA,IAAM,aAAa,CAAC,SAAiB,SAAS,MAAM;AAClD,UAAQ,IAAI,aAAAA,QAAM,KAAK,aAAAA,QAAM,OAAO,GAAG,KAAK,OAAO,MAAM,CAAC,cAAS,CAAC,GAAG,OAAO;AAChF;AAEA,IAAM,WAAW,CAAC,SAAiB,SAAS,MAAM;AAChD,UAAQ,IAAI,aAAAA,QAAM,IAAI,GAAG,KAAK,OAAO,MAAM,CAAC,UAAK,OAAO,EAAE,CAAC;AAC7D;AAEA,IAAM,gBAAgB,CAAC,cAAsB,YAAoB;AAC/D,QAAM,aAAa,GAAG,YAAY,YAAY,OAAO,EAAE;AACvD,aAAW,GAAG,UAAU,IAAI,aAAAA,QAAM,KAAK,OAAO,CAAC,EAAE;AACnD;;;ACjPA,IAAAC,gBAAkB;AAClB,IAAAC,gBAA+C;AAE/C,IAAAC,yBAAwC;AAExC,IAAM,6BAA6B,OAAO,QAAmC;AAE3E,QAAM,cAAc,MAAM,IAAI,OAAO,cAAc,+CAAwB,KAAK,oBAAoB;AAEpG,SAAO;AACT;AAEA,IAAM,oBAAoB,OAAO,QAAmC;AAClE,QAAM,cAAc,MAAM,2BAA2B,GAAG;AACxD,SAAO,MAAM,YAAY,OAAO;AAClC;AAEA,IAAM,oBAAoB,OAAO,KAAgC,YAAqB;AACpF,QAAM,cAAc,MAAM,2BAA2B,GAAG;AACxD,QAAM,YAAY,UAAU,OAAO;AACrC;AAGA,IAAM,0BAA0B,CAAC,mBAA2B;AAC1D,UAAQ,IAAI,saAAwE;AACpF,UAAQ,IAAI,kCAAwB,cAAc,EAAE;AACpD,UAAQ,IAAI,saAAwE;AACtF;AAGA,IAAM,oBAAoB,MAAM;AAC9B,UAAQ,IAAI,saAAwE;AACtF;AAEO,IAAM,yBAAyB,OACpC,KACA,SACA,gBACG;AACH,MAAI;AACF,UAAM,mBAAmB,MAAM,kBAAkB,GAAG;AAEpD,UAAM,kBAAkB,KAAK,OAAO;AAGpC,QAAI,SAAS;AACX,8BAAwB,GAAG,cAAc,IAAI,cAAAC,QAAM,KAAK,WAAW,CAAC,YAAY,OAAO,EAAE;AAAA,IAC3F;AAGA,QAAI,CAAC,WAAW,kBAAkB;AAChC,wBAAkB;AAAA,IACpB;AAAA,EACF,SAAS,OAAO;AACd,YAAQ,IAAI,cAAAA,QAAM,IAAI,8BAA8B,GAAG,KAAK;AAAA,EAC9D;AACF;AAEO,IAAM,gBAAgB,OAC3B,KACA,aACA,YACG;AACH,QAAM,mBAAmB,MAAM,kBAAkB,GAAG;AAEpD,QAAM,kBAAkB,KAAK,IAAI;AACjC,0BAAwB,IAAI,cAAAA,QAAM,KAAK,WAAW,CAAC,SAAS;AAC5D,QAAM,QAAQ;AACd,oBAAkB;AAGlB,MAAI,CAAC,kBAAkB;AACrB,UAAM,kBAAkB,KAAK,KAAK;AAAA,EACpC;AACF;;;ACzEA,kBAAuB;AACvB,IAAAC,iBAAuB;AACvB,qCAA2C;AAE3C,IAAM,sBAAsB,OAAO,QAAgB,aAA6D;AAI9G,QAAM,WAAW,MAAM,SAAS,QAAQ,yBAAyB;AAGjE,QAAM,YAAY,SAAS,WAAW;AAGtC,MAAI,WAAW;AACb,YAAQ,IAAI,GAAG,MAAM,iCAAiC;AAAA,EACxD;AAEA,SAAO;AACT;AAEO,IAAM,oBAAoB,OAAO,UAA0D,WAAmB;AAEnH,MAAI,MAAM,oBAAoB,kBAAkB,MAAM,QAAQ;AAAG;AAGjE,QAAM,cAAc,IAAI,sBAAO;AAAA,IAC7B;AAAA,IACA,CAAC,sDAAsD;AAAA,IACvD;AAAA,EACF;AAGA,QAAM,YAAY,MAAM,YAAY,YAAY,MAAM;AAEtD,SAAO;AACT;AAEO,IAAM,0BAA0B,OACrC,UACA,WACG;AAEH,MAAI,MAAM,oBAAoB,wBAAwB,MAAM,QAAQ;AAAG;AAGvE,QAAM,cAAc,IAAI,sBAAO;AAAA,IAC7B;AAAA,IACA,CAAC,qDAAqD;AAAA,IACtD;AAAA,EACF;AAGA,QAAM,kBAAkB,MAAM,YAAY,cAAc,MAAM;AAE9D,SAAO;AACT;AAEO,IAAM,uBAAuB,OAClC,UACA,QACA,kBACG;AAEH,MAAI,MAAM,oBAAoB,qBAAqB,MAAM,QAAQ;AAAG;AAGpE,QAAM,kBAAkB,MAAM,wBAAwB,UAAU,MAAM;AACtE,0BAAO,eAAe,EAAE,MAAM,MAAM,0BAA0B;AAG9D,QAAM,YAAY,MAAM,kBAAkB,UAAU,MAAM;AAC1D,0BAAO,SAAS,EAAE,MAAM,eAAe,8BAA8B;AACvE;;;AC1EA,iBAA4B;AAC5B,IAAAC,eAAuB;AAEhB,IAAM,oBAAoB,CAAI,QAAmB,iBAAyB;AAC/E,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,OAAO,2BAA2B;AAC/D,2BAAO,OAAO,KAAK,EAAE,GAAG,QAAQ,cAAc,uCAAuC,YAAY,EAAE;AACrG;AAEO,IAAM,sBAAsB,CAAI,WAAyB;AAC9D,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,SAAO,OAAO;AAChB;AAEO,IAAM,oBAAoB,CAAI,QAAmB,UAAgB;AACtE,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,2BAAO,OAAO,IAAI,EAAE,GAAG,GAAG,OAAO,yCAAyC,KAAK,EAAE;AACjF,SAAO,OAAO;AAChB;AAEO,IAAM,2BAA2B,CAAI,QAAmB,YAA2B;AACxF,2BAAO,OAAO,OAAO,EAAE,GAAG,GAAG,MAAM,4BAA4B;AAC/D,2BAAO,OAAO,IAAI,EAAE,GAAG,QAAQ,SAAS,2CAA2C,OAAO,EAAE;AAC5F,SAAO,OAAO;AAChB;;;ANMA,IAAAC,yBAMO;AAAA,IA0BP,4BAAa,CAAC,QAAQ,eAAe;AAEnC,MAAI,WAAW,YAAY,WAAW,SAAS,YAAY;AACzD;AAAA,EACF;AAGA,SAAO,SAAS,aAAa;AAAA,IAC3B,KAAK;AAAA,IACL,eAAe;AAAA,IACf,UAAU;AAAA,IACV,SAAS;AAAA,IACT,aAAa,CAAC;AAAA,IACd,KAAK;AAAA,IACL,UAAU;AAAA,MACR;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACF;AAAA,EACF;AAGA,MAAI,CAAC,WAAW,WAAW,aAAa,GAAG;AACzC,WAAO,SAAS,aAAa,IAAI;AAAA,MAC/B,KAAK,QAAQ,IAAI,mBAAmB;AAAA,MACpC,UAAU,QAAQ,IAAI,cAAc,CAAC,QAAQ,IAAI,WAAW,IAAI,CAAC;AAAA,MACjE,SAAS;AAAA,MACT,KAAK;AAAA,MACL,eAAe;AAAA,MACf,UAAU;AAAA,MACV,SAAS;AAAA,MACT,aAAa,CAAC;AAAA,IAChB;AAAA,EACF;AAGA,MAAI,CAAC,WAAW,WAAW,aAAa,GAAG;AACzC,WAAO,SAAS,aAAa,IAAI;AAAA,MAC/B,KAAK,QAAQ,IAAI,4BAA4B;AAAA,MAC7C,UAAU,QAAQ,IAAI,cAAc,CAAC,QAAQ,IAAI,WAAW,IAAI,CAAC;AAAA,MACjE,SAAS;AAAA,MACT,KAAK;AAAA,MACL,eAAe;AAAA,MACf,UAAU;AAAA,MACV,SAAS;AAAA,MACT,aAAa,CAAC;AAAA,IAChB;AAAA,EACF;AAGA,SAAO,WAAW;AAAA,IAChB,UAAU,WAAW,UAAU,YAAY;AAAA,IAC3C,YAAY,WAAW,UAAU,cAAc;AAAA,EACjD;AACF,CAAC;AAAA,IAMD,oBAAK,uBAAuB,iCAAiC,EAC1D,iBAAiB,WAAW,mBAAmB,QAAW,oBAAM,MAAM,EACtE,UAAU,OAAO,EAAE,QAAQ,GAAkB,QAAQ;AACpD,QAAM,EAAE,QAAQ,IAAI;AACpB,QAAM,EAAE,MAAM,YAAY,IAAI;AAE9B,MAAI,gBAAgB,cAAc;AAChC,YAAQ,KAAK,cAAAC,QAAM,OAAO,mDAAmD,CAAC;AAC9E;AAAA,EACF;AAEA,MAAI,CAAC,SAAS;AACZ,YAAQ,KAAK,cAAAA,QAAM,IAAI,+BAA+B,CAAC;AACvD;AAAA,EACF;AAEA,UAAQ,KAAK,cAAAA,QAAM,MAAM,iCAAiC,OAAO,EAAE,CAAC;AAEpE,MAAI;AACF,UAAM,sBAAsB,KAAK,OAAO;AAAA,EAC1C,SAAS,GAAG;AACV,YAAQ,KAAK,cAAAA,QAAM,IAAI,2CAA2C,OAAO,KAAK,CAAC,EAAE,CAAC;AAAA,EACpF;AACF,CAAC;AAAA,IAIH,oBAAK,yBAAyB,mDAAmD,EAC9E,iBAAiB,iBAAiB,kCAAkC,MAAM,oBAAM,OAAO,EACvF,iBAAiB,UAAU,8BAA8B,OAAO,oBAAM,OAAO,EAC7E,UAAU,OAAO,EAAE,eAAe,OAAO,GAAoB,QAAQ;AACpE,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe,iBAAiB;AAAA,IAChC,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,IAC9C,QAAQ,UAAU;AAAA,EACpB,CAAC;AACH,CAAC;AAAA,IAEH,oBAAK,6BAAW,kCAAkC,EAAE,UAAU,OAAO,CAAC,GAAG,KAAK,aAAa;AACzF,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe;AAAA,IACf,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,EAChD,CAAC;AACD,SAAO,SAAS;AAClB,CAAC;AAAA,IAED,oBAAK,6BAAW,kCAAkC,EAAE,UAAU,OAAO,CAAC,GAAG,KAAK,aAAa;AACzF,QAAM,YAAY,KAAK;AAAA,IACrB,eAAe;AAAA,IACf,YAAY,IAAI,OAAO,SAAS,cAAc;AAAA,EAChD,CAAC;AACD,SAAO,SAAS;AAClB,CAAC;AAAA,IAID,oBAAK,8BAA8B,0CAA0C,EAC1E,SAAS,UAAU,6BAA6B,OAAO,oBAAM,OAAO,EACpE,UAAU,OAAO,EAAE,OAAO,GAAG,QAAQ;AACpC,QAAM,uBAAuB,KAAK,MAAM;AAC1C,CAAC;AAAA,IA+IH,iCAAkB,CAAC,QAAQ;AACzB,MAAI,WAAW;AAAA,IACb,sBAAsB,OAAO,WAAgC;AAG3D,YAAM,cAAc,MAAM,IAAI,OAAO,sBAAsB,gCAAgC;AAC3F,YAAM,EAAE,cAAc,gBAAgB,IAAI,UAAM,sCAAqB,WAAW;AAGhF,YAAM,4BAA4B;AAAA,QAChC,GAAG;AAAA,QACH,WAAW;AAAA,UACT,GAAG,OAAO;AAAA,UACV;AAAA,QACF;AAAA,MACF;AAEA,iBAAO,kCAAqB,yBAAyB;AAAA,IACvD;AAAA,IACA,sBAAsB,CAAC,WAA2B;AAChD,iBAAO,kCAAqB,MAAM;AAAA,IACpC;AAAA,IACA,sBAAsB,OAAO,WAAgC;AAC3D,iBAAO,sCAAqB,MAAM;AAAA,IACpC;AAAA,IACA,qBAAqB,OAAU,WAA2C;AACxE,YAAM,gBAAgB,MAAM;AAC5B,aAAO,oBAAoB,aAAa;AAAA,IAC1C;AAAA,IACA,mBAAmB,OAAU,QAAwC,iBAAyB;AAC5F,YAAM,gBAAgB,MAAM;AAC5B,aAAO,kBAAkB,eAAe,YAAY;AAAA,IACtD;AAAA,IACA,mBAAmB,OAAU,QAAwC,UAAa;AAChF,YAAM,gBAAgB,MAAM;AAC5B,aAAO,kBAAkB,eAAe,KAAK;AAAA,IAC/C;AAAA,IACA,0BAA0B,OAAU,QAAwC,YAAwB;AAClG,YAAM,gBAAgB,MAAM;AAC5B,aAAO,yBAAyB,eAAe,OAAO;AAAA,IACxD;AAAA,IACA,OAAO;AAAA,MACL,UAAU,OAAO,aAAqB,YAAiC;AACrE,eAAO,cAAc,KAAK,aAAa,OAAO;AAAA,MAChD;AAAA,MACA,YAAY,OAAO,gBAAyB;AAC1C,eAAO,uBAAuB,KAAK,MAAM,WAAW;AAAA,MACtD;AAAA,MACA,aAAa,YAAY;AACvB,eAAO,uBAAuB,KAAK,KAAK;AAAA,MAC1C;AAAA,MACA,aAAa,OAAO,YAA6B;AAC/C,eAAO,YAAY,KAAK,OAAO;AAAA,MACjC;AAAA,MACA,cAAc,OAAO,WAAmB;AACtC,cAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAC7C,eAAO,kBAAkB,OAAO,UAAU,MAAM;AAAA,MAClD;AAAA,MACA,iBAAiB,OAAO,QAAgB,kBAA0B;AAChE,cAAM,CAAC,MAAM,IAAI,MAAM,IAAI,OAAO,WAAW;AAC7C,eAAO,qBAAqB,OAAO,UAAU,QAAQ,aAAa;AAAA,MACpE;AAAA,IACF;AAAA,EACF;AACF,CAAC;","names":["import_chalk","import_sdk","import_types","chalk","import_chalk","import_types","import_mock_contracts","chalk","import_ethers","import_chai","import_mock_contracts","chalk"]}
package/dist/index.mjs ADDED
@@ -0,0 +1,526 @@
1
+ // src/index.ts
2
+ import chalk3 from "chalk";
3
+ import "viem";
4
+ import { extendConfig, extendEnvironment, task, types } from "hardhat/config";
5
+ import { TASK_TEST, TASK_NODE } from "hardhat/builtin-tasks/task-names";
6
+ import "@nomicfoundation/hardhat-ethers/signers";
7
+ import "@cofhe/sdk";
8
+ import { createCofhesdkClient, createCofhesdkConfig } from "@cofhe/sdk/node";
9
+ import { HardhatSignerAdapter } from "@cofhe/sdk/adapters";
10
+
11
+ // src/fund.ts
12
+ import "hardhat/types";
13
+ import "@nomicfoundation/hardhat-ethers";
14
+ async function localcofheFundAccount(hre, toAddress, amount = "10") {
15
+ const privateKey = 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
+ try {
21
+ const wallet = new hre.ethers.Wallet(privateKey, hre.ethers.provider);
22
+ const balance = await hre.ethers.provider.getBalance(wallet.address);
23
+ console.log(`Funder wallet address: ${wallet.address}`);
24
+ console.log(`Funder wallet balance: ${hre.ethers.formatEther(balance)} ETH`);
25
+ const amountToSend = hre.ethers.parseEther(amount);
26
+ if (balance < amountToSend) {
27
+ console.error(
28
+ `Error: Funder wallet doesn't have enough funds. Current balance: ${hre.ethers.formatEther(balance)} ETH`
29
+ );
30
+ return null;
31
+ }
32
+ console.log(`Sending ${amount} ETH to ${toAddress}...`);
33
+ const tx = await wallet.sendTransaction({
34
+ to: toAddress,
35
+ value: amountToSend
36
+ });
37
+ console.log(`Transaction sent! Hash: ${tx.hash}`);
38
+ console.log("Waiting for confirmation...");
39
+ const receipt = await tx.wait();
40
+ console.log(`Transaction confirmed in block ${receipt?.blockNumber}`);
41
+ console.log(`Successfully sent ${amount} ETH to ${toAddress}`);
42
+ return receipt;
43
+ } catch (error) {
44
+ console.error("Error sending funds:", error);
45
+ return null;
46
+ }
47
+ }
48
+ async function localcofheFundWalletIfNeeded(hre, walletAddress) {
49
+ const walletBalance = await hre.ethers.provider.getBalance(walletAddress);
50
+ console.log(`Wallet balance: ${hre.ethers.formatEther(walletBalance)} ETH`);
51
+ if (walletBalance < hre.ethers.parseEther("1")) {
52
+ console.log(`Wallet balance is less than 1 ETH. Funding ${walletAddress}...`);
53
+ const receipt = await localcofheFundAccount(hre, walletAddress);
54
+ if (receipt) {
55
+ const newBalance = await hre.ethers.provider.getBalance(walletAddress);
56
+ console.log(`Wallet new balance: ${hre.ethers.formatEther(newBalance)} ETH`);
57
+ } else {
58
+ console.error(`Failed to fund ${walletAddress}`);
59
+ }
60
+ }
61
+ }
62
+
63
+ // src/consts.ts
64
+ var TASK_COFHE_USE_FAUCET = "task:cofhe:usefaucet";
65
+ var TASK_COFHE_MOCKS_SET_LOG_OPS = "task:cofhe-mocks:setlogops";
66
+ var TASK_COFHE_MOCKS_DEPLOY = "task:cofhe-mocks:deploy";
67
+ var TASK_MANAGER_ADDRESS = "0xeA30c4B8b44078Bbf8a6ef5b9f1eC1626C7848D9";
68
+ var MOCKS_ACL_ADDRESS = "0x0000000000000000000000000000000000000400";
69
+ var MOCKS_ZK_VERIFIER_ADDRESS = "0x0000000000000000000000000000000000000100";
70
+ var MOCKS_QUERY_DECRYPTER_ADDRESS = "0x0000000000000000000000000000000000000200";
71
+ var TEST_BED_ADDRESS = "0x0000000000000000000000000000000000000300";
72
+ var MOCKS_ZK_VERIFIER_SIGNER_ADDRESS = "0x6E12D8C87503D4287c294f2Fdef96ACd9DFf6bd2";
73
+
74
+ // src/deploy.ts
75
+ import "hardhat/types";
76
+ import chalk from "chalk";
77
+ import "ethers";
78
+ import {
79
+ MockTaskManagerArtifact,
80
+ MockACLArtifact,
81
+ MockZkVerifierArtifact,
82
+ MockQueryDecrypterArtifact,
83
+ TestBedArtifact
84
+ } from "@cofhe/mock-contracts";
85
+ var deployMockTaskManager = async (hre) => {
86
+ const [signer] = await hre.ethers.getSigners();
87
+ await hardhatSetCode(hre, TASK_MANAGER_ADDRESS, MockTaskManagerArtifact.deployedBytecode);
88
+ const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact.abi, TASK_MANAGER_ADDRESS);
89
+ const initTx = await taskManager.initialize(signer.address);
90
+ await initTx.wait();
91
+ const tmExists = await taskManager.exists();
92
+ if (!tmExists) {
93
+ throw new Error("MockTaskManager does not exist");
94
+ }
95
+ return taskManager;
96
+ };
97
+ var deployMockACL = async (hre) => {
98
+ await hardhatSetCode(hre, MOCKS_ACL_ADDRESS, MockACLArtifact.deployedBytecode);
99
+ const acl = await hre.ethers.getContractAt(MockACLArtifact.abi, MOCKS_ACL_ADDRESS);
100
+ const exists = await acl.exists();
101
+ if (!exists) {
102
+ logError("MockACL does not exist", 2);
103
+ throw new Error("MockACL does not exist");
104
+ }
105
+ return acl;
106
+ };
107
+ var deployMockZkVerifier = async (hre) => {
108
+ await hardhatSetCode(hre, MOCKS_ZK_VERIFIER_ADDRESS, MockZkVerifierArtifact.deployedBytecode);
109
+ const zkVerifier = await hre.ethers.getContractAt(MockZkVerifierArtifact.abi, MOCKS_ZK_VERIFIER_ADDRESS);
110
+ const zkVerifierExists = await zkVerifier.exists();
111
+ if (!zkVerifierExists) {
112
+ logError("MockZkVerifier does not exist", 2);
113
+ throw new Error("MockZkVerifier does not exist");
114
+ }
115
+ return zkVerifier;
116
+ };
117
+ var deployMockQueryDecrypter = async (hre, acl) => {
118
+ await hardhatSetCode(hre, MOCKS_QUERY_DECRYPTER_ADDRESS, MockQueryDecrypterArtifact.deployedBytecode);
119
+ const queryDecrypter = await hre.ethers.getContractAt(MockQueryDecrypterArtifact.abi, MOCKS_QUERY_DECRYPTER_ADDRESS);
120
+ const initTx = await queryDecrypter.initialize(TASK_MANAGER_ADDRESS, await acl.getAddress());
121
+ await initTx.wait();
122
+ const queryDecrypterExists = await queryDecrypter.exists();
123
+ if (!queryDecrypterExists) {
124
+ logError("MockQueryDecrypter does not exist", 2);
125
+ throw new Error("MockQueryDecrypter does not exist");
126
+ }
127
+ return queryDecrypter;
128
+ };
129
+ var deployTestBedContract = async (hre) => {
130
+ await hardhatSetCode(hre, TEST_BED_ADDRESS, TestBedArtifact.deployedBytecode);
131
+ const testBed = await hre.ethers.getContractAt(TestBedArtifact.abi, TEST_BED_ADDRESS);
132
+ await testBed.waitForDeployment();
133
+ return testBed;
134
+ };
135
+ var fundZkVerifierSigner = async (hre) => {
136
+ const zkVerifierSigner = await hre.ethers.getSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
137
+ await hre.network.provider.send("hardhat_setBalance", [
138
+ zkVerifierSigner.address,
139
+ "0x" + hre.ethers.parseEther("10").toString(16)
140
+ ]);
141
+ };
142
+ var setTaskManagerACL = async (taskManager, acl) => {
143
+ const setAclTx = await taskManager.setACLContract(await acl.getAddress());
144
+ await setAclTx.wait();
145
+ };
146
+ var deployMocks = async (hre, options = {
147
+ deployTestBed: true,
148
+ gasWarning: true,
149
+ silent: false
150
+ }) => {
151
+ const isHardhat = await checkNetworkAndSkip(hre);
152
+ if (!isHardhat)
153
+ return;
154
+ const logEmptyIfNoisy = () => {
155
+ if (!options.silent) {
156
+ logEmpty();
157
+ }
158
+ };
159
+ const logSuccessIfNoisy = (message, indent = 0) => {
160
+ if (!options.silent) {
161
+ logSuccess(message, indent);
162
+ }
163
+ };
164
+ const logDeploymentIfNoisy = (contractName, address) => {
165
+ if (!options.silent) {
166
+ logDeployment(contractName, address);
167
+ }
168
+ };
169
+ const logWarningIfNoisy = (message, indent = 0) => {
170
+ if (!options.silent) {
171
+ logWarning(message, indent);
172
+ }
173
+ };
174
+ logEmptyIfNoisy();
175
+ logSuccessIfNoisy(chalk.bold("cofhe-hardhat-plugin :: deploy mocks"), 0);
176
+ logEmptyIfNoisy();
177
+ logEmptyIfNoisy();
178
+ logSuccessIfNoisy("Mock contracts compiled", 1);
179
+ const taskManager = await deployMockTaskManager(hre);
180
+ logDeploymentIfNoisy("MockTaskManager", await taskManager.getAddress());
181
+ const acl = await deployMockACL(hre);
182
+ logDeploymentIfNoisy("MockACL", await acl.getAddress());
183
+ await setTaskManagerACL(taskManager, acl);
184
+ logSuccessIfNoisy("ACL address set in TaskManager", 2);
185
+ await fundZkVerifierSigner(hre);
186
+ logSuccessIfNoisy(`ZkVerifier signer (${MOCKS_ZK_VERIFIER_SIGNER_ADDRESS}) funded`, 1);
187
+ const zkVerifierSignerBalance = await hre.ethers.provider.getBalance(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
188
+ logSuccessIfNoisy(`ETH balance: ${zkVerifierSignerBalance.toString()}`, 2);
189
+ const zkVerifier = await deployMockZkVerifier(hre);
190
+ logDeploymentIfNoisy("MockZkVerifier", await zkVerifier.getAddress());
191
+ const queryDecrypter = await deployMockQueryDecrypter(hre, acl);
192
+ logDeploymentIfNoisy("MockQueryDecrypter", await queryDecrypter.getAddress());
193
+ if (options.deployTestBed) {
194
+ logSuccessIfNoisy("TestBed deployment enabled", 2);
195
+ const testBed = await deployTestBedContract(hre);
196
+ logDeploymentIfNoisy("TestBed", await testBed.getAddress());
197
+ }
198
+ logEmptyIfNoisy();
199
+ logSuccessIfNoisy(chalk.bold("cofhe-hardhat-plugin :: mocks deployed successfully"), 0);
200
+ if (options.gasWarning) {
201
+ logEmptyIfNoisy();
202
+ logWarningIfNoisy(
203
+ "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",
204
+ 0
205
+ );
206
+ }
207
+ logEmptyIfNoisy();
208
+ };
209
+ var hardhatSetCode = async (hre, address, bytecode) => {
210
+ await hre.network.provider.send("hardhat_setCode", [address, bytecode]);
211
+ };
212
+ var checkNetworkAndSkip = async (hre) => {
213
+ const network = hre.network.name;
214
+ const isHardhat = network === "hardhat";
215
+ if (!isHardhat)
216
+ logSuccess(`cofhe-hardhat-plugin - deploy mocks - skipped on non-hardhat network ${network}`, 0);
217
+ return isHardhat;
218
+ };
219
+ var logEmpty = () => {
220
+ console.log("");
221
+ };
222
+ var logSuccess = (message, indent = 1) => {
223
+ console.log(chalk.green(`${" ".repeat(indent)}\u2713 ${message}`));
224
+ };
225
+ var logWarning = (message, indent = 1) => {
226
+ console.log(chalk.bold(chalk.yellow(`${" ".repeat(indent)}\u26A0 NOTE:`)), message);
227
+ };
228
+ var logError = (message, indent = 1) => {
229
+ console.log(chalk.red(`${" ".repeat(indent)}\u2717 ${message}`));
230
+ };
231
+ var logDeployment = (contractName, address) => {
232
+ const paddedName = `${contractName} deployed`.padEnd(36);
233
+ logSuccess(`${paddedName} ${chalk.bold(address)}`);
234
+ };
235
+
236
+ // src/logging.ts
237
+ import chalk2 from "chalk";
238
+ import "hardhat/types";
239
+ import { MockTaskManagerArtifact as MockTaskManagerArtifact2 } from "@cofhe/mock-contracts";
240
+ var getDeployedMockTaskManager = async (hre) => {
241
+ const taskManager = await hre.ethers.getContractAt(MockTaskManagerArtifact2.abi, TASK_MANAGER_ADDRESS);
242
+ return taskManager;
243
+ };
244
+ var getLoggingEnabled = async (hre) => {
245
+ const taskManager = await getDeployedMockTaskManager(hre);
246
+ return await taskManager.logOps();
247
+ };
248
+ var setLoggingEnabled = async (hre, enabled) => {
249
+ const taskManager = await getDeployedMockTaskManager(hre);
250
+ await taskManager.setLogOps(enabled);
251
+ };
252
+ var printLogsEnabledMessage = (closureMessage) => {
253
+ console.log("\u250C\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u252C\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500");
254
+ console.log(`\u2502 [COFHE-MOCKS] \u2502 ${closureMessage}`);
255
+ console.log("\u251C\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2534\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500");
256
+ };
257
+ var printLogsBlockEnd = () => {
258
+ console.log("\u2514\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500\u2500");
259
+ };
260
+ var mock_setLoggingEnabled = async (hre, enabled, closureName) => {
261
+ try {
262
+ const initiallyEnabled = await getLoggingEnabled(hre);
263
+ await setLoggingEnabled(hre, enabled);
264
+ if (enabled) {
265
+ printLogsEnabledMessage(`${closureName ? `"${chalk2.bold(closureName)}" logs:` : "Logs:"}`);
266
+ }
267
+ if (!enabled && initiallyEnabled) {
268
+ printLogsBlockEnd();
269
+ }
270
+ } catch (error) {
271
+ console.log(chalk2.red("mock_setLoggingEnabled error"), error);
272
+ }
273
+ };
274
+ var mock_withLogs = async (hre, closureName, closure) => {
275
+ const initiallyEnabled = await getLoggingEnabled(hre);
276
+ await setLoggingEnabled(hre, true);
277
+ printLogsEnabledMessage(`"${chalk2.bold(closureName)}" logs:`);
278
+ await closure();
279
+ printLogsBlockEnd();
280
+ if (!initiallyEnabled) {
281
+ await setLoggingEnabled(hre, false);
282
+ }
283
+ };
284
+
285
+ // src/utils.ts
286
+ import { expect } from "chai";
287
+ import { ethers } from "ethers";
288
+ import "@nomicfoundation/hardhat-ethers/internal/hardhat-ethers-provider";
289
+ var mock_checkIsTestnet = async (fnName, provider) => {
290
+ const bytecode = await provider.getCode(MOCKS_ZK_VERIFIER_ADDRESS);
291
+ const isTestnet = bytecode.length === 0;
292
+ if (isTestnet) {
293
+ console.log(`${fnName} - skipped on non-testnet chain`);
294
+ }
295
+ return isTestnet;
296
+ };
297
+ var mock_getPlaintext = async (provider, ctHash) => {
298
+ if (await mock_checkIsTestnet(mock_getPlaintext.name, provider))
299
+ return;
300
+ const taskManager = new ethers.Contract(
301
+ TASK_MANAGER_ADDRESS,
302
+ ["function mockStorage(uint256) view returns (uint256)"],
303
+ provider
304
+ );
305
+ const plaintext = await taskManager.mockStorage(ctHash);
306
+ return plaintext;
307
+ };
308
+ var mock_getPlaintextExists = async (provider, ctHash) => {
309
+ if (await mock_checkIsTestnet(mock_getPlaintextExists.name, provider))
310
+ return;
311
+ const taskManager = new ethers.Contract(
312
+ TASK_MANAGER_ADDRESS,
313
+ ["function inMockStorage(uint256) view returns (bool)"],
314
+ provider
315
+ );
316
+ const plaintextExists = await taskManager.inMockStorage(ctHash);
317
+ return plaintextExists;
318
+ };
319
+ var mock_expectPlaintext = async (provider, ctHash, expectedValue) => {
320
+ if (await mock_checkIsTestnet(mock_expectPlaintext.name, provider))
321
+ return;
322
+ const plaintextExists = await mock_getPlaintextExists(provider, ctHash);
323
+ expect(plaintextExists).equal(true, "Plaintext does not exist");
324
+ const plaintext = await mock_getPlaintext(provider, ctHash);
325
+ expect(plaintext).equal(expectedValue, "Plaintext value is incorrect");
326
+ };
327
+
328
+ // src/expectResultUtils.ts
329
+ import "@cofhe/sdk";
330
+ import { expect as expect2 } from "chai";
331
+ var expectResultError = (result, errorPartial) => {
332
+ expect2(result.success).to.eq(false, "Result should be an error");
333
+ expect2(result.error).to.include(errorPartial, `Error should contain error partial: ${errorPartial}`);
334
+ };
335
+ var expectResultSuccess = (result) => {
336
+ expect2(result.success).to.eq(true, "Result should be a success");
337
+ return result.data;
338
+ };
339
+ var expectResultValue = (result, value) => {
340
+ expect2(result.success).to.eq(true, "Result should be a success");
341
+ expect2(result.data).to.eq(value, `Result should have the expected value ${value}`);
342
+ return result.data;
343
+ };
344
+ var expectResultPartialValue = (result, partial) => {
345
+ expect2(result.success).to.eq(true, "Result should be a success");
346
+ expect2(result.data).to.include(partial, `Result should have the expected partial ${partial}`);
347
+ return result.data;
348
+ };
349
+
350
+ // src/index.ts
351
+ import {
352
+ MockACLArtifact as MockACLArtifact2,
353
+ MockQueryDecrypterArtifact as MockQueryDecrypterArtifact2,
354
+ MockTaskManagerArtifact as MockTaskManagerArtifact3,
355
+ MockZkVerifierArtifact as MockZkVerifierArtifact2,
356
+ TestBedArtifact as TestBedArtifact2
357
+ } from "@cofhe/mock-contracts";
358
+ extendConfig((config, userConfig) => {
359
+ if (userConfig.networks && userConfig.networks.localcofhe) {
360
+ return;
361
+ }
362
+ config.networks.localcofhe = {
363
+ gas: "auto",
364
+ gasMultiplier: 1.2,
365
+ gasPrice: "auto",
366
+ timeout: 1e4,
367
+ httpHeaders: {},
368
+ url: "http://127.0.0.1:42069",
369
+ accounts: [
370
+ "0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80",
371
+ "0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d",
372
+ "0x5de4111afa1a4b94908f83103eb1f1706367c2e68ca870fc3fb9a804cdab365a",
373
+ "0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6"
374
+ ]
375
+ };
376
+ if (!userConfig.networks?.["eth-sepolia"]) {
377
+ config.networks["eth-sepolia"] = {
378
+ url: process.env.SEPOLIA_RPC_URL ?? "https://ethereum-sepolia.publicnode.com",
379
+ accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],
380
+ chainId: 11155111,
381
+ gas: "auto",
382
+ gasMultiplier: 1.2,
383
+ gasPrice: "auto",
384
+ timeout: 6e4,
385
+ httpHeaders: {}
386
+ };
387
+ }
388
+ if (!userConfig.networks?.["arb-sepolia"]) {
389
+ config.networks["arb-sepolia"] = {
390
+ url: process.env.ARBITRUM_SEPOLIA_RPC_URL ?? "https://sepolia-rollup.arbitrum.io/rpc",
391
+ accounts: process.env.PRIVATE_KEY ? [process.env.PRIVATE_KEY] : [],
392
+ chainId: 421614,
393
+ gas: "auto",
394
+ gasMultiplier: 1.2,
395
+ gasPrice: "auto",
396
+ timeout: 6e4,
397
+ httpHeaders: {}
398
+ };
399
+ }
400
+ config.cofhesdk = {
401
+ logMocks: userConfig.cofhesdk?.logMocks ?? true,
402
+ gasWarning: userConfig.cofhesdk?.gasWarning ?? true
403
+ };
404
+ });
405
+ task(TASK_COFHE_USE_FAUCET, "Fund an account from the funder").addOptionalParam("address", "Address to fund", void 0, types.string).setAction(async ({ address }, hre) => {
406
+ const { network } = hre;
407
+ const { name: networkName } = network;
408
+ if (networkName !== "localcofhe") {
409
+ console.info(chalk3.yellow(`Programmatic faucet only supported for localcofhe`));
410
+ return;
411
+ }
412
+ if (!address) {
413
+ console.info(chalk3.red(`Failed to get address to fund`));
414
+ return;
415
+ }
416
+ console.info(chalk3.green(`Getting funds from faucet for ${address}`));
417
+ try {
418
+ await localcofheFundAccount(hre, address);
419
+ } catch (e) {
420
+ console.info(chalk3.red(`failed to get funds from localcofhe for ${address}: ${e}`));
421
+ }
422
+ });
423
+ task(TASK_COFHE_MOCKS_DEPLOY, "Deploys the mock contracts on the Hardhat network").addOptionalParam("deployTestBed", "Whether to deploy the test bed", true, types.boolean).addOptionalParam("silent", "Whether to suppress output", false, types.boolean).setAction(async ({ deployTestBed, silent }, hre) => {
424
+ await deployMocks(hre, {
425
+ deployTestBed: deployTestBed ?? true,
426
+ gasWarning: hre.config.cofhesdk.gasWarning ?? true,
427
+ silent: silent ?? false
428
+ });
429
+ });
430
+ task(TASK_TEST, "Deploy mock contracts on hardhat").setAction(async ({}, hre, runSuper) => {
431
+ await deployMocks(hre, {
432
+ deployTestBed: true,
433
+ gasWarning: hre.config.cofhesdk.gasWarning ?? true
434
+ });
435
+ return runSuper();
436
+ });
437
+ task(TASK_NODE, "Deploy mock contracts on hardhat").setAction(async ({}, hre, runSuper) => {
438
+ await deployMocks(hre, {
439
+ deployTestBed: true,
440
+ gasWarning: hre.config.cofhesdk.gasWarning ?? true
441
+ });
442
+ return runSuper();
443
+ });
444
+ task(TASK_COFHE_MOCKS_SET_LOG_OPS, "Set logging for the Mock CoFHE contracts").addParam("enable", "Whether to enable logging", false, types.boolean).setAction(async ({ enable }, hre) => {
445
+ await mock_setLoggingEnabled(hre, enable);
446
+ });
447
+ extendEnvironment((hre) => {
448
+ hre.cofhesdk = {
449
+ createCofhesdkConfig: async (config) => {
450
+ const zkvHhSigner = await hre.ethers.getImpersonatedSigner(MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
451
+ const { walletClient: zkvWalletClient } = await HardhatSignerAdapter(zkvHhSigner);
452
+ const configWithZkvWalletClient = {
453
+ ...config,
454
+ _internal: {
455
+ ...config._internal,
456
+ zkvWalletClient
457
+ }
458
+ };
459
+ return createCofhesdkConfig(configWithZkvWalletClient);
460
+ },
461
+ createCofhesdkClient: (config) => {
462
+ return createCofhesdkClient(config);
463
+ },
464
+ hardhatSignerAdapter: async (signer) => {
465
+ return HardhatSignerAdapter(signer);
466
+ },
467
+ expectResultSuccess: async (result) => {
468
+ const awaitedResult = await result;
469
+ return expectResultSuccess(awaitedResult);
470
+ },
471
+ expectResultError: async (result, errorPartial) => {
472
+ const awaitedResult = await result;
473
+ return expectResultError(awaitedResult, errorPartial);
474
+ },
475
+ expectResultValue: async (result, value) => {
476
+ const awaitedResult = await result;
477
+ return expectResultValue(awaitedResult, value);
478
+ },
479
+ expectResultPartialValue: async (result, partial) => {
480
+ const awaitedResult = await result;
481
+ return expectResultPartialValue(awaitedResult, partial);
482
+ },
483
+ mocks: {
484
+ withLogs: async (closureName, closure) => {
485
+ return mock_withLogs(hre, closureName, closure);
486
+ },
487
+ enableLogs: async (closureName) => {
488
+ return mock_setLoggingEnabled(hre, true, closureName);
489
+ },
490
+ disableLogs: async () => {
491
+ return mock_setLoggingEnabled(hre, false);
492
+ },
493
+ deployMocks: async (options) => {
494
+ return deployMocks(hre, options);
495
+ },
496
+ getPlaintext: async (ctHash) => {
497
+ const [signer] = await hre.ethers.getSigners();
498
+ return mock_getPlaintext(signer.provider, ctHash);
499
+ },
500
+ expectPlaintext: async (ctHash, expectedValue) => {
501
+ const [signer] = await hre.ethers.getSigners();
502
+ return mock_expectPlaintext(signer.provider, ctHash, expectedValue);
503
+ }
504
+ }
505
+ };
506
+ });
507
+ export {
508
+ MockACLArtifact2 as MockACLArtifact,
509
+ MockQueryDecrypterArtifact2 as MockQueryDecrypterArtifact,
510
+ MockTaskManagerArtifact3 as MockTaskManagerArtifact,
511
+ MockZkVerifierArtifact2 as MockZkVerifierArtifact,
512
+ TestBedArtifact2 as TestBedArtifact,
513
+ deployMocks,
514
+ expectResultError,
515
+ expectResultPartialValue,
516
+ expectResultSuccess,
517
+ expectResultValue,
518
+ localcofheFundAccount,
519
+ localcofheFundWalletIfNeeded,
520
+ mock_expectPlaintext,
521
+ mock_getPlaintext,
522
+ mock_getPlaintextExists,
523
+ mock_setLoggingEnabled,
524
+ mock_withLogs
525
+ };
526
+ //# sourceMappingURL=index.mjs.map