openswarm/.venv/lib/python3.12/site-packages/rich_toolkit/styles/base.py

from __future__ import annotations

from typing import Any, Dict, Optional, Type, TypeVar, Union

from rich.color import Color
from rich.console import Console, ConsoleRenderable, Group, RenderableType
from rich.text import Text
from rich.theme import Theme
from typing_extensions import Literal

from rich_toolkit.button import Button
from rich_toolkit.container import Container
from rich_toolkit.element import CursorOffset, Element
from rich_toolkit.input import Input
from rich_toolkit.menu import Menu
from rich_toolkit.progress import Progress, ProgressLine
from rich_toolkit.spacer import Spacer
from rich_toolkit.utils.colors import (
    fade_text,
    get_terminal_background_color,
    get_terminal_text_color,
    lighten,
)

ConsoleRenderableClass = TypeVar(
    "ConsoleRenderableClass", bound=Type[ConsoleRenderable]
)


class BaseStyle:
    brightness_multiplier = 0.1

    base_theme = {
        "tag.title": "bold",
        "tag": "bold",
        "text": "#ffffff",
        "selected": "green",
        "result": "white",
        "progress": "on #893AE3",
        "error": "red",
        "cancelled": "red italic",
        # is there a way to make nested styles?
        # like label.active uses active style if not set?
        "active": "green",
        "title.error": "white",
        "title.cancelled": "white",
        "placeholder": "grey62",
        "placeholder.cancelled": "indian_red strike",
    }

    _should_show_progress_title = True

    def __init__(
        self,
        theme: Optional[Dict[str, str]] = None,
        background_color: str = "#000000",
        text_color: str = "#FFFFFF",
    ):
        self.background_color = get_terminal_background_color(background_color)
        self.text_color = get_terminal_text_color(text_color)
        self.animation_counter = 0

        base_theme = Theme(self.base_theme)
        self.theme = base_theme
        self.console = Console(theme=base_theme)

        if theme:
            self.console.push_theme(Theme(theme))

    def empty_line(self) -> RenderableType:
        return " "

    def render_context_enter(self) -> Optional[RenderableType]:
        return ""

    def render_context_exit(self) -> Optional[RenderableType]:
        return ""

    def _get_animation_colors(
        self,
        steps: int = 5,
        breathe: bool = False,
        animation_status: Literal["started", "stopped", "error"] = "started",
        **metadata: Any,
    ) -> list[Color]:
        animated = animation_status == "started"

        if animation_status == "error":
            base_color = self.console.get_style("error").color

            if base_color is None:
                base_color = Color.parse("red")

        else:
            base_color = self.console.get_style("progress").bgcolor

        if not base_color:
            base_color = Color.from_rgb(255, 255, 255)

        if breathe:
            steps = steps // 2

        if animated and base_color.triplet is not None:
            colors = [
                lighten(base_color, self.brightness_multiplier * i)
                for i in range(0, steps)
            ]

        else:
            colors = [base_color] * steps

        if breathe:
            colors = colors + colors[::-1]

        return colors

    def _count_label_lines(self, label: str, decoration_width: int = 0) -> int:
        available_width = self.console.width - decoration_width
        if available_width <= 0:
            return 1
        renderable = Text.from_markup(label) if isinstance(label, str) else label
        lines = self.console.render_lines(
            renderable,
            self.console.options.update_width(available_width),
            pad=False,
        )
        return len(lines)

    def get_cursor_offset_for_element(
        self, element: Element, parent: Optional[Element] = None
    ) -> CursorOffset:
        offset = element.cursor_offset
        if isinstance(element, Input) and not element.inline and element.label:
            label_lines = self._count_label_lines(element.label)
            return CursorOffset(top=label_lines + 1, left=offset.left)
        return offset

    def render_element(
        self,
        element: Any,
        is_active: bool = False,
        done: bool = False,
        parent: Optional[Element] = None,
        **kwargs: Any,
    ) -> RenderableType:
        if isinstance(element, str):
            return self.render_string(element, is_active, done, parent)
        elif isinstance(element, Button):
            return self.render_button(element, is_active, done, parent)
        elif isinstance(element, Container):
            return self.render_container(element, is_active, done, parent)
        elif isinstance(element, Input):
            return self.render_input(element, is_active, done, parent)
        elif isinstance(element, Menu):
            return self.render_menu(element, is_active, done, parent)
        elif isinstance(element, Progress):
            self.animation_counter += 1

            return self.render_progress(element, is_active, done, parent)
        elif isinstance(element, ProgressLine):
            return self.render_progress_log_line(
                element.text,
                parent=parent,
                index=kwargs.get("index", 0),
                max_lines=kwargs.get("max_lines", -1),
                total_lines=kwargs.get("total_lines", -1),
            )
        elif isinstance(element, Spacer):
            return self.render_spacer()
        elif isinstance(element, ConsoleRenderable):
            return element

        raise ValueError(f"Unknown element type: {type(element)}")

    def render_string(
        self,
        string: str,
        is_active: bool = False,
        done: bool = False,
        parent: Optional[Element] = None,
    ) -> RenderableType:
        return string

    def render_button(
        self,
        element: Button,
        is_active: bool = False,
        done: bool = False,
        parent: Optional[Element] = None,
    ) -> RenderableType:
        style = "black on blue" if is_active else "white on black"
        return Text(f" {element.label} ", style=style)

    def render_spacer(self) -> RenderableType:
        return ""

    def render_container(
        self,
        container: Container,
        is_active: bool = False,
        done: bool = False,
        parent: Optional[Element] = None,
    ) -> RenderableType:
        content = []

        for i, element in enumerate(container.elements):
            content.append(
                self.render_element(
                    element,
                    is_active=i == container.active_element_index,
                    done=done,
                    parent=container,
                )
            )

        return Group(*content, "\n" if not done else "")

    def render_input(
        self,
        element: Input,
        is_active: bool = False,
        done: bool = False,
        parent: Optional[Element] = None,
    ) -> RenderableType:
        label = self.render_input_label(element, is_active=is_active, parent=parent)
        text = self.render_input_value(
            element, is_active=is_active, parent=parent, done=done
        )

        contents = []

        if element.inline or done:
            if done and element.password:
                text = "*" * len(element.text)
            if label:
                text = f"{label} {text}"

            contents.append(text)
        else:
            if label:
                contents.append(label)

            contents.append(text)

        if validation_message := self.render_validation_message(element):
            contents.extend(validation_message)

        # TODO: do we need this?
        element._height = len(contents)

        return Group(*contents)

    def render_validation_message(
        self, element: Union[Input, Menu]
    ) -> Optional[list[RenderableType]]:
        if element._cancelled:
            return [Text(""), "[cancelled]Cancelled.[/]"]

        if element.valid is False:
            return [Text(""), f"[error]{element.validation_message}[/]"]

        return None

    # TODO: maybe don't reuse this for menus
    def render_input_value(
        self,
        input: Union[Menu, Input],
        is_active: bool = False,
        parent: Optional[Element] = None,
        done: bool = False,
    ) -> RenderableType:
        text = input.text

        if isinstance(input, Input) and input.password and text:
            text = "*" * len(text)

        if input._cancelled:
            if not text:
                return ""

            return f"[placeholder.cancelled]{text}[/]"

        if done:
            if (
                not text
                and isinstance(input, Input)
                and input.default_as_placeholder
                and input.default
            ):
                text = input.default

            return f"[result]{text}[/]"

        if not text:
            placeholder = input.placeholder if isinstance(input, Input) else ""

            # Use zero-width space when placeholder is empty to prevent
            # the line from being stripped as a trailing blank line
            placeholder = placeholder or "\u200b"
            return f"[placeholder]{placeholder}[/]"

        return f"[text]{text}[/]"

    def render_input_label(
        self,
        input: Union[Input, Menu],
        is_active: bool = False,
        parent: Optional[Element] = None,
    ) -> Union[str, Text, None]:
        from rich_toolkit.form import Form

        label: Union[str, Text, None] = None

        if input.label:
            label = input.label

            if isinstance(parent, Form):
                if is_active:
                    label = f"[active]{label}[/]"
                elif input.valid is False:
                    label = f"[error]{label}[/]"

        return label

    def _build_menu_options(self, element: Menu, separator: Text) -> Text:
        """Build the menu Text containing scroll indicators and option items."""
        menu = Text(justify="left")

        checked_prefix = Text(element.active_prefix + " ")
        unchecked_prefix = Text(element.inactive_prefix + " ")

        start, end = element.visible_options_range
        visible_options = element.options[start:end]
        needs_scrolling = element._needs_scrolling()

        # Reserve space for scroll indicators to prevent layout shift
        if needs_scrolling:
            if element.has_more_above:
                menu.append(Text(element.MORE_ABOVE_INDICATOR + "\n", style="dim"))
            else:
                menu.append(Text(" " * len(element.MORE_ABOVE_INDICATOR) + "\n"))

        for idx, option in enumerate(visible_options):
            actual_idx = start + idx
            is_at_cursor = actual_idx == element.selected

            # Prefix reflects checked state (multi-select) or cursor (single-select)
            if element.multiple:
                is_marked = element.is_option_checked_by_ref(option)
            else:
                is_marked = is_at_cursor
            prefix = checked_prefix if is_marked else unchecked_prefix

            # Style reflects cursor position regardless of checked state
            style = self.console.get_style("selected" if is_at_cursor else "text")

            is_last = idx == len(visible_options) - 1

            menu.append(
                Text.assemble(
                    prefix,
                    option["name"],
                    separator if not is_last else "",
                    style=style,
                )
            )

        if needs_scrolling:
            if element.has_more_below:
                menu.append(Text("\n" + element.MORE_BELOW_INDICATOR, style="dim"))
            else:
                menu.append(Text("\n" + " " * len(element.MORE_BELOW_INDICATOR)))

        if not element.options:
            menu = Text("No results found", style=self.console.get_style("text"))

        return menu

    def _build_filter_parts(self, element: Menu) -> list[RenderableType]:
        if not element.allow_filtering:
            return []

        filter_parts: list[RenderableType] = []

        filter_line = Text.assemble(
            (element.filter_prompt, self.console.get_style("text")),
            (element.text, self.console.get_style("text")),
        )

        if hint := element.selection_count_hint:
            filter_line.append(f" {hint}", style="dim")

        filter_line.append("\n")
        filter_parts.append(filter_line)

        return filter_parts

    def render_menu(
        self,
        element: Menu,
        is_active: bool = False,
        done: bool = False,
        parent: Optional[Element] = None,
    ) -> RenderableType:
        label = self.render_input_label(element, is_active=is_active, parent=parent)

        if done:
            result_content = Text()

            if label:
                result_content.append(label)
                result_content.append(" ")

            # For single-select menus, check if selection is valid
            # For multi-select menus, result_display_name only uses checked items
            should_show_cancelled = element._cancelled
            if not element.multiple:
                selection_is_valid = 0 <= element.selected < len(element.options)
                should_show_cancelled = should_show_cancelled or not selection_is_valid

            if should_show_cancelled:
                result_content.append(
                    "Cancelled.",
                    style=self.console.get_style("cancelled"),
                )
            else:
                result_content.append(
                    element.result_display_name,
                    style=self.console.get_style("result"),
                )

            return result_content

        separator = Text("  " if element.inline else "\n")
        menu = self._build_menu_options(element, separator)
        filter_parts = self._build_filter_parts(element)

        content: list[RenderableType] = []

        if label:
            content.append(label)

        content.extend(filter_parts)
        content.append(menu)

        if message := self.render_validation_message(element):
            content.extend(message)

        return Group(*content)

    def _render_cancelled_progress_message(self) -> Text:
        return Text("Cancelled.", style=self.console.get_style("cancelled"))

    def _render_progress_content(self, element: Progress) -> RenderableType:
        content: str | Group | Text = element.current_message

        if element.logs and element._inline_logs:
            lines_to_show = (
                element.logs[-element.lines_to_show :]
                if element.lines_to_show > 0
                else element.logs
            )

            start_content = [element.title, ""]

            if not self._should_show_progress_title:
                start_content = []

            content = Group(
                *start_content,
                *[
                    self.render_element(
                        line,
                        index=index,
                        max_lines=element.lines_to_show,
                        total_lines=len(element.logs),
                        parent=element,
                    )
                    for index, line in enumerate(lines_to_show)
                ],
            )

        return content

    def _progress_has_content_beyond_title(self, element: Progress) -> bool:
        if element.logs and element._inline_logs:
            return True

        if isinstance(element.current_message, Text):
            return element.current_message.plain != element.title

        return element.current_message != element.title

    def render_progress(
        self,
        element: Progress,
        is_active: bool = False,
        done: bool = False,
        parent: Optional[Element] = None,
    ) -> RenderableType:
        content = self._render_progress_content(element)

        if done and element._cancelled:
            return Group(content, self._render_cancelled_progress_message())

        return content

    def render_progress_log_line(
        self,
        line: str | Text,
        index: int,
        max_lines: int = -1,
        total_lines: int = -1,
        parent: Optional[Element] = None,
    ) -> Text:
        line = Text.from_markup(line) if isinstance(line, str) else line
        if max_lines == -1:
            return line

        shown_lines = min(total_lines, max_lines)

        # this is the minimum brightness based on the max_lines
        min_brightness = 0.4
        # but we want to have a slightly higher brightness if there's less than max_lines
        # otherwise you could get the something like this:

        # line 1 -> very dark
        # line 2 -> slightly darker
        # line 3 -> normal

        # which is ok, but not great, so we we increase the brightness if there's less than max_lines
        # so that the last line is always the brightest
        current_min_brightness = min_brightness + abs(shown_lines - max_lines) * 0.1
        current_min_brightness = min(max(current_min_brightness, min_brightness), 1.0)

        brightness_multiplier = ((index + 1) / shown_lines) * (
            1.0 - current_min_brightness
        ) + current_min_brightness

        return fade_text(
            line,
            text_color=Color.parse(self.text_color),
            background_color=self.background_color,
            brightness_multiplier=brightness_multiplier,
        )