93 lines
3.5 KiB
Python
93 lines
3.5 KiB
Python
from __future__ import annotations
|
|
|
|
from fastapi import APIRouter, Depends, File, HTTPException, Request, UploadFile, status
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core.database.session import get_session
|
|
from app.models import Action, ActionIngestStatus, User
|
|
from app.schemas.capability_sch import ActionIngestWithCapabilitiesResponse
|
|
from app.services.capability_service import CapabilityService
|
|
from app.services.openapi_service import OpenAPIService
|
|
from app.utils.business_logger import log_business_event
|
|
from app.utils.token_manager import get_current_user
|
|
|
|
|
|
router = APIRouter(tags=["Actions"])
|
|
|
|
|
|
@router.post("/ingest", response_model=ActionIngestWithCapabilitiesResponse, status_code=status.HTTP_201_CREATED)
|
|
async def ingest_actions(
|
|
request: Request,
|
|
file: UploadFile = File(...),
|
|
session: AsyncSession = Depends(get_session),
|
|
current_user: User = Depends(get_current_user),
|
|
):
|
|
trace_id = getattr(request.state, "traceId", None)
|
|
payload = await file.read()
|
|
try:
|
|
document = OpenAPIService.load_document(payload)
|
|
ingestion_result = OpenAPIService.extract_actions_with_failures(document, source_filename=file.filename)
|
|
except ValueError as exc:
|
|
log_business_event(
|
|
"actions_ingest_rejected",
|
|
trace_id=trace_id,
|
|
user_id=str(current_user.id),
|
|
source_filename=file.filename,
|
|
file_size_bytes=len(payload),
|
|
reason="invalid_openapi_document",
|
|
details=str(exc),
|
|
)
|
|
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail=str(exc)) from exc
|
|
|
|
action_payloads = ingestion_result["succeeded"] + ingestion_result["failed"]
|
|
if not action_payloads:
|
|
log_business_event(
|
|
"actions_ingest_rejected",
|
|
trace_id=trace_id,
|
|
user_id=str(current_user.id),
|
|
source_filename=file.filename,
|
|
file_size_bytes=len(payload),
|
|
reason="no_supported_operations",
|
|
)
|
|
raise HTTPException(status_code=status.HTTP_400_BAD_REQUEST, detail="No supported HTTP operations found in OpenAPI file")
|
|
|
|
actions = [Action(user_id=current_user.id, **action_payload) for action_payload in action_payloads]
|
|
session.add_all(actions)
|
|
await session.flush()
|
|
|
|
succeeded_actions = [action for action in actions if action.ingest_status == ActionIngestStatus.SUCCEEDED]
|
|
failed_actions = [action for action in actions if action.ingest_status == ActionIngestStatus.FAILED]
|
|
|
|
capability_service = CapabilityService(session)
|
|
capabilities = await capability_service.create_from_actions(
|
|
succeeded_actions,
|
|
owner_user_id=current_user.id,
|
|
refresh=False,
|
|
)
|
|
await session.commit()
|
|
|
|
for action in actions:
|
|
await session.refresh(action)
|
|
for capability in capabilities:
|
|
await session.refresh(capability)
|
|
|
|
log_business_event(
|
|
"actions_ingested",
|
|
trace_id=trace_id,
|
|
user_id=str(current_user.id),
|
|
source_filename=file.filename,
|
|
file_size_bytes=len(payload),
|
|
succeeded_count=len(succeeded_actions),
|
|
failed_count=len(failed_actions),
|
|
created_capabilities_count=len(capabilities),
|
|
)
|
|
|
|
return ActionIngestWithCapabilitiesResponse(
|
|
succeeded_count=len(succeeded_actions),
|
|
failed_count=len(failed_actions),
|
|
created_capabilities_count=len(capabilities),
|
|
succeeded_actions=succeeded_actions,
|
|
failed_actions=failed_actions,
|
|
capabilities=capabilities,
|
|
)
|