CineSync.py

import os
import sys
import subprocess
import random
import getpass
import logging
import subprocess
import sys
import pkg_resources
import platform

# Script Metadata
SCRIPT_VERSION = "2.1"
SCRIPT_DATE = "2024-12-14"

# Define variables
SCRIPTS_FOLDER = "MediaHub"
BROKEN_LINK_FOLDER = "BrokenLinkVault"
MONITOR_SCRIPT = os.path.join(SCRIPTS_FOLDER, "utils/service_manager.py")
ENV_FILE = ".env"
LIBRARY_SCRIPT = os.path.join(SCRIPTS_FOLDER, "main.py")
BROKEN_LINKS_SCRIPT = os.path.join(SCRIPTS_FOLDER, "utils/broken_links.py")
RENAMER_SCRIPT = os.path.join(SCRIPTS_FOLDER, "utils/tmdb_renamer.py")

# Setup logging
logging.basicConfig(filename='script.log', level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# Determine the Python command based on the OS
python_command = 'python' if platform.system() == 'Windows' else 'python3'

# Function to print text with color
def print_color(text, color):
    colors = {
        "red": "\033[91m",
        "green": "\033[92m",
        "yellow": "\033[93m",
        "blue": "\033[94m",
        "end": "\033[0m",
    }
    print(f"{colors.get(color, '')}{text}{colors.get('end', '')}")

# Function to check and install required Python packages
def check_python_and_dependencies(required_version="3.6"):
    try:
        version_output = subprocess.run([sys.executable, '--version'], check=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
        version_str = version_output.stdout.decode().strip().split()[1]
        major, minor, micro = map(int, version_str.split('.'))
        current_version = f"{major}.{minor}"
        print(f"Current Python version: {current_version}")

        if (major, minor) < tuple(map(int, required_version.split('.'))):
            print_color(f"Python {required_version} or higher is required. Current version is {current_version}.", "red")
            input("Press Enter to exit the script...")
            sys.exit(1)
    except FileNotFoundError:
        print_color("Python is not installed. Please install Python 3.6 or higher.", "red")
        input("Press Enter to exit the script...")
        sys.exit(1)
    except subprocess.CalledProcessError as e:
        print_color(f"Error checking Python version: {e}", "red")
        input("Press Enter to exit the script...")
        sys.exit(1)

    # Check and install dependencies
    required_packages = []
    if not os.path.exists('requirements.txt'):
        print_color("Error: requirements.txt file not found.", "red")
        input("Press Enter to exit the script...")
        sys.exit(1)

    with open('requirements.txt', 'r') as file:
        for line in file:
            package = line.strip()
            if package:
                required_packages.append(package)

    installed_packages = [pkg.key for pkg in pkg_resources.working_set]

    missing_packages = [pkg for pkg in required_packages if pkg.lower() not in installed_packages]

    if missing_packages:
        print_color("Missing packages detected. Installing...", "yellow")
        try:
            subprocess.check_call([sys.executable, '-m', 'pip', 'install', *missing_packages])
            print_color("All missing packages installed successfully.", "green")
        except subprocess.CalledProcessError as e:
            print_color(f"Error installing packages: {e}", "red")
            input("Press Enter to exit the script...")
            sys.exit(1)
    else:
        print_color("All required packages are already installed.", "green")

check_python_and_dependencies()

# Function to clear the screen
def clear_screen():
    os.system('cls' if os.name == 'nt' else 'clear')

# Function to print the banner
def print_banner():
    print("""
    a88888b. oo                   .d88888b
   d8'   `88                      88.    "'
   88        dP 88d888b. .d8888b. `Y88888b. dP    dP 88d888b. .d8888b.
   88        88 88'  `88 88ooood8       `8b 88    88 88'  `88 88'  `"`
   Y8.   .88 88 88    88 88.  ... d8'   .8P 88.  .88 88    88 88.  ...
    Y88888P' dP dP    dP `88888P'  Y88888P  `8888P88 dP    dP `88888P'
                                                 .88
                                             d8888P
    """)
    print(f"\nVersion {SCRIPT_VERSION} - Last updated on {SCRIPT_DATE}\n")

# Function to print a random welcome message
def print_random_welcome(username):
    welcome_messages = [
        f"Welcome back, {username}! Ready to manage your library?",
        f"Hey there, {username}! Your library adventure begins now!",
        f"Hello, {username}! Let's dive into your library management system!",
        f"Greetings, {username}! Get ready to organize your library!",
    ]
    random_messages = [
        "Did you know? Organizing your library can reduce stress!",
        "Tip of the day: Keep your library updated to avoid confusion.",
        "Fact: Well-organized libraries can improve productivity!",
        "Remember: Regular scans help keep your library tidy.",
    ]
    print(f"\n{random.choice(welcome_messages)}")
    print(f"Random Tip: {random.choice(random_messages)}\n")

# Function to greet the user
def greet_user():
    username = getpass.getuser()
    print_random_welcome(username)

# Function to edit the .env file
def edit_env_file():
    try:
        if os.path.exists(ENV_FILE):
            subprocess.run([python_command, '-m', 'nano', ENV_FILE], check=True)
            print("\n.env file editing completed.")
        else:
            print("The .env file does not exist. Creating a new one.")
            with open(ENV_FILE, 'w') as f:
                pass
            subprocess.run([python_command, '-m', 'nano', ENV_FILE], check=True)
            print("\n.env file created and edited.")
    except subprocess.CalledProcessError as e:
        logging.error(f"Error editing .env file: {e}")
        print_color("Error editing .env file. Check the log for details.", "red")
    input("Press Enter to return to the main menu...")

# Function for Real-Time Monitoring
def real_time_monitoring():
    if os.uname().sysname != "Linux" and os.geteuid() != 0:
        print_color("Error: This function requires root privileges. Please run the script with sudo.", "red")
        input("Press Enter to return to the main menu...")
        return

    while True:
        clear_screen()
        print_banner()
        print("\nReal-Time Monitoring Options:")
        print("1) Enable Real-Time Monitoring Service")
        print("2) Disable Real-Time Monitoring Service")
        print("3) Exit to Main Menu")
        choice = input("Select an option: ")

        try:
            if choice == '1':
                subprocess.run([python_command, MONITOR_SCRIPT, 'enable'], check=True)
                input("Press Enter to continue...")
            elif choice == '2':
                subprocess.run([python_command, MONITOR_SCRIPT, 'disable'], check=True)
                input("Press Enter to continue...")
            elif choice == '3':
                break
            else:
                print_color("Invalid option. Please select again.", "red")
                input("Press Enter to continue...")
        except subprocess.CalledProcessError as e:
            logging.error(f"Error executing real-time monitoring command: {e}")
            print_color("Error executing command. Check the log for details.", "red")

    if os.uname().sysname != "Linux":
        print_color("Warning: Real-Time Monitoring is only available on Linux OS.", "yellow")
        input("Press Enter to return to the main menu...")

# Function to execute full library scan
def execute_full_library_scan():
    while True:
        clear_screen()
        print_banner()
        print("\nFull Library Scan Options:")
        print("1) Auto Scan")
        print("2) Manual Scan (Use only when TMDB ID is enabled)")
        print("3) Back to Main Menu")
        choice = input("Select an option: ")

        try:
            if choice == '1':
                if os.path.exists(LIBRARY_SCRIPT):
                        subprocess.run([python_command, LIBRARY_SCRIPT, '--auto-select'], check=True)
                        input("Auto scan completed. Press Enter to return to the main menu...")
                else:
                    print_color("Error: The script does not exist.", "red")
                    input("Press Enter to return to the main menu...")
            elif choice == '2':
                if os.path.exists(LIBRARY_SCRIPT):
                        subprocess.run([python_command, LIBRARY_SCRIPT], check=True)
                        input("Manual scan completed. Press Enter to return to the main menu...")
                else:
                    print_color("Error: The library.py script does not exist.", "red")
                    input("Press Enter to return to the main menu...")
            elif choice == '3':
                break
            else:
                print_color("Invalid option. Please select again.", "red")
                input("Press Enter to continue...")
        except subprocess.CalledProcessError as e:
            logging.error(f"Error executing library scan: {e}")
            print_color("Error executing scan. Check the log for details.", "red")

# Function to configure broken symlinks
def configure_broken_symlinks():
    config_file = os.path.join(BROKEN_LINK_FOLDER, "broken_links_config.txt")

    # Create the BrokenLinkVault folder if it doesn't exist
    os.makedirs(BROKEN_LINK_FOLDER, exist_ok=True)

    # Create the config file if it doesn't exist
    if not os.path.exists(config_file):
        open(config_file, 'w').close()

    while True:
        clear_screen()
        print_banner()
        print("\nRemove Broken Symlinks:")
        print("1) Run Scan")
        print("2) Add Directory")
        print("3) Remove Directory")
        print("4) Show Current Directories")
        print("5) Back to Main Menu")
        choice = input("Select an option: ")

        if choice == '1':
            execute_vault_scan()
        elif choice == '2':
            directory = input("Enter directory path to add: ").replace('\\', '/')
            if directory and os.path.isdir(directory):
                with open(config_file, 'r') as file:
                    if directory in file.read():
                        print_color("Directory already exists.", "yellow")
                    else:
                        with open(config_file, 'a') as file:
                            file.write(directory + '\n')
                        print_color("Directory added successfully.", "green")
                input("Press Enter to continue...")
            else:
                print_color("Invalid directory path. Please enter a valid directory.", "red")
                input("Press Enter to continue...")
        elif choice == '3':
            if os.path.getsize(config_file) == 0:
                print_color("No directories available.", "yellow")
                input("Press Enter to continue...")
                continue

            print("\nCurrent Directories:")
            with open(config_file, 'r') as file:
                for i, line in enumerate(file, 1):
                    print(f"{i}. {line.strip()}")

            try:
                index = int(input("Enter the number of the directory to remove: ")) - 1
                with open(config_file, 'r') as file:
                    lines = file.readlines()
                if 0 <= index < len(lines):
                    with open(config_file, 'w') as file:
                        for i, line in enumerate(lines):
                            if i != index:
                                file.write(line)
                    print_color("Directory removed successfully.", "green")
                else:
                    print_color("Invalid index. Please try again.", "red")
            except ValueError:
                print_color("Invalid input. Please enter a number.", "red")
            input("Press Enter to continue...")
        elif choice == '4':
            print("\nCurrent Directories:")
            with open(config_file, 'r') as file:
                if os.path.getsize(config_file) == 0:
                    print("No directories found.")
                else:
                    for i, line in enumerate(file, 1):
                        print(f"{i}. {line.strip()}")
            input("Press Enter to continue...")
        elif choice == '5':
            break
        else:
            print_color("Invalid option. Please select again.", "red")
            input("Press Enter to continue...")

# Function to execute vault scan
def execute_vault_scan():
    if os.path.exists(BROKEN_LINKS_SCRIPT):
        logging.info(f"Executing script at: {BROKEN_LINKS_SCRIPT}")
        try:
            result = subprocess.run([python_command, BROKEN_LINKS_SCRIPT], check=True, text=True, capture_output=True)
            print("Broken symlinks scan completed.")
            print("Output:\n", result.stdout)
            logging.info(f"Script output:\n{result.stdout}")
        except subprocess.CalledProcessError as e:
            logging.error(f"Error executing broken links script: {e.stderr}")
            print(f"Error executing scan. Check the log for details: {e.stderr}")
            logging.error(f"Command '{e.cmd}' returned non-zero exit status {e.returncode}.")
    else:
        print("Error: The script does not exist.")
    input("Press Enter to return to the main menu...")

# Main function
def main():
    while True:
        clear_screen()
        print_banner()
        greet_user()
        print("\nMain Menu:")
        print("1) Edit .env file")
        print("2) Full Library Scan")
        print("3) Configure Broken Symlinks")
        print("4) Exit")

        choice = input("Select an option: ")

        if choice == '1':
            edit_env_file()
        elif choice == '2':
            execute_full_library_scan()
        elif choice == '3':
            configure_broken_symlinks()
        elif choice == '4':
            print("Exiting the script. Have a great day!")
            break
        else:
            print_color("Invalid option. Please select again.", "red")
            input("Press Enter to continue...")

if __name__ == "__main__":
    main()