from rest_framework.response import Response


def ok(data, message="ok"):
    return Response({"code": 200, "data": data, "message": message})


class EnvelopeMixin:
    """错误信封统一收口（Task7 同款裁剪，无 ACAO——管理台与 API 同源，跨域由 nginx 层面不存在）。

    simplejwt 认证失败 401 / 权限 403 / 限流 429 等由 DRF 异常处理器产出的响应
    body 是 {detail}，管理台前端（Task17）按 {code,data,message} 信封对接——
    在此统一转换，视图自身的 ok/4xx 响应不受影响（只动含 detail 的 dict）。
    """

    def finalize_response(self, request, response, *args, **kwargs):
        response = super().finalize_response(request, response, *args, **kwargs)
        if response.status_code >= 400 and isinstance(response.data, dict):
            if "detail" in response.data:
                response.data = {"code": response.status_code, "data": None,
                                 "message": response.data["detail"]}
            elif "code" not in response.data:   # simplejwt 序列化器字段错误 {"password": [...]} 等
                response.data = {"code": response.status_code, "data": response.data,
                                 "message": "invalid_input"}
        return response