from __future__ import annotations import logging from datetime import datetime from fastapi import APIRouter, Depends, HTTPException from pydantic import BaseModel, Field from yuxi.channel.container import ChannelContainer, get_channel from yuxi.channel.domain.exception.duplicate_binding import DuplicateBindingException from yuxi.channel.domain.exception.invalid_agent_config import InvalidAgentConfigException from yuxi.channel.interfaces.rest.auth.depends import channel_auth_depends logger = logging.getLogger(__name__) router = APIRouter() class BindingCreateRequest(BaseModel): channel_type: str = Field(..., min_length=1, max_length=20) account_id: str = Field(..., min_length=1, max_length=64) group_id: str = Field("", max_length=128) agent_config_id: int = Field(..., gt=0) class BindingUpdateRequest(BaseModel): agent_config_id: int | None = None is_enabled: bool | None = None class BindingResponse(BaseModel): id: int channel_type: str account_id: str group_id: str agent_config_id: int is_enabled: bool created_at: datetime class PaginatedMeta(BaseModel): total: int offset: int limit: int class BindingListResponse(BaseModel): items: list[BindingResponse] meta: PaginatedMeta @router.post("/channel/bindings", response_model=BindingResponse, status_code=201) async def create_binding( req: BindingCreateRequest, channel: ChannelContainer = Depends(get_channel), _: bool = Depends(channel_auth_depends), ): try: created = await channel.binding_service.create( channel_type=req.channel_type, account_id=req.account_id, group_id=req.group_id, agent_config_id=req.agent_config_id, ) except DuplicateBindingException as e: raise HTTPException(status_code=409, detail=str(e)) except InvalidAgentConfigException as e: raise HTTPException(status_code=400, detail=str(e)) return BindingResponse( id=created.id, channel_type=created.channel_type, account_id=created.account_id, group_id=created.group_id, agent_config_id=created.agent_config_id, is_enabled=created.is_enabled, created_at=created.created_at, ) @router.get("/channel/bindings", response_model=BindingListResponse) async def list_bindings( channel_type: str | None = None, offset: int = 0, limit: int = 50, channel: ChannelContainer = Depends(get_channel), _: bool = Depends(channel_auth_depends), ): bindings, total = await channel.binding_service.list( channel_type=channel_type, offset=offset, limit=limit, ) return BindingListResponse( items=[ BindingResponse( id=b.id, channel_type=b.channel_type, account_id=b.account_id, group_id=b.group_id, agent_config_id=b.agent_config_id, is_enabled=b.is_enabled, created_at=b.created_at, ) for b in bindings ], meta=PaginatedMeta(total=total, offset=offset, limit=limit), ) @router.patch("/channel/bindings/{binding_id}", response_model=BindingResponse) async def update_binding( binding_id: int, req: BindingUpdateRequest, channel: ChannelContainer = Depends(get_channel), _: bool = Depends(channel_auth_depends), ): if req.agent_config_id is None and req.is_enabled is None: raise HTTPException(status_code=400, detail="no fields to update") updated = await channel.binding_service.update( binding_id, agent_config_id=req.agent_config_id, is_enabled=req.is_enabled ) if not updated: raise HTTPException(status_code=404, detail="binding not found") return BindingResponse( id=updated.id, channel_type=updated.channel_type, account_id=updated.account_id, group_id=updated.group_id, agent_config_id=updated.agent_config_id, is_enabled=updated.is_enabled, created_at=updated.created_at, ) @router.delete("/channel/bindings/{binding_id}") async def delete_binding( binding_id: int, channel: ChannelContainer = Depends(get_channel), _: bool = Depends(channel_auth_depends), ): deleted = await channel.binding_service.delete(binding_id) if not deleted: raise HTTPException(status_code=404, detail="binding not found") return {"ok": True}