[BLD-143] Dashboard: Fix distribute funds in split contract

Author: MananTank

apps/dashboard/src/@/hooks/useSplit.ts

@@ -7,6 +7,7 @@ import {
 import { toast } from "sonner";
 import {
   type Chain,
+  getAddress,
   NATIVE_TOKEN_ADDRESS,
   sendAndConfirmTransaction,
   type ThirdwebClient,
@@ -98,7 +99,8 @@ export function useSplitDistributeFunds(contract: ThirdwebContract) {
         .filter((token) => token.value !== 0n)
         .map(async (currency) => {
           const transaction =
-            currency.name === "Native Token"
+            getAddress(currency.tokenAddress) ===
+            getAddress(NATIVE_TOKEN_ADDRESS)
               ? distribute({ contract })
               : distributeByToken({
                   contract,
@@ -108,6 +110,7 @@ export function useSplitDistributeFunds(contract: ThirdwebContract) {
             account,
             transaction,
           });
+
           toast.promise(promise, {
             error: (err) => ({
               message: `Error distributing ${currency.name}`,
@@ -116,6 +119,8 @@ export function useSplitDistributeFunds(contract: ThirdwebContract) {
             loading: `Distributing ${currency.name}`,
             success: `Successfully distributed ${currency.name}`,
           });
+
+          await promise;
         });
 
       return await Promise.all(distributions);

apps/dashboard/src/app/(app)/(dashboard)/(chain)/[chain_id]/[contractAddress]/split/components/distribute-button.tsx

@@ -2,14 +2,12 @@
 
 import { SplitIcon } from "lucide-react";
 import { useMemo } from "react";
-import { toast } from "sonner";
 import type { ThirdwebContract } from "thirdweb";
 import type { GetBalanceResult } from "thirdweb/extensions/erc20";
 import { TransactionButton } from "@/components/tx-button";
 import { Button } from "@/components/ui/button";
 import { ToolTipLabel } from "@/components/ui/tooltip";
 import { useSplitDistributeFunds } from "@/hooks/useSplit";
-import { parseError } from "@/utils/errorParser";
 
 export const DistributeButton = ({
   contract,
@@ -25,6 +23,7 @@ export const DistributeButton = ({
   isLoggedIn: boolean;
 }) => {
   const validBalances = balances.filter((item) => item.value !== 0n);
+
   const numTransactions = useMemo(() => {
     if (
       validBalances.length === 1 &&
@@ -38,28 +37,18 @@ export const DistributeButton = ({
     return validBalances?.filter((b) => b.value !== 0n).length;
   }, [validBalances, balancesIsPending]);
 
-  const mutation = useSplitDistributeFunds(contract);
+  const distributeFundsMutation = useSplitDistributeFunds(contract);
 
   const distributeFunds = () => {
-    mutation.mutate(undefined, {
-      onError: (error) => {
-        toast.error("Failed to process transaction", {
-          description: parseError(error),
-        });
-        console.error(error);
-      },
-      onSuccess: () => {
-        toast.success("Funds splitted successfully");
-      },
-    });
+    distributeFundsMutation.mutateAsync();
   };
 
   if (balancesIsError) {
     return (
       <TransactionButton
         client={contract.client}
         isLoggedIn={isLoggedIn}
-        isPending={mutation.isPending}
+        isPending={distributeFundsMutation.isPending}
         onClick={distributeFunds}
         size="sm"
         transactionCount={undefined}
@@ -86,7 +75,7 @@ export const DistributeButton = ({
     <TransactionButton
       client={contract.client}
       isLoggedIn={isLoggedIn}
-      isPending={mutation.isPending}
+      isPending={distributeFundsMutation.isPending}
       onClick={distributeFunds}
       transactionCount={numTransactions === 1 ? undefined : numTransactions}
       txChainID={contract.chain.id}