@clonegod/ttd-base-common 1.0.9 → 1.0.10

package/dist/trade/abstract_dex_trade_plus.d.ts

@@ -1,6 +1,7 @@
 import { AbastrcatTrade, AppConfig, TradeContext } from "@clonegod/ttd-core/dist";
 import { ethers } from "ethers";
 import { DexConfig, EvmChainConfig } from "../types";
+import { SimpleRedisClient } from "../redis";
 export declare abstract class AbstractEvmDexTradePlus extends AbastrcatTrade {
     protected appConfig: AppConfig;
     protected group_wallets: ethers.Wallet[];
@@ -13,6 +14,8 @@ export declare abstract class AbstractEvmDexTradePlus extends AbastrcatTrade {
     protected routerContract: ethers.Contract;
     protected chainConfig: EvmChainConfig;
     protected dexConfig: DexConfig;
+    protected redisClient: SimpleRedisClient;
+    protected chainNameLower: string;
     constructor(appConfig: AppConfig);
     protected abstract initConfigs(): void;
     init(): Promise<void>;
@@ -32,4 +35,9 @@ export declare abstract class AbstractEvmDexTradePlus extends AbastrcatTrade {
     protected getWrappedNativeAddress(): string;
     abstract execute(context: TradeContext, retryCount?: number): Promise<string>;
     protected buildTipTransferTx(to: string, transfer_amount_gwei: string, gas_price_gwei: string, transfer_nonce: number, wallet: ethers.Wallet): Promise<string>;
+    protected chooseWallet(context: TradeContext): Promise<ethers.Wallet>;
+    protected determineInputOutputTokens(order_msg: any, pool_info: any): {
+        inputToken: any;
+        outputToken: any;
+    };
 }

package/dist/trade/abstract_dex_trade_plus.js

@@ -8,11 +8,16 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
         step((generator = generator.apply(thisArg, _arguments || [])).next());
     });
 };
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AbstractEvmDexTradePlus = void 0;
 const dist_1 = require("@clonegod/ttd-core/dist");
 const ethers_1 = require("ethers");
 const common_1 = require("../common");
+const decimal_js_1 = __importDefault(require("decimal.js"));
+const redis_1 = require("../redis");
 class AbstractEvmDexTradePlus extends dist_1.AbastrcatTrade {
     constructor(appConfig) {
         super();
@@ -25,6 +30,8 @@ class AbstractEvmDexTradePlus extends dist_1.AbastrcatTrade {
         this.pairContracts = new Map();
         this.tokenContracts = new Map();
         this.initConfigs();
+        this.chainNameLower = this.appConfig.env_args.chain_id.toLowerCase();
+        this.redisClient = new redis_1.SimpleRedisClient(`${this.chainNameLower}:tx`);
     }
     init() {
         return __awaiter(this, void 0, void 0, function* () {
@@ -276,5 +283,139 @@ class AbstractEvmDexTradePlus extends dist_1.AbastrcatTrade {
             return signedTx;
         });
     }
+    chooseWallet(context) {
+        return __awaiter(this, void 0, void 0, function* () {
+            var _a, _b;
+            if (this.getWalletMode() === 'single') {
+                return this.getSingleWallet();
+            }
+            const { inputToken } = this.determineInputOutputTokens(context.order_msg, context.pool_info);
+            console.log('inputToken', inputToken);
+            const requiredAmount = new decimal_js_1.default(context.order_msg.amount);
+            const wallet_assets = yield this.redisClient.hgetall(`${this.chainNameLower}:wallet:assets`);
+            if (!wallet_assets || Object.keys(wallet_assets).length === 0) {
+                throw new Error('没有找到钱包资产信息，请确保钱包监控服务正在运行');
+            }
+            const wallet_assets_map = {};
+            for (const [walletAddress, assetsJson] of Object.entries(wallet_assets)) {
+                try {
+                    wallet_assets_map[walletAddress] = JSON.parse(assetsJson);
+                }
+                catch (error) {
+                    console.warn(`解析钱包资产数据失败: ${walletAddress}`, error);
+                }
+            }
+            const availableWallets = [];
+            const allWallets = [];
+            for (const [walletAddress, assets] of Object.entries(wallet_assets_map)) {
+                try {
+                    const wallet = (_a = this.group_wallets) === null || _a === void 0 ? void 0 : _a.find(w => w.address.toLowerCase() === walletAddress.toLowerCase());
+                    if (!wallet) {
+                        continue;
+                    }
+                    const tokenAsset = (_b = assets.tokens) === null || _b === void 0 ? void 0 : _b.find((token) => token.tokenaddr.toLowerCase() === inputToken.address.toLowerCase());
+                    let tokenBalance = (tokenAsset === null || tokenAsset === void 0 ? void 0 : tokenAsset.balance) || '0';
+                    const balanceDecimal = new decimal_js_1.default(tokenBalance);
+                    if (balanceDecimal.gte(requiredAmount)) {
+                        availableWallets.push({
+                            wallet,
+                            balance: tokenBalance
+                        });
+                    }
+                    else {
+                        console.log(`钱包 ${walletAddress} 余额不足: ${tokenBalance} ${inputToken.symbol} < ${requiredAmount} ${inputToken.symbol}`);
+                    }
+                    allWallets.push({
+                        wallet,
+                        balance: tokenBalance
+                    });
+                }
+                catch (error) {
+                    console.warn(`处理钱包资产时出错: ${walletAddress}`, error);
+                }
+            }
+            if (availableWallets.length === 0) {
+                let error_msg = `all wallets are insufficient, required: ${requiredAmount} ${inputToken.symbol}`;
+                (0, dist_1.log_warn)(error_msg);
+                allWallets.forEach((walletInfo, index) => {
+                    const shortAddress = walletInfo.wallet.address.substring(0, 6) + '...' + walletInfo.wallet.address.substring(walletInfo.wallet.address.length - 4);
+                    console.log(`${index + 1}. ${shortAddress}: ${walletInfo.balance} ${inputToken.symbol}`);
+                });
+                throw new Error(error_msg);
+            }
+            console.log('availableWallets', availableWallets.map(w => w.wallet.address));
+            const wallet_last_used = yield this.redisClient.hgetall(`${this.chainNameLower}:wallet:last_used`);
+            for (const availableWallet of availableWallets) {
+                const lastUsedData = wallet_last_used === null || wallet_last_used === void 0 ? void 0 : wallet_last_used[availableWallet.wallet.address];
+                if (lastUsedData) {
+                    try {
+                        const lastUsedInfo = JSON.parse(lastUsedData);
+                        availableWallet.lastUsedAt = lastUsedInfo.lastUsedAt;
+                    }
+                    catch (error) {
+                        console.warn(`解析钱包最后使用时间失败: ${availableWallet.wallet.address}`, error);
+                    }
+                }
+                else {
+                    yield this.redisClient.hsetValue(`${this.chainNameLower}:wallet:last_used`, availableWallet.wallet.address, JSON.stringify({
+                        lastUsedAt: 0,
+                        lastUsedTime: new Date(0).toISOString()
+                    }), 24 * 60 * 60);
+                    availableWallet.lastUsedAt = 0;
+                }
+            }
+            let selectedWallet;
+            let selectionStrategy;
+            if (availableWallets.length === 1) {
+                selectedWallet = availableWallets[0].wallet;
+                selectionStrategy = 'OnlyOne';
+            }
+            else {
+                const hasUsageHistory = wallet_last_used && Object.keys(wallet_last_used).length > 0;
+                if (hasUsageHistory) {
+                    availableWallets.sort((a, b) => {
+                        const aTime = a.lastUsedAt || 0;
+                        const bTime = b.lastUsedAt || 0;
+                        return aTime - bTime;
+                    });
+                    selectedWallet = availableWallets[0].wallet;
+                    selectionStrategy = 'MaxLastUsed';
+                }
+                else {
+                    availableWallets.sort((a, b) => {
+                        const aBalance = new decimal_js_1.default(a.balance);
+                        const bBalance = new decimal_js_1.default(b.balance);
+                        return bBalance.cmp(aBalance);
+                    });
+                    selectedWallet = availableWallets[0].wallet;
+                    selectionStrategy = 'MaxBalance';
+                }
+            }
+            yield this.redisClient.hsetValue(`${this.chainNameLower}:wallet:last_used`, selectedWallet.address, JSON.stringify({
+                lastUsedAt: Date.now(),
+                lastUsedTime: new Date().toISOString()
+            }), 24 * 60 * 60);
+            const selectedWalletInfo = availableWallets.find(w => w.wallet.address === selectedWallet.address);
+            (0, dist_1.log_info)(`Choose wallet: ${selectionStrategy} -> ${selectedWallet.address}, balance: ${selectedWalletInfo === null || selectedWalletInfo === void 0 ? void 0 : selectedWalletInfo.balance} ${inputToken.symbol}, required: ${requiredAmount} ${inputToken.symbol}`);
+            return selectedWallet;
+        });
+    }
+    determineInputOutputTokens(order_msg, pool_info) {
+        const { aToB } = order_msg;
+        const { tokenA, tokenB, quote_token } = pool_info;
+        const quoteToken = [tokenA, tokenB].find(token => token.symbol === quote_token);
+        const nonQuoteToken = [tokenA, tokenB].find(token => token.symbol !== quote_token);
+        const inputToken = aToB ? nonQuoteToken : quoteToken;
+        const outputToken = aToB ? quoteToken : nonQuoteToken;
+        if (!inputToken || !outputToken) {
+            throw new Error(`无法确定输入/输出代币: ${JSON.stringify({
+                tokenA: tokenA.symbol,
+                tokenB: tokenB.symbol,
+                quote: quote_token,
+                aToB
+            })}`);
+        }
+        return { inputToken, outputToken };
+    }
 }
 exports.AbstractEvmDexTradePlus = AbstractEvmDexTradePlus;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "@clonegod/ttd-base-common",
-    "version": "1.0.9",
+    "version": "1.0.10",
     "description": "Base common library",
     "license": "UNLICENSED",
     "main": "dist/index.js",