Allow to use useTransactor hook with contract.write.method (#623)

Co-authored-by: Shiv Bhonde <[email protected]>
scaffold-eth · Nov 25, 2023 · 55ae5a4 · 55ae5a4
1 parent 6053ad5
commit 55ae5a4
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/packages/nextjs/hooks/scaffold-eth/useScaffoldContractWrite.ts b/packages/nextjs/hooks/scaffold-eth/useScaffoldContractWrite.ts
@@ -67,7 +67,7 @@ export const useScaffoldContractWrite = <
     if (wagmiContractWrite.writeAsync) {
       try {
         setIsMining(true);
-        await writeTx(
+        const writeTxResult = await writeTx(
           () =>
             wagmiContractWrite.writeAsync({
               args: newArgs ?? args,
@@ -76,6 +76,8 @@ export const useScaffoldContractWrite = <
             }),
           { onBlockConfirmation, blockConfirmations },
         );
+
+        return writeTxResult;
       } catch (e: any) {
         const message = getParsedError(e);
         notification.error(message);

diff --git a/packages/nextjs/hooks/scaffold-eth/useTransactor.tsx b/packages/nextjs/hooks/scaffold-eth/useTransactor.tsx
@@ -5,7 +5,7 @@ import { getParsedError } from "~~/components/scaffold-eth";
 import { getBlockExplorerTxLink, notification } from "~~/utils/scaffold-eth";
 
 type TransactionFunc = (
-  tx: (() => Promise<WriteContractResult>) | SendTransactionParameters,
+  tx: (() => Promise<WriteContractResult>) | (() => Promise<Hash>) | SendTransactionParameters,
   options?: {
     onBlockConfirmation?: (txnReceipt: TransactionReceipt) => void;
     blockConfirmations?: number;
@@ -57,7 +57,12 @@ export const useTransactor = (_walletClient?: WalletClient): TransactionFunc =>
       notificationId = notification.loading(<TxnNotification message="Awaiting for user confirmation" />);
       if (typeof tx === "function") {
         // Tx is already prepared by the caller
-        transactionHash = (await tx()).hash;
+        const result = await tx();
+        if (typeof result === "string") {
+          transactionHash = result;
+        } else {
+          transactionHash = result.hash;
+        }
       } else if (tx != null) {
         transactionHash = await walletClient.sendTransaction(tx);
       } else {