"""Define Anker Solix API errors."""

from __future__ import annotations


class AnkerSolixError(Exception):
    """Define a base error."""


class AuthorizationError(AnkerSolixError):
    """Authorization error."""


class ConnectError(AnkerSolixError):
    """Connection error."""


class NetworkError(AnkerSolixError):
    """Network error."""


class ServerError(AnkerSolixError):
    """Server error."""


class RequestError(AnkerSolixError):
    """Request error."""


class VerifyCodeError(AnkerSolixError):
    """Verify code error."""


class VerifyCodeExpiredError(AnkerSolixError):
    """Verification code has expired."""


class NeedVerifyCodeError(AnkerSolixError):
    """Need verification code error."""


class VerifyCodeMaxError(AnkerSolixError):
    """Maximum attempts of verications error."""


class VerifyCodeNoneMatchError(AnkerSolixError):
    """Verify code none match error."""


class VerifyCodePasswordError(AnkerSolixError):
    """Verify code password error."""


class ClientPublicKeyError(AnkerSolixError):
    """Define an error for client public key error."""


class TokenKickedOutError(AnkerSolixError):
    """Define an error for token does not exist because it was kicked out."""


class InvalidCredentialsError(AnkerSolixError):
    """Define an error for unauthenticated accounts."""


class RetryExceeded(AnkerSolixError):
    """Define an error for exceeded retry attempts. Please try again in 24 hours."""


ERRORS: dict[int, type[AnkerSolixError]] = {
    401: AuthorizationError,
    403: AuthorizationError,
    997: ConnectError,
    998: NetworkError,
    999: ServerError,
    10000: RequestError,
    10003: RequestError,
    10007: RequestError,
    26050: VerifyCodeError,
    26051: VerifyCodeExpiredError,
    26052: NeedVerifyCodeError,
    26053: VerifyCodeMaxError,
    26054: VerifyCodeNoneMatchError,
    26055: VerifyCodePasswordError,
    26070: ClientPublicKeyError,
    26084: TokenKickedOutError,
    26108: InvalidCredentialsError,
    26156: InvalidCredentialsError,
    100053: RetryExceeded,
}


def raise_error(data: dict) -> None:
    """Raise the appropriate error based upon a response code."""
    code = data.get("code", -1)
    if code in [0]:
        return
    cls = ERRORS.get(code, AnkerSolixError)
    raise cls(f'({code}) {data.get("msg","Error msg not found")}')