aegis-stack 0.2.0rc2__py3-none-any.whl

aegis/commands/add_service.py

@@ -0,0 +1,332 @@
+"""
+Add service command implementation.
+
+Adds services (auth, AI, etc.) to an existing Aegis Stack project using Copier's update mechanism.
+"""
+
+from pathlib import Path
+
+import typer
+
+from ..cli.utils import detect_scheduler_backend
+from ..core.components import COMPONENTS, CORE_COMPONENTS, SchedulerBackend
+from ..core.copier_manager import is_copier_project, load_copier_answers
+from ..core.manual_updater import ManualUpdater
+from ..core.service_resolver import ServiceResolver
+from ..core.services import SERVICES
+
+
+def add_service_command(
+    services: str | None = typer.Argument(
+        None,
+        help="Comma-separated list of services to add (auth,ai)",
+    ),
+    interactive: bool = typer.Option(
+        False,
+        "--interactive",
+        "-i",
+        help="Use interactive service selection",
+    ),
+    project_path: str = typer.Option(
+        ".",
+        "--project-path",
+        "-p",
+        help="Path to the Aegis Stack project (default: current directory)",
+    ),
+    yes: bool = typer.Option(False, "--yes", "-y", help="Skip confirmation prompt"),
+) -> None:
+    """
+    Add services to an existing Aegis Stack project.
+
+    This command uses Copier's update mechanism to add new services (auth, AI, etc.)
+    to a project that was generated with 'aegis init'.
+
+    Examples:
+
+        - aegis add-service auth
+
+        - aegis add-service auth,ai
+
+        - aegis add-service auth --project-path ../my-project
+
+    Note: This command only works with projects generated using Copier
+    (the default since v0.2.0). Services may auto-add required components.
+    """
+
+    typer.echo("🛡️  Aegis Stack - Add Services")
+    typer.echo("=" * 50)
+
+    # Resolve project path
+    target_path = Path(project_path).resolve()
+
+    # Validate it's a Copier project
+    if not is_copier_project(target_path):
+        typer.echo(
+            f"❌ Project at {target_path} was not generated with Copier.", err=True
+        )
+        typer.echo(
+            "   The 'aegis add service' command only works with Copier-generated projects.",
+            err=True,
+        )
+        typer.echo(
+            "   To add services, regenerate the project with the services included.",
+            err=True,
+        )
+        raise typer.Exit(1)
+
+    typer.echo(f"📁 Project: {target_path}")
+
+    # Validate services argument or interactive mode
+    if not interactive and not services:
+        typer.echo(
+            "❌ Error: services argument is required (or use --interactive)", err=True
+        )
+        typer.echo("   Usage: aegis add service auth,ai", err=True)
+        typer.echo("   Or: aegis add service --interactive", err=True)
+        raise typer.Exit(1)
+
+    # Interactive mode
+    if interactive:
+        if services:
+            typer.echo(
+                "⚠️  Warning: --interactive flag ignores service arguments",
+                err=False,
+            )
+
+        from ..cli.interactive import interactive_service_selection
+
+        selected_services = interactive_service_selection(target_path)
+
+        if not selected_services:
+            typer.echo("\n✅ No services selected")
+            raise typer.Exit(0)
+
+        # Convert to comma-separated string for existing logic
+        services = ",".join(selected_services)
+
+        # Auto-confirm in interactive mode
+        yes = True
+
+    # Verify project is in a git repository (required for Copier updates)
+    git_dir = target_path / ".git"
+    if not git_dir.exists():
+        typer.echo("\n❌ Project is not in a git repository", err=True)
+        typer.echo("   Copier updates require git for change tracking", err=True)
+        typer.echo(
+            "   💡 Projects created with 'aegis init' should have git initialized automatically",
+            err=True,
+        )
+        typer.echo(
+            "   💡 If you created this project manually, run: git init && git add . && git commit -m 'Initial commit'",
+            err=True,
+        )
+        raise typer.Exit(1)
+
+    # Parse and validate services
+    assert services is not None  # Already validated by check above
+    services_raw = [s.strip() for s in services.split(",")]
+
+    # Check for empty services
+    if any(not s for s in services_raw):
+        typer.echo("❌ Empty service name is not allowed", err=True)
+        raise typer.Exit(1)
+
+    selected_services = [s for s in services_raw if s]
+
+    # Validate services exist
+    try:
+        errors = ServiceResolver.validate_services(selected_services)
+        if errors:
+            for error in errors:
+                typer.echo(f"❌ {error}", err=True)
+            raise typer.Exit(1)
+    except Exception as e:
+        typer.echo(f"❌ Service validation failed: {e}", err=True)
+        raise typer.Exit(1)
+
+    # Load existing project configuration
+    try:
+        existing_answers = load_copier_answers(target_path)
+    except Exception as e:
+        typer.echo(f"❌ Failed to load project configuration: {e}", err=True)
+        raise typer.Exit(1)
+
+    # Check which services are already enabled
+    already_enabled = []
+    for service in selected_services:
+        # Check if service is already enabled in answers
+        include_key = f"include_{service}"
+        if existing_answers.get(include_key) is True:
+            already_enabled.append(service)
+
+    if already_enabled:
+        typer.echo(f"ℹ️  Already enabled: {', '.join(already_enabled)}", err=False)
+
+    # Filter out already enabled services
+    services_to_add = [s for s in selected_services if s not in already_enabled]
+
+    if not services_to_add:
+        typer.echo("✅ All requested services are already enabled!")
+        raise typer.Exit(0)
+
+    # Resolve service dependencies to components
+    try:
+        required_components, _ = ServiceResolver.resolve_service_dependencies(
+            services_to_add
+        )
+    except ValueError as e:
+        typer.echo(f"❌ Failed to resolve service dependencies: {e}", err=True)
+        raise typer.Exit(1)
+
+    # Check which components are already enabled
+    enabled_components = []
+    missing_components = []
+
+    for component in required_components:
+        include_key = f"include_{component}"
+        if existing_answers.get(include_key) is True or component in CORE_COMPONENTS:
+            enabled_components.append(component)
+        else:
+            missing_components.append(component)
+
+    # Show what will be added
+    typer.echo("\n🔧 Services to add:")
+    for service in services_to_add:
+        if service in SERVICES:
+            desc = SERVICES[service].description
+            typer.echo(f"   • {service}: {desc}")
+
+    # Show component requirements
+    if missing_components:
+        typer.echo("\n📦 Required components (will be auto-added):")
+        for component in missing_components:
+            if component in COMPONENTS:
+                desc = COMPONENTS[component].description
+                typer.echo(f"   • {component}: {desc}")
+
+    if enabled_components:
+        # Filter out core components from display
+        non_core_enabled = [c for c in enabled_components if c not in CORE_COMPONENTS]
+        if non_core_enabled:
+            typer.echo(
+                f"\n✅ Already have required components: {', '.join(non_core_enabled)}"
+            )
+
+    # Confirm before proceeding
+    typer.echo()
+    if not yes and not typer.confirm("🚀 Add these services?"):
+        typer.echo("❌ Operation cancelled")
+        raise typer.Exit(0)
+
+    # Prepare update data for ManualUpdater
+    update_data: dict[str, bool | str] = {}
+
+    # Add service flags
+    for service in services_to_add:
+        include_key = f"include_{service}"
+        update_data[include_key] = True
+
+    # Add missing component flags
+    for component in missing_components:
+        include_key = f"include_{component}"
+        update_data[include_key] = True
+
+    # Add services using ManualUpdater
+    typer.echo("\n🔄 Updating project...")
+    try:
+        updater = ManualUpdater(target_path)
+
+        # Add missing components first
+        for component in missing_components:
+            typer.echo(f"\n📦 Adding required component: {component}...")
+
+            # Prepare component-specific data
+            component_data: dict[str, bool | str] = {}
+
+            # Handle scheduler backend if needed
+            if component == "scheduler":
+                scheduler_backend = detect_scheduler_backend([component])
+                component_data["scheduler_backend"] = scheduler_backend
+                component_data["scheduler_with_persistence"] = (
+                    scheduler_backend == SchedulerBackend.SQLITE.value
+                )
+            elif component == "database":
+                component_data["database_engine"] = "sqlite"
+
+            # Add the component
+            result = updater.add_component(component, component_data)
+
+            if not result.success:
+                typer.echo(
+                    f"❌ Failed to add component {component}: {result.error_message}",
+                    err=True,
+                )
+                raise typer.Exit(1)
+
+            if result.files_modified:
+                typer.echo(f"   ✅ Added {len(result.files_modified)} files")
+            if result.files_skipped:
+                typer.echo(f"   ⚠️  Skipped {len(result.files_skipped)} existing files")
+
+        # Now add each service sequentially
+        for service in services_to_add:
+            typer.echo(f"\n🔧 Adding service: {service}...")
+
+            # Prepare service-specific data
+            service_data: dict[str, bool | str] = {}
+
+            # For AI service, set default providers
+            if service == "ai":
+                service_data["ai_providers"] = "openai"
+
+            # Add the service (services are added like components)
+            result = updater.add_component(service, service_data)
+
+            if not result.success:
+                typer.echo(
+                    f"❌ Failed to add service {service}: {result.error_message}",
+                    err=True,
+                )
+                raise typer.Exit(1)
+
+            # Show results
+            if result.files_modified:
+                typer.echo(f"   ✅ Added {len(result.files_modified)} files")
+            if result.files_skipped:
+                typer.echo(f"   ⚠️  Skipped {len(result.files_skipped)} existing files")
+
+        typer.echo("\n✅ Services added successfully!")
+
+        # Provide next steps
+        if len(services_to_add) > 0 or len(missing_components) > 0:
+            typer.echo("\n💡 Review changes:")
+            typer.echo("   git diff docker-compose.yml")
+            typer.echo("   git diff pyproject.toml")
+
+        typer.echo("\n📝 Next steps:")
+        typer.echo("   1. Run 'make check' to verify the update")
+        typer.echo("   2. Test your application")
+        typer.echo("   3. Commit the changes with: git add . && git commit")
+
+        # Service-specific guidance
+        if "auth" in services_to_add:
+            project_slug = existing_answers.get("project_slug", "my-project")
+            typer.echo("\n🔐 Auth Service Setup:")
+            typer.echo("   1. Run 'make migrate' to apply auth migrations")
+            typer.echo(
+                f"   2. Create test users with CLI: '{project_slug} auth create-test-users'"
+            )
+            typer.echo("   3. Check auth routes at /api/auth/docs")
+
+        if "ai" in services_to_add:
+            project_slug = existing_answers.get("project_slug", "my-project")
+            typer.echo("\n🤖 AI Service Setup:")
+            typer.echo(
+                "   1. Set AI_PROVIDER in .env (openai, anthropic, google, groq)"
+            )
+            typer.echo("   2. Set provider API key (OPENAI_API_KEY, etc.)")
+            typer.echo(f"   3. Test with CLI: '{project_slug} ai chat'")
+
+    except Exception as e:
+        typer.echo(f"\n❌ Failed to add services: {e}", err=True)
+        raise typer.Exit(1)

aegis/commands/components.py

@@ -0,0 +1,35 @@
+"""
+Components command implementation.
+"""
+
+import typer
+
+from ..core.components import CORE_COMPONENTS, ComponentType, get_components_by_type
+
+
+def components_command() -> None:
+    """List available components and their dependencies."""
+
+    typer.echo("\n📦 CORE COMPONENTS")
+    typer.echo("=" * 40)
+    for component in CORE_COMPONENTS:
+        if component == "backend":
+            typer.echo("  backend      - FastAPI backend server (always included)")
+        elif component == "frontend":
+            typer.echo("  frontend     - Flet frontend interface (always included)")
+
+    typer.echo("\n🏗️  INFRASTRUCTURE COMPONENTS")
+    typer.echo("=" * 40)
+
+    infra_components = get_components_by_type(ComponentType.INFRASTRUCTURE)
+    for name, spec in infra_components.items():
+        typer.echo(f"  {name:12} - {spec.description}")
+        if spec.requires:
+            typer.echo(f"               Requires: {', '.join(spec.requires)}")
+        if spec.recommends:
+            typer.echo(f"               Recommends: {', '.join(spec.recommends)}")
+
+    typer.echo(
+        "\n💡 Use 'aegis init PROJECT_NAME --components redis,worker' "
+        "to select components"
+    )