python_market_data_service/app/api/admin_routes.py

"""管理后台API路由 - 对应Go的api/admin_router.go"""
from fastapi import APIRouter, Depends, HTTPException, Header, Query
from typing import Optional

from app.models import (
    Response, ConfigListRequest, ConfigUpdateRequest,
    ReloadRequest, AdapterToggleRequest, AdapterConfigUpdateRequest,
    APITestRequest, WSTestRequest, TestHistoryRequest
)
from app.services import ConfigService, AdapterService, TestService
from app.core.config import get_config

admin_router = APIRouter()

# 服务实例
config_service = ConfigService()
adapter_service = AdapterService()
test_service = TestService()


def verify_admin_token(x_admin_token: Optional[str] = Header(None)):
    """验证Admin Token"""
    # TODO: 实现Token验证
    return x_admin_token


# ============================================
# 系统管理接口
# ============================================

@admin_router.get("/admin/system/status", response_model=Response)
def get_system_status(
    token: str = Depends(verify_admin_token)
):
    """获取系统状态"""
    try:
        data = config_service.get_system_status()
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.post("/admin/system/reload", response_model=Response)
def reload_config(
    req: Optional[ReloadRequest] = None,
    token: str = Depends(verify_admin_token)
):
    """热加载配置"""
    try:
        if req is None:
            req = ReloadRequest()
        data = config_service.reload_config(req)
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.post("/admin/system/restart", response_model=Response)
def restart_service(
    token: str = Depends(verify_admin_token)
):
    """重启服务"""
    # TODO: 实现服务重启逻辑
    return Response(
        code=0,
        message="重启命令已发送",
        data={"status": "restarting"}
    )


# ============================================
# 配置管理接口
# ============================================

@admin_router.get("/admin/config", response_model=Response)
def get_config_list(
    type: Optional[str] = Query(None, description="配置类型筛选"),
    token: str = Depends(verify_admin_token)
):
    """获取配置列表"""
    try:
        from app.models import ConfigType
        req = ConfigListRequest()
        if type:
            req.type = ConfigType(type)
        
        data = config_service.get_config_list(req)
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.put("/admin/config", response_model=Response)
def update_config(
    req: ConfigUpdateRequest,
    token: str = Depends(verify_admin_token)
):
    """更新配置"""
    try:
        data = config_service.update_config(req)
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.post("/admin/config/reload", response_model=Response)
def reload_config_endpoint(
    req: Optional[ReloadRequest] = None,
    token: str = Depends(verify_admin_token)
):
    """热加载配置"""
    return reload_config(req, token)


# ============================================
# 适配器管理接口
# ============================================

@admin_router.get("/admin/adapters", response_model=Response)
def get_adapter_list(
    token: str = Depends(verify_admin_token)
):
    """获取适配器列表"""
    try:
        data = adapter_service.get_adapter_list()
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.post("/admin/adapters/toggle", response_model=Response)
def toggle_adapter(
    req: AdapterToggleRequest,
    token: str = Depends(verify_admin_token)
):
    """切换适配器状态"""
    try:
        adapter_service.toggle_adapter(req)
        return Response(code=0, message="success")
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.put("/admin/adapters/config", response_model=Response)
def update_adapter_config(
    req: AdapterConfigUpdateRequest,
    token: str = Depends(verify_admin_token)
):
    """更新适配器配置"""
    try:
        adapter_service.update_adapter_config(req)
        return Response(code=0, message="success")
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


# ============================================
# 测试管理接口
# ============================================

@admin_router.get("/admin/tests/api", response_model=Response)
def get_api_test_list(
    token: str = Depends(verify_admin_token)
):
    """获取API测试列表"""
    try:
        data = test_service.get_api_test_list()
        # 设置基础URL
        config = get_config()
        data.base_url = f"http://localhost:{config.server.port}"
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.post("/admin/tests/api/run", response_model=Response)
async def run_api_test(
    req: APITestRequest,
    token: str = Depends(verify_admin_token)
):
    """执行API测试"""
    try:
        config = get_config()
        base_url = f"http://localhost:{config.server.port}"
        data = await test_service.run_api_test(base_url, req)
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.get("/admin/tests/ws", response_model=Response)
def get_ws_test_list(
    token: str = Depends(verify_admin_token)
):
    """获取WebSocket测试列表"""
    try:
        data = test_service.get_ws_test_list()
        config = get_config()
        data.ws_url = f"ws://localhost:{config.server.port}/v1/stream"
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.post("/admin/tests/ws/run", response_model=Response)
async def run_ws_test(
    req: WSTestRequest,
    token: str = Depends(verify_admin_token)
):
    """执行WebSocket测试"""
    try:
        config = get_config()
        ws_url = f"ws://localhost:{config.server.port}/v1/stream"
        data = await test_service.run_ws_test(ws_url, req)
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))


@admin_router.get("/admin/tests/history", response_model=Response)
def get_test_history(
    type: Optional[str] = Query(None, description="测试类型"),
    limit: int = Query(default=20, ge=1, le=100, description="数量限制"),
    token: str = Depends(verify_admin_token)
):
    """获取测试历史"""
    try:
        req = TestHistoryRequest(type=type, limit=limit)
        data = test_service.get_test_history(req)
        return Response(code=0, message="success", data=data)
    except Exception as e:
        raise HTTPException(status_code=500, detail=str(e))