@cofhe/hardhat-plugin 0.7.0 → 0.8.0

package/dist/index.js

@@ -30,6 +30,7 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var src_exports = {};
 __export(src_exports, {
+  MOCK_GAS_CONSUMED_TOPIC: () => MOCK_GAS_CONSUMED_TOPIC,
   TASK_COFHE_MOCKS_DEPLOY: () => TASK_COFHE_MOCKS_DEPLOY,
   TASK_COFHE_MOCKS_SET_LOG_OPS: () => TASK_COFHE_MOCKS_SET_LOG_OPS,
   TASK_COFHE_USE_FAUCET: () => TASK_COFHE_USE_FAUCET,
@@ -40,20 +41,23 @@ __export(src_exports, {
   localcofheFundAccount: () => localcofheFundAccount,
   localcofheFundWalletIfNeeded: () => localcofheFundWalletIfNeeded,
   mock_expectPlaintext: () => mock_expectPlaintext,
+  mock_getAdjustedGasBreakdown: () => mock_getAdjustedGasBreakdown,
+  mock_getAdjustedGasUsed: () => mock_getAdjustedGasUsed,
   mock_getPlaintext: () => mock_getPlaintext,
   mock_getPlaintextExists: () => mock_getPlaintextExists,
   mock_setLoggingEnabled: () => mock_setLoggingEnabled,
-  mock_withLogs: () => mock_withLogs
+  mock_withLogs: () => mock_withLogs,
+  printMockGasSummary: () => printMockGasSummary
 });
 module.exports = __toCommonJS(src_exports);
-var import_chalk3 = __toESM(require("chalk"));
+var import_chalk4 = __toESM(require("chalk"));
 var import_path = __toESM(require("path"));
 var import_fs = __toESM(require("fs"));
 var import_viem = require("viem");
 var import_config = require("hardhat/config");
 var import_task_names = require("hardhat/builtin-tasks/task-names");
 var import_signers = require("@nomicfoundation/hardhat-ethers/signers");
-var import_sdk3 = require("@cofhe/sdk");
+var import_sdk4 = require("@cofhe/sdk");
 var import_node = require("@cofhe/sdk/node");
 var import_adapters = require("@cofhe/sdk/adapters");
 
@@ -147,7 +151,7 @@ var getFixedMockContract = async (hre, artifact) => {
 };
 var mock_checkIsTestnet = async (fnName, provider) => {
   const bytecode = await provider.getCode(import_sdk.MOCKS_ZK_VERIFIER_ADDRESS);
-  const isTestnet = bytecode.length === 0;
+  const isTestnet = !bytecode || bytecode.length <= 2;
   if (isTestnet) {
     console.log(`${fnName} - skipped on non-testnet chain`);
   }
@@ -157,7 +161,7 @@ var mock_getPlaintext = async (provider, ctHash) => {
   if (await mock_checkIsTestnet(mock_getPlaintext.name, provider))
     return;
   const taskManager = new import_ethers.ethers.Contract(
-    import_sdk.TASK_MANAGER_ADDRESS,
+    import_sdk.MOCKS_TASK_MANAGER_ADDRESS,
     ["function mockStorage(uint256) view returns (uint256)"],
     provider
   );
@@ -168,7 +172,7 @@ var mock_getPlaintextExists = async (provider, ctHash) => {
   if (await mock_checkIsTestnet(mock_getPlaintextExists.name, provider))
     return;
   const taskManager = new import_ethers.ethers.Contract(
-    import_sdk.TASK_MANAGER_ADDRESS,
+    import_sdk.MOCKS_TASK_MANAGER_ADDRESS,
     ["function inMockStorage(uint256) view returns (bool)"],
     provider
   );
@@ -197,8 +201,12 @@ var deployMocks = async (hre, options = {
   }
   await hre.run("compile", { quiet: true });
   log("vv", import_chalk.default.bold("cofhe-hardhat-plugin :: deploy mocks"), 0);
+  const addressBook = await deployMockAddressBook(hre);
+  logDeployment("MockCoFHEAddressBook", await addressBook.getAddress());
   const taskManager = await deployMockTaskManager(hre);
   logDeployment("MockTaskManager", await taskManager.getAddress());
+  await registerTaskManager(addressBook, taskManager);
+  log("vv", "TaskManager registered in CoFHEAddressBook", 2);
   const acl = await deployMockACL(hre);
   logDeployment("MockACL", await acl.getAddress());
   const acpRevoker = await deployMockContractFromArtifact(hre, import_mock_contracts.ACPTimestampRevokerArtifact);
@@ -210,7 +218,7 @@ var deployMocks = async (hre, options = {
   await (await acl.setShareRegistry(await acpShareRegistry.getAddress())).wait();
   log("vv", "Share registry set in ACL", 2);
   await linkTaskManagerAndACL(taskManager, acl);
-  log("vv", "ACL address set in TaskManager", 2);
+  log("vv", "ACL address set in TaskManager, TaskManager address set in ACL", 2);
   await setVerifierSigner(taskManager);
   log("vv", "Verifier signer set", 2);
   await setDecryptResultSigner(taskManager);
@@ -221,13 +229,13 @@ var deployMocks = async (hre, options = {
   log("vv", `ETH balance: ${zkVerifierSignerBalance.toString()}`, 2);
   const zkVerifier = await deployMockZkVerifier(hre);
   logDeployment("MockZkVerifier", await zkVerifier.getAddress());
-  const thresholdNetwork = await deployMockThresholdNetwork(hre, acl);
+  const thresholdNetwork = await deployMockThresholdNetwork(hre, taskManager, acl);
   logDeployment("MockThresholdNetwork", await thresholdNetwork.getAddress());
   log("v", import_chalk.default.bold("cofhe-hardhat-plugin :: mocks deployed"), 0);
   if (options.gasWarning) {
     logEmpty("v");
     logWarning(
-      "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.gasWarning: false' in your hardhat config)",
+      "When using mocks, FHE operations (eg FHE.add / FHE.mul) consume extra gas replicating off-chain CoFHE work on-chain. Use hre.cofhe.getAdjustedGasUsed(receipt) or set 'cofhe.gasSummary: true' to see gas numbers with that overhead excluded.\n(Disable this warning by setting 'cofhe.gasWarning: false' in your hardhat config)",
       0
     );
   }
@@ -236,11 +244,25 @@ var deployMocks = async (hre, options = {
 var getIsHardhat = async (hre) => {
   return hre.network.name === "hardhat";
 };
+var deployMockAddressBook = async (hre) => {
+  const addressBook = await deployMockContractFromArtifact(hre, import_mock_contracts.MockCoFHEAddressBookArtifact);
+  const bookExists = await addressBook.exists();
+  if (!bookExists) {
+    throw new Error("MockCoFHEAddressBook does not exist");
+  }
+  return addressBook;
+};
+var registerTaskManager = async (addressBook, taskManager) => {
+  const setTmTx = await addressBook.setTm(import_sdk2.TASK_MANAGER_ID, await taskManager.getAddress());
+  await setTmTx.wait();
+};
 var deployMockTaskManager = async (hre) => {
   const [signer] = await hre.ethers.getSigners();
   const taskManager = await deployMockContractFromArtifact(hre, import_mock_contracts.MockTaskManagerArtifact);
   const initTx = await taskManager.initialize(signer.address);
   await initTx.wait();
+  const securityZonesTx = await taskManager.setSecurityZones(0, 1);
+  await securityZonesTx.wait();
   const tmExists = await taskManager.exists();
   if (!tmExists) {
     throw new Error("MockTaskManager does not exist");
@@ -269,6 +291,8 @@ var linkTaskManagerAndACL = async (taskManager, acl) => {
   const aclAddress = await acl.getAddress();
   const linkAclTx = await taskManager.setACLContract(aclAddress);
   await linkAclTx.wait();
+  const linkTaskManagerTx = await acl.setTaskManager(await taskManager.getAddress());
+  await linkTaskManagerTx.wait();
 };
 var setVerifierSigner = async (taskManager) => {
   const signer = new import_ethers2.Wallet(import_sdk2.MOCKS_ZK_VERIFIER_SIGNER_PRIVATE_KEY);
@@ -288,9 +312,9 @@ var deployMockZkVerifier = async (hre) => {
   }
   return zkVerifier;
 };
-var deployMockThresholdNetwork = async (hre, acl) => {
+var deployMockThresholdNetwork = async (hre, taskManager, acl) => {
   const thresholdNetwork = await deployMockContractFromArtifact(hre, import_mock_contracts.MockThresholdNetworkArtifact);
-  const initTx = await thresholdNetwork.initialize(import_sdk2.TASK_MANAGER_ADDRESS, await acl.getAddress());
+  const initTx = await thresholdNetwork.initialize(await taskManager.getAddress(), await acl.getAddress());
   await initTx.wait();
   const exists = await thresholdNetwork.exists();
   if (!exists) {
@@ -374,6 +398,154 @@ var mock_withLogs = async (hre, closureName, closure) => {
   }
 };
 
+// src/gas.ts
+var import_chalk3 = __toESM(require("chalk"));
+var import_ethers3 = require("ethers");
+var import_sdk3 = require("@cofhe/sdk");
+var MOCK_GAS_CONSUMED_TOPIC = import_ethers3.ethers.id("MockGasConsumed(uint256)");
+var mock_getAdjustedGasBreakdown = (receipt) => {
+  let mockGas = 0n;
+  let mockGasEvents = 0;
+  for (const log2 of receipt.logs ?? []) {
+    if (log2.address.toLowerCase() === import_sdk3.MOCKS_TASK_MANAGER_ADDRESS.toLowerCase() && log2.topics[0] === MOCK_GAS_CONSUMED_TOPIC) {
+      mockGas += BigInt(log2.data);
+      mockGasEvents += 1;
+    }
+  }
+  const gasUsed = BigInt(receipt.gasUsed);
+  return { gasUsed, mockGas, adjustedGasUsed: mockGas > gasUsed ? 0n : gasUsed - mockGas, mockGasEvents };
+};
+var mock_getAdjustedGasUsed = (receipt) => mock_getAdjustedGasBreakdown(receipt).adjustedGasUsed;
+var shortAddress = (address) => `${address.slice(0, 6)}..${address.slice(-4)}`;
+var buildSelectorNameMap = async (hre) => {
+  const selectorNames = /* @__PURE__ */ new Map();
+  for (const fqn of await hre.artifacts.getAllFullyQualifiedNames()) {
+    try {
+      const artifact = await hre.artifacts.readArtifact(fqn);
+      const iface = new import_ethers3.ethers.Interface(artifact.abi);
+      iface.forEachFunction((fn) => {
+        if (!selectorNames.has(fn.selector))
+          selectorNames.set(fn.selector, fn.format("sighash"));
+      });
+    } catch {
+    }
+  }
+  return selectorNames;
+};
+var buildAddressNameMap = async (hre, addresses) => {
+  const codeToName = /* @__PURE__ */ new Map();
+  for (const fqn of await hre.artifacts.getAllFullyQualifiedNames()) {
+    try {
+      const artifact = await hre.artifacts.readArtifact(fqn);
+      if (artifact.deployedBytecode && artifact.deployedBytecode !== "0x") {
+        codeToName.set(artifact.deployedBytecode.toLowerCase(), artifact.contractName);
+      }
+    } catch {
+    }
+  }
+  const addressNames = /* @__PURE__ */ new Map();
+  for (const address of addresses) {
+    try {
+      const code = await hre.network.provider.send("eth_getCode", [address, "latest"]);
+      addressNames.set(address, codeToName.get(code.toLowerCase()) ?? shortAddress(address));
+    } catch {
+      addressNames.set(address, shortAddress(address));
+    }
+  }
+  return addressNames;
+};
+var printMockGasSummary = async (hre) => {
+  if (hre.network.name !== "hardhat")
+    return;
+  let logs;
+  try {
+    logs = await hre.network.provider.send("eth_getLogs", [
+      { fromBlock: "0x0", toBlock: "latest", address: import_sdk3.MOCKS_TASK_MANAGER_ADDRESS, topics: [MOCK_GAS_CONSUMED_TOPIC] }
+    ]);
+  } catch {
+    return;
+  }
+  if (!logs || logs.length === 0)
+    return;
+  const mockGasPerTx = /* @__PURE__ */ new Map();
+  for (const log2 of logs) {
+    mockGasPerTx.set(log2.transactionHash, (mockGasPerTx.get(log2.transactionHash) ?? 0n) + BigInt(log2.data));
+  }
+  const aggregates = /* @__PURE__ */ new Map();
+  for (const [txHash, mockGas] of mockGasPerTx) {
+    try {
+      const receipt = await hre.network.provider.send("eth_getTransactionReceipt", [txHash]);
+      const tx = await hre.network.provider.send("eth_getTransactionByHash", [txHash]);
+      if (!receipt)
+        continue;
+      const target = tx?.to ?? receipt.contractAddress;
+      if (!target)
+        continue;
+      const to = import_ethers3.ethers.getAddress(target);
+      const selector = tx?.to ? (tx.input ?? "0x").slice(0, 10) : "(deployment)";
+      const key = `${to}:${selector}`;
+      const aggregate = aggregates.get(key) ?? { to, selector, calls: 0, totalGasUsed: 0n, totalMockGas: 0n };
+      aggregate.calls += 1;
+      aggregate.totalGasUsed += BigInt(receipt.gasUsed);
+      aggregate.totalMockGas += mockGas;
+      aggregates.set(key, aggregate);
+    } catch {
+    }
+  }
+  if (aggregates.size === 0)
+    return;
+  const selectorNames = await buildSelectorNameMap(hre);
+  const addressNames = await buildAddressNameMap(hre, [...new Set([...aggregates.values()].map((a) => a.to))]);
+  const merged = /* @__PURE__ */ new Map();
+  for (const a of aggregates.values()) {
+    const contract = addressNames.get(a.to) ?? shortAddress(a.to);
+    const method = selectorNames.get(a.selector) ?? a.selector;
+    const key = `${contract}:${method}`;
+    const row = merged.get(key) ?? { contract, method, calls: 0, totalGasUsed: 0n, totalMockGas: 0n };
+    row.calls += a.calls;
+    row.totalGasUsed += a.totalGasUsed;
+    row.totalMockGas += a.totalMockGas;
+    merged.set(key, row);
+  }
+  const rows = [...merged.values()].map((r) => {
+    const avgGasUsed = r.totalGasUsed / BigInt(r.calls);
+    const avgMockGas = r.totalMockGas / BigInt(r.calls);
+    return {
+      contract: r.contract,
+      method: r.method,
+      calls: r.calls,
+      avgGasUsed,
+      avgAdjusted: avgMockGas > avgGasUsed ? 0n : avgGasUsed - avgMockGas,
+      overheadPct: avgGasUsed === 0n ? 0 : Math.min(100, Number(avgMockGas * 100n / avgGasUsed))
+    };
+  }).sort((a, b) => a.contract.localeCompare(b.contract) || a.method.localeCompare(b.method));
+  const headers = ["Contract", "Method", "Calls", "Avg gas (mocks)", "Avg gas (adjusted)", "Mock overhead"];
+  const cells = rows.map((r) => [
+    r.contract,
+    r.method,
+    String(r.calls),
+    r.avgGasUsed.toLocaleString(),
+    r.avgAdjusted.toLocaleString(),
+    `${r.overheadPct}%`
+  ]);
+  const widths = headers.map((h, i) => Math.max(h.length, ...cells.map((row) => row[i].length)));
+  const line = (row) => `\u2502 ${row.map((cell, i) => cell.padEnd(widths[i])).join(" \u2502 ")} \u2502`;
+  const divider = (l, m, r) => `${l}${widths.map((w) => "\u2500".repeat(w + 2)).join(m)}${r}`;
+  console.log("");
+  console.log(import_chalk3.default.bold("[COFHE-MOCKS] Gas summary") + import_chalk3.default.dim(" \u2014 adjusted \u2248 cost excluding mock-only overhead"));
+  console.log(divider("\u250C", "\u252C", "\u2510"));
+  console.log(line(headers));
+  console.log(divider("\u251C", "\u253C", "\u2524"));
+  for (const row of cells)
+    console.log(line(row));
+  console.log(divider("\u2514", "\u2534", "\u2518"));
+  console.log(
+    import_chalk3.default.dim(
+      "Raw numbers include on-chain replication of off-chain CoFHE work; adjusted numbers estimate real-network cost. estimateGas remains unadjusted."
+    )
+  );
+};
+
 // src/index.ts
 var import_chains = require("@cofhe/sdk/chains");
 var import_mock_contracts3 = require("@cofhe/mock-contracts");
@@ -414,23 +586,22 @@ function generateMockContractsStubContent() {
   return [.../* @__PURE__ */ new Set([...existing, stubPath])];
 });
 (0, import_config.extendConfig)((config, userConfig) => {
-  if (userConfig.networks && userConfig.networks.localcofhe) {
-    return;
+  if (!userConfig.networks?.localcofhe) {
+    config.networks.localcofhe = {
+      gas: "auto",
+      gasMultiplier: 1.2,
+      gasPrice: "auto",
+      timeout: 1e4,
+      httpHeaders: {},
+      url: "http://127.0.0.1:42069",
+      accounts: [
+        "0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80",
+        "0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d",
+        "0x5de4111afa1a4b94908f83103eb1f1706367c2e68ca870fc3fb9a804cdab365a",
+        "0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6"
+      ]
+    };
   }
-  config.networks.localcofhe = {
-    gas: "auto",
-    gasMultiplier: 1.2,
-    gasPrice: "auto",
-    timeout: 1e4,
-    httpHeaders: {},
-    url: "http://127.0.0.1:42069",
-    accounts: [
-      "0xac0974bec39a17e36ba4a6b4d238ff944bacb478cbed5efcae784d7bf4f2ff80",
-      "0x59c6995e998f97a5a0044966f0945389dc9e86dae88c7a8412f4603b6b78690d",
-      "0x5de4111afa1a4b94908f83103eb1f1706367c2e68ca870fc3fb9a804cdab365a",
-      "0x7c852118294e51e653712a81e05800f419141751be58f605c371e15141b007a6"
-    ]
-  };
   if (!userConfig.networks?.["eth-sepolia"]) {
     config.networks["eth-sepolia"] = {
       url: process.env.SEPOLIA_RPC_URL ?? "https://ethereum-sepolia.publicnode.com",
@@ -458,6 +629,7 @@ function generateMockContractsStubContent() {
   config.cofhe = {
     logMocks: userConfig.cofhe?.logMocks ?? false,
     gasWarning: userConfig.cofhe?.gasWarning ?? false,
+    gasSummary: userConfig.cofhe?.gasSummary ?? false,
     mocksDeployVerbosity: userConfig.cofhe?.mocksDeployVerbosity ?? "v"
   };
 });
@@ -465,18 +637,18 @@ function generateMockContractsStubContent() {
   const { network } = hre;
   const { name: networkName } = network;
   if (networkName !== "localcofhe") {
-    console.info(import_chalk3.default.yellow(`Programmatic faucet only supported for localcofhe`));
+    console.info(import_chalk4.default.yellow(`Programmatic faucet only supported for localcofhe`));
     return;
   }
   if (!address) {
-    console.info(import_chalk3.default.red(`Failed to get address to fund`));
+    console.info(import_chalk4.default.red(`Failed to get address to fund`));
     return;
   }
-  console.info(import_chalk3.default.green(`Getting funds from faucet for ${address}`));
+  console.info(import_chalk4.default.green(`Getting funds from faucet for ${address}`));
   try {
     await localcofheFundAccount(hre, address);
   } catch (e) {
-    console.info(import_chalk3.default.red(`failed to get funds from localcofhe for ${address}: ${e}`));
+    console.info(import_chalk4.default.red(`failed to get funds from localcofhe for ${address}: ${e}`));
   }
 });
 (0, import_config.task)(TASK_COFHE_MOCKS_DEPLOY, "Deploys the mock contracts on the Hardhat network").setAction(
@@ -499,7 +671,11 @@ function generateMockContractsStubContent() {
       mocksDeployVerbosity: hre.config.cofhe.mocksDeployVerbosity
     });
   }
-  return runSuper();
+  const result = await runSuper();
+  if (hre.config.cofhe.gasSummary) {
+    await printMockGasSummary(hre);
+  }
+  return result;
 });
 (0, import_config.task)(import_task_names.TASK_NODE, "Deploy mock contracts on hardhat").setAction(async ({}, hre, runSuper) => {
   const skipAutoDeploy = (() => {
@@ -527,7 +703,7 @@ function buildHardhatPluginMocksConfig(mocksConfig) {
 (0, import_config.extendEnvironment)((hre) => {
   hre.cofhe = {
     createConfig: async (config) => {
-      const zkvHhSigner = await hre.ethers.getImpersonatedSigner(import_sdk3.MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
+      const zkvHhSigner = await hre.ethers.getImpersonatedSigner(import_sdk4.MOCKS_ZK_VERIFIER_SIGNER_ADDRESS);
       const { walletClient: zkvWalletClient } = await (0, import_adapters.HardhatSignerAdapter)(zkvHhSigner);
       const configWithZkvWalletClient = {
         environment: "hardhat",
@@ -565,6 +741,8 @@ function buildHardhatPluginMocksConfig(mocksConfig) {
       });
       return client;
     },
+    getAdjustedGasUsed: mock_getAdjustedGasUsed,
+    getAdjustedGasBreakdown: mock_getAdjustedGasBreakdown,
     mocks: {
       withLogs: async (closureName, closure) => {
         return mock_withLogs(hre, closureName, closure);
@@ -586,6 +764,7 @@ function buildHardhatPluginMocksConfig(mocksConfig) {
         const [signer] = await hre.ethers.getSigners();
         return mock_expectPlaintext(signer.provider, ctHash, expectedValue);
       },
+      getMockCoFHEAddressBook: async () => getFixedMockContract(hre, import_mock_contracts3.MockCoFHEAddressBookArtifact),
       getMockTaskManager: async () => getFixedMockContract(hre, import_mock_contracts3.MockTaskManagerArtifact),
       getMockACL: async () => {
         const taskManager = await getFixedMockContract(hre, import_mock_contracts3.MockTaskManagerArtifact);
@@ -600,6 +779,7 @@ function buildHardhatPluginMocksConfig(mocksConfig) {
 });
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  MOCK_GAS_CONSUMED_TOPIC,
   TASK_COFHE_MOCKS_DEPLOY,
   TASK_COFHE_MOCKS_SET_LOG_OPS,
   TASK_COFHE_USE_FAUCET,
@@ -610,9 +790,12 @@ function buildHardhatPluginMocksConfig(mocksConfig) {
   localcofheFundAccount,
   localcofheFundWalletIfNeeded,
   mock_expectPlaintext,
+  mock_getAdjustedGasBreakdown,
+  mock_getAdjustedGasUsed,
   mock_getPlaintext,
   mock_getPlaintextExists,
   mock_setLoggingEnabled,
-  mock_withLogs
+  mock_withLogs,
+  printMockGasSummary
 });
 //# sourceMappingURL=index.js.map