#!/usr/bin/env python3
"""Installer / Uninstaller for Heimdall Token Saver extension.

Cross-platform: macOS, Linux, Windows.

Usage:
    python3 install.py --target claude        # Install for Claude Code
    python3 install.py --target antigravity   # Install for Antigravity CLI
    python3 install.py --target both          # Install for both
    python3 install.py --link                 # Use symlinks (development mode)
    python3 install.py --uninstall            # Remove from both platforms
    python3 install.py --uninstall --target claude  # Remove from Claude Code only
"""

import argparse
import platform
import sys


def _force_utf8_stdio():
    """Force UTF-8 output so emoji never crash on Windows (cp1252)."""
    for stream in (sys.stdout, sys.stderr):
        reconfigure = getattr(stream, "reconfigure", None)
        if reconfigure is not None:
            try:
                reconfigure(encoding="utf-8", errors="replace")
            except (AttributeError, OSError, ValueError):
                pass


_force_utf8_stdio()

from installers import antigravity, claude, universal
from installers.common import (
    install_cli,
    install_core,
    migrate_from_legacy,
    uninstall_cli,
    uninstall_core,
    uninstall_data_dir,
)


def main():
    parser = argparse.ArgumentParser(
        description="Install or uninstall Heimdall Token Saver extension",
        formatter_class=argparse.RawDescriptionHelpFormatter,
        epilog="""
Examples:
  python3 install.py --target claude          Install for Claude Code
  python3 install.py --target antigravity     Install for Antigravity CLI
  python3 install.py --target both            Install for both
  python3 install.py --link --target claude   Dev mode (symlinks)
  python3 install.py --uninstall              Uninstall from both (default)
  python3 install.py --uninstall --target claude  Uninstall from Claude only
  python3 install.py --uninstall --keep-data  Uninstall but keep stats DB
""",
    )
    parser.add_argument(
        "--target",
        choices=["claude", "antigravity", "both", "all"],
        default="all",
        help="Target platform (default: all)",
    )
    parser.add_argument(
        "--link",
        action="store_true",
        help="Use symlinks instead of copies (development mode)",
    )
    parser.add_argument(
        "--uninstall",
        action="store_true",
        help="Remove the extension completely",
    )
    parser.add_argument(
        "--keep-data",
        action="store_true",
        help="When uninstalling, keep the ~/.token-saver data directory (stats, config)",
    )
    args = parser.parse_args()

    if args.uninstall:
        # Default to 'all' for uninstall so nothing is left behind
        target = args.target or "all"
        print(f"Uninstalling token-saver from: {target}")

        print("\n--- Legacy cleanup ---")
        migrate_from_legacy()

        print("\n--- CLI ---")
        uninstall_cli()

        if target in ("claude", "both", "all"):
            claude.uninstall()
        if target in ("antigravity", "both", "all"):
            antigravity.uninstall()
        if target in ("both", "all"):
            universal.uninstall()

        print("\n--- Core ---")
        uninstall_core()

        if not args.keep_data:
            print("\n--- Data ---")
            uninstall_data_dir()

        print("\nUninstallation complete.")
        return

    # --- Install ---
    target = args.target or "all"
    print(f"Installing token-saver for: {target}")
    print(f"Platform: {platform.system()}")
    print(f"Mode: {'symlink' if args.link else 'copy'}")

    # Clean up any leftover "token-saving" installation before proceeding
    print("\n--- Legacy cleanup ---")
    migrate_from_legacy()

    if target in ("claude", "both", "all"):
        claude.install(use_symlink=args.link)
    if target in ("antigravity", "both", "all"):
        antigravity.install(use_symlink=args.link)
    if target in ("both", "all"):
        universal.install(use_symlink=args.link)

    install_core(use_symlink=args.link)

    print("\n--- CLI ---")
    install_cli(use_symlink=args.link)

    print("\nInstallation complete.")


if __name__ == "__main__":
    main()