sqlsaber 0.28.0__tar.gz → 0.29.0__tar.gz

{sqlsaber-0.28.0 → sqlsaber-0.29.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: sqlsaber
-Version: 0.28.0
+Version: 0.29.0
 Summary: SQLsaber - Open-source agentic SQL assistant
 License-File: LICENSE
 Requires-Python: >=3.12

{sqlsaber-0.28.0 → sqlsaber-0.29.0}/docs/src/content/docs/changelog.md

@@ -7,15 +7,26 @@ All notable changes to SQLsaber will be documented here.
 
 ### Unreleased
 
+### v0.29.0 - 2025-10-04
+
+#### Added
+
+- Theme management commands
+  - `saber theme set` - Interactively select from all available Pygments themes with searchable list
+  - `saber theme reset` - Reset to default theme (nord)
+  - Theme configuration persists across sessions
+  - Environment variable `SQLSABER_THEME` can override configured theme
+
+#### Changed
+
+- Theme manager now derives semantic colors directly from selected Pygments styles, enabling out-of-the-box support for any upstream theme while retaining user overrides and fallbacks.
+
 ### v0.28.0 - 2025-10-03
 
 #### Added
 
 - Unified theming system
-  - 7 built-in themes with exact Pygments color matching
-    - Dark themes: `nord` (default), `dracula`, `one-dark`, `material`, `lightbulb`
-    - Light themes: `solarized-light`, `vs`
-  - Easy theme switching via `SQLSABER_THEME` environment variable or config file
+- Easy theme switching via `SQLSABER_THEME` environment variable or config file
 
 #### Changed
 

{sqlsaber-0.28.0 → sqlsaber-0.29.0}/docs/src/content/docs/reference/commands.md

@@ -310,6 +310,39 @@ saber models reset
 
 ---
 
+### `saber theme`
+
+Manage syntax highlighting theme settings.
+
+#### `saber theme set`
+
+Interactively select a syntax highlighting theme from all available Pygments themes.
+
+**Usage:**
+
+```bash
+saber theme set
+```
+
+You can also set themes via environment variable:
+
+```bash
+export SQLSABER_THEME=dracula
+saber
+```
+
+#### `saber theme reset`
+
+Reset to the default theme (nord).
+
+**Usage:**
+
+```bash
+saber theme reset
+```
+
+---
+
 ### `saber threads`
 
 Manage conversation threads.

{sqlsaber-0.28.0 → sqlsaber-0.29.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "sqlsaber"
-version = "0.28.0"
+version = "0.29.0"
 description = "SQLsaber - Open-source agentic SQL assistant"
 readme = "README.md"
 requires-python = ">=3.12"

{sqlsaber-0.28.0 → sqlsaber-0.29.0}/src/sqlsaber/cli/commands.py

@@ -11,6 +11,7 @@ from sqlsaber.cli.database import create_db_app
 from sqlsaber.cli.memory import create_memory_app
 from sqlsaber.cli.models import create_models_app
 from sqlsaber.cli.onboarding import needs_onboarding, run_onboarding
+from sqlsaber.cli.theme import create_theme_app
 from sqlsaber.cli.threads import create_threads_app
 
 # Lazy imports - only import what's needed for CLI parsing
@@ -35,6 +36,7 @@ app.command(create_auth_app(), name="auth")
 app.command(create_db_app(), name="db")
 app.command(create_memory_app(), name="memory")
 app.command(create_models_app(), name="models")
+app.command(create_theme_app(), name="theme")
 app.command(create_threads_app(), name="threads")
 
 console = create_console()

sqlsaber-0.29.0/src/sqlsaber/cli/theme.py

@@ -0,0 +1,146 @@
+"""Theme management CLI commands."""
+
+import asyncio
+import json
+import os
+import sys
+from pathlib import Path
+
+import cyclopts
+import questionary
+from platformdirs import user_config_dir
+from pygments.styles import get_all_styles
+
+from sqlsaber.theme.manager import DEFAULT_THEME_NAME, create_console
+
+console = create_console()
+
+# Create the theme management CLI app
+theme_app = cyclopts.App(
+    name="theme",
+    help="Manage theme settings",
+)
+
+
+class ThemeManager:
+    """Manages theme configuration persistence."""
+
+    def __init__(self):
+        self.config_dir = Path(user_config_dir("sqlsaber"))
+        self.config_file = self.config_dir / "theme.json"
+
+    def _ensure_config_dir(self) -> None:
+        """Ensure config directory exists."""
+        self.config_dir.mkdir(parents=True, exist_ok=True)
+
+    def _load_config(self) -> dict:
+        """Load theme configuration from file."""
+        if not self.config_file.exists():
+            return {}
+
+        try:
+            with open(self.config_file, "r") as f:
+                return json.load(f)
+        except Exception:
+            return {}
+
+    def _save_config(self, config: dict) -> None:
+        """Save theme configuration to file."""
+        self._ensure_config_dir()
+
+        with open(self.config_file, "w") as f:
+            json.dump(config, f, indent=2)
+
+    def get_current_theme(self) -> str:
+        """Get the currently configured theme."""
+        config = self._load_config()
+        env_theme = os.getenv("SQLSABER_THEME")
+        if env_theme:
+            return env_theme
+        return config.get("theme", {}).get("pygments_style") or DEFAULT_THEME_NAME
+
+    def set_theme(self, theme_name: str) -> bool:
+        """Set the current theme."""
+        try:
+            config = self._load_config()
+            if "theme" not in config:
+                config["theme"] = {}
+            config["theme"]["name"] = theme_name
+            config["theme"]["pygments_style"] = theme_name
+            self._save_config(config)
+            return True
+        except Exception as e:
+            console.print(f"[error]Error setting theme: {e}[/error]")
+            return False
+
+    def reset_theme(self) -> bool:
+        """Reset to default theme."""
+        try:
+            if self.config_file.exists():
+                self.config_file.unlink()
+            return True
+        except Exception as e:
+            console.print(f"[error]Error resetting theme: {e}[/error]")
+            return False
+
+    def get_available_themes(self) -> list[str]:
+        """Get list of available Pygments themes."""
+        return sorted(get_all_styles())
+
+
+theme_manager = ThemeManager()
+
+
+@theme_app.command
+def set():
+    """Set the theme to use for syntax highlighting."""
+
+    async def interactive_set():
+        themes = theme_manager.get_available_themes()
+        current_theme = theme_manager.get_current_theme()
+
+        # Create choices with current theme highlighted
+        choices = [
+            questionary.Choice(
+                title=f"{theme} (current)" if theme == current_theme else theme,
+                value=theme,
+            )
+            for theme in themes
+        ]
+
+        selected_theme = await questionary.select(
+            "Select a theme:",
+            choices=choices,
+            default=current_theme,
+            use_search_filter=True,
+            use_jk_keys=False,
+        ).ask_async()
+
+        if selected_theme:
+            if theme_manager.set_theme(selected_theme):
+                console.print(f"[success]✓ Theme set to: {selected_theme}[/success]")
+            else:
+                console.print("[error]✗ Failed to set theme[/error]")
+                sys.exit(1)
+        else:
+            console.print("[warning]Operation cancelled[/warning]")
+
+    asyncio.run(interactive_set())
+
+
+@theme_app.command
+def reset():
+    """Reset to the default theme."""
+
+    if theme_manager.reset_theme():
+        console.print(
+            f"[success]✓ Theme reset to default: {DEFAULT_THEME_NAME}[/success]"
+        )
+    else:
+        console.print("[error]✗ Failed to reset theme[/error]")
+        sys.exit(1)
+
+
+def create_theme_app() -> cyclopts.App:
+    """Return the theme management CLI app."""
+    return theme_app

{sqlsaber-0.28.0 → sqlsaber-0.29.0}/src/sqlsaber/theme/manager.py

@@ -1,7 +1,7 @@
 """Theme management for unified theming across Rich and prompt_toolkit."""
 
+import json
 import os
-import tomllib
 from dataclasses import dataclass
 from functools import lru_cache
 from typing import Dict
@@ -10,6 +10,8 @@ from platformdirs import user_config_dir
 from prompt_toolkit.styles import Style as PTStyle
 from prompt_toolkit.styles.pygments import style_from_pygments_cls
 from pygments.styles import get_style_by_name
+from pygments.token import Token
+from pygments.util import ClassNotFound
 from rich.console import Console
 from rich.theme import Theme
 
@@ -43,89 +45,97 @@ DEFAULT_ROLE_PALETTE = {
     "title": "bold $success",
 }
 
-# Theme presets using exact Pygments colors
-THEME_PRESETS = {
-    # Nord - exact colors from pygments nord theme
-    "nord": {
-        "primary": "#81a1c1",  # Keyword (frost)
-        "accent": "#b48ead",  # Number (aurora purple)
-        "success": "#a3be8c",  # String (aurora green)
-        "warning": "#ebcb8b",  # String.Escape (aurora yellow)
-        "error": "#bf616a",  # Error/Generic.Error (aurora red)
-        "info": "#88c0d0",  # Name.Function (frost cyan)
-        "muted": "dim",
-    },
-    # Dracula - exact colors from pygments dracula theme
-    "dracula": {
-        "primary": "#bd93f9",  # purple
-        "accent": "#ff79c6",  # pink
-        "success": "#50fa7b",  # green
-        "warning": "#f1fa8c",  # yellow
-        "error": "#ff5555",  # red
-        "info": "#8be9fd",  # cyan
-        "muted": "dim",
-    },
-    # Solarized Light - exact colors from pygments solarized-light theme
-    "solarized-light": {
-        "primary": "#268bd2",  # blue
-        "accent": "#d33682",  # magenta
-        "success": "#859900",  # green
-        "warning": "#b58900",  # yellow
-        "error": "#dc322f",  # red
-        "info": "#2aa198",  # cyan
-        "muted": "dim",
-    },
-    # VS (Visual Studio Light) - exact colors from pygments vs theme
-    "vs": {
-        "primary": "#0000ff",  # Keyword (blue)
-        "accent": "#2b91af",  # Keyword.Type/Name.Class
-        "success": "#008000",  # Comment (green)
-        "warning": "#b58900",  # (using solarized yellow as fallback)
-        "error": "#dc322f",  # (using solarized red as fallback)
-        "info": "#2aa198",  # (using solarized cyan as fallback)
-        "muted": "dim",
-    },
-    # Material (approximation based on material design colors)
-    "material": {
-        "primary": "#89ddff",  # cyan
-        "accent": "#f07178",  # pink/red
-        "success": "#c3e88d",  # green
-        "warning": "#ffcb6b",  # yellow
-        "error": "#ff5370",  # red
-        "info": "#82aaff",  # blue
-        "muted": "dim",
-    },
-    # One Dark - exact colors from pygments one-dark theme
-    "one-dark": {
-        "primary": "#c678dd",  # Keyword (purple)
-        "accent": "#e06c75",  # Name (red)
-        "success": "#98c379",  # String (green)
-        "warning": "#e5c07b",  # Keyword.Type (yellow)
-        "error": "#e06c75",  # Name (red, used for errors)
-        "info": "#61afef",  # Name.Function (blue)
-        "muted": "dim",
-    },
-    # Lightbulb - exact colors from pygments lightbulb theme (minimal dark)
-    "lightbulb": {
-        "primary": "#73d0ff",  # Keyword.Type/Name.Class (blue_1)
-        "accent": "#dfbfff",  # Number (magenta_1)
-        "success": "#d5ff80",  # String (green_1)
-        "warning": "#ffd173",  # Name.Function (yellow_1)
-        "error": "#f88f7f",  # Error (red_1)
-        "info": "#95e6cb",  # Name.Entity (cyan_1)
-        "muted": "dim",
-    },
+ROLE_TOKEN_PREFERENCES: dict[str, tuple] = {
+    "primary": (
+        Token.Keyword,
+        Token.Keyword.Namespace,
+        Token.Name.Tag,
+    ),
+    "accent": (
+        Token.Name.Tag,
+        Token.Keyword.Type,
+        Token.Literal.Number,
+        Token.Operator.Word,
+    ),
+    "success": (
+        Token.Literal.String,
+        Token.Generic.Inserted,
+        Token.Name.Attribute,
+    ),
+    "warning": (
+        Token.Literal.String.Escape,
+        Token.Name.Constant,
+        Token.Generic.Emph,
+    ),
+    "error": (
+        Token.Error,
+        Token.Generic.Error,
+        Token.Generic.Deleted,
+        Token.Name.Exception,
+    ),
+    "info": (
+        Token.Name.Function,
+        Token.Name.Builtin,
+        Token.Keyword.Type,
+    ),
+    "muted": (
+        Token.Comment,
+        Token.Generic.Subheading,
+        Token.Text,
+    ),
 }
 
 
+def _normalize_hex(color: str | None) -> str | None:
+    if not color:
+        return None
+    color = color.strip()
+    if not color:
+        return None
+    if color.startswith("#"):
+        color = color[1:]
+    if len(color) == 3:
+        color = "".join(ch * 2 for ch in color)
+    if len(color) != 6:
+        return None
+    return f"#{color.lower()}"
+
+
+def _build_role_palette_from_style(style_name: str) -> dict[str, str]:
+    try:
+        style_cls = get_style_by_name(style_name)
+    except ClassNotFound:
+        return {}
+
+    palette: dict[str, str] = {}
+    try:
+        base_color = _normalize_hex(style_cls.style_for_token(Token.Text).get("color"))
+    except KeyError:
+        base_color = None
+    for role, tokens in ROLE_TOKEN_PREFERENCES.items():
+        for token in tokens:
+            try:
+                style_def = style_cls.style_for_token(token)
+            except KeyError:
+                continue
+            color = _normalize_hex(style_def.get("color"))
+            if not color or color == base_color:
+                continue
+            if role == "accent" and color == palette.get("primary"):
+                continue
+            palette[role] = color
+            break
+    return palette
+
+
 def _load_user_theme_config() -> dict:
     """Load theme configuration from user config directory."""
     cfg_dir = user_config_dir("sqlsaber")
-    path = os.path.join(cfg_dir, "theme.toml")
+    path = os.path.join(cfg_dir, "theme.json")
     if not os.path.exists(path):
         return {}
-    with open(path, "rb") as f:
-        return tomllib.load(f)
+    with open(path, "r") as f:
+        return json.load(f)
 
 
 def _resolve_refs(palette: dict[str, str]) -> dict[str, str]:
@@ -204,7 +214,7 @@ def get_theme_manager() -> ThemeManager:
     pygments_style = user_cfg.get("theme", {}).get("pygments_style") or name
 
     roles = dict(DEFAULT_ROLE_PALETTE)
-    roles.update(THEME_PRESETS.get(name, {}))
+    roles.update(_build_role_palette_from_style(pygments_style))
     roles.update(user_cfg.get("roles", {}))
 
     cfg = ThemeConfig(name=name, pygments_style=pygments_style, roles=roles)

sqlsaber-0.29.0/tests/test_theme/test_manager.py

@@ -0,0 +1,38 @@
+import pytest
+
+from sqlsaber.theme import manager
+
+
+@pytest.fixture(autouse=True)
+def clear_theme_cache():
+    manager.get_theme_manager.cache_clear()
+    yield
+    manager.get_theme_manager.cache_clear()
+
+
+def test_dynamic_palette_uses_pygments_colors(monkeypatch):
+    monkeypatch.setenv("SQLSABER_THEME", "zenburn")
+
+    tm = manager.get_theme_manager()
+
+    assert tm.pygments_style_name == "zenburn"
+    assert tm.style("primary") == "#efdcbc"
+    assert tm.style("accent") == "#e89393"
+    assert tm.style("success") == "#cc9393"
+    assert tm.style("error") == "#e37170"
+    assert tm.style("info") == "#efef8f"
+    assert tm.style("muted") == "#7f9f7f"
+
+    monkeypatch.delenv("SQLSABER_THEME", raising=False)
+
+
+def test_unknown_theme_falls_back_to_defaults(monkeypatch):
+    monkeypatch.setenv("SQLSABER_THEME", "does-not-exist")
+
+    tm = manager.get_theme_manager()
+
+    assert tm.pygments_style_name == "does-not-exist"
+    assert tm.style("primary") == manager.DEFAULT_ROLE_PALETTE["primary"]
+    assert tm.style("accent") == manager.DEFAULT_ROLE_PALETTE["accent"]
+
+    monkeypatch.delenv("SQLSABER_THEME", raising=False)

{sqlsaber-0.28.0 → sqlsaber-0.29.0}/uv.lock

@@ -1888,7 +1888,7 @@ wheels = [
 
 [[package]]
 name = "sqlsaber"
-version = "0.28.0"
+version = "0.29.0"
 source = { editable = "." }
 dependencies = [
     { name = "aiomysql" },