# -*- coding: utf-8 -*-

from typing import Any, Mapping, Optional
from fastapi import status
from fastapi.responses import JSONResponse, StreamingResponse
from starlette.background import BackgroundTask
from pydantic import Field, BaseModel

from app.common.constant import RET

class ResponseSchema(BaseModel):
    """响应模型"""
    code: int = Field(default=RET.OK.code, description="业务状态码")
    msg: str = Field(default=RET.OK.msg, description="响应消息")
    data: Optional[Any] = Field(default=None, description="响应数据")
    status_code: int = Field(default=status.HTTP_200_OK, description="HTTP状态码")

class SuccessResponse(JSONResponse):
    """成功响应类"""

    def __init__(
            self,
            data: Optional[Any] = None,
            msg: Optional[str] = RET.OK.msg,
            code: int = RET.OK.code,
            status_code: int = status.HTTP_200_OK,
    ) -> None:
        """
        初始化成功响应类
        
        :param data: 响应数据
        :param msg: 响应消息
        :param code: 业务状态码
        :param status_code: HTTP状态码
        """
        content = ResponseSchema(
            code=code,
            msg=msg,
            data=data,
            status_code=status_code
        ).model_dump()
        super().__init__(content=content, status_code=status_code)


class ErrorResponse(JSONResponse):
    """错误响应类"""

    def __init__(
            self,
            data: Optional[Any] = None,
            msg: Optional[str] = RET.ERROR.msg,
            code: int = RET.ERROR.code,
            status_code: int = status.HTTP_400_BAD_REQUEST,
    ) -> None:
        """
        初始化错误响应类
        
        :param data: 响应数据
        :param msg: 响应消息
        :param code: 业务状态码
        :param status_code: HTTP状态码
        """
        content = ResponseSchema(
            code=code,
            msg=msg,
            data=data,
            status_code=status_code
        ).model_dump()
        super().__init__(content=content, status_code=status_code)


class StreamResponse(StreamingResponse):
    """流式响应类"""

    def __init__(
            self,
            data: Any = None,
            status_code: int = status.HTTP_200_OK,
            headers: Mapping[str, str] | None = None,
            media_type: str | None = None,
            background: BackgroundTask | None = None
    ) -> None:
        """
        初始化流式响应类
        
        :param data: 响应数据
        :param msg: 响应消息
        :param code: 业务状态码
        :param status_code: HTTP状态码
        """
        super().__init__(
            content=data, 
            status_code=status_code, 
            media_type=media_type, # 文件类型
            headers=headers, # 文件名
            background=background # 文件大小
        )