LEDMatrix/scripts/utils/start_web_conditionally.py

import json
import os
import sys
import subprocess
from pathlib import Path

# Get project root directory (parent of scripts/utils/)
PROJECT_DIR = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
CONFIG_FILE = os.path.join(PROJECT_DIR, 'config', 'config.json')
WEB_INTERFACE_SCRIPT = os.path.join(PROJECT_DIR, 'web_interface', 'start.py')
# Marker file created by first_time_install.sh to indicate dependencies are installed
DEPS_MARKER = os.path.join(PROJECT_DIR, '.web_deps_installed')


def dependencies_installed():
    """Check if dependencies were installed during first-time setup."""
    return os.path.exists(DEPS_MARKER)

def install_dependencies():
    """Install required dependencies using system Python."""
    print("Installing dependencies...")
    try:
        requirements_file = os.path.join(PROJECT_DIR, 'web_interface', 'requirements.txt')
        # Use --ignore-installed to handle system packages (like psutil) that can't be uninstalled
        # This allows pip to install even if a system package version conflicts
        result = subprocess.run([
            sys.executable, '-m', 'pip', 'install', '--break-system-packages', '--ignore-installed', '-r', requirements_file
        ], capture_output=True, text=True)
        
        if result.returncode != 0:
            # Check if the error is just about psutil version conflict
            if 'psutil' in result.stderr.lower() and ('uninstall' in result.stderr.lower() or 'cannot uninstall' in result.stderr.lower()):
                print("Warning: psutil version conflict detected (system package vs requirements).")
                print("Attempting to install other dependencies without psutil...")
                # Try installing without psutil
                with open(requirements_file, 'r') as f:
                    lines = f.readlines()
                # Filter out psutil line
                filtered_lines = [line for line in lines if 'psutil' not in line.lower()]
                temp_reqs = os.path.join(PROJECT_DIR, 'web_interface', 'requirements_temp.txt')
                with open(temp_reqs, 'w') as f:
                    f.writelines(filtered_lines)
                try:
                    subprocess.check_call([
                        sys.executable, '-m', 'pip', 'install', '--break-system-packages', '--ignore-installed', '-r', temp_reqs
                    ])
                    print("Dependencies installed successfully (psutil skipped - using system version)")
                finally:
                    if os.path.exists(temp_reqs):
                        os.remove(temp_reqs)
            else:
                # Re-raise the error if it's not about psutil
                print(f"Failed to install dependencies: {result.stderr}")
                return False
        else:
            print("Dependencies installed successfully")
        
        # Install rgbmatrix module from local source (optional - not required for web interface)
        print("Installing rgbmatrix module (optional)...")
        rgbmatrix_path = Path(PROJECT_DIR) / 'rpi-rgb-led-matrix-master' / 'bindings' / 'python'
        if rgbmatrix_path.exists():
            try:
                subprocess.check_call([
                    sys.executable, '-m', 'pip', 'install', '--break-system-packages', '-e', str(rgbmatrix_path)
                ], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
                print("rgbmatrix module installed successfully")
            except subprocess.CalledProcessError:
                print("Warning: rgbmatrix module installation failed (not required for web interface, continuing...)")
        else:
            print("rgbmatrix module path not found (not required for web interface, continuing...)")
        
        return True
    except subprocess.CalledProcessError as e:
        print(f"Failed to install dependencies: {e}")
        return False

def main():
    try:
        with open(CONFIG_FILE, 'r') as f:
            config_data = json.load(f)
    except FileNotFoundError:
        print(f"Config file {CONFIG_FILE} not found. Web interface will not start.")
        sys.exit(0) # Exit gracefully, don't start
    except Exception as e:
        print(f"Error reading config file {CONFIG_FILE}: {e}. Web interface will not start.")
        sys.exit(1) # Exit with error, service might restart depending on config

    autostart_enabled = config_data.get("web_display_autostart", False)

    # Handle both boolean True and string "on"/"true" values
    is_enabled = (autostart_enabled is True) or (isinstance(autostart_enabled, str) and autostart_enabled.lower() in ("on", "true", "yes", "1"))

    if is_enabled:
        print("Configuration 'web_display_autostart' is enabled. Starting web interface...")

        # Only install dependencies if not already done during first-time setup
        if not dependencies_installed():
            print("First run detected: Installing dependencies...")
            if not install_dependencies():
                print("Failed to install dependencies. Exiting.")
                sys.exit(1)
            # Create marker file after successful install
            Path(DEPS_MARKER).touch()
            print("Dependencies installed and marker file created.")
        else:
            print("Dependencies already installed (marker file found). Skipping installation.")
        
        try:
            # Replace the current process with web_interface.py using system Python
            # This is important for systemd to correctly manage the web server process.
            # Ensure PYTHONPATH is set correctly if web_interface.py has relative imports to src
            # The WorkingDirectory in systemd service should handle this for web_interface.py
            print(f"Launching web interface v3: {sys.executable} {WEB_INTERFACE_SCRIPT}")
            os.execvp(sys.executable, [sys.executable, WEB_INTERFACE_SCRIPT])
        except Exception as e:
            print(f"Failed to exec web interface: {e}")
            sys.exit(1) # Failed to start
    else:
        print("Configuration 'web_display_autostart' is false or not set. Web interface will not be started.")
        sys.exit(0) # Exit gracefully, service considered successful

if __name__ == '__main__':
    main()