[SDK] get ERC20Value for claimToBatch

Author: kien-ngo

packages/thirdweb/src/extensions/erc721/drops/write/claimToBatch.ts

@@ -1,4 +1,5 @@
 import type { Address } from "abitype";
+import { resolvePromisedValue } from "src/utils/promise/resolve-promised-value.js";
 import { multicall } from "../../../../extensions/common/__generated__/IMulticall/write/multicall.js";
 import type {
   BaseTransactionOptions,
@@ -49,7 +50,10 @@ export function claimToBatch(
   return multicall({
     contract: options.contract,
     asyncParams: () => getClaimToBatchParams(options),
-    overrides: options.overrides,
+    overrides: {
+      erc20Value: () => getERC20Value(options),
+      ...options.content,
+    },
   });
 }
 
@@ -96,6 +100,51 @@ async function getClaimToBatchParams(
   return { data };
 }
 
+/**
+ * @internal
+ */
+async function getERC20Value(
+  options: BaseTransactionOptions<ClaimToBatchParams>,
+): Promise<
+  | {
+      amountWei: bigint;
+      tokenAddress: string;
+    }
+  | undefined
+> {
+  const data = await Promise.all(
+    options.content.map(async (item) => {
+      const claimParams = await getClaimParams({
+        type: "erc721",
+        contract: options.contract,
+        to: item.to,
+        from: options.from,
+        quantity: item.quantity,
+      });
+      const erc20Value = await resolvePromisedValue(
+        claimParams.overrides.erc20Value,
+      );
+      return erc20Value;
+    }),
+  );
+
+  const totalAmountWei = data
+    .filter((item) => item !== undefined)
+    .reduce(
+      (accumulator, currentValue) => accumulator + currentValue.amountWei,
+      BigInt(0),
+    );
+
+  if (!data.length || !data[0]) {
+    return undefined;
+  }
+
+  return {
+    amountWei: totalAmountWei,
+    tokenAddress: data[0].tokenAddress,
+  };
+}
+
 /**
  * Optimization
  * For identical addresses that stays next to each other in the array,

packages/thirdweb/src/transaction/prepare-transaction.ts

@@ -25,10 +25,20 @@ export type StaticPrepareTransactionOptions = {
   client: ThirdwebClient;
   // extras
   extraCallData?: Hex;
-  erc20Value?: {
-    amountWei: bigint;
-    tokenAddress: Address;
-  };
+  erc20Value?:
+    | {
+        amountWei: bigint;
+        tokenAddress: Address;
+      }
+    | (() => Promise<
+        Readonly<
+          | {
+              amountWei: bigint;
+              tokenAddress: Address;
+            }
+          | undefined
+        >
+      >);
 };
 
 export type EIP712TransactionOptions = {