from typing import Annotated

from fastapi import APIRouter, Body, Depends
from fastapi.responses import JSONResponse
from redis.asyncio.client import Redis

from app.common.request import PaginationService
from app.common.response import ErrorResponse, ResponseSchema, SuccessResponse
from app.core.base_params import PaginationQueryParam
from app.core.dependencies import AuthPermission, redis_getter
from app.core.logger import log
from app.core.router_class import OperationLogRoute

from .schema import OnlineOutSchema, OnlineQueryParam
from .service import OnlineService

OnlineRouter = APIRouter(route_class=OperationLogRoute, prefix="/online", tags=["在线用户"])


@OnlineRouter.get(
    "/list",
    dependencies=[Depends(AuthPermission(["module_monitor:online:query"]))],
    summary="获取在线用户列表",
    description="获取在线用户列表",
    response_model=ResponseSchema[list[OnlineOutSchema]],
)
async def get_online_list_controller(
    redis: Annotated[Redis, Depends(redis_getter)],
    paging_query: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[OnlineQueryParam, Depends()],
) -> JSONResponse:
    """
    获取在线用户列表

    参数:
    - redis (Redis): Redis异步客户端实例。
    - paging_query (PaginationQueryParam): 分页查询参数模型。
    - search (OnlineQueryParam): 查询参数模型。

    返回:
    - JSONResponse: 包含在线用户列表的JSON响应。
    """
    result_dict_list = await OnlineService.get_online_list_service(redis=redis, search=search)
    # 在线用户来自 Redis，无法在数据库层 OFFSET/LIMIT
    result_dict = await PaginationService.paginate(
        data_list=result_dict_list,
        page_no=paging_query.page_no,
        page_size=paging_query.page_size,
    )
    log.info("获取成功")

    return SuccessResponse(data=result_dict, msg="获取成功")


@OnlineRouter.delete(
    "/delete",
    dependencies=[Depends(AuthPermission(["module_monitor:online:delete"]))],
    summary="强制下线",
    description="强制下线",
    response_model=ResponseSchema[None],
)
async def delete_online_controller(
    session_id: Annotated[str, Body(description="会话编号")],
    redis: Annotated[Redis, Depends(redis_getter)],
) -> JSONResponse:
    """
    强制下线指定在线用户

    参数:
    - session_id (str): 在线用户会话ID。
    - redis (Redis): Redis异步客户端实例。

    返回:
    - JSONResponse: 包含操作结果的JSON响应。
    """
    is_ok = await OnlineService.delete_online_service(redis=redis, session_id=session_id)
    if is_ok:
        log.info("强制下线成功")
        return SuccessResponse(msg="强制下线成功")
    log.info("强制下线失败")
    return ErrorResponse(msg="强制下线失败")


@OnlineRouter.delete(
    "/clear",
    dependencies=[Depends(AuthPermission(["module_monitor:online:delete"]))],
    summary="清除所有在线用户",
    description="清除所有在线用户",
    response_model=ResponseSchema[None],
)
async def clear_online_controller(
    redis: Annotated[Redis, Depends(redis_getter)],
) -> JSONResponse:
    """
    清除所有在线用户

    参数:
    - redis (Redis): Redis异步客户端实例。

    返回:
    - JSONResponse: 包含操作结果的JSON响应。
    """
    is_ok = await OnlineService.clear_online_service(redis=redis)
    if is_ok:
        log.info("清除所有在线用户成功")
        return SuccessResponse(msg="清除所有在线用户成功")
    log.info("清除所有在线用户失败")
    return ErrorResponse(msg="清除所有在线用户失败")