ChristianLempa_Boilerplates/cli/core/repo.py

"""Repository management module for syncing library repositories."""

from __future__ import annotations

import logging
import subprocess
from pathlib import Path
from typing import Optional

from rich.console import Console
from rich.progress import Progress, SpinnerColumn, TextColumn
from rich.table import Table
from typer import Argument, Option, Typer

from ..core.config import ConfigManager
from ..core.display import DisplayManager
from ..core.exceptions import ConfigError

logger = logging.getLogger(__name__)
console = Console()
console_err = Console(stderr=True)
display = DisplayManager()

app = Typer(help="Manage library repositories")


def _run_git_command(
    args: list[str], cwd: Optional[Path] = None
) -> tuple[bool, str, str]:
    """Run a git command and return the result.

    Args:
        args: Git command arguments (without 'git' prefix)
        cwd: Working directory for the command

    Returns:
        Tuple of (success, stdout, stderr)
    """
    try:
        result = subprocess.run(
            ["git"] + args,
            cwd=cwd,
            capture_output=True,
            text=True,
            timeout=300,  # 5 minute timeout
        )
        return result.returncode == 0, result.stdout, result.stderr
    except subprocess.TimeoutExpired:
        return False, "", "Command timed out after 5 minutes"
    except FileNotFoundError:
        return False, "", "Git command not found. Please install git."
    except Exception as e:
        return False, "", str(e)


def _clone_or_pull_repo(
    name: str,
    url: str,
    target_path: Path,
    branch: Optional[str] = None,
    sparse_dir: Optional[str] = None,
) -> tuple[bool, str]:
    """Clone or pull a git repository with optional sparse-checkout.

    Args:
        name: Library name
        url: Git repository URL
        target_path: Target directory for the repository
        branch: Git branch to clone/pull (optional)
        sparse_dir: Directory to sparse-checkout (optional, use None or "." for full clone)

    Returns:
        Tuple of (success, message)
    """
    if target_path.exists() and (target_path / ".git").exists():
        # Repository exists, pull updates
        logger.debug(f"Pulling updates for library '{name}' at {target_path}")

        # Determine which branch to pull
        pull_branch = branch if branch else "main"

        # Pull updates from specific branch
        success, stdout, stderr = _run_git_command(
            ["pull", "--ff-only", "origin", pull_branch], cwd=target_path
        )

        if success:
            # Check if anything was updated
            if "Already up to date" in stdout or "Already up-to-date" in stdout:
                return True, "Already up to date"
            else:
                return True, "Updated successfully"
        else:
            error_msg = stderr or stdout
            logger.error(f"Failed to pull library '{name}': {error_msg}")
            return False, f"Pull failed: {error_msg}"
    else:
        # Repository doesn't exist, clone it
        logger.debug(f"Cloning library '{name}' from {url} to {target_path}")

        # Ensure parent directory exists
        target_path.parent.mkdir(parents=True, exist_ok=True)

        # Determine if we should use sparse-checkout
        use_sparse = sparse_dir and sparse_dir != "."

        if use_sparse:
            # Use sparse-checkout to clone only specific directory
            logger.debug(f"Using sparse-checkout for directory: {sparse_dir}")

            # Initialize empty repo
            success, stdout, stderr = _run_git_command(["init"], cwd=None)
            if success:
                # Create target directory
                target_path.mkdir(parents=True, exist_ok=True)

                # Initialize git repo
                success, stdout, stderr = _run_git_command(["init"], cwd=target_path)
                if not success:
                    return False, f"Failed to initialize repo: {stderr or stdout}"

                # Add remote
                success, stdout, stderr = _run_git_command(
                    ["remote", "add", "origin", url], cwd=target_path
                )
                if not success:
                    return False, f"Failed to add remote: {stderr or stdout}"

                # Enable sparse-checkout (non-cone mode to exclude root files)
                success, stdout, stderr = _run_git_command(
                    ["sparse-checkout", "init", "--no-cone"], cwd=target_path
                )
                if not success:
                    return (
                        False,
                        f"Failed to enable sparse-checkout: {stderr or stdout}",
                    )

                # Set sparse-checkout to specific directory (non-cone uses patterns)
                success, stdout, stderr = _run_git_command(
                    ["sparse-checkout", "set", f"{sparse_dir}/*"], cwd=target_path
                )
                if not success:
                    return (
                        False,
                        f"Failed to set sparse-checkout directory: {stderr or stdout}",
                    )

                # Fetch specific branch (without attempting to update local ref)
                fetch_args = ["fetch", "--depth", "1", "origin"]
                if branch:
                    fetch_args.append(branch)
                else:
                    fetch_args.append("main")

                success, stdout, stderr = _run_git_command(fetch_args, cwd=target_path)
                if not success:
                    return False, f"Fetch failed: {stderr or stdout}"

                # Checkout the branch
                checkout_branch = branch if branch else "main"
                success, stdout, stderr = _run_git_command(
                    ["checkout", checkout_branch], cwd=target_path
                )
                if not success:
                    return False, f"Checkout failed: {stderr or stdout}"

                # Done! Files are in target_path/sparse_dir/
                return True, "Cloned successfully (sparse)"
            else:
                return False, f"Failed to initialize: {stderr or stdout}"
        else:
            # Regular full clone
            clone_args = ["clone", "--depth", "1"]
            if branch:
                clone_args.extend(["--branch", branch])
            clone_args.extend([url, str(target_path)])

            success, stdout, stderr = _run_git_command(clone_args)

            if success:
                return True, "Cloned successfully"
            else:
                error_msg = stderr or stdout
                logger.error(f"Failed to clone library '{name}': {error_msg}")
                return False, f"Clone failed: {error_msg}"


@app.command()
def update(
    library_name: Optional[str] = Argument(
        None, help="Name of specific library to update (updates all if not specified)"
    ),
    verbose: bool = Option(False, "--verbose", "-v", help="Show detailed output"),
) -> None:
    """Update library repositories by cloning or pulling from git.

    This command syncs all configured libraries from their git repositories.
    If a library doesn't exist locally, it will be cloned. If it exists, it will be pulled.
    """
    config = ConfigManager()
    libraries = config.get_libraries()

    if not libraries:
        display.display_warning("No libraries configured")
        console.print(
            "Libraries are auto-configured on first run with a default library."
        )
        return

    # Filter to specific library if requested
    if library_name:
        libraries = [lib for lib in libraries if lib.get("name") == library_name]
        if not libraries:
            console_err.print(
                f"[red]Error:[/red] Library '{library_name}' not found in configuration"
            )
            return

    libraries_path = config.get_libraries_path()

    # Create results table
    results = []

    with Progress(
        SpinnerColumn(),
        TextColumn("[progress.description]{task.description}"),
        console=console,
    ) as progress:
        for lib in libraries:
            name = lib.get("name")
            lib_type = lib.get("type", "git")
            enabled = lib.get("enabled", True)

            if not enabled:
                if verbose:
                    console.print(f"[dim]Skipping disabled library: {name}[/dim]")
                results.append((name, "Skipped (disabled)", False))
                continue

            # Skip static libraries (no sync needed)
            if lib_type == "static":
                if verbose:
                    console.print(
                        f"[dim]Skipping static library: {name} (no sync needed)[/dim]"
                    )
                results.append((name, "N/A (static)", True))
                continue

            # Handle git libraries
            url = lib.get("url")
            branch = lib.get("branch")
            directory = lib.get("directory", "library")

            task = progress.add_task(f"Updating {name}...", total=None)

            # Target path: ~/.config/boilerplates/libraries/{name}/
            target_path = libraries_path / name

            # Clone or pull the repository with sparse-checkout if directory is specified
            success, message = _clone_or_pull_repo(
                name, url, target_path, branch, directory
            )

            results.append((name, message, success))
            progress.remove_task(task)

            if verbose:
                if success:
                    display.display_success(f"{name}: {message}")
                else:
                    display.display_error(f"{name}: {message}")

    # Display summary table
    if not verbose:
        display.display_status_table(
            "Library Update Summary", results, columns=("Library", "Status")
        )

    # Summary
    total = len(results)
    successful = sum(1 for _, _, success in results if success)

    if successful == total:
        console.print(
            f"\n[green]All libraries updated successfully ({successful}/{total})[/green]"
        )
    elif successful > 0:
        console.print(
            f"\n[yellow]Partially successful: {successful}/{total} libraries updated[/yellow]"
        )
    else:
        console.print("\n[red]Failed to update libraries[/red]")


@app.command()
def list() -> None:
    """List all configured libraries."""
    config = ConfigManager()
    libraries = config.get_libraries()

    if not libraries:
        console.print("[yellow]No libraries configured.[/yellow]")
        return

    table = Table(title="Configured Libraries", show_header=True)
    table.add_column("Name", style="cyan", no_wrap=True)
    table.add_column("URL/Path", style="blue")
    table.add_column("Branch", style="yellow")
    table.add_column("Directory", style="magenta")
    table.add_column("Type", style="cyan")
    table.add_column("Status", style="green")

    libraries_path = config.get_libraries_path()

    for lib in libraries:
        name = lib.get("name", "")
        lib_type = lib.get("type", "git")
        enabled = lib.get("enabled", True)

        if lib_type == "git":
            url_or_path = lib.get("url", "")
            branch = lib.get("branch", "main")
            directory = lib.get("directory", "library")

            # Check if library exists locally
            library_base = libraries_path / name
            if directory and directory != ".":
                library_path = library_base / directory
            else:
                library_path = library_base
            exists = library_path.exists()

        elif lib_type == "static":
            url_or_path = lib.get("path", "")
            branch = "-"
            directory = "-"

            # Check if static path exists
            from pathlib import Path

            library_path = Path(url_or_path).expanduser()
            if not library_path.is_absolute():
                library_path = (config.config_path.parent / library_path).resolve()
            exists = library_path.exists()

        else:
            # Unknown type
            url_or_path = "<unknown type>"
            branch = "-"
            directory = "-"
            exists = False

        type_display = lib_type

        status_parts = []
        if not enabled:
            status_parts.append("[dim]disabled[/dim]")
        elif exists:
            status_parts.append("[green]available[/green]")
        else:
            status_parts.append("[yellow]not found[/yellow]")

        status = " ".join(status_parts)

        table.add_row(name, url_or_path, branch, directory, type_display, status)

    console.print(table)


@app.command()
def add(
    name: str = Argument(..., help="Unique name for the library"),
    library_type: str = Option(
        "git", "--type", "-t", help="Library type (git or static)"
    ),
    url: Optional[str] = Option(
        None, "--url", "-u", help="Git repository URL (for git type)"
    ),
    branch: str = Option("main", "--branch", "-b", help="Git branch (for git type)"),
    directory: str = Option(
        "library", "--directory", "-d", help="Directory in repo (for git type)"
    ),
    path: Optional[str] = Option(
        None, "--path", "-p", help="Local path (for static type)"
    ),
    enabled: bool = Option(
        True, "--enabled/--disabled", help="Enable or disable the library"
    ),
    sync: bool = Option(True, "--sync/--no-sync", help="Sync after adding (git only)"),
) -> None:
    """Add a new library to the configuration.

    Examples:
      # Add a git library
      repo add mylib --type git --url https://github.com/user/templates.git

      # Add a static library
      repo add local --type static --path ~/my-templates
    """
    config = ConfigManager()

    try:
        if library_type == "git":
            if not url:
                display.display_error("--url is required for git libraries")
                return
            config.add_library(
                name,
                library_type="git",
                url=url,
                branch=branch,
                directory=directory,
                enabled=enabled,
            )
        elif library_type == "static":
            if not path:
                display.display_error("--path is required for static libraries")
                return
            config.add_library(name, library_type="static", path=path, enabled=enabled)
        else:
            display.display_error(
                f"Invalid library type: {library_type}. Must be 'git' or 'static'."
            )
            return

        display.display_success(f"Added {library_type} library '{name}'")

        if library_type == "git" and sync and enabled:
            console.print(f"\nSyncing library '{name}'...")
            update(library_name=name, verbose=True)
        elif library_type == "static":
            display.display_info(f"Static library points to: {path}")
    except ConfigError as e:
        display.display_error(str(e))


@app.command()
def remove(
    name: str = Argument(..., help="Name of the library to remove"),
    keep_files: bool = Option(
        False, "--keep-files", help="Keep the local library files (don't delete)"
    ),
) -> None:
    """Remove a library from the configuration and delete its local files."""
    config = ConfigManager()

    try:
        # Remove from config
        config.remove_library(name)
        display.display_success(f"Removed library '{name}' from configuration")

        # Delete local files unless --keep-files is specified
        if not keep_files:
            libraries_path = config.get_libraries_path()
            library_path = libraries_path / name

            if library_path.exists():
                import shutil

                shutil.rmtree(library_path)
                display.display_success(f"Deleted local files at {library_path}")
            else:
                display.display_info(f"No local files found at {library_path}")
    except ConfigError as e:
        display.display_error(str(e))


# Register the repo command with the CLI
def register_cli(parent_app: Typer) -> None:
    """Register the repo command with the parent Typer app."""
    parent_app.add_typer(app, name="repo")