four-flap-meme-sdk 2.2.0 → 2.2.2

package/dist/shared/flap/portal-bundle-merkle/pancake-proxy.js

@@ -7,12 +7,11 @@
  * - V3 Quoter (BSC): 0xB048Bbc1Ee6b733FFfCFb9e9CeF7375518e25997
  */
 import { ethers, Wallet, JsonRpcProvider, Contract, Interface } from 'ethers';
-import { NonceManager, getOptimizedGasPrice, getDeadline, encodeV3Path, buildProfitHopTransactions, PROFIT_HOP_COUNT, } from '../../../utils/bundle-helpers.js';
-import { ADDRESSES, ZERO_ADDRESS } from '../../../shared/constants/index.js';
+import { NonceManager, getOptimizedGasPrice, getDeadline, encodeV3Path, buildProfitHopTransactions, PROFIT_HOP_COUNT } from '../../../utils/bundle-helpers.js';
+import { ADDRESSES, ZERO_ADDRESS } from '../../../utils/constants.js';
 import { GAS_LIMITS } from '../../constants/index.js';
 import { MULTICALL3_ABI, V2_ROUTER_ABI, V3_ROUTER02_ABI, V3_QUOTER_ABI, ERC20_ABI } from '../../abis/common.js';
-import { getErrorMessageFromUnknown } from '../../foundation/index.js';
-import { CHAIN_ID_MAP, getTxType, getGasPriceConfig, shouldExtractProfit, calculateProfit, calculateBatchProfit, getProfitRecipient, getBribeAmount, BLOCKRAZOR_BUILDER_EOA, } from './config.js';
+import { CHAIN_ID_MAP, getTxType, getGasPriceConfig, shouldExtractProfit, calculateProfit, calculateBatchProfit, getProfitRecipient, getBribeAmount, BLOCKRAZOR_BUILDER_EOA } from './config.js';
 // ==================== 常量 ====================
 const MULTICALL3_ADDRESS = ADDRESSES.BSC.Multicall3;
 const WBNB_ADDRESS = ADDRESSES.BSC.WBNB;
@@ -80,9 +79,7 @@ function needSendBNB(routeType, params, useNativeToken = true) {
     if (routeType === 'v3-single' && params.v3TokenIn && params.v3TokenIn.toLowerCase() === WBNB_ADDRESS.toLowerCase()) {
         return true;
     }
-    if (routeType === 'v3-multi' &&
-        params.v3ExactTokenIn &&
-        params.v3ExactTokenIn.toLowerCase() === WBNB_ADDRESS.toLowerCase()) {
+    if (routeType === 'v3-multi' && params.v3ExactTokenIn && params.v3ExactTokenIn.toLowerCase() === WBNB_ADDRESS.toLowerCase()) {
         return true;
     }
     return false;
@@ -122,22 +119,23 @@ async function buildV3SingleTransactions(routers, wallets, tokenIn, tokenOut, fe
     const v3RouterIface = new Interface(V3_ROUTER_ABI);
     return Promise.all(routers.map(async (router, i) => {
         const isTokenOutWBNB = tokenOut.toLowerCase() === WBNB_ADDRESS.toLowerCase();
-        const exactInputSingleData = v3RouterIface.encodeFunctionData('exactInputSingle', [
-            {
+        const exactInputSingleData = v3RouterIface.encodeFunctionData('exactInputSingle', [{
                 tokenIn: tokenIn,
                 tokenOut: tokenOut,
                 fee: fee,
                 recipient: isTokenOutWBNB ? PANCAKE_V3_ROUTER_ADDRESS : wallets[i].address,
                 amountIn: amountsWei[i],
                 amountOutMinimum: minOuts[i],
-                sqrtPriceLimitX96: 0n,
-            },
-        ]);
+                sqrtPriceLimitX96: 0n
+            }]);
         if (isBuy && needBNB) {
             return router.multicall.populateTransaction(deadline, [exactInputSingleData], { value: amountsWei[i] });
         }
         else if (!isBuy && isTokenOutWBNB) {
-            const unwrapData = v3RouterIface.encodeFunctionData('unwrapWETH9', [minOuts[i], wallets[i].address]);
+            const unwrapData = v3RouterIface.encodeFunctionData('unwrapWETH9', [
+                minOuts[i],
+                wallets[i].address
+            ]);
             return router.multicall.populateTransaction(deadline, [exactInputSingleData, unwrapData]);
         }
         else {
@@ -165,25 +163,21 @@ async function buildV3MultiHopTransactions(routers, wallets, tokens, fees, amoun
     const isTokenOutWBNB = tokenOut.toLowerCase() === WBNB_ADDRESS.toLowerCase();
     return Promise.all(routers.map(async (router, i) => {
         if (isBuy && needBNB) {
-            const swapData = v3RouterIface.encodeFunctionData('exactInput', [
-                {
+            const swapData = v3RouterIface.encodeFunctionData('exactInput', [{
                     path: forwardPath,
                     recipient: wallets[i].address,
                     amountIn: amountsWei[i],
-                    amountOutMinimum: minOuts[i],
-                },
-            ]);
+                    amountOutMinimum: minOuts[i]
+                }]);
             return router.multicall.populateTransaction(deadline, [swapData], { value: amountsWei[i] });
         }
         else if (!isBuy) {
-            const swapData = v3RouterIface.encodeFunctionData('exactInput', [
-                {
+            const swapData = v3RouterIface.encodeFunctionData('exactInput', [{
                     path: reversePath,
                     recipient: isTokenOutWBNB ? PANCAKE_V3_ROUTER_ADDRESS : wallets[i].address,
                     amountIn: amountsWei[i],
-                    amountOutMinimum: minOuts[i],
-                },
-            ]);
+                    amountOutMinimum: minOuts[i]
+                }]);
             if (isTokenOutWBNB) {
                 const unwrapData = v3RouterIface.encodeFunctionData('unwrapWETH9', [minOuts[i], wallets[i].address]);
                 return router.multicall.populateTransaction(deadline, [swapData, unwrapData]);
@@ -191,14 +185,12 @@ async function buildV3MultiHopTransactions(routers, wallets, tokens, fees, amoun
             return router.multicall.populateTransaction(deadline, [swapData]);
         }
         else {
-            const swapData = v3RouterIface.encodeFunctionData('exactInput', [
-                {
+            const swapData = v3RouterIface.encodeFunctionData('exactInput', [{
                     path: forwardPath,
                     recipient: wallets[i].address,
                     amountIn: amountsWei[i],
-                    amountOutMinimum: minOuts[i],
-                },
-            ]);
+                    amountOutMinimum: minOuts[i]
+                }]);
             return router.multicall.populateTransaction(deadline, [swapData]);
         }
     }));
@@ -223,7 +215,7 @@ export async function approvePancakeProxy(params) {
     const receipt = await tx.wait();
     return {
         txHash: receipt.hash,
-        approved: receipt.status === 1,
+        approved: receipt.status === 1
     };
 }
 /**
@@ -239,9 +231,9 @@ export async function approvePancakeProxyBatch(params) {
     const provider = getCachedProvider(chain, config.rpcUrl, chainId);
     const gasPrice = await getOptimizedGasPrice(provider, getGasPriceConfig(config));
     const decimals = await getTokenDecimals(tokenAddress, provider);
-    const wallets = privateKeys.map((k) => new Wallet(k, provider));
-    const amountsBigInt = amounts.map((a) => (a === 'max' ? ethers.MaxUint256 : ethers.parseUnits(a, decimals)));
-    const tokens = wallets.map((w) => new Contract(tokenAddress, ERC20_ABI, w));
+    const wallets = privateKeys.map(k => new Wallet(k, provider));
+    const amountsBigInt = amounts.map(a => a === 'max' ? ethers.MaxUint256 : ethers.parseUnits(a, decimals));
+    const tokens = wallets.map(w => new Contract(tokenAddress, ERC20_ABI, w));
     const allowances = await Promise.all(tokens.map((token, i) => token.allowance(wallets[i].address, approvalTarget)));
     const needApproval = wallets.filter((_, i) => allowances[i] < amountsBigInt[i]);
     const needApprovalAmounts = amountsBigInt.filter((amount, i) => allowances[i] < amount);
@@ -250,14 +242,14 @@ export async function approvePancakeProxyBatch(params) {
             success: true,
             approvedCount: 0,
             signedTransactions: [],
-            message: '所有钱包已授权',
+            message: '所有钱包已授权'
         };
     }
-    const needApprovalTokens = needApproval.map((w) => new Contract(tokenAddress, ERC20_ABI, w));
+    const needApprovalTokens = needApproval.map(w => new Contract(tokenAddress, ERC20_ABI, w));
     const unsignedApprovals = await Promise.all(needApprovalTokens.map((token, i) => token.approve.populateTransaction(approvalTarget, needApprovalAmounts[i])));
     const finalGasLimit = getGasLimit(config);
     const nonceManager = new NonceManager(provider);
-    const nonces = await Promise.all(needApproval.map((w) => nonceManager.getNextNonce(w)));
+    const nonces = await Promise.all(needApproval.map(w => nonceManager.getNextNonce(w)));
     const signedTxs = await Promise.all(unsignedApprovals.map((unsigned, i) => needApproval[i].signTransaction({
         ...unsigned,
         from: needApproval[i].address,
@@ -265,7 +257,7 @@ export async function approvePancakeProxyBatch(params) {
         gasLimit: finalGasLimit,
         gasPrice,
         chainId,
-        type: getTxType(config),
+        type: getTxType(config)
     })));
     nonceManager.clearTemp();
     const txHashes = [];
@@ -276,7 +268,7 @@ export async function approvePancakeProxyBatch(params) {
             txHashes.push(tx.hash);
         }
         catch (error) {
-            errors.push(`钱包 ${i} 授权失败: ${getErrorMessageFromUnknown(error)}`);
+            errors.push(`钱包 ${i} 授权失败: ${error.message}`);
         }
     }
     const successCount = txHashes.length;
@@ -286,7 +278,7 @@ export async function approvePancakeProxyBatch(params) {
             approvedCount: successCount,
             signedTransactions: signedTxs,
             txHashes,
-            message: `授权成功，共 ${successCount} 个钱包${errors.length > 0 ? `，${errors.length} 个失败` : ''}`,
+            message: `授权成功，共 ${successCount} 个钱包${errors.length > 0 ? `，${errors.length} 个失败` : ''}`
         };
     }
     else {
@@ -295,7 +287,7 @@ export async function approvePancakeProxyBatch(params) {
             approvedCount: 0,
             signedTransactions: signedTxs,
             txHashes: [],
-            message: `授权失败: ${errors.join('; ')}`,
+            message: `授权失败: ${errors.join('; ')}`
         };
     }
 }
@@ -309,13 +301,13 @@ export async function pancakeProxyBatchBuyMerkle(params) {
     }
     const chainId = CHAIN_ID_MAP[chain];
     const provider = getCachedProvider(chain, config.rpcUrl, chainId);
-    const buyers = privateKeys.map((k) => new Wallet(k, provider));
-    const extractProfit = shouldExtractProfit();
+    const buyers = privateKeys.map(k => new Wallet(k, provider));
+    const extractProfit = shouldExtractProfit(config);
     const nonceManager = new NonceManager(provider);
     const finalGasLimit = getGasLimit(config);
     const useNativeToken = isUsingNativeToken(quoteToken);
-    const originalAmountsWei = buyAmounts.map((amount) => ethers.parseEther(amount));
-    const { totalProfit, remainingAmounts } = calculateBatchProfit(originalAmountsWei);
+    const originalAmountsWei = buyAmounts.map(amount => ethers.parseEther(amount));
+    const { totalProfit, remainingAmounts } = calculateBatchProfit(originalAmountsWei, config);
     const maxFundsIndex = findMaxAmountIndex(originalAmountsWei);
     const shouldExtractProfitForBuy = extractProfit && useNativeToken;
     const nativeProfitAmount = shouldExtractProfitForBuy ? totalProfit : 0n;
@@ -323,17 +315,18 @@ export async function pancakeProxyBatchBuyMerkle(params) {
     if (!useNativeToken && quoteTokenDecimals !== undefined && quoteTokenDecimals !== 18) {
         const decimalsDiff = 18 - quoteTokenDecimals;
         const divisor = BigInt(10 ** decimalsDiff);
-        actualAmountsWei = remainingAmounts.map((amount) => amount / divisor);
+        actualAmountsWei = remainingAmounts.map(amount => amount / divisor);
     }
     const presetGasPrice = config.gasPrice;
     const presetNonces = config.nonces;
-    const [gasPrice, nonces] = await Promise.all([
+    const [gasPrice, tokenDecimals, nonces] = await Promise.all([
         presetGasPrice !== undefined
             ? Promise.resolve(presetGasPrice)
             : getOptimizedGasPrice(provider, getGasPriceConfig(config)),
+        getTokenDecimals(tokenAddress, provider),
         presetNonces && presetNonces.length === buyers.length
             ? Promise.resolve(presetNonces)
-            : allocateProfitAwareNonces(buyers, shouldExtractProfitForBuy, maxFundsIndex, nativeProfitAmount, nonceManager),
+            : allocateProfitAwareNonces(buyers, shouldExtractProfitForBuy, maxFundsIndex, nativeProfitAmount, nonceManager)
     ]);
     const minOuts = new Array(buyers.length).fill(0n);
     const needBNB = needSendBNB(routeType, params, useNativeToken);
@@ -344,14 +337,14 @@ export async function pancakeProxyBatchBuyMerkle(params) {
         if (!params.v2Path || params.v2Path.length < 2) {
             throw new Error('v2Path is required for V2 routing');
         }
-        routers = buyers.map((w) => new Contract(PANCAKE_V2_ROUTER_ADDRESS, V2_ROUTER_ABI, w));
+        routers = buyers.map(w => new Contract(PANCAKE_V2_ROUTER_ADDRESS, V2_ROUTER_ABI, w));
         unsignedBuys = await buildV2Transactions(routers, buyers, actualAmountsWei, minOuts, params.v2Path, true, needBNB);
     }
     else if (routeType === 'v3-single') {
         if (!params.v3TokenIn || !params.v3Fee) {
             throw new Error('v3TokenIn and v3Fee are required for V3 single-hop');
         }
-        routers = buyers.map((w) => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
+        routers = buyers.map(w => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
         unsignedBuys = await buildV3SingleTransactions(routers, buyers, params.v3TokenIn, tokenAddress, params.v3Fee, actualAmountsWei, minOuts, true, needBNB);
     }
     else if (routeType === 'v3-multi') {
@@ -362,7 +355,7 @@ export async function pancakeProxyBatchBuyMerkle(params) {
         if (!params.v3Fees || params.v3Fees.length !== params.v3Tokens.length - 1) {
             throw new Error(`v3Fees 长度必须等于 v3Tokens 长度 - 1`);
         }
-        routers = buyers.map((w) => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
+        routers = buyers.map(w => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
         unsignedBuys = await buildV3MultiHopTransactions(routers, buyers, params.v3Tokens, params.v3Fees, actualAmountsWei, minOuts, true, needBNB);
     }
     else {
@@ -386,11 +379,11 @@ export async function pancakeProxyBatchBuyMerkle(params) {
             gasPrice,
             gasLimit: GAS_LIMITS.BRIBE,
             chainId,
-            type: txType,
+            type: txType
         }));
     }
     unsignedBuys.forEach((unsigned, i) => {
-        const txValue = useNativeToken ? (unsigned.value ?? 0n) : 0n;
+        const txValue = useNativeToken ? unsigned.value : 0n;
         signPromises.push(buyers[i].signTransaction({
             ...unsigned,
             from: buyers[i].address,
@@ -399,7 +392,7 @@ export async function pancakeProxyBatchBuyMerkle(params) {
             gasPrice,
             chainId,
             type: txType,
-            value: txValue,
+            value: txValue
         }));
     });
     const signedTxs = await Promise.all(signPromises);
@@ -416,7 +409,7 @@ export async function pancakeProxyBatchBuyMerkle(params) {
             gasPrice,
             chainId,
             txType,
-            startNonce: profitNonce,
+            startNonce: profitNonce
         });
         signedTxs.push(...profitHopResult.signedTransactions);
         profitHopWallets = profitHopResult.hopWallets; // ✅ 收集利润多跳钱包
@@ -424,7 +417,7 @@ export async function pancakeProxyBatchBuyMerkle(params) {
     nonceManager.clearTemp();
     return {
         signedTransactions: signedTxs,
-        profitHopWallets, // ✅ 返回利润多跳钱包
+        profitHopWallets // ✅ 返回利润多跳钱包
     };
 }
 /**
@@ -437,9 +430,9 @@ export async function pancakeProxyBatchSellMerkle(params) {
     }
     const chainId = CHAIN_ID_MAP[chain];
     const provider = getCachedProvider(chain, config.rpcUrl, chainId);
-    const sellers = privateKeys.map((k) => new Wallet(k, provider));
+    const sellers = privateKeys.map(k => new Wallet(k, provider));
     const finalGasLimit = getGasLimit(config);
-    const extractProfit = shouldExtractProfit();
+    const extractProfit = shouldExtractProfit(config);
     const nonceManager = new NonceManager(provider);
     const presetGasPrice = config.gasPrice;
     const presetNonces = config.nonces;
@@ -447,9 +440,9 @@ export async function pancakeProxyBatchSellMerkle(params) {
         presetGasPrice !== undefined
             ? Promise.resolve(presetGasPrice)
             : getOptimizedGasPrice(provider, getGasPriceConfig(config)),
-        getTokenDecimals(tokenAddress, provider),
+        getTokenDecimals(tokenAddress, provider)
     ]);
-    const amountsWei = sellAmounts.map((amount) => ethers.parseUnits(amount, tokenDecimals));
+    const amountsWei = sellAmounts.map(amount => ethers.parseUnits(amount, tokenDecimals));
     // 获取报价
     let quotedOutputs;
     if (routeType === 'v2' && params.v2Path && params.v2Path.length >= 2) {
@@ -464,7 +457,7 @@ export async function pancakeProxyBatchSellMerkle(params) {
                     tokenOut: params.v3TokenOut,
                     amountIn: amount,
                     fee: params.v3Fee,
-                    sqrtPriceLimitX96: 0,
+                    sqrtPriceLimitX96: 0
                 });
                 return result[0];
             }
@@ -485,7 +478,7 @@ export async function pancakeProxyBatchSellMerkle(params) {
         for (let i = 0; i < sellers.length; i++) {
             const quoted = quotedOutputs[i];
             if (quoted > 0n) {
-                const { profit } = calculateProfit(quoted);
+                const { profit } = calculateProfit(quoted, config);
                 totalProfit += profit;
                 if (quoted > maxRevenue) {
                     maxRevenue = quoted;
@@ -512,7 +505,9 @@ export async function pancakeProxyBatchSellMerkle(params) {
     else if (maxRevenueNonceCount > 1 && maxRevenueIndex >= 0) {
         const maxRevenueNonces = await nonceManager.getNextNonceBatch(sellers[maxRevenueIndex], maxRevenueNonceCount);
         const otherSellers = sellers.filter((_, i) => i !== maxRevenueIndex);
-        const otherNonces = otherSellers.length > 0 ? await nonceManager.getNextNoncesForWallets(otherSellers) : [];
+        const otherNonces = otherSellers.length > 0
+            ? await nonceManager.getNextNoncesForWallets(otherSellers)
+            : [];
         nonces = [];
         let otherIdx = 0;
         let nonceIdx = 0;
@@ -541,14 +536,14 @@ export async function pancakeProxyBatchSellMerkle(params) {
         if (!params.v2Path || params.v2Path.length < 2) {
             throw new Error('v2Path is required for V2 routing');
         }
-        routers = sellers.map((w) => new Contract(PANCAKE_V2_ROUTER_ADDRESS, V2_ROUTER_ABI, w));
+        routers = sellers.map(w => new Contract(PANCAKE_V2_ROUTER_ADDRESS, V2_ROUTER_ABI, w));
         unsignedSells = await buildV2Transactions(routers, sellers, amountsWei, minOuts, params.v2Path, false, false);
     }
     else if (routeType === 'v3-single') {
         if (!params.v3TokenOut || !params.v3Fee) {
             throw new Error('v3TokenOut and v3Fee are required for V3 single-hop');
         }
-        routers = sellers.map((w) => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
+        routers = sellers.map(w => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
         unsignedSells = await buildV3SingleTransactions(routers, sellers, tokenAddress, params.v3TokenOut, params.v3Fee, amountsWei, minOuts, false, false);
     }
     else if (routeType === 'v3-multi') {
@@ -559,7 +554,7 @@ export async function pancakeProxyBatchSellMerkle(params) {
         if (!params.v3Fees || params.v3Fees.length !== params.v3Tokens.length - 1) {
             throw new Error(`v3Fees 长度必须等于 v3Tokens 长度 - 1`);
         }
-        routers = sellers.map((w) => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
+        routers = sellers.map(w => new Contract(PANCAKE_V3_ROUTER_ADDRESS, V3_ROUTER_ABI, w));
         unsignedSells = await buildV3MultiHopTransactions(routers, sellers, params.v3Tokens, params.v3Fees, amountsWei, minOuts, false, false);
     }
     else {
@@ -575,7 +570,7 @@ export async function pancakeProxyBatchSellMerkle(params) {
             gasPrice,
             gasLimit: GAS_LIMITS.BRIBE,
             chainId,
-            type: txType,
+            type: txType
         }));
     }
     unsignedSells.forEach((unsigned, i) => {
@@ -588,7 +583,7 @@ export async function pancakeProxyBatchSellMerkle(params) {
             gasPrice,
             chainId,
             type: txType,
-            value: txValue,
+            value: txValue
         }));
     });
     const signedTxs = await Promise.all(signPromises);
@@ -604,7 +599,7 @@ export async function pancakeProxyBatchSellMerkle(params) {
             gasPrice,
             chainId,
             txType,
-            startNonce: profitNonce,
+            startNonce: profitNonce
         });
         signedTxs.push(...profitHopResult.signedTransactions);
         profitHopWallets = profitHopResult.hopWallets; // ✅ 收集利润多跳钱包
@@ -612,7 +607,7 @@ export async function pancakeProxyBatchSellMerkle(params) {
     nonceManager.clearTemp();
     return {
         signedTransactions: signedTxs,
-        profitHopWallets, // ✅ 返回利润多跳钱包
+        profitHopWallets // ✅ 返回利润多跳钱包
     };
 }
 // ==================== 内部工具函数 ====================
@@ -635,10 +630,10 @@ async function batchGetV2Quotes(provider, amountsWei, v2Path) {
     try {
         const v2RouterIface = new Interface(V2_ROUTER_ABI);
         const multicall = new Contract(MULTICALL3_ADDRESS, MULTICALL3_ABI, provider);
-        const calls = amountsWei.map((amount) => ({
+        const calls = amountsWei.map(amount => ({
             target: PANCAKE_V2_ROUTER_ADDRESS,
             allowFailure: true,
-            callData: v2RouterIface.encodeFunctionData('getAmountsOut', [amount, v2Path]),
+            callData: v2RouterIface.encodeFunctionData('getAmountsOut', [amount, v2Path])
         }));
         const results = await multicall.aggregate3.staticCall(calls);
         return results.map((r) => {
@@ -689,7 +684,9 @@ async function allocateProfitAwareNonces(wallets, extractProfit, maxIndex, total
     }
     const maxIndexNonces = await nonceManager.getNextNonceBatch(wallets[maxIndex], 2);
     const otherWallets = wallets.filter((_, i) => i !== maxIndex);
-    const otherNonces = otherWallets.length > 0 ? await nonceManager.getNextNoncesForWallets(otherWallets) : [];
+    const otherNonces = otherWallets.length > 0
+        ? await nonceManager.getNextNoncesForWallets(otherWallets)
+        : [];
     const nonces = [];
     let otherIdx = 0;
     for (let i = 0; i < wallets.length; i++) {