Merge pull request #15 from lidofinance/feat/hardfork

feat: use hardfork from env variable

Author: tamtamchik

.github/workflows/build_and_push_image.yml

@@ -29,6 +29,8 @@ jobs:
           username: ${{ github.repository_owner }}
           password: ${{ secrets.GITHUB_TOKEN }}
 
+      # Mainnet
+
       - name: Build and push mainnet fork image
         uses: docker/build-push-action@v6.1.0
         with:
@@ -39,16 +41,6 @@ jobs:
           build-args: |
             HH_CONFIG=hardhat.config.mainnet-fork.ts
 
-      - name: Build and push hoodi fork image
-        uses: docker/build-push-action@v6.1.0
-        with:
-          context: .
-          platforms: linux/amd64,linux/arm64
-          push: true
-          tags: ghcr.io/lidofinance/hardhat-node:${{ github.ref_name }}-hoodi-fork
-          build-args: |
-            HH_CONFIG=hardhat.config.hoodi-fork.ts
-
       - name: Build and push mainnet scratch image
         uses: docker/build-push-action@v6.1.0
         with:
@@ -59,12 +51,14 @@ jobs:
           build-args: |
             HH_CONFIG=hardhat.config.mainnet-scratch.ts
 
-      - name: Build and push mainnet shanghai image
+      # Hoodi
+
+      - name: Build and push hoodi fork image
         uses: docker/build-push-action@v6.1.0
         with:
           context: .
           platforms: linux/amd64,linux/arm64
           push: true
-          tags: ghcr.io/lidofinance/hardhat-node:${{ github.ref_name }}-shanghai
+          tags: ghcr.io/lidofinance/hardhat-node:${{ github.ref_name }}-hoodi-fork
           build-args: |
-            HH_CONFIG=hardhat.config.mainnet-fork-shanghai.ts
+            HH_CONFIG=hardhat.config.hoodi-fork.ts

base.config.ts

@@ -0,0 +1,28 @@
+import { HardhatUserConfig } from "hardhat/config";
+
+const baseConfig: HardhatUserConfig = {
+  solidity: "0.8.25",
+  mocha: { timeout: 5 * 60 * 1000 },
+  keystores: { path: "keystores" },
+  networks: {
+    hardhat: {
+      initialBaseFeePerGas: 0,
+      accounts: { count: 30 },
+    },
+  },
+};
+
+export const createConfig = (
+  overrides: Partial<HardhatUserConfig> = {}
+): HardhatUserConfig => ({
+  ...baseConfig,
+  ...overrides,
+  networks: {
+    ...baseConfig.networks,
+    ...overrides.networks,
+    hardhat: {
+      ...baseConfig.networks?.hardhat,
+      ...overrides.networks?.hardhat,
+    },
+  },
+});

hardhat.config.hoodi-fork.ts

@@ -1,28 +1,15 @@
 import "dotenv/config";
-import { HardhatUserConfig } from "hardhat/config";
-import "@nomicfoundation/hardhat-ethers";
 
 import networks from "./src/networks";
 
-const config: HardhatUserConfig = {
-  solidity: "0.8.25",
+import { createConfig } from "./base.config";
+
+export default createConfig({
   networks: {
     hardhat: {
       ...networks.maybeChainIdConfig(560048),
-      initialBaseFeePerGas: 0,
-      accounts: {
-        count: 30,
-      },
-      forking: {
-        url: networks.rpcUrl("eth", "hoodi"),
-      },
+      ...networks.hardforkConfig(),
+      forking: { url: networks.rpcUrl("eth", "hoodi") },
     },
   },
-  mocha: {
-    timeout: 5 * 60 * 1000,
-  },
-  keystores: {
-    path: "keystores",
-  },
-};
-export default config;
+});

hardhat.config.mainnet-fork-shanghai.ts

@@ -1,28 +1,15 @@
 import "dotenv/config";
-import { HardhatUserConfig } from "hardhat/config";
-import "@nomicfoundation/hardhat-ethers";
 
 import networks from "./src/networks";
 
-const config: HardhatUserConfig = {
-  solidity: "0.8.25",
+import { createConfig } from "./base.config";
+
+export default createConfig({
   networks: {
     hardhat: {
       ...networks.maybeChainIdConfig(1),
-      initialBaseFeePerGas: 0,
-      accounts: {
-        count: 30,
-      },
-      forking: {
-        url: networks.rpcUrl("eth", "mainnet"),
-      },
+      ...networks.hardforkConfig(),
+      forking: { url: networks.rpcUrl("eth", "mainnet") },
     },
   },
-  mocha: {
-    timeout: 5 * 60 * 1000,
-  },
-  keystores: {
-    path: "keystores",
-  },
-};
-export default config;
+});

hardhat.config.mainnet-fork.ts

@@ -1,18 +1,14 @@
 import "dotenv/config";
-import { HardhatUserConfig } from "hardhat/config";
-import "@nomicfoundation/hardhat-ethers";
+
 import networks from "./src/networks";
 
-const config: HardhatUserConfig = {
-  solidity: "0.8.25",
+import { createConfig } from "./base.config";
+
+export default createConfig({
   networks: {
     hardhat: {
       ...networks.maybeChainIdConfig(1),
-      initialBaseFeePerGas: 0,
-      accounts: {
-        count: 30,
-      },
+      ...networks.hardforkConfig(),
     },
   },
-};
-export default config;
+});

hardhat.config.mainnet-scratch.ts

@@ -34,6 +34,10 @@ function DONT_SET_CHAIN_ID() {
   return Boolean(process.env["DONT_SET_CHAIN_ID"]);
 }
 
+function HARDFORK() {
+  return process.env["HARDFORK"] || "cancun";
+}
+
 export default {
   ETH_RPC_URL,
   ARB_RPC_URL,
@@ -44,4 +48,5 @@ export default {
   LOCAL_OPT_RPC_URL,
   ALCHEMY_TOKEN,
   DONT_SET_CHAIN_ID,
+  HARDFORK,
 };

src/env.ts

@@ -3,36 +3,13 @@ import env from "./env";
 export type ChainName = "eth" | "arb" | "opt";
 export type NetworkName = "mainnet" | "hoodi";
 
-/**
- * Returns the RPC url for the given network name
- */
-function url(chainName: ChainName, networkName: NetworkName): string {
-  const url =
-    rpcUrl(chainName) ||
-    infuraUrl(chainName, networkName) ||
-    alchemyUrl(chainName, networkName);
-  if (!url) {
-    throw new Error(
-      "RPC node credential was not provided. Please, set one of RPC_URL, INFURA_TOKEN or ALCHEMY_TOKEN env variables"
-    );
-  }
-  return url;
-}
-
-function rpcUrl(chainName: ChainName) {
+function getChainRpcUrl(chainName: ChainName) {
   if (chainName === "eth") return env.ETH_RPC_URL();
   if (chainName === "arb") return env.ARB_RPC_URL();
   if (chainName === "opt") return env.OPT_RPC_URL();
 }
 
-function maybeChainIdConfig(chainId: number) {
-  if (env.DONT_SET_CHAIN_ID()) {
-    return {};
-  }
-  return { chainId };
-}
-
-function infuraUrl(chainName: ChainName, networkName: NetworkName) {
+function getInfuraUrl(chainName: ChainName, networkName: NetworkName) {
   const infuraToken = env.INFURA_TOKEN();
   const infuraNetworkDomain =
     chainName === "eth" && networkName === "mainnet"
@@ -43,14 +20,42 @@ function infuraUrl(chainName: ChainName, networkName: NetworkName) {
     : undefined;
 }
 
-function alchemyUrl(chainName: ChainName, networkName: NetworkName) {
+function getAlchemyUrl(chainName: ChainName, networkName: NetworkName) {
   const alchemyToken = env.ALCHEMY_TOKEN();
   return alchemyToken
     ? `https://${chainName}-${networkName}.g.alchemy.com/v2/${alchemyToken}`
     : undefined;
 }
 
+/**
+ * Returns the RPC url for the given network name
+ */
+function rpcUrl(chainName: ChainName, networkName: NetworkName): string {
+  const url =
+    getChainRpcUrl(chainName) ||
+    getInfuraUrl(chainName, networkName) ||
+    getAlchemyUrl(chainName, networkName);
+  if (!url) {
+    throw new Error(
+      "RPC node credential was not provided. Please, set one of RPC_URL, INFURA_TOKEN or ALCHEMY_TOKEN env variables"
+    );
+  }
+  return url;
+}
+
+function maybeChainIdConfig(chainId: number) {
+  if (env.DONT_SET_CHAIN_ID()) {
+    return {};
+  }
+  return { chainId };
+}
+
+function hardforkConfig() {
+  return { hardfork: env.HARDFORK() };
+}
+
 export default {
-  rpcUrl: url,
+  rpcUrl,
+  hardforkConfig,
   maybeChainIdConfig,
 };