"""Domain errors for AI Studio interactions."""

from enum import IntEnum


class ErrorCode(IntEnum):
    USAGE_LIMIT_EXCEEDED = 429
    UNAUTHORIZED = 401
    FORBIDDEN = 403
    NOT_FOUND = 404
    RATE_LIMITED = 429
    INTERNAL_ERROR = 500
    BAD_REQUEST = 400


class AistudioError(Exception):
    pass


class AuthError(AistudioError):
    pass


class UsageLimitExceeded(AistudioError):
    pass


class SnapshotExpired(AistudioError):
    pass


class ModelNotFoundError(AistudioError):
    pass


class RequestError(AistudioError):
    def __init__(self, status: int, message: str = ""):
        self.status = status
        super().__init__(f"HTTP {status}: {message}")


def classify_error(status: int, body: str) -> AistudioError:
    if status == 429:
        return UsageLimitExceeded(f"配额用完: {body[:200]}")
    if status == 401:
        return AuthError(f"认证失败: {body[:200]}")
    if status == 403:
        return AuthError(f"禁止访问: {body[:200]}")
    return RequestError(status, body[:200])