# +----------------------------------------------------------------------
# | WaitAdmin(fastapi)快速开发后台管理系统
# +----------------------------------------------------------------------
# | 欢迎阅读学习程序代码,建议反馈是我们前进的动力
# | 程序完全开源可支持商用,允许去除界面版权信息
# | gitee:   https://gitee.com/wafts/waitadmin-python
# | github:  https://github.com/topwait/waitadmin-python
# | 官方网站: https://www.waitadmin.cn
# | WaitAdmin团队版权所有并拥有最终解释权
# +----------------------------------------------------------------------
# | Author: WaitAdmin Team <2474369941@qq.com>
# +----------------------------------------------------------------------
import math
import inspect
from functools import wraps
from typing import Generic, TypeVar, Sequence, Callable, Any
from fastapi.encoders import jsonable_encoder
from fastapi.responses import JSONResponse
from pydantic import BaseModel, Field
from common.enums.errors import ErrorEnum

T = TypeVar("T")


class R(BaseModel, Generic[T]):
    """ 响应工具类 """
    code: int = Field(default=ErrorEnum.SUCCESS.code, description="状态")
    msg: str = Field(default=ErrorEnum.SUCCESS.msg, description="提示")
    data: T = Field(default_factory=list, description="数据")

    @classmethod
    def success(cls, msg: str = "OK", data: Any = None, code: int = 0):
        if data is None:
            data = []
        return cls(code=code, msg=msg, data=data)

    @classmethod
    def fail(cls, msg: str = "FAIL", data: Any = None, code: int = 1):
        if data is None:
            data = []
        return cls(code=code, msg=msg, data=data)


class PagingResult(BaseModel, Generic[T]):
    """ 分页结果封装 """
    current_page: int = Field(default=1, description="当前页码")
    last_page: int = Field(default=1, description="最后页码")
    per_page: int = Field(default=20, description="每页数据")
    total: int = Field(default=1, description="数据总量")
    extend: Sequence[T] = Field(default=[], description="扩展信息")
    lists: Sequence[T] = Field(description="分页数据")

    @classmethod
    def create(cls, data: Sequence[T], total: int, page_no: int, page_size: int) -> "PagingResult[T]":
        last_page = math.ceil(total / page_size)
        return cls(lists=data, total=total, current_page=page_no, per_page=page_size, last_page=last_page)


def response_json(func: Callable[..., T]) -> Callable[..., T]:
    """ 统一响应格式 """
    @wraps(func)
    async def wrapper(*args, **kwargs) -> T:
        if inspect.iscoroutinefunction(func):
            resp = await func(*args, **kwargs) or []
        else:
            resp = func(*args, **kwargs) or []
        return JSONResponse(
            content=jsonable_encoder(
                R.success(data=resp).dict(),
                by_alias=False
            ),
            media_type="application/json;charset=utf-8"
        )

    return wrapper