"""外部系统限界上下文 - IntegrationTool 子域 Router。 挂载到聚合 router 的 /integration-tools 前缀下,覆盖厂商集成(PRD FR-07/08) 的工具生成流程编排(list / discover / preview / generate / persist)。 Request Schema 与 Input DTO 不共享类,Router 内显式构造 DTO, 操作人字段由 current_user.uid 填充。 静态路径优先:/discover / /preview / /generate 必须在 POST ""(persist)之前声明, 避免被空路径匹配。 discover / preview / generate 三个端点请求体字段完全一致,共用 ``IntegrationToolFlowRequest``,对应 use_cases 层 ``IntegrationToolFlowInput``。 操作语义由端点路径与服务方法名区分。 """ from __future__ import annotations from typing import Any from fastapi import APIRouter, Depends, Query from pydantic import BaseModel, ConfigDict, Field from sqlalchemy.ext.asyncio import AsyncSession from yuxi.external_systems.infrastructure.container import create_use_cases_from_db from yuxi.external_systems.use_cases.dto.tool import ( ExternalToolCreateInput, IntegrationToolFlowInput, ListToolsInput, PersistGeneratedToolsInput, ) from yuxi.storage.postgres.models_business import User from server.utils.auth_middleware import get_admin_user, get_db, get_required_user integration_tool_router = APIRouter( prefix="/integration-tools", tags=["external-systems-integration-tool"], ) # ============================================================================= # === Request Schemas(与 Input DTO 不共享类) === # ============================================================================= class ExternalToolItemRequest(BaseModel): """外部工具项请求体。字段对齐 ``ExternalToolCreateInput``。 用于持久化集成生成工具(``PersistIntegrationToolsRequest.tools``)。 """ model_config = ConfigDict(frozen=True) slug: str = Field(..., min_length=1, max_length=128, pattern=r"^[a-zA-Z_][a-zA-Z0-9_-]{0,127}$") name: str = Field(..., min_length=1, max_length=128) description: str = Field(..., min_length=1) category: str = Field(default="default", max_length=64) adapter_type: str = Field(default="http", max_length=32) enabled: bool = True timeout: int = Field(default=30, ge=1, le=300) retry_policy: dict[str, Any] = Field(default_factory=dict) auth_type: str = Field(default="none", max_length=32) auth_config: dict[str, Any] = Field(default_factory=dict) adapter_config: dict[str, Any] = Field(default_factory=dict) class PersistIntegrationToolsRequest(BaseModel): """持久化集成生成工具请求体。字段对齐 ``PersistGeneratedToolsInput``(不含 created_by)。""" model_config = ConfigDict(frozen=True) system_id: int tools: list[ExternalToolItemRequest] = Field(..., min_length=1) class IntegrationToolFlowRequest(BaseModel): """集成工具生成流程请求体(discover / preview / generate 共用)。 字段对齐 ``IntegrationToolFlowInput``。三个操作的请求参数完全一致, 共用一个 Schema,操作语义由端点路径区分。 ``source_type`` 为空时由 use_cases 层按 ``system.source_type`` 兜底解析。 ``env_key`` 为空时由 use_cases 层解析系统默认环境。 ``discovery_options`` 为用户选择,透传到 ``system_config``,各厂商字段名不同。 """ model_config = ConfigDict(frozen=True) system_id: int source_type: str | None = None env_key: str | None = None discovery_options: dict[str, Any] | None = None # ============================================================================= # === Endpoints === # ============================================================================= @integration_tool_router.get("", response_model=dict) async def list_generated_tools( limit: int = Query(100, ge=1, le=500), offset: int = Query(0, ge=0), system_id: int | None = Query(None), category: str | None = Query(None, max_length=64), adapter_type: str | None = Query(None, max_length=32), enabled: bool | None = Query(None), keyword: str | None = Query(None, max_length=128), db: AsyncSession = Depends(get_db), current_user: User = Depends(get_required_user), ) -> dict[str, Any]: """分页列出已生成的集成工具。 与 ``tool_router.GET /tools`` 功能等价,保持 Router 边界清晰: ``integration_tool_router`` 仅依赖 ``integration_tool_service`` 端口。 ``limit`` / ``offset`` 内部转换为 ``page`` / ``page_size``。 """ use_cases = create_use_cases_from_db(db) input_dto = ListToolsInput( page=offset // limit + 1, page_size=limit, system_id=system_id, category=category, adapter_type=adapter_type, enabled=enabled, keyword=keyword, ) output = await use_cases.integration_tool_service.list_generated_tools(input_dto) return {"success": True, "data": output.model_dump()} @integration_tool_router.post("/discover", response_model=dict) async def discover_resources( body: IntegrationToolFlowRequest, db: AsyncSession = Depends(get_db), current_user: User = Depends(get_admin_user), ) -> dict[str, Any]: """发现外部系统可用资源(调用厂商集成的 ``discover`` handler)。""" use_cases = create_use_cases_from_db(db) input_dto = IntegrationToolFlowInput( system_id=body.system_id, source_type=body.source_type, env_key=body.env_key, discovery_options=body.discovery_options, ) output = await use_cases.integration_tool_service.discover_resources(input_dto) return {"success": True, "data": output.model_dump()} @integration_tool_router.post("/preview", response_model=dict) async def preview_tools( body: IntegrationToolFlowRequest, db: AsyncSession = Depends(get_db), current_user: User = Depends(get_admin_user), ) -> dict[str, Any]: """预览将生成的工具列表(调用厂商集成的 ``preview_tools`` handler)。""" use_cases = create_use_cases_from_db(db) input_dto = IntegrationToolFlowInput( system_id=body.system_id, source_type=body.source_type, env_key=body.env_key, discovery_options=body.discovery_options, ) output = await use_cases.integration_tool_service.preview_tools(input_dto) return {"success": True, "data": output.model_dump()} @integration_tool_router.post("/generate", response_model=dict) async def generate_tools_draft( body: IntegrationToolFlowRequest, db: AsyncSession = Depends(get_db), current_user: User = Depends(get_admin_user), ) -> dict[str, Any]: """生成工具草稿(调用厂商集成的 ``create_tools`` handler,不持久化)。 草稿由前端确认后调用 ``POST /integration-tools``(persist)落库。 """ use_cases = create_use_cases_from_db(db) input_dto = IntegrationToolFlowInput( system_id=body.system_id, source_type=body.source_type, env_key=body.env_key, discovery_options=body.discovery_options, ) output = await use_cases.integration_tool_service.generate_tools_draft(input_dto) return {"success": True, "data": output.model_dump()} @integration_tool_router.post("", response_model=dict) async def persist_generated_tools( body: PersistIntegrationToolsRequest, db: AsyncSession = Depends(get_db), current_user: User = Depends(get_admin_user), ) -> dict[str, Any]: """持久化厂商集成生成的工具草稿。created_by 由服务端以 current_user.uid 填充。""" use_cases = create_use_cases_from_db(db) input_dto = PersistGeneratedToolsInput( system_id=body.system_id, tools=[ExternalToolCreateInput(**item.model_dump()) for item in body.tools], created_by=current_user.uid, ) output = await use_cases.integration_tool_service.persist_generated_tools(input_dto) return {"success": True, "data": output.model_dump()}