main.py

"""
main.py
--------
Entry point for Horton.

On first run (no .env file), launches an interactive setup screen that
lets the user choose their LLM provider and paste their API key.
The key is written to .env and never logged.

On subsequent runs, loads the existing .env and launches the game directly.

Usage:
    python main.py                     # launch game
    python main.py --setup             # re-run provider setup
    python main.py --scenario PATH     # use a custom scenario file
"""

from __future__ import annotations

import argparse
import os
import sys
from pathlib import Path


def _load_env() -> None:
    """Load .env if it exists."""
    env_path = Path(__file__).parent / ".env"
    if env_path.exists():
        from dotenv import load_dotenv
        load_dotenv(env_path)


def _run_setup() -> None:
    """
    Interactive provider setup — runs before the main TUI.
    Uses plain terminal I/O (not Textual) so it works before dependencies
    are fully initialized.
    """
    import getpass

    CYAN = "\033[96m"
    MAGENTA = "\033[95m"
    GREEN = "\033[92m"
    DIM = "\033[2m"
    RESET = "\033[0m"
    BOLD = "\033[1m"

    print()
    print(f"{MAGENTA}{BOLD}╔══════════════════════════════════════════════╗{RESET}")
    print(f"{MAGENTA}{BOLD}║        HORTON — FIRST RUN SETUP              ║{RESET}")
    print(f"{MAGENTA}{BOLD}╚══════════════════════════════════════════════╝{RESET}")
    print()
    print(f"{CYAN}Choose your LLM provider:{RESET}")
    print(f"  {BOLD}[1]{RESET} Anthropic Claude  (claude-sonnet-4-6 + haiku-4-5)")
    print(f"  {BOLD}[2]{RESET} Google Gemini     (gemini-1.5-pro + flash)  {GREEN}← free tier available{RESET}")
    print()

    while True:
        choice = input(f"{MAGENTA}>{RESET} ").strip()
        if choice in ("1", "2"):
            break
        print("  Enter 1 or 2")

    if choice == "1":
        provider = "anthropic"
        key_name = "ANTHROPIC_API_KEY"
        key_hint = "sk-ant-..."
        docs_hint = "https://console.anthropic.com"
    else:
        provider = "gemini"
        key_name = "GOOGLE_API_KEY"
        key_hint = "AIza..."
        docs_hint = "https://aistudio.google.com — free tier, no credit card"

    print()
    print(f"{CYAN}Paste your {key_name}:{RESET}")
    print(f"{DIM}(input hidden)  Get one at: {docs_hint}{RESET}")
    print()

    api_key = getpass.getpass(f"{MAGENTA}>{RESET} ").strip()
    if not api_key:
        print(f"\n{MAGENTA}No key entered. Exiting.{RESET}")
        sys.exit(1)

    # Write .env
    env_path = Path(__file__).parent / ".env"
    with open(env_path, "w") as f:
        f.write(f"LLM_PROVIDER={provider}\n")
        f.write(f"{key_name}={api_key}\n")

    print()
    print(f"{GREEN}✓ Config saved to .env{RESET}")
    print(f"{DIM}  To change provider later, run: python main.py --setup{RESET}")
    print()

    # Set in current process env so we don't need to reload
    os.environ["LLM_PROVIDER"] = provider
    os.environ[key_name] = api_key


def _validate_env() -> bool:
    """Check that required env vars are set for the chosen provider."""
    provider = os.environ.get("LLM_PROVIDER", "").lower()
    if provider == "anthropic":
        return bool(os.environ.get("ANTHROPIC_API_KEY"))
    elif provider == "gemini":
        return bool(os.environ.get("GOOGLE_API_KEY"))
    return False


def main() -> None:
    parser = argparse.ArgumentParser(description="Horton — Social Engineering CTF")
    parser.add_argument("--setup", action="store_true", help="Re-run provider setup")
    parser.add_argument(
        "--scenario",
        default=str(Path(__file__).parent / "scenarios" / "github_pat_heist.json"),
        help="Path to scenario JSON file",
    )
    args = parser.parse_args()

    # Load existing env
    _load_env()

    # Run setup if requested or if no valid config found
    env_path = Path(__file__).parent / ".env"
    if args.setup or not env_path.exists() or not _validate_env():
        _run_setup()

    # Validate once more after setup
    if not _validate_env():
        print("\n[ERROR] No valid API key configured. Run: python main.py --setup")
        sys.exit(1)

    # Launch game
    from engine.game import GameEngine
    from ui.app import HortonApp

    try:
        engine = GameEngine.from_scenario_file(args.scenario)
        app = HortonApp(engine=engine)
        app.run()
    except KeyboardInterrupt:
        pass
    except Exception as e:
        print(f"\n[ERROR] {e}")
        sys.exit(1)


if __name__ == "__main__":
    main()