#!/bin/bash

################################################################################
# Plex Media Server Management Script
################################################################################
#
# Author: Peter Wood <peter@peterwood.dev>
# Description: Modern, user-friendly Plex Media Server management script with
#              styled output and comprehensive service control capabilities.
#              Provides an interactive interface for common Plex operations.
#
# Features:
#   - Service start/stop/restart/status operations
#   - Styled console output with Unicode symbols
#   - Service health monitoring
#   - Process management and monitoring
#   - Interactive menu system
#
# Related Scripts:
#   - backup-plex.sh: Comprehensive backup solution
#   - restore-plex.sh: Backup restoration utilities
#   - monitor-plex-backup.sh: Backup system monitoring
#   - validate-plex-backups.sh: Backup validation tools
#   - test-plex-backup.sh: Testing framework
#
# Usage:
#   ./plex.sh start     # Start Plex service
#   ./plex.sh stop      # Stop Plex service
#   ./plex.sh restart   # Restart Plex service
#   ./plex.sh status    # Show service status
#   ./plex.sh           # Interactive menu
#
# Dependencies:
#   - systemctl (systemd service management)
#   - Plex Media Server package
#   - Web browser (for web interface launching)
#
# Exit Codes:
#   0 - Success
#   1 - General error
#   2 - Service operation failure
#   3 - Invalid command or option
#
################################################################################

# 🎬 Plex Media Server Management Script
# A modern script for managing Plex Media Server with style
# Author: acedanger
# Version: 2.0

set -euo pipefail

# 🎨 Color definitions for sexy output
readonly RED='\033[0;31m'
readonly GREEN='\033[0;32m'
readonly YELLOW='\033[1;33m'
readonly BLUE='\033[0;34m'
readonly PURPLE='\033[0;35m'
readonly CYAN='\033[0;36m'
readonly WHITE='\033[1;37m'
readonly BOLD='\033[1m'
readonly DIM='\033[2m'
readonly RESET='\033[0m'

# 🔧 Configuration
readonly PLEX_SERVICE="plexmediaserver"
SCRIPT_NAME="$(basename "$0")"
readonly SCRIPT_NAME
SCRIPT_DIR="$(dirname "$(readlink -f "$0")")"
readonly SCRIPT_DIR

# 🎭 ASCII symbols for compatible output
readonly CHECKMARK="[✓]"
readonly CROSS="[✗]"
readonly ROCKET="[>]"
readonly STOP_SIGN="[■]"
readonly RECYCLE="[~]"
readonly INFO="[i]"
readonly HOURGLASS="[*]"
readonly SPARKLES="[*]"

# 🎉 Function to print completion footer
print_footer() {
    echo -e "\n${DIM}${CYAN}--- Operation completed [*] ---${RESET}\n"
}

# 🎯 Function to print status with style
print_status() {
    local status="$1"
    local message="$2"
    local color="$3"
    echo -e "${color}${BOLD}[${status}]${RESET} ${message}"
}

# ⏱️ Function to show loading animation
show_loading() {
    local message="$1"
    local pid="$2"
    local spin='-\|/'
    local i=0

    echo -ne "${CYAN}${HOURGLASS} ${message}${RESET}"
    while kill -0 "$pid" 2>/dev/null; do
        i=$(( (i+1) %4 ))
        printf "\r%s%s %s %s%s" "${CYAN}" "${HOURGLASS}" "${message}" "${spin:$i:1}" "${RESET}"
        sleep 0.1
    done
    printf "\r%s%s %s %s%s\n" "${CYAN}" "${HOURGLASS}" "${message}" "${CHECKMARK}" "${RESET}"
}

# 🔧 Enhanced function to repair database issues
repair_database() {
    print_status "${INFO}" "Attempting to repair Plex database..." "${BLUE}"
    
    local db_dir="/var/lib/plexmediaserver/Library/Application Support/Plex Media Server/Plug-in Support/Databases"
    local main_db="$db_dir/com.plexapp.plugins.library.db"
    local backup_db="$db_dir/com.plexapp.plugins.library.db.backup.$(date +%Y%m%d_%H%M%S)"
    local corrupted_dir="$db_dir/corrupted-$(date +%Y%m%d_%H%M%S)"
    
    if [[ ! -f "$main_db" ]]; then
        print_status "${CROSS}" "Main database not found at: $main_db" "${RED}"
        return 1
    fi
    
    # Stop Plex service first
    print_status "${INFO}" "Stopping Plex service..." "${BLUE}"
    sudo systemctl stop "$PLEX_SERVICE" 2>/dev/null || true
    sleep 3
    
    # Check if critical tables exist
    print_status "${INFO}" "Checking database structure..." "${BLUE}"
    local has_metadata_table=false
    if sudo -u plex sqlite3 "$main_db" "SELECT name FROM sqlite_master WHERE type='table' AND name='metadata_items';" 2>/dev/null | grep -q metadata_items; then
        has_metadata_table=true
    fi
    
    if [[ "$has_metadata_table" == "false" ]]; then
        print_status "${CROSS}" "Critical table 'metadata_items' is missing! Database is severely corrupted." "${RED}"
        print_status "${INFO}" "Attempting recovery from available backups..." "${YELLOW}"
        
        # Find the best recovery candidate
        local recovery_db=""
        local recovery_candidates=(
            "$db_dir/com.plexapp.plugins.library.db.recovery-"*
            "$db_dir/com.plexapp.plugins.library.db.20"*
        )
        
        for candidate in "${recovery_candidates[@]}"; do
            if [[ -f "$candidate" && "$candidate" != *"tmp"* && "$candidate" != *"empty"* ]]; then
                # Test if this candidate has the metadata_items table
                if sudo -u plex sqlite3 "$candidate" "SELECT name FROM sqlite_master WHERE type='table' AND name='metadata_items';" 2>/dev/null | grep -q metadata_items; then
                    recovery_db="$candidate"
                    break
                fi
            fi
        done
        
        if [[ -n "$recovery_db" ]]; then
            print_status "${CHECKMARK}" "Found recovery database: $(basename "$recovery_db")" "${GREEN}"
            
            # Move corrupted database to backup location
            print_status "${INFO}" "Moving corrupted database to backup location..." "${BLUE}"
            sudo mkdir -p "$corrupted_dir"
            sudo mv "$main_db" "$corrupted_dir/"
            sudo mv "$main_db-shm" "$corrupted_dir/" 2>/dev/null || true
            sudo mv "$main_db-wal" "$corrupted_dir/" 2>/dev/null || true
            
            # Copy recovery database as new main database
            print_status "${INFO}" "Restoring database from recovery file..." "${BLUE}"
            if sudo cp "$recovery_db" "$main_db"; then
                sudo chown plex:plex "$main_db"
                sudo chmod 644 "$main_db"
                print_status "${CHECKMARK}" "Database restored successfully!" "${GREEN}"
                
                # Verify the restored database
                print_status "${INFO}" "Verifying restored database..." "${BLUE}"
                local integrity_result
                integrity_result=$(sudo -u plex sqlite3 "$main_db" "PRAGMA integrity_check;" 2>&1)
                
                if echo "$integrity_result" | grep -q "ok"; then
                    print_status "${CHECKMARK}" "Restored database integrity verified!" "${GREEN}"
                    return 0
                elif echo "$integrity_result" | grep -q "no such collation sequence: icu"; then
                    print_status "${CROSS}" "ICU collation sequence issue detected!" "${YELLOW}"
                    print_status "${INFO}" "Attempting ICU-aware recovery..." "${BLUE}"
                    
                    # Try ICU-aware recovery script
                    local icu_script="${SCRIPT_DIR}/icu-aware-recovery.sh"
                    if [[ -f "$icu_script" ]]; then
                        if "$icu_script" --auto; then
                            print_status "${CHECKMARK}" "ICU-aware recovery completed!" "${GREEN}"
                            return 0
                        else
                            print_status "${CROSS}" "ICU-aware recovery failed!" "${RED}"
                        fi
                    else
                        print_status "${INFO}" "ICU recovery script not found, trying manual fix..." "${YELLOW}"
                        
                        # Try to recreate database without ICU dependencies
                        local temp_db="/tmp/plex_temp_$(date +%Y%m%d_%H%M%S).db"
                        print_status "${INFO}" "Attempting to dump and recreate database..." "${BLUE}"
                        
                        if sudo -u plex sqlite3 "$recovery_db" ".dump" | grep -v "COLLATE icu_" | sudo -u plex sqlite3 "$temp_db"; then
                            print_status "${INFO}" "Database dump successful, replacing main database..." "${BLUE}"
                            sudo mv "$temp_db" "$main_db"
                            sudo chown plex:plex "$main_db"
                            sudo chmod 644 "$main_db"
                            
                            # Verify the recreated database
                            if sudo -u plex sqlite3 "$main_db" "PRAGMA integrity_check;" 2>/dev/null | grep -q "ok"; then
                                print_status "${CHECKMARK}" "Database recreated successfully without ICU!" "${GREEN}"
                                return 0
                            fi
                        fi
                        
                        # Clean up temp file if it exists
                        sudo rm -f "$temp_db" 2>/dev/null || true
                    fi
                    
                    print_status "${CROSS}" "Failed to resolve ICU collation issues!" "${RED}"
                    return 1
                else
                    print_status "${CROSS}" "Restored database failed integrity check!" "${RED}"
                    print_status "${INFO}" "Integrity check result:" "${YELLOW}"
                    echo -e "${DIM}${YELLOW}   $integrity_result${RESET}"
                    return 1
                fi
            else
                print_status "${CROSS}" "Failed to restore database!" "${RED}"
                return 1
            fi
        else
            print_status "${CROSS}" "No valid recovery databases found!" "${RED}"
            print_status "${INFO}" "Available options:" "${YELLOW}"
            echo -e "${DIM}${YELLOW}   1. Check manual backups in /mnt/share/media/backups/plex/${RESET}"
            echo -e "${DIM}${YELLOW}   2. Let Plex rebuild database (will lose all metadata)${RESET}"
            echo -e "${DIM}${YELLOW}   3. Run: sudo rm '$main_db' && sudo systemctl start plexmediaserver${RESET}"
            return 1
        fi
    fi
    
    # Create backup of current database
    print_status "${INFO}" "Creating backup of current database..." "${BLUE}"
    if ! sudo cp "$main_db" "$backup_db"; then
        print_status "${CROSS}" "Failed to create database backup!" "${RED}"
        return 1
    fi
    
    print_status "${CHECKMARK}" "Database backed up to: $backup_db" "${GREEN}"
    
    # Try to vacuum the database
    print_status "${INFO}" "Running VACUUM on database..." "${BLUE}"
    if sudo -u plex sqlite3 "$main_db" "VACUUM;" 2>/dev/null; then
        print_status "${CHECKMARK}" "Database VACUUM completed successfully" "${GREEN}"
        
        # Test integrity again
        if sudo -u plex sqlite3 "$main_db" "PRAGMA integrity_check;" 2>/dev/null | grep -q "ok"; then
            print_status "${CHECKMARK}" "Database integrity restored!" "${GREEN}"
            print_status "${INFO}" "You can now try starting Plex again" "${BLUE}"
            return 0
        else
            print_status "${CROSS}" "Database still corrupted after VACUUM" "${RED}"
        fi
    else
        print_status "${CROSS}" "VACUUM operation failed" "${RED}"
    fi
    
    # Try reindex as last resort
    print_status "${INFO}" "Attempting REINDEX operation..." "${BLUE}"
    if sudo -u plex sqlite3 "$main_db" "REINDEX;" 2>/dev/null; then
        print_status "${CHECKMARK}" "Database REINDEX completed" "${GREEN}"
        
        # Test integrity one more time
        if sudo -u plex sqlite3 "$main_db" "PRAGMA integrity_check;" 2>/dev/null | grep -q "ok"; then
            print_status "${CHECKMARK}" "Database integrity restored after REINDEX!" "${GREEN}"
            return 0
        fi
    fi
    
    print_status "${CROSS}" "All repair attempts failed" "${RED}"
    print_status "${INFO}" "Manual intervention required. Consider:" "${YELLOW}"
    echo -e "${DIM}${YELLOW}   1. Restore from external backup using restore-plex.sh${RESET}"
    echo -e "${DIM}${YELLOW}   2. Use nuclear recovery: ./nuclear-plex-recovery.sh${RESET}"
    echo -e "${DIM}${YELLOW}   3. Check corrupted database moved to: $corrupted_dir${RESET}"
    
    return 1
}

# 🔍 Function to check database integrity
check_database_integrity() {
    print_status "${INFO}" "Checking database integrity..." "${BLUE}"
    
    local db_dir="/var/lib/plexmediaserver/Library/Application Support/Plex Media Server/Plug-in Support/Databases"
    local main_db="$db_dir/com.plexapp.plugins.library.db"
    local repair_script="${SCRIPT_DIR}/plex-database-repair.sh"
    
    if [[ ! -f "$main_db" ]]; then
        print_status "${CROSS}" "Main database not found at: $main_db" "${RED}"
        return 1
    fi
    
    # Use shared repair script for integrity checking if available
    if [[ -f "$repair_script" ]]; then
        if "$repair_script" check "$main_db" >/dev/null 2>&1; then
            print_status "${CHECKMARK}" "Database integrity check passed" "${GREEN}"
            return 0
        else
            print_status "${CROSS}" "Database integrity check failed!" "${RED}"
            print_status "${INFO}" "Consider running database repair: plex repair" "${YELLOW}"
            return 1
        fi
    else
        # Fallback to basic sqlite3 check
        if ! sudo -u plex sqlite3 "$main_db" "PRAGMA integrity_check;" >/dev/null 2>&1; then
            print_status "${CROSS}" "Database integrity check failed!" "${RED}"
            print_status "${INFO}" "Consider running database repair or restore from backup" "${YELLOW}"
            return 1
        fi
        
        print_status "${CHECKMARK}" "Database integrity check passed" "${GREEN}"
        return 0
    fi
}

# �🚀 Enhanced start function
start_plex() {
    print_status "${ROCKET}" "Starting Plex Media Server..." "${GREEN}"

    if systemctl is-active --quiet "$PLEX_SERVICE"; then
        print_status "${INFO}" "Plex is already running!" "${YELLOW}"
        show_detailed_status
        return 0
    fi

    # Reset any failed state first
    sudo systemctl reset-failed "$PLEX_SERVICE" 2>/dev/null || true
    
    # Check database integrity before starting
    if ! check_database_integrity; then
        print_status "${CROSS}" "Database integrity issues detected. Service may fail to start." "${RED}"
        echo -e "${DIM}${YELLOW}   Try: sudo systemctl stop plexmediaserver && sudo -u plex sqlite3 /var/lib/plexmediaserver/Library/Application\ Support/Plex\ Media\ Server/Plug-in\ Support/Databases/com.plexapp.plugins.library.db 'VACUUM;'${RESET}"
        return 1
    fi

    print_status "${INFO}" "Attempting to start service..." "${BLUE}"
    
    if ! sudo systemctl start "$PLEX_SERVICE"; then
        print_status "${CROSS}" "Failed to start Plex Media Server!" "${RED}"
        print_status "${INFO}" "Checking service logs..." "${BLUE}"
        
        # Show recent error logs
        echo -e "\n${DIM}${RED}Recent error logs:${RESET}"
        sudo journalctl -u "$PLEX_SERVICE" --no-pager -n 5 --since "1 minute ago" | tail -5
        
        return 1
    fi

    # Wait and verify startup
    sleep 3
    local timeout=30
    local elapsed=0
    
    print_status "${HOURGLASS}" "Waiting for service to initialize..." "${CYAN}"
    
    while [[ $elapsed -lt $timeout ]]; do
        if systemctl is-active --quiet "$PLEX_SERVICE"; then
            print_status "${CHECKMARK}" "Plex Media Server started successfully!" "${GREEN}"
            print_footer
            return 0
        fi
        
        sleep 2
        elapsed=$((elapsed + 2))
        echo -ne "${DIM}${CYAN}   Waiting... ${elapsed}s/${timeout}s${RESET}\r"
    done
    
    echo ""
    print_status "${CROSS}" "Service startup timeout or failed!" "${RED}"
    
    # Show current status
    local status
    status=$(systemctl is-active "$PLEX_SERVICE" 2>/dev/null || echo "unknown")
    print_status "${INFO}" "Current status: $status" "${YELLOW}"
    
    if [[ "$status" == "failed" ]]; then
        echo -e "\n${DIM}${RED}Recent error logs:${RESET}"
        sudo journalctl -u "$PLEX_SERVICE" --no-pager -n 10 --since "2 minutes ago"
    fi
    
    return 1
}

# 🛑 Enhanced stop function
stop_plex() {
    print_status "${STOP_SIGN}" "Stopping Plex Media Server..." "${YELLOW}"

    if ! systemctl is-active --quiet "$PLEX_SERVICE"; then
        print_status "${INFO}" "Plex is already stopped!" "${YELLOW}"
        return 0
    fi

    sudo systemctl stop "$PLEX_SERVICE" &
    local pid=$!
    show_loading "Gracefully shutting down Plex" $pid
    wait $pid

    if ! systemctl is-active --quiet "$PLEX_SERVICE"; then
        print_status "${CHECKMARK}" "Plex Media Server stopped successfully!" "${GREEN}"
        print_footer
    else
        print_status "${CROSS}" "Failed to stop Plex Media Server!" "${RED}"
        return 1
    fi
}

# ♻️ Enhanced restart function
restart_plex() {
    print_status "${RECYCLE}" "Restarting Plex Media Server..." "${BLUE}"

    if systemctl is-active --quiet "$PLEX_SERVICE"; then
        stop_plex
        echo ""
    fi

    start_plex
}

# 📊 Enhanced status function with detailed info
show_detailed_status() {
    local service_status
    service_status=$(systemctl is-active "$PLEX_SERVICE" 2>/dev/null || echo "inactive")

    echo -e "\n${BOLD}${BLUE}+==============================================================+${RESET}"
    echo -e "${BOLD}${BLUE}                         SERVICE STATUS${RESET}"
    echo -e "${BOLD}${BLUE}+==============================================================+${RESET}"

    case "$service_status" in
        "active")
            print_status "${CHECKMARK}" "Service Status: ${GREEN}${BOLD}ACTIVE${RESET}" "${GREEN}"

            # Get additional info
            local uptime
            uptime=$(systemctl show "$PLEX_SERVICE" --property=ActiveEnterTimestamp --value | xargs -I {} date -d {} "+%Y-%m-%d %H:%M:%S" 2>/dev/null || echo "Unknown")

            local memory_usage
            memory_usage=$(systemctl show "$PLEX_SERVICE" --property=MemoryCurrent --value 2>/dev/null || echo "0")
            if [[ "$memory_usage" != "0" ]] && [[ "$memory_usage" =~ ^[0-9]+$ ]]; then
                memory_usage="$(( memory_usage / 1024 / 1024 )) MB"
            else
                memory_usage="Unknown"
            fi

            echo -e "${DIM}${CYAN}   Started: ${WHITE}${uptime}${RESET}"
            echo -e "${DIM}${CYAN}   Memory Usage: ${WHITE}${memory_usage}${RESET}"
            echo -e "${DIM}${CYAN}   Service Name: ${WHITE}${PLEX_SERVICE}${RESET}"
            ;;
        "inactive")
            print_status "${CROSS}" "Service Status: ${RED}${BOLD}INACTIVE${RESET}" "${RED}"
            echo -e "${DIM}${YELLOW}   Use '${SCRIPT_NAME} start' to start the service${RESET}"
            ;;
        "failed")
            print_status "${CROSS}" "Service Status: ${RED}${BOLD}FAILED${RESET}" "${RED}"
            echo -e "${DIM}${RED}   Check logs with: ${WHITE}journalctl -u ${PLEX_SERVICE}${RESET}"
            ;;
        *)
            print_status "${INFO}" "Service Status: ${YELLOW}${BOLD}${service_status^^}${RESET}" "${YELLOW}"
            ;;
    esac

    # Show recent logs
    echo -e "\n${DIM}${CYAN}+--- Recent Service Logs (24h) ---+${RESET}"
    
    # Try to get logs with sudo, fall back to user permissions
    local logs
    if logs=$(sudo journalctl -u "$PLEX_SERVICE" --no-pager -n 5 --since "24 hours ago" --output=short 2>/dev/null); then
        if [[ -n "$logs" && "$logs" != "-- No entries --" ]]; then
            echo -e "${DIM}${logs}${RESET}"
        else
            echo -e "${DIM}${YELLOW}No recent log entries found${RESET}"
        fi
    else
        # Fallback: try without sudo
        logs=$(journalctl -u "$PLEX_SERVICE" --no-pager -n 5 --since "24 hours ago" 2>/dev/null || echo "Unable to access logs")
        if [[ "$logs" == "Unable to access logs" || "$logs" == "-- No entries --" ]]; then
            echo -e "${DIM}${YELLOW}Unable to access recent logs (try: sudo journalctl -u ${PLEX_SERVICE})${RESET}"
        else
            echo -e "${DIM}${logs}${RESET}"
        fi
    fi
    
    echo -e "${DIM}${CYAN}+----------------------------------+${RESET}"
}

# 🔧 Show available commands
show_help() {
    echo -e "${BOLD}${WHITE}Usage:${RESET} ${CYAN}${SCRIPT_NAME}${RESET} ${YELLOW}<command>${RESET}"
    echo ""
    echo -e "${BOLD}${WHITE}Available Commands:${RESET}"
    echo -e "  ${GREEN}${BOLD}start${RESET}     ${ROCKET} Start Plex Media Server"
    echo -e "  ${YELLOW}${BOLD}stop${RESET}      ${STOP_SIGN} Stop Plex Media Server"
    echo -e "  ${BLUE}${BOLD}restart${RESET}   ${RECYCLE} Restart Plex Media Server"
    echo -e "  ${CYAN}${BOLD}status${RESET}    ${INFO} Show detailed service status"
    echo -e "  ${RED}${BOLD}repair${RESET}    [!] Repair database corruption issues"
    echo -e "  ${RED}${BOLD}nuclear${RESET}   [!!] Nuclear database recovery (last resort)"
    echo -e "  ${PURPLE}${BOLD}help${RESET}      [*] Show this help message"
    echo ""
    echo -e "${DIM}${WHITE}Examples:${RESET}"
    echo -e "  ${DIM}${SCRIPT_NAME} start     # Start the Plex service${RESET}"
    echo -e "  ${DIM}${SCRIPT_NAME} status    # Show current status${RESET}"
    echo -e "  ${DIM}${SCRIPT_NAME} repair    # Fix database issues${RESET}"
    echo -e "  ${DIM}${SCRIPT_NAME} nuclear   # Complete database replacement${RESET}"
    echo ""
}

# Nuclear database recovery function
nuclear_recovery() {
    print_status "${INFO}" "Starting nuclear database recovery..." "${RED}"
    
    local nuclear_script="${SCRIPT_DIR}/nuclear-plex-recovery.sh"
    
    if [[ ! -f "$nuclear_script" ]]; then
        print_status "${CROSS}" "Nuclear recovery script not found: $nuclear_script" "${RED}"
        print_status "${INFO}" "This script should be in the same directory as plex.sh" "${YELLOW}"
        return 2
    fi
    
    # Warning message
    echo -e "\n${RED}${BOLD}⚠️  WARNING: NUCLEAR RECOVERY ⚠️${RESET}"
    echo -e "${RED}This will completely replace your Plex database with a backup!${RESET}"
    echo -e "${RED}All changes since the backup was created will be lost!${RESET}"
    echo -e "${YELLOW}This should only be used when standard repair methods have failed.${RESET}\n"
    
    # Get user confirmation
    echo -e "${CYAN}Do you want to proceed with nuclear recovery? ${RESET}"
    echo -e "${DIM}Type 'YES' (uppercase) to confirm: ${RESET}"
    read -r confirmation
    
    if [[ "$confirmation" != "YES" ]]; then
        print_status "${INFO}" "Nuclear recovery cancelled by user" "${YELLOW}"
        return 0
    fi
    
    print_status "${INFO}" "Executing nuclear recovery script..." "${BLUE}"
    
    # Execute the nuclear recovery script
    if sudo "$nuclear_script" --auto; then
        print_status "${CHECKMARK}" "Nuclear recovery completed successfully!" "${GREEN}"
        print_status "${INFO}" "Your Plex server should now be operational" "${GREEN}"
        print_footer
        return 0
    else
        local exit_code=$?
        print_status "${CROSS}" "Nuclear recovery failed!" "${RED}"
        
        case $exit_code in
            2) print_status "${INFO}" "Backup file issues - check backup integrity" "${YELLOW}" ;;
            3) print_status "${INFO}" "Database replacement failure - check permissions" "${YELLOW}" ;;
            4) print_status "${INFO}" "Service management failure - check systemctl" "${YELLOW}" ;;
            5) print_status "${INFO}" "Rollback performed due to failure" "${YELLOW}" ;;
            *) print_status "${INFO}" "Unknown error occurred during recovery" "${YELLOW}" ;;
        esac
        
        print_footer
        return $exit_code
    fi
}

# Database repair function
repair_plex() {
    print_status "${INFO}" "Starting Plex database repair..." "${YELLOW}"
    
    # Run the enhanced repair function
    if repair_database; then
        print_status "${CHECKMARK}" "Database repair completed successfully!" "${GREEN}"
        
        # Try to start the service
        print_status "${INFO}" "Starting Plex service..." "${BLUE}"
        if sudo systemctl start "$PLEX_SERVICE"; then
            # Wait for service to fully start
            sleep 5
            if systemctl is-active --quiet "$PLEX_SERVICE"; then
                print_status "${CHECKMARK}" "Plex service started successfully!" "${GREEN}"
                print_footer
                return 0
            else
                print_status "${CROSS}" "Service started but may not be fully operational!" "${YELLOW}"
                show_detailed_status
                return 1
            fi
        else
            print_status "${CROSS}" "Failed to start Plex service after repair!" "${RED}"
            return 1
        fi
    else
        local repair_exit_code=$?
        print_status "${CROSS}" "Database repair failed!" "${RED}"
        
        # Try to start the service anyway in case partial repair helped
        print_status "${INFO}" "Attempting to start Plex service anyway..." "${BLUE}"
        sudo systemctl start "$PLEX_SERVICE" 2>/dev/null || true
        
        if [[ $repair_exit_code -eq 2 ]]; then
            print_status "${INFO}" "Critical error - manual intervention required" "${YELLOW}"
        else
            print_status "${INFO}" "Repair failed but service may still work with corrupted database" "${YELLOW}"
        fi
        
        print_footer
        return $repair_exit_code
    fi
}

# 🎯 Main script logic
main() {
    # Check if running as root
    if [[ $EUID -eq 0 ]]; then
        print_status "${CROSS}" "Don't run this script as root! Use your regular user account." "${RED}"
        exit 1
    fi

    # Check if no arguments provided
    if [[ $# -eq 0 ]]; then
        show_help
        exit 1
    fi

    case "${1,,}" in  # Convert to lowercase
        "start")
            start_plex
            ;;
        "stop")
            stop_plex
            ;;
        "restart"|"reload")
            restart_plex
            ;;
        "status"|"info")
            show_detailed_status
            ;;
        "repair"|"fix")
            repair_plex
            ;;
        "nuclear"|"nuke")
            nuclear_recovery
            ;;
        "help"|"--help"|"-h")
            show_help
            ;;
        *)
            print_status "${CROSS}" "Unknown command: ${RED}${BOLD}$1${RESET}" "${RED}"
            echo ""
            show_help
            exit 1
            ;;
    esac
}

# 🚀 Execute main function with all arguments
main "$@"