sqlsaber 0.28.0__py3-none-any.whl → 0.29.0__py3-none-any.whl

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/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/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.28.0.dist-info → sqlsaber-0.29.0.dist-info}/METADATA

@@ -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.dist-info → sqlsaber-0.29.0.dist-info}/RECORD

@@ -11,7 +11,7 @@ sqlsaber/application/model_selection.py,sha256=xZI-nvUgYZikaTK38SCmEWvWSfRsDpFu2
 sqlsaber/application/prompts.py,sha256=4rMGcWpYJbNWPMzqVWseUMx0nwvXOkWS6GaTAJ5mhfc,3473
 sqlsaber/cli/__init__.py,sha256=qVSLVJLLJYzoC6aj6y9MFrzZvAwc4_OgxU9DlkQnZ4M,86
 sqlsaber/cli/auth.py,sha256=ysDBXEFR8Jz7wYbIP6X7yWA2ivd8SDnUp_jUg_qYNWk,6088
-sqlsaber/cli/commands.py,sha256=-rTxr-kW7j2rR8wAg0tATKoh284pMDPKVMpQKaJwtqk,8540
+sqlsaber/cli/commands.py,sha256=LiHLMAZF9fNgQXShtF67dzDvp-w4LlSbmeZIa47ntC0,8634
 sqlsaber/cli/completers.py,sha256=g-hLDq5fiBx7gg8Bte1Lq8GU-ZxCYVs4dcPsmHPIcK4,6574
 sqlsaber/cli/database.py,sha256=hh8PdWnhaD0fO2jwvSSQyxsjwk-JyvmcY7f5tuHfnAQ,10663
 sqlsaber/cli/display.py,sha256=WB5JCumhXadziDEX1EZHG3vN1Chol5FNAaTXHieqFK0,17892
@@ -20,6 +20,7 @@ sqlsaber/cli/memory.py,sha256=gKP-JJ0w1ya1YTM_Lk7Gw-7wL9ptyj6cZtg-uoW8K7A,7818
 sqlsaber/cli/models.py,sha256=NozZbnisSjbPKo7PW7CltJMIkGcPqTDpDQEY-C_eLhk,8504
 sqlsaber/cli/onboarding.py,sha256=l6FFWn8J1OVQUxr-xIAzKaFhAz8rFh6IEWwIyPWqR6U,11438
 sqlsaber/cli/streaming.py,sha256=1XoZGPPMoTmBQVgp_Bqk483MR93j9oXxSV6Tx_-TpOg,6923
+sqlsaber/cli/theme.py,sha256=hP0kmsMLCtqaT7b5wB1dk1hW1hV94oP4BHdz8S6887A,4243
 sqlsaber/cli/threads.py,sha256=5EV4ckRzKqhWeTKpTfQSNCBuqs3onsJURKT09g4E4XM,13634
 sqlsaber/config/__init__.py,sha256=olwC45k8Nc61yK0WmPUk7XHdbsZH9HuUAbwnmKe3IgA,100
 sqlsaber/config/api_keys.py,sha256=bjogRmIuxNNGusyKXKi0ZpJWeS5Fyn53zrAD8hsoYx4,3671
@@ -44,7 +45,7 @@ sqlsaber/memory/__init__.py,sha256=GiWkU6f6YYVV0EvvXDmFWe_CxarmDCql05t70MkTEWs,6
 sqlsaber/memory/manager.py,sha256=p3fybMVfH-E4ApT1ZRZUnQIWSk9dkfUPCyfkmA0HALs,2739
 sqlsaber/memory/storage.py,sha256=ne8szLlGj5NELheqLnI7zu21V8YS4rtpYGGC7tOmi-s,5745
 sqlsaber/theme/__init__.py,sha256=qCICX1Cg4B6yCbZ1UrerxglWxcqldRFVSRrSs73na_8,188
-sqlsaber/theme/manager.py,sha256=0DWuVXn7JoC8NvAl5FSqc61eagKFTx5YnoY8SoCTxGM,7236
+sqlsaber/theme/manager.py,sha256=7QXoqPvl-aKfvqTRduWYwnHsySu66Gg8a3-QQkVM5Ss,6551
 sqlsaber/threads/__init__.py,sha256=Hh3dIG1tuC8fXprREUpslCIgPYz8_6o7aRLx4yNeO48,139
 sqlsaber/threads/storage.py,sha256=rsUdxT4CR52D7xtGir9UlsFnBMk11jZeflzDrk2q4ME,11183
 sqlsaber/tools/__init__.py,sha256=x3YdmX_7P0Qq_HtZHAgfIVKTLxYqKk6oc4tGsujQWsc,586
@@ -54,8 +55,8 @@ sqlsaber/tools/instructions.py,sha256=X-x8maVkkyi16b6Tl0hcAFgjiYceZaSwyWTfmrvx8U
 sqlsaber/tools/registry.py,sha256=HWOQMsNIdL4XZS6TeNUyrL-5KoSDH6PHsWd3X66o-18,3211
 sqlsaber/tools/sql_guard.py,sha256=dTDwcZP-N4xPGzcr7MQtKUxKrlDzlc1irr9aH5a4wvk,6182
 sqlsaber/tools/sql_tools.py,sha256=ujmAcfLkNaBrb5LWEgWcINQEQSX0LRPX3VK5Dag1Sj4,9178
-sqlsaber-0.28.0.dist-info/METADATA,sha256=oY9Awl9jLkdeJLa1oeTPHPGH94KktJ_HmBVUVLQs-do,7174
-sqlsaber-0.28.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-sqlsaber-0.28.0.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
-sqlsaber-0.28.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
-sqlsaber-0.28.0.dist-info/RECORD,,
+sqlsaber-0.29.0.dist-info/METADATA,sha256=5Ocm_GD3MIdPLji_VpknTjiQ0ndDd2waD14dr6_IOjg,7174
+sqlsaber-0.29.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+sqlsaber-0.29.0.dist-info/entry_points.txt,sha256=qEbOB7OffXPFgyJc7qEIJlMEX5RN9xdzLmWZa91zCQQ,162
+sqlsaber-0.29.0.dist-info/licenses/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
+sqlsaber-0.29.0.dist-info/RECORD,,