mcp-skills

"""
Setup command implementation.
"""

import builtins
import logging
from pathlib import Path

import click
from rich.console import Console
from rich.progress import (
    BarColumn,
    DownloadColumn,
    Progress,
    SpinnerColumn,
    TextColumn,
    TimeRemainingColumn,
    TransferSpeedColumn,
)

from mcp_skills.cli.commands.install import _install_hooks
from mcp_skills.models.repository import Repository
from mcp_skills.services.agent_detector import AgentDetector
from mcp_skills.services.agent_installer import AgentInstaller
from mcp_skills.services.indexing import IndexingEngine
from mcp_skills.services.repository_manager import RepositoryManager
from mcp_skills.services.skill_manager import SkillManager
from mcp_skills.services.toolchain_detector import ToolchainDetector


console = Console()
logger = logging.getLogger(__name__)

# Store builtin list to avoid shadowing in annotations
ListType = builtins.list


@click.command()
@click.option(
    "--project-dir",
    default=".",
    type=click.Path(exists=True),
    help="Project directory to analyze",
)
@click.option(
    "--config",
    default="~/.mcp-skillset/config.yaml",
    type=click.Path(),
    help="Config file location",
)
@click.option("--auto", is_flag=True, help="Non-interactive setup with defaults")
@click.option(
    "--skip-agents",
    is_flag=True,
    help="Skip automatic installation for AI agents",
)
def setup(project_dir: str, config: str, auto: bool, skip_agents: bool) -> None:
    """Auto-configure mcp-skillset for your project.

    This command will:
    1. Detect your project's toolchain
    2. Clone relevant skill repositories
    3. Index skills with vector + KG
    4. Configure MCP server
    5. Validate setup
    6. Install for AI agents (Claude Code, Auggie by default; excludes Claude Desktop)
    7. Optionally install Claude Code hooks for automatic skill hints
    """
    console.print("🚀 [bold green]Starting mcp-skillset setup...[/bold green]")
    console.print(f"📁 Project directory: {project_dir}")
    console.print(f"⚙️  Config location: {config}\n")

    try:
        # 1. Toolchain detection
        console.print("[bold cyan]Step 1/7:[/bold cyan] Detecting project toolchain...")
        detector = ToolchainDetector()
        project_path = Path(project_dir).resolve()
        toolchain = detector.detect(project_path)

        console.print(
            f"  ✓ Primary language: [bold]{toolchain.primary_language}[/bold]"
        )
        if toolchain.frameworks:
            console.print(f"  ✓ Frameworks: {', '.join(toolchain.frameworks)}")
        if toolchain.test_frameworks:
            console.print(
                f"  ✓ Test frameworks: {', '.join(toolchain.test_frameworks)}"
            )
        console.print(f"  ✓ Confidence: {toolchain.confidence:.0%}\n")

        # 2. Repository cloning
        console.print(
            "[bold cyan]Step 2/7:[/bold cyan] Setting up skill repositories..."
        )
        repo_manager = RepositoryManager()

        # Get default repos or prompt user
        if auto:
            repos_to_add = RepositoryManager.DEFAULT_REPOS
            console.print("  Using default repositories (--auto mode)")
        else:
            console.print("\n  Available default repositories:")
            for i, repo in enumerate(RepositoryManager.DEFAULT_REPOS, 1):
                console.print(f"    {i}. {repo['url']} (priority: {repo['priority']})")

            if click.confirm("\n  Clone default repositories?", default=True):
                repos_to_add = RepositoryManager.DEFAULT_REPOS
            else:
                repos_to_add = []
                console.print(
                    "  [dim]You can add repositories later with: mcp-skillset repo add <url>[/dim]"
                )

        # Clone repositories with per-repository progress bars
        added_repos: ListType[Repository] = []
        with Progress(
            SpinnerColumn(),
            TextColumn("[bold blue]{task.description}"),
            BarColumn(bar_width=40),
            DownloadColumn(),
            TransferSpeedColumn(),
            TimeRemainingColumn(),
            console=console,
        ) as progress:
            for repo_config in repos_to_add:
                # Extract repo name from URL for display
                repo_name = repo_config["url"].split("/")[-1].replace(".git", "")

                try:
                    # Check if already exists
                    repo_id = repo_manager._generate_repo_id(repo_config["url"])
                    existing = repo_manager.get_repository(repo_id)

                    if existing:
                        console.print(
                            f"  ⊙ Repository already exists: {repo_config['url']}"
                        )
                        added_repos.append(existing)
                    else:
                        # Create task for this repository
                        task_id = progress.add_task(
                            f"Cloning {repo_name}",
                            total=100,  # Will be updated by callback
                            start=False,
                        )

                        # Progress callback updates this specific task
                        def make_callback(tid: int):  # type: ignore[misc]
                            def update_progress(
                                current: int, total: int, _message: str
                            ) -> None:
                                if total > 0:
                                    progress.update(tid, completed=current, total=total)
                                    if not progress.tasks[tid].started:
                                        progress.start_task(tid)

                            return update_progress

                        new_repo = repo_manager.add_repository_with_progress(
                            url=repo_config["url"],
                            priority=repo_config["priority"],
                            license=repo_config.get("license", "Unknown"),
                            progress_callback=make_callback(task_id),
                        )
                        progress.update(
                            task_id, description=f"✓ {repo_name}", completed=100
                        )
                        added_repos.append(new_repo)
                        console.print(f"  ✓ Cloned {new_repo.skill_count} skills")
                except Exception as e:
                    console.print(
                        f"  [red]✗ Failed to clone {repo_config['url']}: {e}[/red]"
                    )
                    logger.error(f"Repository clone failed: {e}")

        if not added_repos:
            console.print(
                "\n  [yellow]No repositories configured. Add some with: mcp-skillset repo add <url>[/yellow]"
            )

        console.print()

        # 3. Indexing
        console.print("[bold cyan]Step 3/7:[/bold cyan] Building skill indices...")
        skill_manager = SkillManager()
        indexing_engine = IndexingEngine(skill_manager=skill_manager)

        with Progress(
            SpinnerColumn(),
            TextColumn("[progress.description]{task.description}"),
            console=console,
        ) as progress:
            task = progress.add_task("  Indexing skills...", total=None)
            try:
                stats = indexing_engine.reindex_all(force=True)
                progress.update(task, completed=True)
                console.print(f"  ✓ Indexed {stats.total_skills} skills")
                console.print(f"  ✓ Vector store: {stats.vector_store_size // 1024} KB")
                console.print(
                    f"  ✓ Knowledge graph: {stats.graph_nodes} nodes, {stats.graph_edges} edges\n"
                )
            except Exception as e:
                progress.stop()
                console.print(f"  [red]✗ Indexing failed: {e}[/red]")
                logger.error(f"Indexing failed: {e}")

        # 4. MCP configuration
        console.print("[bold cyan]Step 4/7:[/bold cyan] Configuring MCP server...")
        base_dir = Path.home() / ".mcp-skillset"
        console.print(f"  ✓ Base directory: {base_dir}")
        console.print(f"  ✓ ChromaDB: {base_dir / 'chromadb'}")
        console.print(f"  ✓ Repositories: {base_dir / 'repos'}\n")

        # 5. Validation
        console.print("[bold cyan]Step 5/7:[/bold cyan] Validating setup...")
        repos = repo_manager.list_repositories()
        skills = skill_manager.discover_skills()

        validation_ok = True
        if not repos:
            console.print("  [red]✗ No repositories configured[/red]")
            validation_ok = False
        else:
            console.print(f"  ✓ {len(repos)} repositories configured")

        if not skills:
            console.print("  [red]✗ No skills discovered[/red]")
            validation_ok = False
        else:
            console.print(f"  ✓ {len(skills)} skills available")

        if stats.total_skills == 0:
            console.print("  [red]✗ No skills indexed[/red]")
            validation_ok = False
        else:
            console.print(f"  ✓ {stats.total_skills} skills indexed")

        console.print()

        # 6. Agent installation (if not skipped)
        agents_installed = 0
        if not skip_agents:
            console.print()
            console.print(
                "[bold cyan]Step 6/7:[/bold cyan] Installing for AI agents..."
            )

            agent_detector = AgentDetector()
            # Exclude Claude Desktop by default (config path conflicts with Claude Code)
            all_agents = agent_detector.detect_all()
            found_agents = [
                a for a in all_agents if a.exists and a.id != "claude-desktop"
            ]

            if found_agents:
                console.print(f"  ✓ Detected {len(found_agents)} agent(s):")
                for agent in found_agents:
                    console.print(f"    • {agent.name}")

                # Ask for confirmation (respect auto flag)
                should_install = auto or click.confirm(
                    "\n  Install mcp-skillset for these agents?", default=True
                )

                if should_install:
                    installer = AgentInstaller()
                    success_count = 0

                    for agent in found_agents:
                        try:
                            result = installer.install(agent)
                            if result.success:
                                console.print(f"    ✓ {agent.name} configured")
                                success_count += 1
                            else:
                                console.print(
                                    f"    ✗ {agent.name} failed: {result.error or 'Unknown error'}"
                                )
                        except Exception as e:
                            console.print(f"    ✗ {agent.name} error: {str(e)}")

                    console.print(
                        f"\n  Installed for {success_count}/{len(found_agents)} agent(s)"
                    )
                    agents_installed = success_count
                else:
                    console.print("  Skipped agent installation")
            else:
                console.print(
                    "  No AI agents detected (Claude Desktop, Claude Code, Auggie)"
                )
                console.print(
                    "  You can install manually later with: mcp-skillset install"
                )
        else:
            console.print(
                "\n[dim]Skipped agent installation (--skip-agents flag)[/dim]"
            )

        # 7. Hook installation (optional)
        hooks_installed = False
        if not skip_agents and agents_installed > 0:
            console.print()
            console.print(
                "[bold cyan]Step 7/7:[/bold cyan] Claude Code hooks (optional)..."
            )
            console.print(
                "  Hooks automatically suggest relevant skills when you type prompts."
            )

            # Ask for confirmation (respect auto flag)
            should_install_hooks = auto or click.confirm(
                "\n  Install Claude Code hooks for automatic skill hints?",
                default=True,
            )

            if should_install_hooks:
                try:
                    if _install_hooks(dry_run=False, force=False):
                        console.print("  ✓ Hooks installed successfully")
                        hooks_installed = True
                    else:
                        console.print(
                            "  [dim]Hooks not installed (already configured or Claude Code not found)[/dim]"
                        )
                except Exception as e:
                    console.print(f"  [yellow]⚠ Hook installation failed: {e}[/yellow]")
                    logger.warning(f"Hook installation failed: {e}")
            else:
                console.print("  Skipped hook installation")
                console.print(
                    "  [dim]You can install later with: mcp-skillset install --with-hooks[/dim]"
                )

        console.print()

        # Summary
        if validation_ok:
            console.print("[bold green]✓ Setup complete![/bold green]\n")
            console.print("Next steps:")
            if agents_installed > 0:
                console.print(
                    "  1. [cyan]Restart your AI agent[/cyan] to load mcp-skillset"
                )
                console.print("  2. [cyan]Explore skills:[/cyan] mcp-skillset demo")
                console.print(
                    "  3. [cyan]Search skills:[/cyan] mcp-skillset search 'python testing'"
                )
                if hooks_installed:
                    console.print(
                        "  4. [cyan]Skills will be suggested automatically[/cyan] when you type prompts!"
                    )
                else:
                    console.print(
                        "  4. [cyan]Enable auto-hints:[/cyan] mcp-skillset install --with-hooks"
                    )
            else:
                console.print(
                    "  1. [cyan]Install for agents:[/cyan] mcp-skillset install"
                )
                console.print("  2. [cyan]Explore skills:[/cyan] mcp-skillset demo")
                console.print(
                    "  3. [cyan]Search skills:[/cyan] mcp-skillset search 'python testing'"
                )
                console.print(
                    "  4. [cyan]Show skill:[/cyan] mcp-skillset show <skill-id>"
                )
            console.print()
            console.print(
                "[dim]💡 Tip: Try 'mcp-skillset demo' to see example questions for each skill![/dim]"
            )
        else:
            console.print(
                "[bold yellow]⚠ Setup completed with warnings[/bold yellow]\n"
            )
            console.print(
                "Please check the errors above and run setup again if needed."
            )

    except KeyboardInterrupt:
        console.print("\n[yellow]Setup cancelled by user[/yellow]")
        raise SystemExit(1)
    except Exception as e:
        console.print(f"\n[red]Setup failed: {e}[/red]")
        logger.exception("Setup failed")
        raise SystemExit(1)