cli.py

"""
Command Line Interface for the Component Generator.
"""

from pathlib import Path
import typer
from rich.console import Console
from rich.table import Table
from rich.progress import track
from rich.syntax import Syntax

from src.services.component_loader import ComponentLoader
from src.services.vector_store import VectorStore
from src.services.anthropic_service import AnthropicService

app = typer.Typer(help="Component Generator CLI")
console = Console()


@app.command()
def load_analysis(
    analysis_file: Path = typer.Argument(
        ...,
        exists=True,
        dir_okay=False,
        help="JSON file containing component analysis"
    )
):
    """Load component analysis into vector store."""
    try:
        # Initialize services
        loader = ComponentLoader()
        vector_store = VectorStore()

        console.print("\n📚 Loading component analysis...", style="blue")

        # Load components from analysis
        components = loader.load_analysis(analysis_file)
        console.print(f"✅ Found {len(components)} components", style="green")

        # Create summary table
        table = Table(title="Loaded Components")
        table.add_column("Name", style="cyan")
        table.add_column("Patterns", style="magenta")
        table.add_column("State Complexity", style="blue")
        table.add_column("Features", style="green")

        # Store components
        for component in track(components, description="Storing components..."):
            vector_store.add_component(component)
            table.add_row(
                component.name,
                ", ".join(component.patterns.type),
                str(component.patterns.indicators.state_management.complexity),
                ", ".join(
                    feature for feature, enabled in {
                        "Content Projection": component.patterns.indicators.content_projection,
                        "Service Usage": component.patterns.indicators.service_usage,
                        "User Interaction": component.patterns.indicators.user_interaction,
                        "Form Integration": component.patterns.indicators.form_integration
                    }.items() if enabled
                )
            )

        console.print(table)
        console.print("\n✅ All components stored successfully!", style="green")

    except Exception as e:
        console.print(f"\n❌ Error: {str(e)}", style="red")
        raise typer.Exit(1)


@app.command()
def find_similar(
    description: str = typer.Argument(
        ...,
        help="Description of the component to find"
    ),
    count: int = typer.Option(
        3,
        help="Number of similar components to return"
    )
):
    """Find similar components based on description."""
    try:
        vector_store = VectorStore()

        console.print("\n🔍 Searching for similar components...", style="blue")
        results = vector_store.find_similar(description, count)

        table = Table(title="Similar Components")
        table.add_column("Name", style="cyan")
        table.add_column("Patterns", style="magenta")
        table.add_column("Similarity", style="green")
        table.add_column("Features", style="blue")

        for result in results:
            component = result['component']
            table.add_row(
                component.name,
                ", ".join(component.patterns.type),
                f"{result['similarity_score']:.2%}",
                ", ".join(
                    feature for feature, enabled in {
                        "Content Projection": component.patterns.indicators.content_projection,
                        "Service Usage": component.patterns.indicators.service_usage,
                        "User Interaction": component.patterns.indicators.user_interaction,
                        "Form Integration": component.patterns.indicators.form_integration
                    }.items() if enabled
                )
            )

        console.print(table)

    except Exception as e:
        console.print(f"\n❌ Error: {str(e)}", style="red")
        raise typer.Exit(1)


@app.command()
def generate(
    description: str = typer.Argument(
        ...,
        help="Description of the component to generate using our existing components"
    ),
    output_dir: Path = typer.Option(
        Path("generated"),
        help="Directory to save generated component"
    ),
    num_references: int = typer.Option(
        5,  # Increased from 3 to get more component references
        help="Number of similar components to use as reference"
    )
):
    """Generate a new component composition using our existing components."""
    try:
        # Initialize services
        vector_store = VectorStore()
        anthropic = AnthropicService()

        console.print(
            "\n🔍 Finding relevant components to use...", style="blue")
        similar = vector_store.find_similar(description, num_references)

        # Show reference components
        table = Table(title="Components Available for Use")
        table.add_column("Component", style="cyan")
        table.add_column("Type", style="magenta")
        table.add_column("Inputs", style="green")
        table.add_column("Template Patterns", style="blue")

        for result in similar:
            component = result['component']
            table.add_row(
                component.name,
                ", ".join(component.patterns.type),
                ", ".join(component.patterns.code_markers.inputs or ["None"]),
                ", ".join(
                    component.patterns.code_markers.template_patterns or ["None"])
            )

        console.print(table)
        console.print("\n🤖 Generating component composition...", style="blue")

        # Generate component using updated anthropic service
        generated_code = anthropic.generate_component(
            description,
            [result['component'] for result in similar]
        )

        # Create output directory
        output_dir.mkdir(parents=True, exist_ok=True)

        # Generate component name from description
        component_name = "_".join(
            word.lower() for word in description.split()[:3]
            if word.isalnum()
        )
        component_file = output_dir / f"{component_name}.component.ts"

        # Save generated code
        component_file.write_text(generated_code)

        # Show generated code with syntax highlighting
        console.print("\n✨ Generated Component:", style="green")
        syntax = Syntax(
            generated_code,
            "typescript",
            theme="monokai",
            line_numbers=True
        )
        console.print(syntax)

        # Add usage examples
        console.print("\n📝 Example Usage:", style="blue")
        console.print("Add this to your module:", style="cyan")
        console.print("```typescript")
        console.print(
            f"import {{ {component_name.title()}Component }} from './generated/{component_name}.component';")
        console.print("```")

        console.print(f"\n💾 Saved to: {component_file}", style="blue")

    except Exception as e:
        console.print(f"\n❌ Error: {str(e)}", style="red")
        raise typer.Exit(1)


if __name__ == "__main__":
    app()