import sys
import utime
from micropython import const
from typing import Any

_DEBUG = const(10)
_INFO = const(20)
_WARNING = const(30)
_ERROR = const(40)
_CRITICAL = const(50)

_leveldict = {
    _DEBUG: ("DEBUG", "32"),
    _INFO: ("INFO", "36"),
    _WARNING: ("WARNING", "33"),
    _ERROR: ("ERROR", "31"),
    _CRITICAL: ("CRITICAL", "1;31"),
}

level = _DEBUG
color = True


def _log(name: str, mlevel: int, msg: str, *args: Any) -> None:
    if __debug__ and mlevel >= level:
        if color:
            fmt = (
                "%d \x1b[35m%s\x1b[0m \x1b["
                + _leveldict[mlevel][1]
                + "m%s\x1b[0m "
                + msg
            )
        else:
            fmt = "%d %s %s " + msg
        print(fmt % ((utime.ticks_us(), name, _leveldict[mlevel][0]) + args))


def debug(name: str, msg: str, *args: Any) -> None:
    _log(name, _DEBUG, msg, *args)


def info(name: str, msg: str, *args: Any) -> None:
    _log(name, _INFO, msg, *args)


def warning(name: str, msg: str, *args: Any) -> None:
    _log(name, _WARNING, msg, *args)


def error(name: str, msg: str, *args: Any) -> None:
    _log(name, _ERROR, msg, *args)


def exception(name: str, exc: BaseException) -> None:
    # we are using `__class__.__name__` to avoid importing ui module
    # we also need to instruct typechecker to ignore the missing argument
    # in ui.Result exception
    if exc.__class__.__name__ == "Result":
        _log(
            name,
            _DEBUG,
            "ui.Result: %s",
            exc.value,  # type: ignore[Cannot access member "value" for type "BaseException"]
        )
    elif exc.__class__.__name__ == "Cancelled":
        _log(name, _DEBUG, "ui.Cancelled")
    else:
        _log(name, _ERROR, "exception:")
        # since mypy 0.770 we cannot override sys, so print_exception is unknown
        sys.print_exception(exc)  # type: ignore ["print_exception" is not a known member of module]