smart-support/backend/app/openapi/review_api.py

"""FastAPI router for OpenAPI import review workflow.

Exposes endpoints for:
- Starting an import job
- Querying job status
- Reviewing and editing classifications
- Approving a job to trigger tool generation
"""

from __future__ import annotations

import uuid

from fastapi import APIRouter, HTTPException
from pydantic import BaseModel, field_validator

from app.openapi.models import ClassificationResult

router = APIRouter(prefix="/api/openapi", tags=["openapi"])

# In-memory store: job_id -> job dict
_job_store: dict[str, dict] = {}


# --- Request / Response schemas ---


class ImportRequest(BaseModel):
    url: str

    @field_validator("url")
    @classmethod
    def url_must_not_be_empty(cls, value: str) -> str:
        if not value.strip():
            raise ValueError("url must not be empty")
        return value


class JobResponse(BaseModel):
    job_id: str
    status: str
    spec_url: str
    total_endpoints: int = 0
    classified_count: int = 0
    error_message: str | None = None


class ClassificationResponse(BaseModel):
    index: int
    access_type: str
    needs_interrupt: bool
    agent_group: str
    confidence: float
    customer_params: list[str]
    endpoint: dict


class UpdateClassificationRequest(BaseModel):
    access_type: str
    needs_interrupt: bool
    agent_group: str


# --- Helpers ---


def _job_to_response(job: dict) -> dict:
    return {
        "job_id": job["job_id"],
        "status": job["status"],
        "spec_url": job["spec_url"],
        "total_endpoints": job.get("total_endpoints", 0),
        "classified_count": job.get("classified_count", 0),
        "error_message": job.get("error_message"),
    }


def _classification_to_response(idx: int, clf: ClassificationResult) -> dict:
    ep = clf.endpoint
    return {
        "index": idx,
        "access_type": clf.access_type,
        "needs_interrupt": clf.needs_interrupt,
        "agent_group": clf.agent_group,
        "confidence": clf.confidence,
        "customer_params": list(clf.customer_params),
        "endpoint": {
            "path": ep.path,
            "method": ep.method,
            "operation_id": ep.operation_id,
            "summary": ep.summary,
            "description": ep.description,
        },
    }


# --- Endpoints ---


@router.post("/import", status_code=202)
async def start_import(request: ImportRequest) -> dict:
    """Start an OpenAPI import job for the given spec URL."""
    job_id = str(uuid.uuid4())
    job: dict = {
        "job_id": job_id,
        "status": "pending",
        "spec_url": request.url,
        "total_endpoints": 0,
        "classified_count": 0,
        "error_message": None,
        "classifications": [],
    }
    _job_store[job_id] = job
    return _job_to_response(job)


@router.get("/jobs/{job_id}")
async def get_job(job_id: str) -> dict:
    """Get the status of an import job."""
    job = _job_store.get(job_id)
    if job is None:
        raise HTTPException(status_code=404, detail=f"Job '{job_id}' not found")
    return _job_to_response(job)


@router.get("/jobs/{job_id}/classifications")
async def get_classifications(job_id: str) -> list:
    """Get all classifications for an import job."""
    job = _job_store.get(job_id)
    if job is None:
        raise HTTPException(status_code=404, detail=f"Job '{job_id}' not found")
    classifications: list[ClassificationResult] = job.get("classifications", [])
    return [_classification_to_response(i, clf) for i, clf in enumerate(classifications)]


@router.put("/jobs/{job_id}/classifications/{idx}")
async def update_classification(
    job_id: str,
    idx: int,
    request: UpdateClassificationRequest,
) -> dict:
    """Update a specific classification by index."""
    job = _job_store.get(job_id)
    if job is None:
        raise HTTPException(status_code=404, detail=f"Job '{job_id}' not found")

    classifications: list[ClassificationResult] = job.get("classifications", [])
    if idx < 0 or idx >= len(classifications):
        raise HTTPException(
            status_code=404,
            detail=f"Classification index {idx} out of range",
        )

    original = classifications[idx]
    updated = ClassificationResult(
        endpoint=original.endpoint,
        access_type=request.access_type,
        customer_params=original.customer_params,
        agent_group=request.agent_group,
        confidence=original.confidence,
        needs_interrupt=request.needs_interrupt,
    )
    new_classifications = list(classifications)
    new_classifications[idx] = updated
    _job_store[job_id] = {**job, "classifications": new_classifications}

    return _classification_to_response(idx, updated)


@router.post("/jobs/{job_id}/approve")
async def approve_job(job_id: str) -> dict:
    """Approve a job's classifications and trigger tool generation."""
    job = _job_store.get(job_id)
    if job is None:
        raise HTTPException(status_code=404, detail=f"Job '{job_id}' not found")

    updated_job = {**job, "status": "approved"}
    _job_store[job_id] = updated_job
    return _job_to_response(updated_job)