import logging
from typing import Union

from rich.console import Console
from rich.logging import RichHandler


def setup_logging(
    terminal_width: Union[int, None] = None, level: int = logging.INFO
) -> None:
    logger = logging.getLogger("fastapi_cli")
    console = Console(width=terminal_width) if terminal_width else None
    rich_handler = RichHandler(
        show_time=False,
        rich_tracebacks=True,
        tracebacks_show_locals=True,
        markup=True,
        show_path=False,
        console=console,
    )
    rich_handler.setFormatter(logging.Formatter("%(message)s"))
    logger.addHandler(rich_handler)

    logger.setLevel(level)
    logger.propagate = False