tracer-cloud--opensre/surfaces/cli/lifecycle/uninstall.py

from __future__ import annotations

import shutil
import subprocess
import sys
from pathlib import Path

from config.constants.paths import OPENSRE_HOME_DIR


def _is_windows() -> bool:
    return sys.platform == "win32"


def _is_binary_install() -> bool:
    return bool(getattr(sys, "frozen", False))


def _remove_path(p: Path) -> tuple[bool, str | None]:
    if not p.exists() and not p.is_symlink():
        return True, None
    try:
        if p.is_dir():
            shutil.rmtree(p)
        else:
            p.unlink()
        return True, None
    except OSError as exc:
        return False, str(exc)


def _pip_uninstall() -> int:
    result = subprocess.run(
        [sys.executable, "-m", "pip", "uninstall", "--yes", "opensre"],
        check=False,
        capture_output=True,
    )
    return result.returncode


def _data_dirs() -> list[Path]:
    return [
        OPENSRE_HOME_DIR,
        Path.home() / ".config" / "opensre",
    ]


def _is_onedir_binary(exe_path: Path) -> bool:
    return exe_path.parent.name == ".opensre-app" and (exe_path.parent / "_internal").is_dir()


def _launcher_for_binary(exe_path: Path) -> Path | None:
    launcher = shutil.which("opensre")
    if not launcher:
        return None
    launcher_path = Path(launcher)
    try:
        if launcher_path.resolve() == exe_path.resolve():
            return launcher_path
    except OSError:
        return None
    return None


def _binary_install_paths(exe_path: Path | None = None) -> list[Path]:
    exe = exe_path or Path(sys.executable)
    paths: list[Path] = []
    if launcher := _launcher_for_binary(exe):
        paths.append(launcher)
    if _is_onedir_binary(exe):
        paths.append(exe.parent)
    else:
        paths.append(exe)

    deduped: list[Path] = []
    seen: set[str] = set()
    for path in paths:
        key = str(path)
        if key in seen:
            continue
        seen.add(key)
        deduped.append(path)
    return deduped


def run_uninstall(*, yes: bool = False) -> int:
    dirs = _data_dirs()
    binary = _is_binary_install()
    binary_paths = _binary_install_paths() if binary else []

    print()
    print("  The following will be permanently deleted:")
    print()
    for d in dirs:
        tag = "found" if d.exists() else "not found"
        print(f"    {d}  ({tag})")
    if binary:
        for path in binary_paths:
            print(f"    {path}  (binary)")
    else:
        print("    pip package: opensre")
    print()

    if not yes:
        try:
            import questionary

            confirmed = questionary.confirm(
                "  Uninstall opensre from this machine?", default=False
            ).ask()
        except (EOFError, KeyboardInterrupt):
            print("\n  Aborted.")
            return 1
        if not confirmed:
            print("  Cancelled.")
            return 0

    print()

    any_error = False

    for d in dirs:
        if not d.exists():
            print(f"  skipped  {d}  (not found)")
            continue
        ok, err = _remove_path(d)
        if ok:
            print(f"  deleted  {d}")
        else:
            print(f"  error    {d}: {err}", file=sys.stderr)
            any_error = True

    if binary:
        for path in binary_paths:
            ok, err = _remove_path(path)
            if ok:
                print(f"  deleted  {path}")
            else:
                print(f"  error    {path}: {err}", file=sys.stderr)
                any_error = True
    else:
        print("  running  pip uninstall opensre")
        rc = _pip_uninstall()
        if rc == 0:
            print("  deleted  pip package opensre")
        else:
            print(f"  error    pip uninstall failed (exit {rc})", file=sys.stderr)
            if _is_windows():
                hint = "pip uninstall opensre"
            else:
                hint = "pip uninstall opensre  (or: pipx uninstall opensre)"
            print(f"           retry manually: {hint}", file=sys.stderr)
            any_error = True

    print()

    if any_error:
        print("  Uninstall finished with errors. See above for details.", file=sys.stderr)
        return 1

    print("  opensre has been uninstalled.")
    print()
    print("  Your config and data have been removed.")
    print("  To reinstall: curl -fsSL https://install.opensre.com | bash")
    return 0