bakery-ia/services/orchestrator/app/api/internal_demo.py

"""
Internal Demo API Endpoints for Orchestrator Service
Used by demo_session service to clone data for virtual demo tenants
"""

from fastapi import APIRouter, Depends, HTTPException, Header
from typing import Dict, Any
from uuid import UUID
import structlog
import os
import json

from app.core.database import get_db
from sqlalchemy.ext.asyncio import AsyncSession
from sqlalchemy import select, delete, func
from app.models.orchestration_run import OrchestrationRun, OrchestrationStatus
import uuid
from datetime import datetime, timezone, timedelta
from typing import Optional
import sys
from pathlib import Path

# Add shared utilities to path
sys.path.insert(0, str(Path(__file__).parent.parent.parent.parent))
from shared.utils.demo_dates import adjust_date_for_demo

from app.core.config import settings

router = APIRouter()
logger = structlog.get_logger()


async def ensure_unique_run_number(db: AsyncSession, base_run_number: str) -> str:
    """Ensure the run number is unique by appending a suffix if needed"""
    proposed_run_number = base_run_number

    # Check if the proposed run number already exists in the database
    while True:
        result = await db.execute(
            select(OrchestrationRun)
            .where(OrchestrationRun.run_number == proposed_run_number)
        )
        existing_run = result.scalar_one_or_none()

        if not existing_run:
            # Run number is unique, return it
            return proposed_run_number

        # Generate a new run number with an additional random suffix
        random_suffix = str(uuid.uuid4())[:4].upper()
        proposed_run_number = f"{base_run_number[:50-len(random_suffix)-1]}-{random_suffix}"


def verify_internal_api_key(x_internal_api_key: str = Header(...)):
    """Verify internal API key for service-to-service communication"""
    if x_internal_api_key != settings.INTERNAL_API_KEY:
        raise HTTPException(status_code=403, detail="Invalid internal API key")
    return True


async def load_fixture_data_for_tenant(
    db: AsyncSession,
    tenant_uuid: UUID,
    demo_account_type: str,
    reference_time: datetime,
    base_tenant_id: Optional[str] = None
) -> int:
    """
    Load orchestration run data from JSON fixture directly into the virtual tenant.
    Returns the number of runs created.
    """
    from shared.utils.seed_data_paths import get_seed_data_path
    from shared.utils.demo_dates import resolve_time_marker, adjust_date_for_demo

    # Load fixture data
    if demo_account_type == "enterprise_child" and base_tenant_id:
        json_file = get_seed_data_path("enterprise", "11-orchestrator.json", child_id=base_tenant_id)
    else:
        json_file = get_seed_data_path(demo_account_type, "11-orchestrator.json")

    with open(json_file, 'r', encoding='utf-8') as f:
        fixture_data = json.load(f)

    orchestration_run_data = fixture_data.get("orchestration_run")
    if not orchestration_run_data:
        logger.warning("No orchestration_run data in fixture")
        return 0

    # Parse and adjust dates from fixture to reference_time
    base_started_at = resolve_time_marker(orchestration_run_data.get("started_at"), reference_time)
    base_completed_at = resolve_time_marker(orchestration_run_data.get("completed_at"), reference_time)

    # Adjust dates to make them appear recent relative to session creation
    started_at = adjust_date_for_demo(base_started_at, reference_time) if base_started_at else reference_time - timedelta(hours=2)
    completed_at = adjust_date_for_demo(base_completed_at, reference_time) if base_completed_at else started_at + timedelta(minutes=15)

    # Generate unique run number with session context
    current_year = reference_time.year
    unique_suffix = str(uuid.uuid4())[:8].upper()
    run_number = f"ORCH-DEMO-PROF-{current_year}-001-{unique_suffix}"

    # Create orchestration run for virtual tenant
    new_run = OrchestrationRun(
        id=uuid.uuid4(),  # Generate new UUID
        tenant_id=tenant_uuid,
        run_number=run_number,
        status=OrchestrationStatus[orchestration_run_data["status"]],
        run_type=orchestration_run_data.get("run_type", "daily"),
        priority="normal",
        started_at=started_at,
        completed_at=completed_at,
        duration_seconds=orchestration_run_data.get("duration_seconds", 900),

        # Step statuses from orchestration_results
        forecasting_status="success",
        forecasting_started_at=started_at,
        forecasting_completed_at=started_at + timedelta(minutes=2),

        production_status="success",
        production_started_at=started_at + timedelta(minutes=2),
        production_completed_at=started_at + timedelta(minutes=5),

        procurement_status="success",
        procurement_started_at=started_at + timedelta(minutes=5),
        procurement_completed_at=started_at + timedelta(minutes=8),

        notification_status="success",
        notification_started_at=started_at + timedelta(minutes=8),
        notification_completed_at=completed_at,

        # Results from orchestration_results
        forecasts_generated=fixture_data.get("orchestration_results", {}).get("forecasts_generated", 10),
        production_batches_created=fixture_data.get("orchestration_results", {}).get("production_batches_created", 18),
        procurement_plans_created=0,
        purchase_orders_created=fixture_data.get("orchestration_results", {}).get("purchase_orders_created", 6),
        notifications_sent=fixture_data.get("orchestration_results", {}).get("notifications_sent", 8),

        # Metadata
        triggered_by="system",
        created_at=started_at,
        updated_at=completed_at
    )

    db.add(new_run)
    await db.flush()

    logger.info(
        "Loaded orchestration run from fixture",
        tenant_id=str(tenant_uuid),
        run_number=new_run.run_number,
        started_at=started_at.isoformat()
    )

    return 1


@router.post("/internal/demo/clone")
async def clone_demo_data(
    base_tenant_id: str,
    virtual_tenant_id: str,
    demo_account_type: str,
    session_id: Optional[str] = None,
    session_created_at: Optional[str] = None,
    db: AsyncSession = Depends(get_db),
    _: bool = Depends(verify_internal_api_key)
):
    """
    Clone orchestration run demo data from base tenant to virtual tenant

    This endpoint is called by the demo_session service during session initialization.
    It clones orchestration runs with date adjustments to make them appear recent.

    If the base tenant has no orchestration runs, it will first seed them from the fixture.
    """

    start_time = datetime.now(timezone.utc)

    # Parse session_created_at or use current time
    if session_created_at:
        try:
            reference_time = datetime.fromisoformat(session_created_at.replace('Z', '+00:00'))
        except:
            reference_time = datetime.now(timezone.utc)
    else:
        reference_time = datetime.now(timezone.utc)

    logger.info(
        "Starting orchestration runs cloning with date adjustment",
        base_tenant_id=base_tenant_id,
        virtual_tenant_id=virtual_tenant_id,
        demo_account_type=demo_account_type,
        session_id=session_id,
        reference_time=reference_time.isoformat()
    )

    try:
        virtual_uuid = uuid.UUID(virtual_tenant_id)

        # Load fixture data directly into virtual tenant (no base tenant cloning)
        runs_created = await load_fixture_data_for_tenant(
            db,
            virtual_uuid,
            demo_account_type,
            reference_time,
            base_tenant_id
        )

        await db.commit()

        duration_ms = int((datetime.now(timezone.utc) - start_time).total_seconds() * 1000)

        logger.info(
            "Orchestration runs loaded from fixture successfully",
            virtual_tenant_id=str(virtual_tenant_id),
            runs_created=runs_created,
            duration_ms=duration_ms
        )

        return {
            "service": "orchestrator",
            "status": "completed",
            "success": True,
            "records_cloned": runs_created,
            "runs_cloned": runs_created,
            "duration_ms": duration_ms
        }

    except Exception as e:
        logger.error("Failed to clone orchestration runs", error=str(e), exc_info=True)
        await db.rollback()
        raise HTTPException(status_code=500, detail=f"Failed to clone orchestration runs: {str(e)}")


@router.delete("/internal/demo/tenant/{virtual_tenant_id}")
async def delete_demo_data(
    virtual_tenant_id: str,
    db: AsyncSession = Depends(get_db),
    _: bool = Depends(verify_internal_api_key)
):
    """Delete all orchestration runs for a virtual demo tenant"""
    logger.info("Deleting orchestration runs for virtual tenant", virtual_tenant_id=virtual_tenant_id)
    start_time = datetime.now(timezone.utc)

    try:
        virtual_uuid = uuid.UUID(virtual_tenant_id)

        # Count records
        run_count = await db.scalar(
            select(func.count(OrchestrationRun.id))
            .where(OrchestrationRun.tenant_id == virtual_uuid)
        )

        # Delete orchestration runs
        await db.execute(
            delete(OrchestrationRun)
            .where(OrchestrationRun.tenant_id == virtual_uuid)
        )
        await db.commit()

        duration_ms = int((datetime.now(timezone.utc) - start_time).total_seconds() * 1000)
        logger.info(
            "Orchestration runs deleted successfully",
            virtual_tenant_id=virtual_tenant_id,
            duration_ms=duration_ms
        )

        return {
            "service": "orchestrator",
            "status": "deleted",
            "virtual_tenant_id": virtual_tenant_id,
            "records_deleted": {
                "orchestration_runs": run_count,
                "total": run_count
            },
            "duration_ms": duration_ms
        }
    except Exception as e:
        logger.error("Failed to delete orchestration runs", error=str(e), exc_info=True)
        await db.rollback()
        raise HTTPException(status_code=500, detail=str(e))


@router.get("/internal/demo/clone/health")
async def health_check(_: bool = Depends(verify_internal_api_key)):
    """Health check for demo cloning endpoint"""
    return {"status": "healthy", "service": "orchestrator"}