Adds validateBeforeWrite param to createReadWriteContract (#69)

* Adds validate before write Co-authored-by: Danny Delott <[email protected]> * adds changeset * updates variable names per PR comments --------- Co-authored-by: Danny Delott <[email protected]>
delvtech · May 13, 2024 · b224f38 · b224f38
1 parent 93531e6
commit b224f38
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/.changeset/rare-tools-drum.md b/.changeset/rare-tools-drum.md
@@ -0,0 +1,5 @@
+---
+"@delvtech/evm-client-viem": minor
+---
+
+This added a validateBeforeWrite param to createReadWriteContract.
diff --git a/packages/evm-client-viem/src/contract/createReadWriteContract.ts b/packages/evm-client-viem/src/contract/createReadWriteContract.ts
@@ -8,12 +8,16 @@ import {
   CreateReadContractOptions,
 } from 'src/contract/createReadContract';
 import { createSimulateContractParameters } from 'src/contract/utils/createSimulateContractParameters';
-import { Abi, WalletClient } from 'viem';
+import { Abi, Address, WalletClient } from 'viem';
 
 export interface ReadWriteContractOptions<TAbi extends Abi = Abi>
   extends CreateReadContractOptions<TAbi> {
   walletClient: WalletClient;
   readContract?: ReadContract<TAbi>;
+  /**
+   * Turn this off for debugging
+   */
+  simulateBeforeWrite?: boolean;
 }
 
 /**
@@ -26,6 +30,7 @@ export function createReadWriteContract<TAbi extends Abi = Abi>({
   publicClient,
   walletClient,
   readContract = createReadContract({ abi, address, publicClient }),
+  simulateBeforeWrite: simulateBeforeWrite = true,
 }: ReadWriteContractOptions<TAbi>): ReadWriteContract<TAbi> {
   return {
     ...readContract,
@@ -58,6 +63,17 @@ export function createReadWriteContract<TAbi extends Abi = Abi>({
         value: args,
       });
 
+      if (!simulateBeforeWrite) {
+        return walletClient.writeContract({
+          address,
+          abi: abi as Abi,
+          functionName,
+          args: arrayArgs,
+          account: account as Address,
+          chain: undefined,
+        });
+      }
+
       const { request } = await publicClient.simulateContract({
         abi: abi as Abi,
         address: address,