Expose default publicClient in hardhat-viem

v-next/hardhat-viem/src/internal/clients.ts

@@ -1,5 +1,5 @@
 import type {
-  GetPublicClientReturnType,
+  PublicClientType,
   GetWalletClientReturnType,
   TestClient,
 } from "../types.js";
@@ -28,7 +28,7 @@ export async function getPublicClient<ChainTypeT extends ChainType | string>(
   provider: EthereumProvider,
   chainType: ChainTypeT,
   publicClientConfig?: Partial<ViemPublicClientConfig>,
-): Promise<GetPublicClientReturnType<ChainTypeT>> {
+): Promise<PublicClientType<ChainTypeT>> {
   const chain =
     publicClientConfig?.chain ?? (await getChain(provider, chainType));
   const parameters = {
@@ -48,7 +48,7 @@ export async function getPublicClient<ChainTypeT extends ChainType | string>(
 
   /* eslint-disable-next-line @typescript-eslint/consistent-type-assertions --
   We need to assert the type because TS gets confused with the conditional type */
-  return publicClient as GetPublicClientReturnType<ChainTypeT>;
+  return publicClient as PublicClientType<ChainTypeT>;
 }
 
 export async function getWalletClients<ChainTypeT extends ChainType | string>(

v-next/hardhat-viem/src/internal/contracts.ts

@@ -19,11 +19,12 @@ import { resolveLinkedBytecode } from "@ignored/hardhat-vnext-utils/bytecode";
 import { ensureError } from "@ignored/hardhat-vnext-utils/error";
 import { getContractAddress, getContract } from "viem";
 
-import { getDefaultWalletClient, getPublicClient } from "./clients.js";
+import { getDefaultWalletClient } from "./clients.js";
 
 export async function deployContract<ContractName extends string>(
   provider: EthereumProvider,
   artifactManager: ArtifactManager,
+  defaultPublicClient: PublicClient,
   contractName: ContractName,
   constructorArgs: unknown[] = [],
   deployContractConfig: DeployContractConfig = {},
@@ -47,8 +48,8 @@ export async function deployContract<ContractName extends string>(
     });
   }
 
-  const [publicClient, walletClient, { abi, bytecode }] = await Promise.all([
-    client?.public ?? getPublicClient(provider, "l1"),
+  const publicClient = client?.public ?? defaultPublicClient;
+  const [walletClient, { abi, bytecode }] = await Promise.all([
     client?.wallet ?? getDefaultWalletClient(provider, "l1"),
     getContractAbiAndBytecode(artifactManager, contractName, libraries),
   ]);
@@ -104,6 +105,7 @@ export async function deployContract<ContractName extends string>(
 export async function sendDeploymentTransaction<ContractName extends string>(
   provider: EthereumProvider,
   artifactManager: ArtifactManager,
+  defaultPublicClient: PublicClient,
   contractName: ContractName,
   constructorArgs: unknown[] = [],
   sendDeploymentTransactionConfig: SendDeploymentTransactionConfig = {},
@@ -116,8 +118,8 @@ export async function sendDeploymentTransaction<ContractName extends string>(
     libraries = {},
     ...deployContractParameters
   } = sendDeploymentTransactionConfig;
-  const [publicClient, walletClient, { abi, bytecode }] = await Promise.all([
-    client?.public ?? getPublicClient(provider, "l1"),
+  const publicClient = client?.public ?? defaultPublicClient;
+  const [walletClient, { abi, bytecode }] = await Promise.all([
     client?.wallet ?? getDefaultWalletClient(provider, "l1"),
     getContractAbiAndBytecode(artifactManager, contractName, libraries),
   ]);
@@ -167,12 +169,14 @@ export async function sendDeploymentTransaction<ContractName extends string>(
 export async function getContractAt<ContractName extends string>(
   provider: EthereumProvider,
   artifactManager: ArtifactManager,
+  defaultPublicClient: PublicClient,
   contractName: ContractName,
   address: ViemAddress,
   getContractAtConfig: GetContractAtConfig = {},
 ): Promise<ContractReturnType<ContractName>> {
-  const [publicClient, walletClient, artifact] = await Promise.all([
-    getContractAtConfig.client?.public ?? getPublicClient(provider, "l1"),
+  const publicClient =
+    getContractAtConfig.client?.public ?? defaultPublicClient;
+  const [walletClient, artifact] = await Promise.all([
     getContractAtConfig.client?.wallet ??
       getDefaultWalletClient(provider, "l1"),
     artifactManager.readArtifact(contractName),

v-next/hardhat-viem/src/internal/hook-handlers/network.ts

@@ -19,7 +19,7 @@ export default async (): Promise<Partial<NetworkHooks>> => {
     ) {
       const connection: NetworkConnection<ChainTypeT> = await next(context);
 
-      connection.viem = initializeViem(
+      connection.viem = await initializeViem(
         connection.chainType,
         connection.provider,
         context.artifacts,

v-next/hardhat-viem/src/internal/initialization.ts

@@ -15,12 +15,16 @@ import {
   sendDeploymentTransaction,
 } from "./contracts.js";
 
-export function initializeViem<ChainTypeT extends ChainType | string>(
+export async function initializeViem<ChainTypeT extends ChainType | string>(
   chainType: ChainTypeT,
   provider: EthereumProvider,
   artifactManager: ArtifactManager,
-): HardhatViemHelpers<ChainTypeT> {
+): Promise<HardhatViemHelpers<ChainTypeT>> {
+  const defaultPublicClient = await getPublicClient(provider, chainType);
+
   return {
+    publicClient: defaultPublicClient,
+
     getPublicClient: (publicClientConfig) =>
       getPublicClient(provider, chainType, publicClientConfig),
 
@@ -37,6 +41,7 @@ export function initializeViem<ChainTypeT extends ChainType | string>(
       deployContract(
         provider,
         artifactManager,
+        defaultPublicClient,
         contractName,
         constructorArgs,
         deployContractConfig,
@@ -50,6 +55,7 @@ export function initializeViem<ChainTypeT extends ChainType | string>(
       sendDeploymentTransaction(
         provider,
         artifactManager,
+        defaultPublicClient,
         contractName,
         constructorArgs,
         sendDeploymentTransactionConfig,
@@ -59,6 +65,7 @@ export function initializeViem<ChainTypeT extends ChainType | string>(
       getContractAt(
         provider,
         artifactManager,
+        defaultPublicClient,
         contractName,
         address,
         getContractAtConfig,

v-next/hardhat-viem/src/types.ts

@@ -32,6 +32,8 @@ import type {
 export interface HardhatViemHelpers<
   ChainTypeT extends ChainType | string = DefaultChainType,
 > {
+  publicClient: PublicClientType<ChainTypeT>;
+
   /**
    * Creates a public client configured with the provided settings.
    *
@@ -42,7 +44,7 @@ export interface HardhatViemHelpers<
    */
   getPublicClient: (
     publicClientConfig?: Partial<ViemPublicClientConfig>,
-  ) => Promise<GetPublicClientReturnType<ChainTypeT>>;
+  ) => Promise<PublicClientType<ChainTypeT>>;
   /**
    * Creates a wallet client configured with the provided settings for each
    * account in the provider.
@@ -135,7 +137,7 @@ export interface HardhatViemHelpers<
   ) => Promise<ContractReturnType<ContractName>>;
 }
 
-export type GetPublicClientReturnType<ChainTypeT extends ChainType | string> =
+export type PublicClientType<ChainTypeT extends ChainType | string> =
   ChainTypeT extends "optimism" ? OpPublicClient : PublicClient;
 
 export type GetWalletClientReturnType<ChainTypeT extends ChainType | string> =

v-next/hardhat-viem/test/hook-handlers/network.ts

@@ -24,7 +24,10 @@ describe("hook-handlers/network", () => {
 
     it("should be extended with viem", () => {
       assert.ok(isObject(connection.viem), "viem should be defined");
-
+      assert.ok(
+        isObject(connection.viem.publicClient),
+        "viem should have publicClient object",
+      );
       assert.ok(
         typeof connection.viem.getPublicClient === "function",
         "viem should have a getPublicClient function",