"""Error helpers."""

from __future__ import annotations

from typing import TypeVar

from .core import (
    DatabaseError,
    Error,
    InterfaceError,
    InternalError,
    OperationalError,
    ProgrammingError,
)

__all__ = [
    'db_error',
    'error',
    'if_error',
    'int_error',
    'op_error',
    'prg_error'
]

# Error messages

E = TypeVar('E', bound=Error)

def error(msg: str, cls: type[E]) -> E:
    """Return specified error object with empty sqlstate attribute."""
    error = cls(msg)
    if isinstance(error, DatabaseError):
        error.sqlstate = None
    return error


def db_error(msg: str) -> DatabaseError:
    """Return DatabaseError."""
    return error(msg, DatabaseError)


def int_error(msg: str) -> InternalError:
    """Return InternalError."""
    return error(msg, InternalError)


def prg_error(msg: str) -> ProgrammingError:
    """Return ProgrammingError."""
    return error(msg, ProgrammingError)


def if_error(msg: str) -> InterfaceError:
    """Return InterfaceError."""
    return error(msg, InterfaceError)


def op_error(msg: str) -> OperationalError:
    """Return OperationalError."""
    return error(msg, OperationalError)