70 lines
2 KiB
Python
70 lines
2 KiB
Python
from fastapi import APIRouter, Depends, HTTPException, status
|
|
|
|
from adapter.di.container import AppContainer
|
|
from adapter.http.fastapi.dependencies import (
|
|
get_container,
|
|
get_create_sandbox,
|
|
)
|
|
from adapter.http.fastapi.schemas import (
|
|
CreateSandboxRequest,
|
|
ErrorResponse,
|
|
HealthResponse,
|
|
SandboxSessionResponse,
|
|
)
|
|
from domain.error import SandboxError, SandboxStartError
|
|
from domain.sandbox import SandboxSession
|
|
from usecase.sandbox import CreateSandbox, CreateSandboxCommand
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get(
|
|
'/health',
|
|
response_model=HealthResponse,
|
|
status_code=status.HTTP_200_OK,
|
|
)
|
|
def health(container: AppContainer = Depends(get_container)) -> HealthResponse:
|
|
return HealthResponse(
|
|
status='ok',
|
|
app=container.config.app.name,
|
|
env=container.config.app.env,
|
|
)
|
|
|
|
|
|
@router.post(
|
|
'/create',
|
|
response_model=SandboxSessionResponse,
|
|
responses={
|
|
status.HTTP_503_SERVICE_UNAVAILABLE: {'model': ErrorResponse},
|
|
status.HTTP_500_INTERNAL_SERVER_ERROR: {'model': ErrorResponse},
|
|
},
|
|
status_code=status.HTTP_200_OK,
|
|
)
|
|
def create_sandbox(
|
|
request: CreateSandboxRequest,
|
|
usecase: CreateSandbox = Depends(get_create_sandbox),
|
|
) -> SandboxSessionResponse:
|
|
try:
|
|
session = usecase.execute(CreateSandboxCommand(chat_id=request.chat_id))
|
|
except SandboxStartError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_503_SERVICE_UNAVAILABLE,
|
|
detail=str(exc),
|
|
) from exc
|
|
except SandboxError as exc:
|
|
raise HTTPException(
|
|
status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
|
|
detail=str(exc),
|
|
) from exc
|
|
|
|
return _to_sandbox_session_response(session)
|
|
|
|
|
|
def _to_sandbox_session_response(session: SandboxSession) -> SandboxSessionResponse:
|
|
return SandboxSessionResponse(
|
|
session_id=session.session_id,
|
|
chat_id=session.chat_id,
|
|
container_id=session.container_id,
|
|
status=session.status.value,
|
|
expires_at=session.expires_at,
|
|
)
|