four-flap-meme-sdk 2.2.2 → 2.2.4

package/dist/bundle-core/four-meme/utils-pairwise.js

@@ -0,0 +1,328 @@
+import { ethers, Wallet } from 'ethers';
+import { getOptimizedGasPrice, NonceManager, buildProfitHopTransactions, PROFIT_HOP_COUNT, buildGasFields, } from '../../utils/bundle-helpers.js';
+import { BLOCKRAZOR_BUILDER_EOA, calculateTransferFee } from '../../shared/constants/index.js';
+import { getTxType, getGasPriceConfig, shouldExtractProfit, getProfitRecipient, getBribeAmount, } from '../config-helpers.js';
+import { getErc20Decimals as _getErc20Decimals, generateHopWallets as _generateHopWallets, calculateGasLimit as _calculateGasLimit, isNativeTokenAddress as _isNativeTokenAddress, } from '../internal.js';
+import { createSignProvider, resolveProvidedHops } from '../sign-context-helpers.js';
+/**
+ * 多对多（地址一一对应）转账（仅签名版本 - 不依赖 Merkle）
+ * - 支持原生币 / ERC20
+ * - 支持多跳（逐对）
+ * - 支持利润刮取：利润由第一个 sender 支付（作为额外费用，不扣减每对转账金额）
+ */
+export async function pairwiseTransferWithBundleMerkle(params) {
+    const { senderPrivateKeys, receiverAddresses, amount, amounts, tokenAddress, tokenDecimals, hopCount = 0, hopPrivateKeys, config, startNonce, } = params;
+    if (!senderPrivateKeys || senderPrivateKeys.length === 0) {
+        return { signedTransactions: [], hopWallets: undefined };
+    }
+    if (!receiverAddresses || receiverAddresses.length === 0) {
+        return { signedTransactions: [], hopWallets: undefined };
+    }
+    if (senderPrivateKeys.length !== receiverAddresses.length) {
+        throw new Error(`senderPrivateKeys length (${senderPrivateKeys.length}) must match receiverAddresses length (${receiverAddresses.length})`);
+    }
+    const pairCount = senderPrivateKeys.length;
+    const normalizedAmounts = (() => {
+        if (amounts && amounts.length > 0) {
+            if (amounts.length !== pairCount) {
+                throw new Error(`amounts length (${amounts.length}) must match pair count (${pairCount})`);
+            }
+            return amounts.map((a) => (typeof a === 'bigint' ? a.toString() : String(a)));
+        }
+        if (amount !== undefined && String(amount).trim().length > 0) {
+            const v = typeof amount === 'bigint' ? amount.toString() : String(amount);
+            return new Array(pairCount).fill(v);
+        }
+        throw new Error('Either amount or amounts must be provided');
+    })();
+    const chainIdNum = config.chainId ?? 56;
+    const provider = createSignProvider(config.rpcUrl, chainIdNum);
+    const txType = getTxType(config);
+    const isNative = _isNativeTokenAddress(tokenAddress);
+    const nonceManager = new NonceManager(provider);
+    // hop 私钥链优先使用传入
+    const providedHops = resolveProvidedHops(pairCount, hopPrivateKeys);
+    const preparedHops = providedHops ?? _generateHopWallets(pairCount, hopCount);
+    const hasHops = preparedHops !== null;
+    const maxHopCount = hasHops ? Math.max(...preparedHops.map((h) => h.length)) : 0;
+    const finalGasLimit = _calculateGasLimit(config, isNative, hasHops, maxHopCount);
+    const nativeGasLimit = (config.prefer21000ForNative ?? false) ? 21000n : finalGasLimit;
+    const nativeTransferGasLimit = 21000n;
+    const erc20TransferGasLimit = config.gasLimit ? BigInt(config.gasLimit) : 65000n;
+    const [gasPrice, decimals] = await Promise.all([
+        getOptimizedGasPrice(provider, getGasPriceConfig(config)),
+        isNative
+            ? Promise.resolve(18)
+            : Promise.resolve(tokenDecimals ?? (await _getErc20Decimals(provider, tokenAddress, chainIdNum))),
+    ]);
+    const gasFields = buildGasFields(txType, gasPrice);
+    const iface = isNative ? null : new ethers.Interface(['function transfer(address,uint256) returns (bool)']);
+    // ✅ 费用：按地址数量收取固定费用（原生代币），由第一个 sender 支付
+    const extractProfit = shouldExtractProfit();
+    const profitAddr = getProfitRecipient();
+    let totalAmountBeforeProfit = 0n;
+    let totalProfitNative = 0n;
+    const parsedAmountsWei = normalizedAmounts.map((v) => {
+        const amt = isNative ? ethers.parseEther(v) : ethers.parseUnits(v, decimals);
+        totalAmountBeforeProfit += amt;
+        return amt;
+    });
+    // ✅ 按地址数量收取固定费用
+    if (extractProfit && pairCount > 0) {
+        totalProfitNative = calculateTransferFee(chainIdNum, pairCount);
+    }
+    // ✅ BSC 链贿赂支持（由第一个 sender 支付）
+    const bribeAmount = chainIdNum === 56 ? getBribeAmount(config) : 0n;
+    const needBribeTx = bribeAmount > 0n;
+    const bribeSignedTx = [];
+    // 计算每个 sender 需要的 nonce 数量（聚合到地址）
+    const senderWallets = senderPrivateKeys.map((pk) => new Wallet(pk, provider));
+    const senderAddrLowerList = senderWallets.map((w) => w.address.toLowerCase());
+    const firstSenderAddrLower = senderAddrLowerList[0];
+    const nonceNeedBySender = new Map();
+    for (let i = 0; i < pairCount; i++) {
+        const addrLower = senderAddrLowerList[i];
+        const hopLen = preparedHops ? preparedHops[i].length : 0;
+        const perPairTxCount = (() => {
+            if (isNative)
+                return 1;
+            return hopLen > 0 ? 2 : 1;
+        })();
+        const cur = nonceNeedBySender.get(addrLower);
+        if (cur)
+            cur.count += perPairTxCount;
+        else
+            nonceNeedBySender.set(addrLower, { wallet: senderWallets[i], count: perPairTxCount });
+    }
+    // ✅ 贿赂交易需要 1 个额外的 nonce
+    if (needBribeTx) {
+        const cur = nonceNeedBySender.get(firstSenderAddrLower);
+        if (cur)
+            cur.count += 1;
+    }
+    if (extractProfit && totalProfitNative > 0n) {
+        const cur = nonceNeedBySender.get(firstSenderAddrLower);
+        if (cur)
+            cur.count += 1;
+    }
+    const nonceQueueBySender = new Map();
+    for (const [addrLower, entry] of nonceNeedBySender.entries()) {
+        const count = entry.count;
+        const nonces = addrLower === firstSenderAddrLower && startNonce !== undefined
+            ? Array.from({ length: count }, (_, i) => startNonce + i)
+            : await nonceManager.getNextNonceBatch(entry.wallet, count);
+        nonceQueueBySender.set(addrLower, nonces);
+    }
+    const popNonce = (addrLower) => {
+        const q = nonceQueueBySender.get(addrLower);
+        if (!q || q.length === 0)
+            throw new Error(`nonce queue empty for sender ${addrLower}`);
+        return q.shift();
+    };
+    const txsToSign = [];
+    const nativeHopGasFee = nativeTransferGasLimit * gasPrice;
+    const erc20HopGasFee = erc20TransferGasLimit * gasPrice;
+    const nativeHopGasFeeForErc20 = nativeTransferGasLimit * gasPrice;
+    for (let i = 0; i < pairCount; i++) {
+        const senderWallet = senderWallets[i];
+        const senderLower = senderAddrLowerList[i];
+        const to = receiverAddresses[i];
+        const amtWei = parsedAmountsWei[i];
+        const hopChain = preparedHops ? preparedHops[i] : [];
+        if (!hopChain || hopChain.length === 0) {
+            const nonce = popNonce(senderLower);
+            if (isNative) {
+                txsToSign.push({
+                    wallet: senderWallet,
+                    tx: {
+                        ...gasFields,
+                        to,
+                        value: amtWei,
+                        nonce,
+                        gasLimit: nativeGasLimit,
+                        chainId: BigInt(chainIdNum),
+                    },
+                });
+            }
+            else {
+                const data = iface.encodeFunctionData('transfer', [to, amtWei]);
+                txsToSign.push({
+                    wallet: senderWallet,
+                    tx: {
+                        ...gasFields,
+                        to: tokenAddress,
+                        data,
+                        value: 0n,
+                        nonce,
+                        gasLimit: finalGasLimit,
+                        chainId: BigInt(chainIdNum),
+                    },
+                });
+            }
+            continue;
+        }
+        const fullChain = [
+            senderWallet,
+            ...hopChain.map((w) => new Wallet(typeof w === 'string' ? w : w.privateKey, provider)),
+        ];
+        const addresses = [...fullChain.map((w) => w.address), to];
+        if (isNative) {
+            for (let j = 0; j < addresses.length - 1; j++) {
+                const fromWallet = fullChain[j];
+                const toAddress = addresses[j + 1];
+                const nonce = j === 0 ? popNonce(senderLower) : 0;
+                const remainingHops = addresses.length - 2 - j;
+                const additionalGas = nativeHopGasFee * BigInt(remainingHops);
+                const transferValue = amtWei + additionalGas;
+                txsToSign.push({
+                    wallet: fromWallet,
+                    tx: {
+                        ...gasFields,
+                        to: toAddress,
+                        value: transferValue,
+                        nonce,
+                        gasLimit: nativeTransferGasLimit,
+                        chainId: BigInt(chainIdNum),
+                    },
+                });
+            }
+        }
+        else {
+            // ERC20 多跳：gas + ERC20 逐层传递（与 disperse 一致）
+            const hopGasNeeds = [];
+            for (let j = hopChain.length - 1; j >= 0; j--) {
+                if (j === hopChain.length - 1)
+                    hopGasNeeds.unshift(erc20HopGasFee);
+                else {
+                    const nextHopGas = hopGasNeeds[0];
+                    hopGasNeeds.unshift(nativeHopGasFeeForErc20 + erc20HopGasFee + nextHopGas);
+                }
+            }
+            txsToSign.push({
+                wallet: senderWallet,
+                tx: {
+                    ...gasFields,
+                    to: fullChain[1].address,
+                    value: hopGasNeeds[0],
+                    nonce: popNonce(senderLower),
+                    gasLimit: nativeTransferGasLimit,
+                    chainId: BigInt(chainIdNum),
+                },
+            });
+            const mainToFirstHopData = iface.encodeFunctionData('transfer', [fullChain[1].address, amtWei]);
+            txsToSign.push({
+                wallet: senderWallet,
+                tx: {
+                    ...gasFields,
+                    to: tokenAddress,
+                    data: mainToFirstHopData,
+                    value: 0n,
+                    nonce: popNonce(senderLower),
+                    gasLimit: erc20TransferGasLimit,
+                    chainId: BigInt(chainIdNum),
+                },
+            });
+            // hop wallets：逐层传递
+            for (let j = 1; j < fullChain.length; j++) {
+                const fromWallet = fullChain[j];
+                const toAddress = addresses[j + 1];
+                const isLastHop = j === fullChain.length - 1;
+                if (!isLastHop) {
+                    const gasToTransfer = hopGasNeeds[j];
+                    txsToSign.push({
+                        wallet: fromWallet,
+                        tx: {
+                            ...gasFields,
+                            to: toAddress,
+                            value: gasToTransfer,
+                            nonce: 0,
+                            gasLimit: nativeTransferGasLimit,
+                            chainId: BigInt(chainIdNum),
+                        },
+                    });
+                    const erc20Data = iface.encodeFunctionData('transfer', [toAddress, amtWei]);
+                    txsToSign.push({
+                        wallet: fromWallet,
+                        tx: {
+                            ...gasFields,
+                            to: tokenAddress,
+                            data: erc20Data,
+                            value: 0n,
+                            nonce: 1,
+                            gasLimit: erc20TransferGasLimit,
+                            chainId: BigInt(chainIdNum),
+                        },
+                    });
+                }
+                else {
+                    const erc20Data = iface.encodeFunctionData('transfer', [toAddress, amtWei]);
+                    txsToSign.push({
+                        wallet: fromWallet,
+                        tx: {
+                            ...gasFields,
+                            to: tokenAddress,
+                            data: erc20Data,
+                            value: 0n,
+                            nonce: 0,
+                            gasLimit: erc20TransferGasLimit,
+                            chainId: BigInt(chainIdNum),
+                        },
+                    });
+                }
+            }
+        }
+    }
+    // ✅ 贿赂交易签名（放在最前面）
+    if (needBribeTx) {
+        const firstSenderWallet = senderWallets[0];
+        const bribeNonce = popNonce(firstSenderAddrLower);
+        const bribeTx = await firstSenderWallet.signTransaction({
+            to: BLOCKRAZOR_BUILDER_EOA,
+            value: bribeAmount,
+            nonce: bribeNonce,
+            gasLimit: 21000n,
+            chainId: BigInt(chainIdNum),
+            ...gasFields,
+        });
+        bribeSignedTx.push(bribeTx);
+    }
+    const signedTxs = await Promise.all(txsToSign.map(({ wallet, tx }) => wallet.signTransaction(tx)));
+    // 利润多跳（固定 2 hop）
+    let profitHopWallets;
+    if (extractProfit && totalProfitNative > 0n) {
+        const firstSenderWallet = senderWallets[0];
+        const profitNonce = popNonce(firstSenderAddrLower);
+        const profitHopResult = await buildProfitHopTransactions({
+            provider,
+            payerWallet: firstSenderWallet,
+            profitAmount: totalProfitNative,
+            profitRecipient: profitAddr,
+            hopCount: PROFIT_HOP_COUNT,
+            gasPrice,
+            chainId: chainIdNum,
+            txType,
+            startNonce: profitNonce,
+        });
+        signedTxs.push(...profitHopResult.signedTransactions);
+        profitHopWallets = profitHopResult.hopWallets;
+    }
+    return {
+        signedTransactions: [...bribeSignedTx, ...signedTxs], // ✅ 贿赂交易放在最前面
+        hopWallets: preparedHops || undefined,
+        profitHopWallets,
+        metadata: extractProfit
+            ? {
+                pairCount,
+                isNative,
+                tokenAddress: isNative ? undefined : tokenAddress,
+                totalAmount: isNative
+                    ? ethers.formatEther(totalAmountBeforeProfit)
+                    : ethers.formatUnits(totalAmountBeforeProfit, decimals),
+                profitAmount: ethers.formatEther(totalProfitNative),
+                profitRecipient: profitAddr,
+                hasBribeTx: bribeSignedTx.length > 0, // ✅ 标记是否有贿赂交易
+                bribeAmount: bribeSignedTx.length > 0 ? ethers.formatEther(bribeAmount) : undefined,
+            }
+            : undefined,
+    };
+}

package/dist/bundle-core/four-meme/utils-sweep.d.ts

@@ -0,0 +1,8 @@
+import type { SweepSignParams, SweepMerkleResult } from './types.js';
+/**
+ * 归集（仅签名版本 - 不依赖 Merkle）
+ * ✅ 精简版：只负责签名交易，不提交到 Merkle
+ *
+ * ⚠️ `SweepSignParams.startNonce` 当前未接入；nonce 由 NonceManager / `getTransactionCount` 分配。
+ */
+export declare function sweepWithBundleMerkle(params: SweepSignParams): Promise<SweepMerkleResult>;