claude-mpm 4.1.6__py3-none-any.whl → 4.24.0__py3-none-any.whl

claude_mpm/scripts/socketio_daemon.py

@@ -1,458 +1,218 @@
 #!/usr/bin/env python3
 """
-Pure Python daemon management for Socket.IO server.
-No external dependencies required.
+SocketIO Daemon Script for Claude MPM.
+
+This script provides a command-line interface to the unified monitor daemon
+that runs the Socket.IO server for event broadcasting and dashboard functionality.
+
+WHY: The pyproject.toml references this as an entry point for claude-mpm-socketio,
+providing a dedicated command for managing the Socket.IO daemon process.
 """
 
+import argparse
 import os
 import signal
-import subprocess
 import sys
 import time
 from pathlib import Path
 
+# Add parent directory to path for imports
+sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
 
-# Detect and use virtual environment Python if available
-def get_python_executable():
-    """
-    Get the appropriate Python executable, preferring virtual environment.
-
-    WHY: The daemon must use the same Python environment as the parent process
-    to ensure all dependencies are available. System Python won't have the
-    required packages installed.
-    """
-    # First, check if we're already in a virtual environment
-    if hasattr(sys, "real_prefix") or (
-        hasattr(sys, "base_prefix") and sys.base_prefix != sys.prefix
-    ):
-        # We're in a virtual environment, use its Python
-        return sys.executable
-
-    # Check for common virtual environment indicators
-    # 1. VIRTUAL_ENV environment variable (most common)
-    venv_path = os.environ.get("VIRTUAL_ENV")
-    if venv_path:
-        venv_python = Path(venv_path) / "bin" / "python"
-        if venv_python.exists():
-            return str(venv_python)
-
-    # 2. Check if current executable is in a venv directory structure
-    exe_path = Path(sys.executable).resolve()
-    for parent in exe_path.parents:
-        # Check for common venv directory names
-        if parent.name in ("venv", ".venv", "env", ".env"):
-            # This looks like a virtual environment
-            return sys.executable
-
-        # Check for typical venv structure (bin/python or Scripts/python.exe)
-        if parent.name == "bin" and (parent.parent / "pyvenv.cfg").exists():
-            return sys.executable
-        if parent.name == "Scripts" and (parent.parent / "pyvenv.cfg").exists():
-            return sys.executable
-
-    # 3. Try to detect project-specific venv
-    # Look for venv in the project root (going up from script location)
-    script_path = Path(__file__).resolve()
-    for parent in script_path.parents:
-        # Stop at src or when we've gone too far up
-        if parent.name == "src" or not (parent / "src").exists():
-            # Check for venv directories
-            for venv_name in ("venv", ".venv", "env", ".env"):
-                venv_dir = parent / venv_name
-                if venv_dir.exists():
-                    venv_python = venv_dir / "bin" / "python"
-                    if venv_python.exists():
-                        return str(venv_python)
-            break
-
-    # Fall back to current Python executable
-    return sys.executable
-
-
-# Store the detected Python executable for daemon usage
-PYTHON_EXECUTABLE = get_python_executable()
-
-import psutil
-
-# Handle imports for both development and installed scenarios
-script_dir = Path(__file__).parent
-try:
-    # When installed as package, this should work directly
-    from claude_mpm.services.port_manager import PortManager
-    from claude_mpm.services.socketio.server.main import SocketIOServer
-except ImportError:
-    # Need to add the appropriate directory to sys.path
-    import sys
-
-    # Get the absolute path of this script
-    script_path = Path(__file__).resolve()
-
-    # Determine if we're in development or installed environment
-    if "site-packages" in str(script_path):
-        # Installed environment: ~/.local/pipx/venvs/claude-mpm/lib/python3.13/site-packages/claude_mpm/scripts/socketio_daemon.py
-        # Need to add site-packages directory to path
-        parts = script_path.parts
-        site_packages_idx = next(
-            i for i, part in enumerate(parts) if part == "site-packages"
-        )
-        site_packages_path = Path(*parts[: site_packages_idx + 1])
-
-        if site_packages_path.exists() and str(site_packages_path) not in sys.path:
-            sys.path.insert(0, str(site_packages_path))
-    else:
-        # Development environment: Project/src/claude_mpm/scripts/socketio_daemon.py
-        # Need to add src directory to path
-        # Go up: scripts -> claude_mpm -> src
-        src_path = script_path.parent.parent.parent
-
-        if (
-            src_path.exists()
-            and (src_path / "claude_mpm").exists()
-            and str(src_path) not in sys.path
-        ):
-            sys.path.insert(0, str(src_path))
-
-    # Try importing again after path modification
-    try:
-        from claude_mpm.services.port_manager import PortManager
-        from claude_mpm.services.socketio.server.main import SocketIOServer
-    except ImportError as e:
-        print(f"❌ Failed to import SocketIOServer after path adjustment: {e}")
-        print(f"📍 Script path: {script_path}")
-        print(f"🐍 Python path entries: {len(sys.path)}")
-        for i, path in enumerate(sys.path):
-            print(f"   [{i}] {path}")
-
-        # Check if claude_mpm directory exists in any path
-        claude_mpm_found = False
-        for path_str in sys.path:
-            claude_mpm_path = Path(path_str) / "claude_mpm"
-            if claude_mpm_path.exists():
-                print(f"✅ Found claude_mpm at: {claude_mpm_path}")
-                claude_mpm_found = True
-
-        if not claude_mpm_found:
-            print("❌ claude_mpm directory not found in any sys.path entry")
-
-        print("\n💡 Troubleshooting tips:")
-        print("   1. Ensure claude-mpm is properly installed: pip install claude-mpm")
-        print("   2. If in development, ensure you're in the project root directory")
-        print("   3. Check that PYTHONPATH includes the package location")
-        sys.exit(1)
-
-# Use deployment root for daemon files to keep everything centralized
-from claude_mpm.core.unified_paths import get_project_root
-
-deployment_root = get_project_root()
-PID_FILE = deployment_root / ".claude-mpm" / "socketio-server.pid"
-LOG_FILE = deployment_root / ".claude-mpm" / "socketio-server.log"
-
-
-def ensure_dirs():
-    """Ensure required directories exist."""
-    PID_FILE.parent.mkdir(parents=True, exist_ok=True)
-
-
-def is_running():
-    """Check if server is already running."""
-    if not PID_FILE.exists():
+from claude_mpm.core.logging_config import get_logger
+from claude_mpm.services.monitor.daemon import UnifiedMonitorDaemon
+from claude_mpm.services.port_manager import PortManager
+
+# Default paths
+DEFAULT_PID_FILE = Path.home() / ".claude-mpm" / "socketio-server.pid"
+DEFAULT_LOG_FILE = Path.home() / ".claude-mpm" / "logs" / "socketio" / "daemon.log"
+DEFAULT_PORT = 8765
+
+logger = get_logger(__name__)
+
+
+def is_running(pid_file: Path) -> bool:
+    """Check if daemon is running."""
+    if not pid_file.exists():
         return False
 
     try:
-        with open(PID_FILE) as f:
+        with pid_file.open() as f:
             pid = int(f.read().strip())
 
         # Check if process exists
-        process = psutil.Process(pid)
-        return process.is_running()
-    except (ValueError, psutil.NoSuchProcess, psutil.AccessDenied):
-        # Clean up stale PID file
-        PID_FILE.unlink(missing_ok=True)
+        os.kill(pid, 0)
+        return True
+    except (OSError, ValueError):
+        # Process doesn't exist or invalid PID
+        if pid_file.exists():
+            pid_file.unlink()
         return False
 
 
-def start_server():
-    """Start the Socket.IO server as a daemon with dynamic port selection."""
-    # Initialize port manager
-    port_manager = PortManager()
+def start_server(port: int = DEFAULT_PORT, daemon: bool = True) -> bool:
+    """Start the Socket.IO server."""
+    pid_file = DEFAULT_PID_FILE
+    log_file = DEFAULT_LOG_FILE
+
+    # Ensure directories exist
+    pid_file.parent.mkdir(parents=True, exist_ok=True)
+    log_file.parent.mkdir(parents=True, exist_ok=True)
+
+    if is_running(pid_file):
+        logger.info("Socket.IO daemon is already running")
+        return False
 
-    # Clean up any dead instances first
-    port_manager.cleanup_dead_instances()
-
-    # Check if we already have a running instance
-    if is_running():
-        print("Socket.IO daemon server is already running.")
-        print(f"Use '{__file__} status' for details")
-        return
-
-    # Find an available port
-    selected_port = port_manager.find_available_port()
-    if not selected_port:
-        print("❌ No available ports in range 8765-8785")
-        print("   All ports are either in use or blocked")
-        return
-
-    print(f"🔍 Selected port: {selected_port}")
-
-    # Check for existing instances on this port
-    existing_instance = port_manager.get_instance_by_port(selected_port)
-    if existing_instance:
-        print(f"⚠️  Port {selected_port} is already used by claude-mpm instance:")
-        print(f"   PID: {existing_instance.get('pid')}")
-        print(f"   Started: {time.ctime(existing_instance.get('start_time', 0))}")
-        return
-
-    ensure_dirs()
-
-    # Fork to create daemon using the correct Python environment
-    pid = os.fork()
-    if pid > 0:
-        # Parent process
-        print(f"Starting Socket.IO server on port {selected_port} (PID: {pid})...")
-        print(f"Using Python: {PYTHON_EXECUTABLE}")
-
-        # Register the instance
-        instance_id = port_manager.register_instance(selected_port, pid)
-
-        # Save PID and port info
-        with open(PID_FILE, "w") as f:
-            f.write(str(pid))
-
-        # Save port info for other tools
-        port_file = PID_FILE.parent / "socketio-port"
-        with open(port_file, "w") as f:
-            f.write(str(selected_port))
-
-        print("Socket.IO server started successfully.")
-        print(f"Port: {selected_port}")
-        print(f"Instance ID: {instance_id}")
-        print(f"PID file: {PID_FILE}")
-        print(f"Log file: {LOG_FILE}")
-        sys.exit(0)
-
-    # Child process - become daemon
-    os.setsid()
-    os.umask(0)
-
-    # Redirect stdout/stderr to log file
-    with open(LOG_FILE, "a") as log:
-        os.dup2(log.fileno(), sys.stdout.fileno())
-        os.dup2(log.fileno(), sys.stderr.fileno())
-
-    # Log environment information for debugging
-    print(
-        f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Starting Socket.IO server on port {selected_port}..."
+    # Use PortManager to find available port
+    port_manager = PortManager()
+    actual_port = port_manager.find_available_port(preferred_port=port)
+
+    if actual_port != port:
+        logger.info(f"Port {port} is in use, using port {actual_port} instead")
+
+    # Create and start daemon
+    monitor_daemon = UnifiedMonitorDaemon(
+        host="localhost",
+        port=actual_port,
+        daemon_mode=daemon,
+        pid_file=str(pid_file),
+        log_file=str(log_file),
     )
-    print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Python executable: {sys.executable}")
-    print(f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Python version: {sys.version}")
-    print(
-        f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Python path: {sys.path[:3]}..."
-    )  # Show first 3 entries
-    server = SocketIOServer(host="localhost", port=selected_port)
-
-    # Handle signals
-    def signal_handler(signum, frame):
-        print(
-            f"[{time.strftime('%Y-%m-%d %H:%M:%S')}] Received signal {signum}, shutting down..."
-        )
-        server.stop_sync()
-
-        # Clean up instance registration
-        port_manager_cleanup = PortManager()
-        instances = port_manager_cleanup.load_instances()
-        for instance_id, instance_info in instances.items():
-            if instance_info.get("pid") == os.getpid():
-                port_manager_cleanup.remove_instance(instance_id)
-                break
 
-        PID_FILE.unlink(missing_ok=True)
-        sys.exit(0)
+    success = monitor_daemon.start()
 
-    signal.signal(signal.SIGTERM, signal_handler)
-    signal.signal(signal.SIGINT, signal_handler)
+    if success:
+        logger.info(f"Socket.IO daemon started on port {actual_port}")
+        # Save the port for clients to discover
+        port_file = pid_file.parent / "socketio-port"
+        with port_file.open("w") as f:
+            f.write(str(actual_port))
+    else:
+        logger.error("Failed to start Socket.IO daemon")
 
-    # Start server using synchronous method
-    server.start_sync()
+    return success
 
-    # Keep running
-    try:
-        while True:
-            time.sleep(1)
-    except KeyboardInterrupt:
-        signal_handler(signal.SIGINT, None)
 
+def stop_server() -> bool:
+    """Stop the Socket.IO server."""
+    pid_file = DEFAULT_PID_FILE
 
-def stop_server():
-    """Stop the Socket.IO daemon server."""
-    if not is_running():
-        print("Socket.IO daemon server is not running.")
-        print("Check for other servers: socketio_server_manager.py status")
-        return
+    if not is_running(pid_file):
+        logger.info("Socket.IO daemon is not running")
+        return False
 
     try:
-        with open(PID_FILE) as f:
+        with pid_file.open() as f:
             pid = int(f.read().strip())
 
-        print(f"Stopping Socket.IO server (PID: {pid})...")
+        # Send SIGTERM for graceful shutdown
         os.kill(pid, signal.SIGTERM)
 
-        # Wait for process to stop
+        # Wait for process to terminate
         for _ in range(10):
-            if not is_running():
-                print("Socket.IO server stopped successfully.")
-                PID_FILE.unlink(missing_ok=True)
-                return
+            if not is_running(pid_file):
+                break
             time.sleep(0.5)
 
         # Force kill if still running
-        print("Server didn't stop gracefully, forcing...")
-        os.kill(pid, signal.SIGKILL)
-        PID_FILE.unlink(missing_ok=True)
+        if is_running(pid_file):
+            os.kill(pid, signal.SIGKILL)
+            time.sleep(0.5)
 
-    except Exception as e:
-        print(f"Error stopping server: {e}")
+        # Clean up files
+        if pid_file.exists():
+            pid_file.unlink()
 
+        port_file = pid_file.parent / "socketio-port"
+        if port_file.exists():
+            port_file.unlink()
 
-def status_server():
-    """Check server status with port manager integration."""
-    port_manager = PortManager()
+        logger.info("Socket.IO daemon stopped")
+        return True
 
-    # Clean up dead instances first
-    port_manager.cleanup_dead_instances()
+    except Exception as e:
+        logger.error(f"Error stopping daemon: {e}")
+        return False
 
-    if is_running():
-        with open(PID_FILE) as f:
-            pid = int(f.read().strip())
-        print(f"Socket.IO daemon server is running (PID: {pid})")
-        print(f"PID file: {PID_FILE}")
 
-        # Get port information
-        port_file = PID_FILE.parent / "socketio-port"
-        current_port = 8765  # default
-        if port_file.exists():
-            try:
-                with open(port_file) as f:
-                    current_port = int(f.read().strip())
-            except:
-                pass
+def restart_server(port: int = DEFAULT_PORT) -> bool:
+    """Restart the Socket.IO server."""
+    logger.info("Restarting Socket.IO daemon...")
 
-        # Check if port is listening
-        try:
-            import socket
-
-            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
-            result = sock.connect_ex(("localhost", current_port))
-            sock.close()
-            if result == 0:
-                print(f"✅ Server is listening on port {current_port}")
-                print("🔧 Management style: daemon")
-            else:
-                print(
-                    f"⚠️ WARNING: Server process exists but port {current_port} is not accessible"
-                )
-        except:
-            pass
-
-        # Show instance information
-        instance = port_manager.get_instance_by_port(current_port)
-        if instance:
-            print("\n📊 Instance Information:")
-            print(f"   • Port: {instance.get('port')}")
-            print(f"   • Started: {time.ctime(instance.get('start_time', 0))}")
-            print(f"   • Instance ID: {instance.get('instance_id')}")
-
-        # Show management commands
-        print("\n🔧 Management Commands:")
-        print(f"   • Stop: {__file__} stop")
-        print(f"   • Restart: {__file__} restart")
-        print(f"   • List all: {__file__} list")
-
-        # Check for manager conflicts
-        try:
-            import requests
-
-            response = requests.get("http://localhost:8765/health", timeout=1.0)
-            if response.status_code == 200:
-                data = response.json()
-                if "server_id" in data and data.get("server_id") != "daemon-socketio":
-                    print("\n⚠️  POTENTIAL CONFLICT: HTTP-managed server also detected")
-                    print(f"   Server ID: {data.get('server_id')}")
-                    print("   Use 'socketio_server_manager.py diagnose' to resolve")
-        except:
-            pass
+    # Stop if running
+    if is_running(DEFAULT_PID_FILE):
+        stop_server()
+        time.sleep(1)  # Brief pause between stop and start
 
-    else:
-        print("Socket.IO daemon server is not running")
-        print("\n🔧 Start Commands:")
-        print(f"   • Daemon: {__file__} start")
-        print("   • HTTP-managed: socketio_server_manager.py start")
+    # Start again
+    return start_server(port=port)
 
 
-def list_instances():
-    """List all active SocketIO instances."""
-    port_manager = PortManager()
+def status_server() -> bool:
+    """Check status of Socket.IO server."""
+    pid_file = DEFAULT_PID_FILE
+    port_file = pid_file.parent / "socketio-port"
+
+    if is_running(pid_file):
+        try:
+            with pid_file.open() as f:
+                pid = int(f.read().strip())
+
+            port = DEFAULT_PORT
+            if port_file.exists():
+                with port_file.open() as f:
+                    port = int(f.read().strip())
+
+            print(f"Socket.IO daemon is running (PID: {pid}, Port: {port})")
+            return True
+        except Exception as e:
+            print(f"Socket.IO daemon status unknown: {e}")
+            return False
+    else:
+        print("Socket.IO daemon is not running")
+        return False
 
-    # Clean up dead instances first
-    cleaned = port_manager.cleanup_dead_instances()
-    if cleaned > 0:
-        print(f"🧹 Cleaned up {cleaned} dead instances")
 
-    instances = port_manager.list_active_instances()
+def main():
+    """Main entry point for socketio daemon management."""
+    parser = argparse.ArgumentParser(description="Manage Claude MPM Socket.IO daemon")
 
-    if not instances:
-        print("No active SocketIO instances found.")
-        return
+    parser.add_argument(
+        "command", choices=["start", "stop", "restart", "status"], help="Daemon command"
+    )
 
-    print(f"📊 Active SocketIO Instances ({len(instances)}):")
-    print()
+    parser.add_argument(
+        "--port",
+        type=int,
+        default=DEFAULT_PORT,
+        help=f"Port to run on (default: {DEFAULT_PORT})",
+    )
 
-    for instance in instances:
-        port = instance.get("port")
-        pid = instance.get("pid")
-        start_time = instance.get("start_time", 0)
-        instance_id = instance.get("instance_id", "unknown")
+    parser.add_argument(
+        "--foreground",
+        action="store_true",
+        help="Run in foreground instead of daemon mode",
+    )
 
-        print(f"🔌 Port {port}:")
-        print(f"   • PID: {pid}")
-        print(f"   • Started: {time.ctime(start_time)}")
-        print(f"   • Instance ID: {instance_id}")
-        print(f"   • Project: {instance.get('project_root', 'unknown')}")
-        print()
+    args = parser.parse_args()
 
+    # Execute command
+    if args.command == "start":
+        success = start_server(port=args.port, daemon=not args.foreground)
+        sys.exit(0 if success else 1)
 
-def main():
-    """Main entry point."""
-    if len(sys.argv) < 2:
-        print("Usage: socketio-daemon.py {start|stop|restart|status|list}")
-        sys.exit(1)
+    elif args.command == "stop":
+        success = stop_server()
+        sys.exit(0 if success else 1)
 
-    command = sys.argv[1]
+    elif args.command == "restart":
+        success = restart_server(port=args.port)
+        sys.exit(0 if success else 1)
 
-    if command == "start":
-        start_server()
-    elif command == "stop":
-        stop_server()
-    elif command == "restart":
-        stop_server()
-        time.sleep(1)
-        start_server()
-    elif command == "status":
-        status_server()
-    elif command == "list":
-        list_instances()
-    else:
-        print(f"Unknown command: {command}")
-        print("Usage: socketio-daemon.py {start|stop|restart|status|list}")
-        sys.exit(1)
+    elif args.command == "status":
+        success = status_server()
+        sys.exit(0 if success else 1)
 
 
 if __name__ == "__main__":
-    # Install psutil if not available (using correct Python)
-    try:
-        import psutil
-    except ImportError:
-        print(f"Installing psutil using {PYTHON_EXECUTABLE}...")
-        subprocess.check_call([PYTHON_EXECUTABLE, "-m", "pip", "install", "psutil"])
-        import psutil
-
     main()