FastapiAdmin/backend/app/api/v1/module_system/notice/controller.py

from typing import Annotated

from fastapi import APIRouter, Body, Depends, Path
from fastapi.responses import JSONResponse, StreamingResponse
from fastapi_cache import FastAPICache
from fastapi_cache.decorator import cache

from app.common.response import ResponseSchema, StreamResponse, SuccessResponse
from app.core.base_params import PaginationQueryParam
from app.core.base_schema import AuthSchema, BatchSetAvailable, PageResultSchema
from app.core.dependencies import AuthPermission, get_current_user
from app.core.logger import logger
from app.core.router_class import OperationLogRoute
from app.utils.common_util import bytes2file_response

from .schema import (
    NoticeCreateSchema,
    NoticeOutSchema,
    NoticeQueryParam,
    NoticeUpdateSchema,
    PanelDataOut,
)
from .service import NoticeService

NoticeRouter = APIRouter(route_class=OperationLogRoute, prefix="/notice", tags=["系统管理/公告通知"])

_NOTICE_NS = "notice"


@NoticeRouter.get(
    "/detail/{id}",
    summary="获取公告详情",
    response_model=ResponseSchema[NoticeOutSchema],
)
async def get_obj_detail_controller(
    id: Annotated[int, Path(description="公告ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(['module_system:notice:detail']))],
) -> JSONResponse:
    """
    获取公告详情。

    参数:
    - id (int): 公告ID。
    - auth (AuthSchema): 认证信息模型。

    返回:
    - JSONResponse: 包含公告详情的响应模型。
    """
    result_dict = await NoticeService.get_notice_detail_service(id=id, auth=auth)
    return SuccessResponse(data=result_dict, msg="获取公告详情成功")


@NoticeRouter.get(
    "/list",
    summary="查询公告",
    response_model=ResponseSchema[PageResultSchema[NoticeOutSchema]],
)
async def get_obj_list_controller(
    page: Annotated[PaginationQueryParam, Depends()],
    search: Annotated[NoticeQueryParam, Depends()],
    auth: Annotated[AuthSchema, Depends(AuthPermission(['module_system:notice:query']))],
) -> JSONResponse:
    """
    查询公告。

    参数:
    - page (PaginationQueryParam): 分页查询参数模型。
    - search (NoticeQueryParam): 查询公告参数模型。
    - auth (AuthSchema): 认证信息模型。

    返回:
    - JSONResponse: 包含分页公告详情的响应模型。
    """
    result_dict = await NoticeService.get_notice_page_service(
        auth=auth,
        page_no=page.page_no,
        page_size=page.page_size,
        search=search,
        order_by=page.order_by,
    )
    return SuccessResponse(data=result_dict, msg="查询公告列表成功")


@NoticeRouter.post(
    "/create",
    summary="创建公告",
    response_model=ResponseSchema[NoticeOutSchema],
)
async def create_obj_controller(
    data: NoticeCreateSchema,
    auth: Annotated[AuthSchema, Depends(AuthPermission(['module_system:notice:create']))],
) -> JSONResponse:
    """
    创建公告。

    参数:
    - data (NoticeCreateSchema): 创建公告负载模型。
    - auth (AuthSchema): 认证信息模型。

    返回:
    - JSONResponse: 包含创建公告结果的响应模型。
    """
    result_dict = await NoticeService.create_notice_service(auth=auth, data=data)
    await FastAPICache.clear(namespace=_NOTICE_NS)
    return SuccessResponse(data=result_dict, msg="创建公告成功")


@NoticeRouter.put(
    "/update/{id}",
    summary="修改公告",
    response_model=ResponseSchema[NoticeOutSchema],
)
async def update_obj_controller(
    data: NoticeUpdateSchema,
    id: Annotated[int, Path(description="公告ID")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(['module_system:notice:update']))],
) -> JSONResponse:
    """
    修改公告。

    参数:
    - data (NoticeUpdateSchema): 修改公告负载模型。
    - id (int): 公告ID。
    - auth (AuthSchema): 认证信息模型。

    返回:
    - JSONResponse: 包含修改公告结果的响应模型。
    """
    result_dict = await NoticeService.update_notice_service(auth=auth, id=id, data=data)
    await FastAPICache.clear(namespace=_NOTICE_NS)
    return SuccessResponse(data=result_dict, msg="修改公告成功")


@NoticeRouter.delete(
    "/delete",
    summary="删除公告",
    response_model=ResponseSchema[None],
)
async def delete_obj_controller(
    ids: Annotated[list[int], Body(description="ID列表")],
    auth: Annotated[AuthSchema, Depends(AuthPermission(['module_system:notice:delete']))],
) -> JSONResponse:
    """
    删除公告。

    参数:
    - ids (list[int]): 公告ID列表。
    - auth (AuthSchema): 认证信息模型。

    返回:
    - JSONResponse: 包含删除公告结果的响应模型。
    """
    await NoticeService.delete_notice_service(auth=auth, ids=ids)
    await FastAPICache.clear(namespace=_NOTICE_NS)
    return SuccessResponse(msg="删除公告成功")


@NoticeRouter.patch(
    "/status/batch",
    summary="批量修改公告状态",
    response_model=ResponseSchema[None],
)
async def batch_set_available_obj_controller(
    data: BatchSetAvailable,
    auth: Annotated[AuthSchema, Depends(AuthPermission(['module_system:notice:patch']))],
) -> JSONResponse:
    """
    批量修改公告状态。

    参数:
    - data (BatchSetAvailable): 批量修改公告状态负载模型。
    - auth (AuthSchema): 认证信息模型。

    返回:
    - JSONResponse: 包含批量修改公告状态结果的响应模型。
    """
    await NoticeService.set_notice_available_service(auth=auth, data=data)
    await FastAPICache.clear(namespace=_NOTICE_NS)
    return SuccessResponse(msg="批量修改公告状态成功")


@NoticeRouter.get(
    "/export",
    summary="导出公告",
    response_model=ResponseSchema[None],
)
async def export_obj_list_controller(
    search: Annotated[NoticeQueryParam, Depends()],
    auth: Annotated[AuthSchema, Depends(AuthPermission(['module_system:notice:export']))],
) -> StreamingResponse:
    """
    导出公告。

    参数:
    - search (NoticeQueryParam): 查询公告参数模型。
    - auth (AuthSchema): 认证信息模型。

    返回:
    - StreamingResponse: 包含导出公告的流式响应模型。
    """
    result_dict_list = await NoticeService.get_notice_list_service(search=search, auth=auth)
    export_data = [item.model_dump() for item in result_dict_list]
    export_result = await NoticeService.export_notice_service(notice_list=export_data)

    return StreamResponse(
        data=bytes2file_response(export_result),
        media_type="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
        headers={"Content-Disposition": "attachment; filename=notice.xlsx"},
    )


@NoticeRouter.get(
    "/available",
    summary="获取全局启用公告",
    response_model=ResponseSchema[list[NoticeOutSchema]],
)
@cache(expire=120, namespace=_NOTICE_NS)
async def get_obj_list_available_controller(
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    """
    获取全局启用公告。

    参数:
    - auth (AuthSchema): 认证信息模型。

    返回:
    - JSONResponse: 包含分页已启用公告详情的响应模型。
    """
    result_dict = await NoticeService.get_notice_available_page_service(auth=auth)
    return SuccessResponse(data=result_dict, msg="查询已启用公告列表成功")


@NoticeRouter.get(
    "/panel",
    summary="通知面板数据（铃铛）",
    response_model=ResponseSchema[PanelDataOut],
)
@cache(expire=30, namespace=_NOTICE_NS)
async def get_notification_panel_controller(
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    """通知面板聚合接口，返回通知、消息、待办三个列表。"""
    result = await NoticeService.get_panel_data_service(auth=auth)
    return SuccessResponse(data=result, msg="获取面板数据成功")

# ============ 已读追踪端点 ============


@NoticeRouter.post(
    "/read/{id}",
    summary="标记已读",
    response_model=ResponseSchema[None],
)
async def mark_read_controller(
    id: Annotated[int, Path(description="通知ID")],
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    """标记已读。通过 `sys_notice_read` 表记录已读时间。"""
    await NoticeService.mark_read_service(auth=auth, notice_id=id)
    await FastAPICache.clear(namespace=_NOTICE_NS)
    logger.info(f"用户[{auth.user.id}]标记通知[{id}]已读")
    return SuccessResponse(msg="标记已读成功")


@NoticeRouter.post(
    "/read-all",
    summary="全部已读",
    response_model=ResponseSchema[int],
)
async def mark_all_read_controller(
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    """全部标记已读。返回本次操作标记的数量。"""
    count = await NoticeService.mark_all_read_service(auth=auth)
    await FastAPICache.clear(namespace=_NOTICE_NS)
    logger.info(f"用户[{auth.user.id}]全部已读, 数量={count}")
    return SuccessResponse(data=count, msg=f"全部标记已读成功，共标记 {count} 条")


@NoticeRouter.get(
    "/unread-count",
    summary="获取未读数量",
    response_model=ResponseSchema[int],
)
@cache(expire=15, namespace=_NOTICE_NS)
async def get_unread_count_controller(
    auth: Annotated[AuthSchema, Depends(get_current_user)],
) -> JSONResponse:
    """获取未读通知数量。通过 LEFT JOIN 统计未读数。"""
    count = await NoticeService.get_unread_count_service(auth=auth)
    return SuccessResponse(data=count, msg="获取未读数量成功")