import logging
import os
from typing import List, Sequence, Tuple, Union

from readchar import key, readkey
from rich.console import Console
from rich.live import Live
from rich.logging import RichHandler
from rich.style import Style
from rich.table import Table

__all__ = ["console", "select_from_list"]

SELECTED = Style(color="green", bold=True)

console = Console(soft_wrap=True)


def init_logging() -> None:
    # TODO: We only enable logging in debug level for now
    #       This is because anaconda-client uses logging for normal
    #       program output and this really warrants a much larger
    #       redesign. In anaconda-cli-base, we currently only use
    #       the logger for debug printing, so we can just do that
    #       here.
    log_level = os.getenv("LOGLEVEL", "INFO").upper()
    if log_level == "DEBUG":
        logging.basicConfig(
            level=log_level,
            format="%(message)s",
            datefmt="[%X]",
            handlers=[RichHandler()],
        )


def _generate_table(header: str, rows: List[str], selected: int) -> Table:
    table = Table(box=None)

    table.add_column(header)

    for i, row in enumerate(rows):
        if i == selected:
            style = SELECTED
            value = f"* {row}"
        else:
            style = None
            value = f"  {row}"
        table.add_row(value, style=style)

    return table


def select_from_list(
    prompt: str, choices: Sequence[Union[str, Tuple[str, str]]]
) -> str:
    """Dynamically select from a list of choices, by using the up/down keys."""
    # inspired by https://github.com/Textualize/rich/discussions/1785#discussioncomment-1883808

    # Construct two lists, one of values, one for display
    # Display names are shown to the user, but values is indexed to return the value.
    display_names: List[str] = []
    values: List[str] = []
    for choice in choices:
        if isinstance(choice, tuple):
            value, display_name = choice
        else:
            value = choice
            display_name = choice

        display_names.append(display_name)
        values.append(value)

    selected = 0
    with Live(
        _generate_table(prompt, display_names, selected), auto_refresh=False
    ) as live:
        while keypress := readkey():
            if keypress == key.UP or keypress == "k":
                selected = max(0, selected - 1)
            if keypress == key.DOWN or keypress == "j":
                selected = min(len(display_names) - 1, selected + 1)
            if keypress in ["\n", "\r", key.ENTER]:
                live.stop()
                return values[selected]
            live.update(_generate_table(prompt, display_names, selected), refresh=True)

    raise ValueError("Unreachable")