import os
import sys

from rich.traceback import install as rich_install

from ...core.constants import PROJECT_NAME


def custom_exception_hook(exc_type, exc_value, exc_traceback):
    print(f"{exc_type.__name__}: {exc_value}")


default_exception_hook = sys.excepthook


def setup_exceptions_handler(
    trace: bool | None,
    dev: bool | None,
    rich_traceback: bool | None,
    rich_traceback_theme: str,
):
    if dev:
        # auto set env
        os.environ[f"{PROJECT_NAME}_DEBUG"] = "1"
    if trace or dev:
        sys.excepthook = default_exception_hook
        if rich_traceback:
            rich_install(show_locals=True, theme=rich_traceback_theme)
    else:
        sys.excepthook = custom_exception_hook