from __future__ import annotations

from typing import Any

from pydantic import BaseModel

from .errors import ERROR_CODE_SUCCESS


class ApiResponse(BaseModel):
    code: int
    msg: str
    data: Any | None = None


def ok(data: Any = None, msg: str = "success") -> ApiResponse:
    return ApiResponse(code=ERROR_CODE_SUCCESS, msg=msg, data=data)