from typing import Any, Dict

from fastapi.responses import JSONResponse


def success_response(data: Any, message: str = "ok") -> Dict[str, Any]:
    return {
        "code": 0,
        "message": message,
        "data": data,
    }


def error_response(message: Any, code: int, status_code: int) -> JSONResponse:
    return JSONResponse(
        status_code=status_code,
        content={
            "code": code,
            "message": message,
            "data": None,
        },
    )