Fix safeTxGas estimation for viem

safe-global · Apr 29, 2024 · 9090d93 · 9090d93
1 parent 707c1ec
commit 9090d93
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/packages/protocol-kit/src/utils/transactions/gas.ts b/packages/protocol-kit/src/utils/transactions/gas.ts
@@ -400,7 +400,11 @@ function decodeSafeTxGas(encodedDataResponse: string): string {
 
 type GnosisChainEstimationError = { info: { error: { data: string | { data: string } } } }
 type EthersEstimationError = { data: string }
-type EstimationError = Error & EthersEstimationError & GnosisChainEstimationError
+type ViemEstimationError = { info: { error: { message: string } } }
+type EstimationError = Error &
+  EthersEstimationError &
+  GnosisChainEstimationError &
+  ViemEstimationError
 
 /**
  * Parses the SafeTxGas estimation response from different providers.
@@ -417,6 +421,12 @@ function parseSafeTxGasErrorResponse(error: EstimationError) {
     return decodeSafeTxGas(ethersData)
   }
 
+  // viem
+  const viemError = error?.info?.error?.message
+  if (viemError) {
+    return decodeSafeTxGas(viemError)
+  }
+
   // gnosis-chain
   const gnosisChainProviderData = error?.info?.error?.data