code-puppy-core-plugins 0.0.2__py3-none-any.whl

code_puppy_core_plugins/theme/themes.py

@@ -0,0 +1,720 @@
+"""Theme definitions and application logic.
+
+Pure module: no I/O beyond reading/writing config via code_puppy.config and
+content_styles. Testable in isolation. Zen: simple is better than complex.
+"""
+
+from __future__ import annotations
+
+import random
+from itertools import cycle
+from typing import Callable
+
+from code_puppy.command_line.colors_menu import BANNER_COLORS, BANNER_DISPLAY_INFO
+from code_puppy.config import set_banner_color
+
+from . import bundled_palettes as bp
+from . import content_styles as cs
+from . import rich_themes as rt
+
+# Palette = the Rich color values from /colors (deduped, sorted for determinism)
+PALETTE: list[str] = sorted(set(BANNER_COLORS.values()))
+BANNER_KEYS: list[str] = list(BANNER_DISPLAY_INFO.keys())
+
+
+# Foreground-safe palette: exclude dark or unparseable Rich colors that work only
+# as background tags.
+def _parseable(name: str) -> bool:
+    from rich.color import Color
+
+    try:
+        Color.parse(name)
+        return True
+    except Exception:
+        return False
+
+
+TEXT_PALETTE: list[str] = sorted(
+    {
+        c
+        for c in PALETTE
+        if _parseable(c)
+        and not any(bad in c for bad in ("black", "navy_blue", "dark_red", "grey0"))
+    }
+)
+
+# Curated themes cycle a small, opinionated color list across banners.
+CURATED_THEMES: dict[str, dict] = {
+    "ocean": {
+        "icon": "🌊",
+        "label": "Ocean",
+        "blurb": "cool blues, cyans & teals",
+        "colors": [
+            "deep_sky_blue4",
+            "cyan4",
+            "dodger_blue3",
+            "dark_turquoise",
+            "steel_blue",
+            "blue",
+            "navy_blue",
+            "dark_cyan",
+            "aquamarine1",
+        ],
+        "content_styles": {
+            "info": "cyan",
+            "warning": "yellow",
+            "success": "bright_cyan",
+            "error": "bold red",
+            "debug": "dim cyan",
+            "diff_add": "bright_cyan",
+            "diff_remove": "red",
+            "diff_context": "dim blue",
+        },
+        "color_remap": rt.make_remap(
+            cyan="deep_sky_blue4",
+            blue="dodger_blue3",
+            magenta="dark_turquoise",
+            bright_cyan="aquamarine1",
+            bright_blue="steel_blue",
+            bright_magenta="dark_cyan",
+        ),
+        "terminal_palette": bp.OCEAN,
+    },
+    "forest": {
+        "icon": "🌲",
+        "label": "Forest",
+        "blurb": "earthy greens & olives",
+        "colors": [
+            "green4",
+            "dark_green",
+            "dark_sea_green4",
+            "spring_green4",
+            "chartreuse4",
+            "dark_olive_green3",
+            "dark_goldenrod",
+        ],
+        "content_styles": {
+            "info": "green",
+            "warning": "dark_goldenrod",
+            "success": "bright_green",
+            "error": "bold red",
+            "debug": "dim green",
+            "diff_add": "bright_green",
+            "diff_remove": "red",
+            "diff_context": "dim green",
+        },
+        "color_remap": rt.make_remap(
+            cyan="spring_green4",
+            blue="green4",
+            magenta="dark_sea_green4",
+            bright_cyan="chartreuse4",
+            bright_blue="dark_green",
+            bright_magenta="dark_olive_green3",
+        ),
+        "terminal_palette": bp.FOREST,
+    },
+    "sunset": {
+        "icon": "🌅",
+        "label": "Sunset",
+        "blurb": "warm reds, oranges & gold",
+        "colors": [
+            "orange_red1",
+            "dark_orange3",
+            "gold3",
+            "red3",
+            "indian_red",
+            "dark_red",
+            "hot_pink3",
+        ],
+        "content_styles": {
+            "info": "gold3",
+            "warning": "dark_orange3",
+            "success": "orange_red1",
+            "error": "bold red",
+            "debug": "dim yellow",
+            "diff_add": "gold3",
+            "diff_remove": "red3",
+            "diff_context": "dim yellow",
+        },
+        "color_remap": rt.make_remap(
+            cyan="gold3",
+            blue="orange_red1",
+            magenta="hot_pink3",
+            bright_cyan="dark_orange3",
+            bright_blue="indian_red",
+            bright_magenta="hot_pink3",
+        ),
+        "terminal_palette": bp.SUNSET,
+    },
+    "vaporwave": {
+        "icon": "🪩",
+        "label": "Vaporwave",
+        "blurb": "neon pinks & purples",
+        "colors": [
+            "hot_pink3",
+            "dark_magenta",
+            "purple",
+            "dark_violet",
+            "medium_purple4",
+            "dark_orchid",
+            "deep_pink4",
+            "pale_violet_red1",
+        ],
+        "content_styles": {
+            "info": "magenta",
+            "warning": "hot_pink3",
+            "success": "bright_magenta",
+            "error": "bold red",
+            "debug": "dim magenta",
+            "diff_add": "bright_magenta",
+            "diff_remove": "red",
+            "diff_context": "dim magenta",
+        },
+        "color_remap": rt.make_remap(
+            cyan="hot_pink3",
+            blue="purple",
+            magenta="dark_magenta",
+            bright_cyan="pale_violet_red1",
+            bright_blue="dark_violet",
+            bright_magenta="medium_orchid",
+        ),
+        "terminal_palette": bp.VAPORWAVE,
+    },
+    "bubblegum-pink": {
+        "icon": "🫧",
+        "label": "Bubblegum Pink",
+        "blurb": "sweet pinks, candy neon & glow",
+        "colors": [
+            "hot_pink3",
+            "deep_pink4",
+            "pale_violet_red1",
+            "magenta",
+            "bright_magenta",
+            "dark_orchid",
+            "medium_purple4",
+            "violet",
+        ],
+        "content_styles": {
+            "info": "hot_pink3",
+            "warning": "deep_pink4",
+            "success": "bright_magenta",
+            "error": "bold red",
+            "debug": "dim magenta",
+            "diff_add": "bright_magenta",
+            "diff_remove": "red",
+            "diff_context": "dim magenta",
+        },
+        "color_remap": rt.make_remap(
+            cyan="hot_pink3",
+            blue="deep_pink4",
+            magenta="pale_violet_red1",
+            bright_cyan="#ff69b4",
+            bright_blue="medium_purple4",
+            bright_magenta="violet",
+        ),
+        "terminal_palette": bp.BUBBLEGUM_PINK,
+    },
+    "purple-puppy": {
+        "icon": "\U0001f436",  # dog face
+        "label": "Purple Puppy",
+        "blurb": "vivid violet fur, white muzzle & pink tongue",
+        "colors": [
+            "#9b30d9",  # highlight fur
+            "#8408b0",  # body fur
+            "#b06be8",  # light fur
+            "#6a0dad",  # shade fur
+            "#d9a7f5",  # lilac glow
+            "#ff7fa8",  # tongue pink
+            "#e8ccff",  # muzzle lavender
+            "#4b1a6e",  # deep shade
+        ],
+        "content_styles": {
+            "info": "#c9a0f0",
+            "warning": "#f5c26b",
+            "success": "#ff8fb3",
+            "error": "bold red",
+            "debug": "dim #b8a0d0",
+            "diff_add": "#ff8fb3",
+            "diff_remove": "red",
+            "diff_context": "dim #9b7fc0",
+        },
+        "color_remap": rt.make_remap(
+            cyan="#c9a0f0",
+            blue="#9b30d9",
+            magenta="#ff7fa8",
+            bright_cyan="#e8ccff",
+            bright_blue="#b06be8",
+            bright_magenta="#ff9fd2",
+        ),
+        "terminal_palette": bp.PURPLE_PUPPY,
+    },
+    # Palette-first themes use OSC ANSI slots so Rich tags follow the new palette;
+    # their Rich-level color_remap remains empty.
+    "catppuccin-mocha": {
+        "icon": "🐱",
+        "label": "Catppuccin Mocha",
+        "blurb": "soothing pastel dark",
+        "colors": [
+            "cyan",
+            "magenta",
+            "blue",
+            "green",
+            "yellow",
+            "red",
+            "bright_cyan",
+            "bright_magenta",
+        ],
+        "content_styles": {
+            "info": "cyan",
+            "warning": "yellow",
+            "success": "green",
+            "error": "bold red",
+            "debug": "dim white",
+            "diff_add": "green",
+            "diff_remove": "red",
+            "diff_context": "dim white",
+        },
+        "color_remap": {},
+        "terminal_palette": bp.CATPPUCCIN_MOCHA,
+    },
+    "catppuccin-latte": {
+        "icon": "☕",
+        "label": "Catppuccin Latte",
+        "blurb": "soft pastel light with lavender & rose",
+        "colors": [
+            "#1e66f5",
+            "#8839ef",
+            "#40a02b",
+            "#df8e1d",
+            "#04a5e5",
+            "#d20f39",
+            "#7287fd",
+            "#ea76cb",
+        ],
+        "content_styles": {
+            "info": "#1e66f5",
+            "warning": "#df8e1d",
+            "success": "#40a02b",
+            "error": "bold #d20f39",
+            "debug": "dim #4c4f69",
+            "diff_add": "#40a02b",
+            "diff_remove": "#d20f39",
+            "diff_context": "dim #5c5f77",
+        },
+        "color_remap": rt.make_remap(
+            cyan="#04a5e5",
+            blue="#1e66f5",
+            magenta="#8839ef",
+            bright_cyan="#179299",
+            bright_blue="#7287fd",
+            bright_magenta="#ea76cb",
+        ),
+        "terminal_palette": bp.CATPPUCCIN_LATTE,
+    },
+    "tokyo-night": {
+        "icon": "🌃",
+        "label": "Tokyo Night",
+        "blurb": "neon-on-navy night",
+        "colors": [
+            "blue",
+            "cyan",
+            "magenta",
+            "green",
+            "yellow",
+            "red",
+            "bright_blue",
+            "bright_cyan",
+        ],
+        "content_styles": {
+            "info": "cyan",
+            "warning": "yellow",
+            "success": "green",
+            "error": "bold red",
+            "debug": "dim white",
+            "diff_add": "green",
+            "diff_remove": "red",
+            "diff_context": "dim white",
+        },
+        "color_remap": {},
+        "terminal_palette": bp.TOKYO_NIGHT,
+    },
+    "green-screen": {
+        "icon": "[CRT]",
+        "label": "Green Screen",
+        "blurb": "llxprt's radioactive green phosphor CRT",
+        # Banner labels render as white, which this theme remaps to #6a9955.
+        # Keep their backgrounds dark enough for readable phosphor-on-glass.
+        "colors": [
+            "#071507",
+            "#0b1f0b",
+            "#102510",
+        ],
+        "content_styles": {
+            "info": "#6a9955",
+            "warning": "#6a9955",
+            "success": "#00ff00",
+            "error": "bold #6a9955",
+            "debug": "dim #4a7035",
+            "diff_add": "#00ff00",
+            "diff_remove": "#6a9955",
+            "diff_context": "dim #4a7035",
+        },
+        "color_remap": rt.make_remap(
+            cyan="#6a9955",
+            blue="#6a9955",
+            magenta="#6a9955",
+            bright_cyan="#6a9955",
+            bright_blue="#6a9955",
+            bright_magenta="#6a9955",
+            white="#6a9955",
+            bright_white="#00ff00",
+        ),
+        "terminal_palette": bp.GREEN_SCREEN,
+    },
+    "deep-black": {
+        "icon": "🌑",
+        "label": "Deep Black",
+        "blurb": "inky noir with subtle neon edge",
+        "colors": [
+            "#050505",
+            "#1f1f1f",
+            "#2b2b2b",
+            "#3a3a3a",
+            "#4dabf7",
+            "#b197fc",
+            "#63e6be",
+            "#ffd166",
+        ],
+        "content_styles": {
+            "info": "#cfd8dc",
+            "warning": "#ffd166",
+            "success": "#96f2d7",
+            "error": "bold #ff6b6b",
+            "debug": "dim #6c757d",
+            "diff_add": "#96f2d7",
+            "diff_remove": "#ff6b6b",
+            "diff_context": "dim #adb5bd",
+        },
+        "color_remap": rt.make_remap(
+            cyan="#cfd8dc",
+            blue="#4dabf7",
+            magenta="#b197fc",
+            bright_cyan="#96f2d7",
+            bright_blue="#74c0fc",
+            bright_magenta="#d0bfff",
+        ),
+        "terminal_palette": bp.DEEP_BLACK,
+    },
+    # Light themes use dim black for debug text so it remains visible.
+    "solarized-light": {
+        "icon": "☀️",
+        "label": "Solarized Light",
+        "blurb": "classic warm beige with calm accents",
+        "colors": [
+            "blue",
+            "magenta",
+            "green",
+            "yellow",
+            "cyan",
+            "red",
+            "bright_blue",
+            "bright_magenta",
+        ],
+        "content_styles": {
+            "info": "blue",
+            "warning": "yellow",
+            "success": "green",
+            "error": "bold red",
+            "debug": "dim black",
+            "diff_add": "green",
+            "diff_remove": "red",
+            "diff_context": "dim black",
+        },
+        "color_remap": {},
+        "terminal_palette": bp.SOLARIZED_LIGHT,
+    },
+    "github-light": {
+        "icon": "📄",
+        "label": "GitHub Light",
+        "blurb": "crisp white, familiar code colors",
+        "colors": [
+            "blue",
+            "magenta",
+            "green",
+            "yellow",
+            "cyan",
+            "red",
+            "bright_blue",
+            "bright_magenta",
+        ],
+        "content_styles": {
+            "info": "blue",
+            "warning": "yellow",
+            "success": "green",
+            "error": "bold red",
+            "debug": "dim black",
+            "diff_add": "green",
+            "diff_remove": "red",
+            "diff_context": "dim black",
+        },
+        "color_remap": {},
+        "terminal_palette": bp.GITHUB_LIGHT,
+    },
+    "rose-pine-dawn": {
+        "icon": "🌸",
+        "label": "Rose Pine Dawn",
+        "blurb": "soft pastel rose light",
+        "colors": [
+            "magenta",
+            "blue",
+            "cyan",
+            "yellow",
+            "green",
+            "red",
+            "bright_magenta",
+            "bright_blue",
+        ],
+        "content_styles": {
+            "info": "blue",
+            "warning": "yellow",
+            "success": "green",
+            "error": "bold red",
+            "debug": "dim black",
+            "diff_add": "green",
+            "diff_remove": "red",
+            "diff_context": "dim black",
+        },
+        "color_remap": {},
+        "terminal_palette": bp.ROSE_PINE_DAWN,
+    },
+}
+
+# Surprise Me is a special case, regenerated each preview/apply.
+SURPRISE = {
+    "icon": "🎲",
+    "label": "Surprise Me",
+    "blurb": "a fresh random remix every time",
+    "colors": None,
+    "content_styles": None,  # generated dynamically
+    "color_remap": None,  # generated dynamically
+    "terminal_palette": None,  # randomized at apply time
+}
+
+# Restore Code Puppy defaults (banners + content).
+DEFAULT = {
+    "icon": "🔄",
+    "label": "Restore Defaults",
+    "blurb": "back to Code Puppy factory colors",
+    "colors": None,
+    "content_styles": None,  # handled specially
+    "color_remap": None,  # handled specially (empty)
+    "terminal_palette": None,  # handled specially (OSC reset)
+}
+
+# Ordered menu: 7 vivid/dark themes, 7 palette-first/light themes, surprise, default.
+MENU: list[tuple[str, dict]] = [
+    ("ocean", CURATED_THEMES["ocean"]),
+    ("forest", CURATED_THEMES["forest"]),
+    ("sunset", CURATED_THEMES["sunset"]),
+    ("vaporwave", CURATED_THEMES["vaporwave"]),
+    ("bubblegum-pink", CURATED_THEMES["bubblegum-pink"]),
+    ("purple-puppy", CURATED_THEMES["purple-puppy"]),
+    ("catppuccin-mocha", CURATED_THEMES["catppuccin-mocha"]),
+    ("catppuccin-latte", CURATED_THEMES["catppuccin-latte"]),
+    ("tokyo-night", CURATED_THEMES["tokyo-night"]),
+    ("green-screen", CURATED_THEMES["green-screen"]),
+    ("deep-black", CURATED_THEMES["deep-black"]),
+    ("solarized-light", CURATED_THEMES["solarized-light"]),
+    ("github-light", CURATED_THEMES["github-light"]),
+    ("rose-pine-dawn", CURATED_THEMES["rose-pine-dawn"]),
+    ("surprise", SURPRISE),
+    ("default", DEFAULT),
+]
+MENU_BY_NAME: dict[str, dict] = {name: theme for name, theme in MENU}
+MENU_BY_INDEX: dict[str, str] = {str(i + 1): name for i, (name, _) in enumerate(MENU)}
+
+# Friendly aliases (no spaces, alt names).
+MENU_BY_NAME["random"] = SURPRISE
+MENU_BY_NAME["reset"] = DEFAULT
+MENU_BY_NAME["defaults"] = DEFAULT
+MENU_BY_NAME["mocha"] = CURATED_THEMES["catppuccin-mocha"]
+MENU_BY_NAME["latte"] = CURATED_THEMES["catppuccin-latte"]
+MENU_BY_NAME["tokyo"] = CURATED_THEMES["tokyo-night"]
+MENU_BY_NAME["green"] = CURATED_THEMES["green-screen"]
+MENU_BY_NAME["crt"] = CURATED_THEMES["green-screen"]
+MENU_BY_NAME["bubblegum"] = CURATED_THEMES["bubblegum-pink"]
+MENU_BY_NAME["pink"] = CURATED_THEMES["bubblegum-pink"]
+MENU_BY_NAME["puppy"] = CURATED_THEMES["purple-puppy"]
+MENU_BY_NAME["purple"] = CURATED_THEMES["purple-puppy"]
+MENU_BY_NAME["solarized"] = CURATED_THEMES["solarized-light"]
+MENU_BY_NAME["github"] = CURATED_THEMES["github-light"]
+MENU_BY_NAME["rose-pine"] = CURATED_THEMES["rose-pine-dawn"]
+MENU_BY_NAME["dawn"] = CURATED_THEMES["rose-pine-dawn"]
+
+
+def colors_for(theme_name: str, rng: random.Random | None = None) -> dict[str, str]:
+    """Build a {banner: color} mapping for the named theme.
+
+    For curated themes, cycles through the theme palette deterministically.
+    For ``surprise``, samples randomly from the full PALETTE.
+    For ``default``, returns the Code Puppy factory banner colors.
+    """
+    if theme_name not in MENU_BY_NAME:
+        raise KeyError(f"Unknown theme: {theme_name!r}")
+
+    if theme_name == "default":
+        from code_puppy.config import DEFAULT_BANNER_COLORS
+
+        return dict(DEFAULT_BANNER_COLORS)
+
+    if theme_name in ("surprise", "random"):
+        chooser = rng.choice if rng is not None else random.choice
+        return {banner: chooser(PALETTE) for banner in BANNER_KEYS}
+
+    palette = cycle(MENU_BY_NAME[theme_name]["colors"])
+    return {banner: next(palette) for banner in BANNER_KEYS}
+
+
+def content_styles_for(
+    theme_name: str, rng: random.Random | None = None
+) -> dict[str, str]:
+    """Build a {content_key: style} mapping for the named theme.
+
+    For curated themes, returns the hand-picked palette.
+    For ``surprise``, generates a random remix (error stays red — it's a UX
+    contract that errors look angry).
+    For ``default``, returns the factory defaults.
+    """
+    if theme_name not in MENU_BY_NAME:
+        raise KeyError(f"Unknown theme: {theme_name!r}")
+
+    if theme_name == "default":
+        return dict(cs.DEFAULT_CONTENT_STYLES)
+
+    if theme_name in ("surprise", "random"):
+        chooser = rng.choice if rng is not None else random.choice
+        mapping = {key: chooser(TEXT_PALETTE) for key in cs.CONTENT_KEYS}
+        # Errors should always be visually distinct + alarming.
+        mapping["error"] = "bold red"
+        return mapping
+
+    return dict(MENU_BY_NAME[theme_name]["content_styles"])
+
+
+def color_remap_for(
+    theme_name: str, rng: random.Random | None = None
+) -> dict[str, str]:
+    """Build a Rich color remap dict for the named theme (Level 2 theming).
+
+    For curated themes, returns the hand-picked remap.
+    For ``surprise``, picks random replacements (semantic colors untouched).
+    For ``default``, returns an empty dict (= no remap, restore).
+    """
+    if theme_name not in MENU_BY_NAME:
+        raise KeyError(f"Unknown theme: {theme_name!r}")
+
+    if theme_name == "default":
+        return {}
+
+    if theme_name in ("surprise", "random"):
+        chooser = rng.choice if rng is not None else random.choice
+        return rt.make_remap(
+            cyan=chooser(TEXT_PALETTE),
+            blue=chooser(TEXT_PALETTE),
+            magenta=chooser(TEXT_PALETTE),
+            bright_cyan=chooser(TEXT_PALETTE),
+            bright_blue=chooser(TEXT_PALETTE),
+            bright_magenta=chooser(TEXT_PALETTE),
+        )
+
+    return dict(MENU_BY_NAME[theme_name]["color_remap"])
+
+
+def terminal_palette_for(
+    theme_name: str, rng: random.Random | None = None
+) -> dict | None:
+    """Build an OSC terminal palette for the named theme.
+
+    Returns None if the theme has no terminal palette (rare). For
+    ``default``, returns None (caller is expected to call osc_palette.reset).
+    For ``surprise``, generates a random bg/fg pair (no full ANSI swap).
+    """
+    if theme_name not in MENU_BY_NAME:
+        raise KeyError(f"Unknown theme: {theme_name!r}")
+
+    if theme_name == "default":
+        return None
+
+    if theme_name in ("surprise", "random"):
+        chooser = rng.choice if rng is not None else random.choice
+        # Random bg/fg pair with enough contrast to read text.
+        LIGHT_BGS = [
+            "#fdf6e3",
+            "#ffffff",
+            "#faf4ed",
+            "#eff1f5",
+            "#f5f5f5",
+            "#f8f0e3",
+        ]
+        DARK_FGS = [
+            "#657b83",
+            "#24292e",
+            "#575279",
+            "#4c4f69",
+            "#2d2d2d",
+            "#3c3c3c",
+        ]
+        DARK_BGS = [
+            "#0a1929",
+            "#1a2310",
+            "#2d1b0e",
+            "#16002a",
+            "#1e1e2e",
+            "#1a1b26",
+            "#282828",
+            "#0e0e2c",
+        ]
+        LIGHT_FGS = [
+            "#d6eaf8",
+            "#e3eecc",
+            "#ffe4cc",
+            "#ffe0ff",
+            "#cdd6f4",
+            "#c0caf5",
+            "#ebdbb2",
+            "#f5f5f5",
+        ]
+        # 30% chance of a light surprise, 70% dark (most users prefer dark TUI).
+        if (rng or random).random() < 0.3:
+            return {"bg": chooser(LIGHT_BGS), "fg": chooser(DARK_FGS)}
+        return {"bg": chooser(DARK_BGS), "fg": chooser(LIGHT_FGS)}
+
+    return MENU_BY_NAME[theme_name].get("terminal_palette")
+
+
+def apply(
+    mapping: dict[str, str], setter: Callable[[str, str], None] = set_banner_color
+) -> None:
+    """Persist a banner→color mapping via the config layer.
+
+    ``setter`` is injectable for tests.
+    """
+    for banner, color in mapping.items():
+        setter(banner, color)
+
+
+def resolve_theme_arg(arg: str) -> str | None:
+    """Resolve menu numbers or a theme name to a canonical theme key.
+
+    Returns None if the argument is not recognized.
+    """
+    if arg in MENU_BY_INDEX:
+        return MENU_BY_INDEX[arg]
+    if arg in MENU_BY_NAME:
+        # Normalize aliases to canonical keys.
+        if arg == "random":
+            return "surprise"
+        if arg in ("reset", "defaults"):
+            return "default"
+        return arg
+    return None