Fix type errors

base-org · Feb 14, 2024 · bd5f860 · bd5f860
1 parent 5196b06
commit bd5f860
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 10 deletions.
diff --git a/src/hooks/L1/useProveWithdrawalArgs.ts b/src/hooks/L1/useProveWithdrawalArgs.ts
@@ -27,7 +27,7 @@ export function useProveWithdrawalArgs({
     throw new Error('L2 chain not configured')
   }
 
-  const l2PublicClient = usePublicClient({ chainId: l2Chain.chainId })
+  const l2PublicClient = usePublicClient({ chainId: l2Chain.chainId })!
 
   const { data: blockNumberOfLatestL2OutputProposal } = useBlockNumberOfLatestL2OutputProposal({
     config: opConfig,

diff --git a/src/hooks/L1/useSimulateProveWithdrawalTransaction.ts b/src/hooks/L1/useSimulateProveWithdrawalTransaction.ts
@@ -56,8 +56,8 @@ export function useSimulateProveWithdrawalTransaction<
   }
 
   const account = useAccount(rest)
-  const l1PublicClient = usePublicClient({ chainId: l2Chain.l1ChainId })
-  const l2PublicClient = usePublicClient({ chainId: l2ChainId })
+  const l1PublicClient = usePublicClient({ chainId: l2Chain.l1ChainId })!
+  const l2PublicClient = usePublicClient({ chainId: l2ChainId })!
   const l1Addresses = opConfig.l2chains[l2ChainId].l1Addresses
 
   const query = {
@@ -103,7 +103,8 @@ export function useSimulateProveWithdrawalTransaction<
     }),
   }
 
-  const enabled = Boolean(account.address) && (queryOverride?.enabled ?? true)
+  const enabled = Boolean(account.address) && (queryOverride?.enabled ?? true) && Boolean(l1PublicClient)
+    && Boolean(l2PublicClient)
   return {
     ...useQuery({ ...query, queryKeyHashFn: hashFn, enabled }),
     queryKey: query.queryKey,

diff --git a/src/hooks/L1/useWriteDepositETH.ts b/src/hooks/L1/useWriteDepositETH.ts
@@ -47,8 +47,8 @@ async function writeMutation(
   { l1ChainId, l2ChainId, args, ...rest }: DepositETHMutationParameters,
 ) {
   const walletClient = await getWalletClient(config, { chainId: l1ChainId })
-  const l1PublicClient = getPublicClient(config, { chainId: l1ChainId })
-  const l2PublicClient = getPublicClient(config, { chainId: l2ChainId })
+  const l1PublicClient = await getPublicClient(config, { chainId: l1ChainId })!
+  const l2PublicClient = await getPublicClient(config, { chainId: l2ChainId })!
   const l1Addresses = config.l2chains[l2ChainId].l1Addresses
 
   const l2GasLimit = args.gasLimit

diff --git a/src/hooks/L1/useWriteFinalizeWithdrawalTransaction.ts b/src/hooks/L1/useWriteFinalizeWithdrawalTransaction.ts
@@ -57,8 +57,8 @@ async function writeMutation(
   { l1ChainId, l2ChainId, args, ...rest }: FinalizeWithdrawalTransactionMutationParameters,
 ) {
   const walletClient = await getWalletClient(config, { chainId: l1ChainId })
-  const l1PublicClient = getPublicClient(config, { chainId: l1ChainId })
-  const l2PublicClient = getPublicClient(config, { chainId: l2ChainId })
+  const l1PublicClient = await getPublicClient(config, { chainId: l1ChainId })!
+  const l2PublicClient = await getPublicClient(config, { chainId: l2ChainId })!
   const l1Addresses = config.l2chains[l2ChainId].l1Addresses
 
   const withdrawalMessages = await getWithdrawalMessages(l2PublicClient, {

diff --git a/src/util/getStateTrieProof.ts b/src/util/getStateTrieProof.ts
@@ -25,15 +25,15 @@ export async function makeStateTrieProof(
 }
 
 export function useMakeStateTrieProof(
-  client: PublicClient,
+  client: PublicClient | undefined,
   blockNumber: bigint | undefined,
   address: `0x${string}`,
   slot: `0x${string}` | undefined,
 ) {
   const [proof, setProof] = useState<StateTrieProof | undefined>(undefined)
 
   useEffect(() => {
-    if (!blockNumber || !slot) {
+    if (!blockNumber || !slot || !client) {
       return undefined
     }