disco-reaper/server-shuttle.py

import sys
import asyncio
import logging
from pathlib import Path
from src.ui.shuttle_app import run_cli
from src.core.configuration import load_config

def setup_logging():
    try:
        config = load_config()
        log_level_str = config.migration.log_level.upper()
        level = getattr(logging, log_level_str, logging.INFO)
    except Exception:
        level = logging.INFO
        
    handlers = [logging.FileHandler('.shuttle.log', mode='a')]
    if level == logging.DEBUG:
        handlers.append(logging.StreamHandler(sys.stdout))

    logging.basicConfig(
        format='%(asctime)s,%(msecs)d %(name)s %(levelname)s %(message)s',
        datefmt='%H:%M:%S',
        level=level,
        handlers=handlers
    )

def select_platform(config):
    from rich.console import Console
    from rich.prompt import Prompt
    from rich.panel import Panel
    
    console = Console()
    
    fillers = [    "DISCORD_BOT_TOKEN", "FLUXER_BOT_TOKEN", "STOAT_BOT_TOKEN",
        "000000000000000000",
        "DISCORD_SERVER_ID", "FLUXER_COMMUNITY_ID", "STOAT_SERVER_ID",
        "", None
    ]
    
    fluxer_set = config.fluxer_bot_token not in fillers and config.fluxer_community_id not in fillers
    stoat_set = config.stoat_bot_token not in fillers and config.stoat_server_id not in fillers
    
    if fluxer_set and not stoat_set:
        return "fluxer"
    elif stoat_set and not fluxer_set:
        return "stoat"
    elif fluxer_set and stoat_set:
        console.print(Panel.fit(
            "[bold]Both Fluxer and Stoat configurations found.[/bold]\n"
            "Which one do you want to use?",
            title="[bold cyan]Platform Selection[/bold cyan]"
        ))
        console.print("(1) [bold blue]Fluxer[/bold blue]")
        console.print("(2) [bold red]Stoat[/bold red]")
        console.print("(Q) [bold dim]Quit[/bold dim]")
        console.print("")
        
        choice = Prompt.ask("Select an option [[bold cyan]1/2/Q[/bold cyan]]", choices=["1", "2", "Q", "q"], show_choices=False).upper()
        if choice == "1":
            return "fluxer"
        elif choice == "2":
            return "stoat"
        else:
            sys.exit(0)
    else:
        # Both are fillers
        console.print(Panel.fit(
            "[bold]First setup, Tool configuration[/bold]\n"
            "Which platform do you want to migrate to?",
            title="[bold cyan]Initial Setup[/bold cyan]"
        ))
        console.print("(1) [bold blue]Fluxer[/bold blue]")
        console.print("(2) [bold red]Stoat[/bold red]")
        console.print("(Q) [bold dim]Quit[/bold dim]")
        console.print("")
        
        choice = Prompt.ask("Select an option [[bold cyan]1/2/Q[/bold cyan]]", choices=["1", "2", "Q", "q"], show_choices=False).upper()
        if choice == "1":
            return "fluxer"
        elif choice == "2":
            return "stoat"
        else:
            sys.exit(0)

def relaunch_in_terminal():
    """Detects if running without a terminal on Linux and relaunches in one."""
    import os
    import sys
    import subprocess
    import shutil

    # Only attempt on Linux
    if sys.platform != "linux":
        return

    # Check if we have a TTY on stdin or stdout, or if already relaunched
    is_tty = sys.stdin.isatty() or sys.stdout.isatty()
    if is_tty or os.environ.get("SERVER_SHUTTLE_RELAUNCHED"):
        return

    # Diagnostic logging to help debug why it fails on some distros
    debug_log = "/tmp/shuttle_terminal_debug.log"
    with open(debug_log, "a") as f:
        f.write(f"Relaunching... isatty={is_tty}, env={os.environ.get('SERVER_SHUTTLE_RELAUNCHED')}\n")

    # List of terminals to try with their specific execution flags
    terminals = [
        ("gnome-terminal", ["--"]),       # Modern GNOME Terminal
        ("ptyxis", ["--"]),               # Fedora/Modern GNOME
        ("x-terminal-emulator", ["-e"]),  # Ubuntu/Debian standard
        ("kgx", ["-e"]),                  # GNOME Console
        ("konsole", ["-e"]),
        ("xfce4-terminal", ["-e"]),
        ("lxterminal", ["-e"]),
        ("mate-terminal", ["-e"]),
        ("alacritty", ["-e"]),
        ("kitty", []),
        ("xterm", ["-e"]),
    ]

    # Resolve the absolute path to ourselves.
    # frozen=True means we are running from a PyInstaller bundle.
    if getattr(sys, 'frozen', False):
        executable = os.path.abspath(sys.argv[0])
    else:
        executable = sys.executable
        
    args = [executable] + sys.argv[1:]
    
    # Set env var to prevent loops
    env = os.environ.copy()
    env["SERVER_SHUTTLE_RELAUNCHED"] = "1"

    for term, cmd_args in terminals:
        if shutil.which(term):
            with open(debug_log, "a") as f:
                f.write(f"Found terminal: {term}\n")
            try:
                # Construct command: term [args] executable [sys.argv]
                subprocess.Popen([term] + cmd_args + args, env=env)
                sys.exit(0)
            except Exception as e:
                with open(debug_log, "a") as f:
                    f.write(f"Failed to launch {term}: {e}\n")
                continue

def main():
    relaunch_in_terminal()
    config = load_config()
    setup_logging()
    platform = select_platform(config)
    try:
        asyncio.run(run_cli(target_platform=platform))
    except KeyboardInterrupt:
        print("\nOperation terminated by user.")
        sys.exit(0)
    except Exception as e:
        print(f"Failed to start tool: {e}")
        sys.exit(1)

if __name__ == "__main__":
    main()