AstrBot/astrbot/dashboard/api/platform.py

from __future__ import annotations

from typing import Any

from fastapi import APIRouter, Depends, Request

from astrbot.dashboard.asgi_runtime import DashboardRequest
from astrbot.dashboard.async_utils import run_maybe_async
from astrbot.dashboard.responses import ApiError, ok
from astrbot.dashboard.schemas import BotRegistrationRequest
from astrbot.dashboard.services.platform_service import (
    PlatformService,
    PlatformServiceError,
)

from .auth import AuthContext, require_dashboard_user, require_scope

router = APIRouter(tags=["Platforms"])
legacy_router = APIRouter(
    prefix="/api/platform",
    tags=["Dashboard Platforms"],
    include_in_schema=False,
)


def get_service(request: Request) -> PlatformService:
    return request.app.state.services.platforms


async def require_config_scope(request: Request) -> AuthContext:
    return await require_scope(request, "config")


async def _json_or_empty(request: Request) -> dict[str, Any]:
    try:
        data = await request.json()
    except Exception:
        return {}
    return data if isinstance(data, dict) else {}


def _raise_platform_error(exc: PlatformServiceError) -> None:
    raise ApiError(str(exc), status_code=exc.status_code) from exc


def _model_dict(payload) -> dict[str, Any]:
    return payload.model_dump(exclude_none=True)


async def _run(operation):
    try:
        result = await run_maybe_async(operation)
        return ok(result)
    except PlatformServiceError as exc:
        _raise_platform_error(exc)


@router.post("/bot-types/{bot_type}/registration")
async def register_bot_type(
    bot_type: str,
    payload: BotRegistrationRequest,
    _auth: AuthContext = Depends(require_config_scope),
    service: PlatformService = Depends(get_service),
):
    return await _run(
        lambda: service.handle_platform_registration(bot_type, _model_dict(payload))
    )


@router.get("/webhooks/platforms/{webhook_uuid}")
async def verify_platform_webhook(
    webhook_uuid: str,
    request: Request,
    service: PlatformService = Depends(get_service),
):
    return await _run(
        lambda: service.handle_webhook_callback(webhook_uuid, DashboardRequest(request))
    )


@router.post("/webhooks/platforms/{webhook_uuid}")
async def receive_platform_webhook(
    webhook_uuid: str,
    request: Request,
    service: PlatformService = Depends(get_service),
):
    return await _run(
        lambda: service.handle_webhook_callback(webhook_uuid, DashboardRequest(request))
    )


@legacy_router.api_route("/webhook/{webhook_uuid}", methods=["GET", "POST"])
async def dashboard_platform_webhook(
    webhook_uuid: str,
    request: Request,
    service: PlatformService = Depends(get_service),
):
    return await _run(
        lambda: service.handle_webhook_callback(webhook_uuid, DashboardRequest(request))
    )


@legacy_router.get("/stats")
async def get_dashboard_platform_stats(
    _username: str = Depends(require_dashboard_user),
    service: PlatformService = Depends(get_service),
):
    return await _run(service.get_platform_stats)


@legacy_router.post("/registration/{platform_type}")
async def handle_dashboard_platform_registration(
    platform_type: str,
    request: Request,
    _username: str = Depends(require_dashboard_user),
    service: PlatformService = Depends(get_service),
):
    payload = await _json_or_empty(request)
    return await _run(
        lambda: service.handle_platform_registration(platform_type, payload)
    )