ui.py

"""
Colorful console UI helpers for Dawn Validator BOT.
"""

from __future__ import annotations

import os
import shutil
from typing import Optional

try:
    from colorama import init as colorama_init, Fore, Style
except ImportError:  # graceful fallback if dependencies not yet installed
    class _Dummy:
        RESET_ALL = ""

        def __getattr__(self, item: str) -> str:
            return ""

    def colorama_init(*_: object, **__: object) -> None:  # type: ignore[override]
        return None

    Fore = _Dummy()  # type: ignore[assignment]
    Style = _Dummy()  # type: ignore[assignment]


colorama_init(autoreset=True)


def clear() -> None:
    """Clear the console screen."""
    os.system("cls" if os.name == "nt" else "clear")


def get_terminal_width(default: int = 80) -> int:
    """Best-effort terminal width detection."""
    try:
        return shutil.get_terminal_size().columns
    except Exception:
        return default


def hr(char: str = "━") -> str:
    """Horizontal rule matching terminal width."""
    width = max(40, get_terminal_width())
    return char * width


def colored(text: str, *, fg: Optional[str] = None, bright: bool = False) -> str:
    """Return text wrapped in color/brightness styles."""
    parts = []
    if bright:
        parts.append(Style.BRIGHT)
    if fg:
        parts.append(fg)
    parts.append(text)
    parts.append(Style.RESET_ALL)
    return "".join(parts)


def print_logo() -> None:
    """Print the Dawn Validator BOT logo with a light gradient effect."""
    logo_lines = [
        r"/$$$$$$$                                            /$$$$$$$   /$$$$$$  /$$$$$$$$",
        r"| $$__  $$                                          | $$__  $$ /$$__  $$|__  $$__/",
        r"| $$  \ $$  /$$$$$$  /$$  /$$  /$$ /$$$$$$$         | $$  \ $$| $$  \ $$   | $$   ",
        r"| $$  | $$ |____  $$| $$ | $$ | $$| $$__  $$ /$$$$$$| $$$$$$$ | $$  | $$   | $$   ",
        r"| $$  | $$  /$$$$$$$| $$ | $$ | $$| $$  \ $$|______/| $$__  $$| $$  | $$   | $$   ",
        r"| $$  | $$ /$$__  $$| $$ | $$ | $$| $$  | $$        | $$  \ $$| $$  | $$   | $$   ",
        r"| $$$$$$$/|  $$$$$$$|  $$$$$/$$$$/| $$  | $$        | $$$$$$$/|  $$$$$$/   | $$   ",
        r"|_______/  \_______/ \_____/\___/ |__/  |__/        |_______/  \______/    |__/",
    ]

    palette = [
        Fore.CYAN,
        Fore.BLUE,
        Fore.MAGENTA,
        Fore.CYAN,
        Fore.BLUE,
        Fore.MAGENTA,
        Fore.CYAN,
        Fore.BLUE,
    ]

    print()
    for idx, line in enumerate(logo_lines):
        color = palette[idx % len(palette)]
        print(colored(line, fg=color, bright=True))
    print()


def print_title(title: str) -> None:
    """Print a colored title block."""
    print(colored(hr(), fg=Fore.MAGENTA, bright=True))
    print(colored(f" {title} ", fg=Fore.CYAN, bright=True))
    print(colored(hr(), fg=Fore.MAGENTA, bright=True))


def print_menu(options: list[str]) -> None:
    """Render a numbered menu."""
    for idx, option in enumerate(options, start=1):
        print(
            colored(
                f"[{idx}] ",
                fg=Fore.YELLOW,
                bright=True,
            )
            + colored(option, fg=Fore.WHITE, bright=False)
        )
    print(
        colored("[0] ", fg=Fore.RED, bright=True)
        + colored("Exit", fg=Fore.WHITE)
    )
    print()


def prompt_choice(prompt: str = "Select menu option") -> str:
    """Prompt user for a menu choice."""
    return input(colored(f"{prompt}: ", fg=Fore.CYAN, bright=True))


def info(msg: str) -> None:
    print(colored(msg, fg=Fore.GREEN, bright=True))


def warn(msg: str) -> None:
    print(colored(msg, fg=Fore.YELLOW, bright=True))


def error(msg: str) -> None:
    print(colored(msg, fg=Fore.RED, bright=True))


def pause(msg: str = "Press Enter to continue...") -> None:
    input(colored(msg, fg=Fore.CYAN))