Merge pull request #4875 from NomicFoundation/upgrade-viem-to-v2

Upgrade hardhat-viem to support viem@2

Author: fvictorio

.changeset/proud-peas-sort.md

@@ -0,0 +1,5 @@
+---
+"@nomicfoundation/hardhat-viem": major
+---
+
+Upgraded hardhat-viem to support viem@2

.github/dependabot.yml

@@ -9,7 +9,7 @@ updates:
       interval: "daily"
     commit-message:
       prefix: "bump"
-  - package-ecosystem: "pnpm"
+  - package-ecosystem: "npm"
     directory: "/"
     schedule:
       interval: "daily"

packages/hardhat-viem/README.md

@@ -154,7 +154,7 @@ const contractA = await hre.viem.deployContract(
   "contractName",
   ["arg1", 50, "arg3"],
   {
-    walletClient: secondWalletClient,
+    client: { wallet: secondWalletClient }
     gas: 1000000,
     value: parseEther("0.0001"),
     confirmations: 5, // 1 by default
@@ -185,7 +185,7 @@ const [_, secondWalletClient] = await hre.viem.getWalletClients();
 const contract = await hre.viem.getContractAt(
   "contractName",
   "0x1234567890123456789012345678901234567890",
-  { walletClient: secondWalletClient }
+  { client: { wallet: secondWalletClient } }
 );
 ```
 
@@ -210,7 +210,7 @@ const { contract: contractName, deploymentTransaction } =
     "contractName",
     ["arg1", 50, "arg3"],
     {
-      walletClient: secondWalletClient,
+      client: { wallet: secondWalletClient },
       gas: 1000000,
       value: parseEther("0.0001"),
     }

packages/hardhat-viem/package.json

@@ -66,12 +66,12 @@
     "sinon": "^9.0.0",
     "ts-node": "^10.8.0",
     "typescript": "~5.0.0",
-    "viem": "^1.15.1"
+    "viem": "^2.7.6"
   },
   "peerDependencies": {
     "hardhat": "workspace:^2.17.0",
     "typescript": "~5.0.0",
-    "viem": "^1.15.1"
+    "viem": "^2.7.6"
   },
   "dependencies": {
     "abitype": "^0.9.8",

packages/hardhat-viem/src/internal/contracts.ts

@@ -27,16 +27,10 @@ export async function deployContract(
   constructorArgs: any[] = [],
   config: DeployContractConfig = {}
 ): Promise<GetContractReturnType> {
-  const {
-    walletClient: configWalletClient,
-    publicClient: configPublicClient,
-    confirmations,
-    ...deployContractParameters
-  } = config;
+  const { client, confirmations, ...deployContractParameters } = config;
   const [publicClient, walletClient, contractArtifact] = await Promise.all([
-    configPublicClient ?? getPublicClient(network.provider),
-    configWalletClient ??
-      getDefaultWalletClient(network.provider, network.name),
+    client?.public ?? getPublicClient(network.provider),
+    client?.wallet ?? getDefaultWalletClient(network.provider, network.name),
     artifacts.readArtifact(contractName),
   ]);
 
@@ -120,15 +114,10 @@ export async function sendDeploymentTransaction(
   contract: GetContractReturnType;
   deploymentTransaction: GetTransactionReturnType;
 }> {
-  const {
-    walletClient: configWalletClient,
-    publicClient: configPublicClient,
-    ...deployContractParameters
-  } = config;
+  const { client, ...deployContractParameters } = config;
   const [publicClient, walletClient, contractArtifact] = await Promise.all([
-    configPublicClient ?? getPublicClient(network.provider),
-    configWalletClient ??
-      getDefaultWalletClient(network.provider, network.name),
+    client?.public ?? getPublicClient(network.provider),
+    client?.wallet ?? getDefaultWalletClient(network.provider, network.name),
     artifacts.readArtifact(contractName),
   ]);
 
@@ -202,8 +191,8 @@ export async function getContractAt(
   config: GetContractAtConfig = {}
 ): Promise<GetContractReturnType> {
   const [publicClient, walletClient, contractArtifact] = await Promise.all([
-    config.publicClient ?? getPublicClient(network.provider),
-    config.walletClient ??
+    config.client?.public ?? getPublicClient(network.provider),
+    config.client?.wallet ??
       getDefaultWalletClient(network.provider, network.name),
     artifacts.readArtifact(contractName),
   ]);
@@ -225,8 +214,10 @@ async function innerGetContractAt(
   const viem = await import("viem");
   const contract = viem.getContract({
     address,
-    publicClient,
-    walletClient,
+    client: {
+      public: publicClient,
+      wallet: walletClient,
+    },
     abi: contractAbi,
   });
 

packages/hardhat-viem/src/types.ts

@@ -13,13 +13,22 @@ export type TestClient = viemT.TestClient<
   viemT.Chain
 >;
 
+export type KeyedClient =
+  | {
+      public?: PublicClient;
+      wallet: WalletClient;
+    }
+  | {
+      public: PublicClient;
+      wallet?: WalletClient;
+    };
+
 export type TestClientMode = Parameters<
   typeof viemT.createTestClient
 >[0]["mode"];
 
 export interface SendTransactionConfig {
-  walletClient?: WalletClient;
-  publicClient?: PublicClient;
+  client?: KeyedClient;
   gas?: bigint;
   gasPrice?: bigint;
   maxFeePerGas?: bigint;
@@ -34,18 +43,12 @@ export interface DeployContractConfig extends SendTransactionConfig {
 export type SendDeploymentTransactionConfig = SendTransactionConfig;
 
 export interface GetContractAtConfig {
-  walletClient?: WalletClient;
-  publicClient?: PublicClient;
+  client?: KeyedClient;
 }
 
 export type GetContractReturnType<
   TAbi extends viemT.Abi | readonly unknown[] = viemT.Abi
-> = viemT.GetContractReturnType<
-  TAbi,
-  PublicClient,
-  WalletClient,
-  viemT.Address
->;
+> = viemT.GetContractReturnType<TAbi, Required<KeyedClient>, viemT.Address>;
 
 export type GetTransactionReturnType = viemT.GetTransactionReturnType<
   viemT.Chain,

packages/hardhat-viem/test/integration.ts

@@ -64,10 +64,10 @@ describe("Integration tests", function () {
         const fromAddress = fromWalletClient.account.address;
         const toAddress = toWalletClient.account.address;
 
-        const fromBalanceBefore: bigint = await publicClient.getBalance({
+        const fromBalanceBefore = await publicClient.getBalance({
           address: fromAddress,
         });
-        const toBalanceBefore: bigint = await publicClient.getBalance({
+        const toBalanceBefore = await publicClient.getBalance({
           address: toAddress,
         });
 
@@ -79,10 +79,10 @@ describe("Integration tests", function () {
         const receipt = await publicClient.waitForTransactionReceipt({ hash });
         const transactionFee = receipt.gasUsed * receipt.effectiveGasPrice;
 
-        const fromBalanceAfter: bigint = await publicClient.getBalance({
+        const fromBalanceAfter = await publicClient.getBalance({
           address: fromAddress,
         });
-        const toBalanceAfter: bigint = await publicClient.getBalance({
+        const toBalanceAfter = await publicClient.getBalance({
           address: toAddress,
         });
 
@@ -141,7 +141,7 @@ describe("Integration tests", function () {
         const contract = await this.hre.viem.deployContract(
           "WithoutConstructorArgs",
           [],
-          { walletClient: secondWalletClient }
+          { client: { wallet: secondWalletClient } }
         );
 
         const owner = await contract.read.getOwner();