theme_manager.py

"""
Theme Manager for NoSQL2SQL
Provides dark/light theme colors and applies them to tkinter widgets
"""

import tkinter as tk
from tkinter import ttk


class ThemeManager:
    """Manages application theming"""
    
    # Light Theme Colors
    LIGHT_THEME = {
        "bg": "#FFFFFF",
        "fg": "#000000",
        "bg_secondary": "#F5F5F5",
        "fg_secondary": "#666666",
        "accent": "#007AFF",
        "accent_hover": "#0051D5",
        "success": "#34C759",
        "error": "#FF3B30",
        "warning": "#FF9500",
        "border": "#D1D1D6",
        "input_bg": "#FFFFFF",
        "input_fg": "#000000",
        "button_bg": "#007AFF",
        "button_fg": "#FFFFFF",
        "disabled": "#C7C7CC",
        "selection": "#007AFF",
        "text_bg": "#FFFFFF",
        "text_fg": "#000000",
        "heading": "#000000",
        "link": "#007AFF"
    }
    
    # Dark Theme Colors
    DARK_THEME = {
        "bg": "#1C1C1E",
        "fg": "#FFFFFF",
        "bg_secondary": "#2C2C2E",
        "fg_secondary": "#ABABAB",
        "accent": "#0A84FF",
        "accent_hover": "#409CFF",
        "success": "#30D158",
        "error": "#FF453A",
        "warning": "#FF9F0A",
        "border": "#38383A",
        "input_bg": "#2C2C2E",
        "input_fg": "#FFFFFF",
        "button_bg": "#0A84FF",
        "button_fg": "#FFFFFF",
        "disabled": "#48484A",
        "selection": "#0A84FF",
        "text_bg": "#2C2C2E",
        "text_fg": "#FFFFFF",
        "heading": "#FFFFFF",
        "link": "#0A84FF"
    }
    
    def __init__(self, root, theme_name="light"):
        self.root = root
        self.theme_name = theme_name
        self.colors = self.LIGHT_THEME if theme_name == "light" else self.DARK_THEME
        self.style = ttk.Style(root)
        self._setup_style()
    
    def _setup_style(self):
        """Setup ttk styles based on theme"""
        
        # Configure root window
        self.root.configure(bg=self.colors["bg"])
        
        # Base ttk style
        self.style.theme_use('clam')
        
        # Configure basic widgets
        self.style.configure(".",
                           background=self.colors["bg"],
                           foreground=self.colors["fg"],
                           bordercolor=self.colors["border"],
                           darkcolor=self.colors["bg_secondary"],
                           lightcolor=self.colors["bg"],
                           troughcolor=self.colors["bg_secondary"],
                           focuscolor=self.colors["accent"],
                           selectbackground=self.colors["selection"],
                           selectforeground=self.colors["button_fg"])
        
        # Frame
        self.style.configure("TFrame",
                           background=self.colors["bg"])
        
        # Label
        self.style.configure("TLabel",
                           background=self.colors["bg"],
                           foreground=self.colors["fg"])
        
        self.style.configure("Heading.TLabel",
                           background=self.colors["bg"],
                           foreground=self.colors["heading"],
                           font=("Arial", 12, "bold"))
        
        self.style.configure("Secondary.TLabel",
                           background=self.colors["bg"],
                           foreground=self.colors["fg_secondary"],
                           font=("Arial", 9))
        
        # Button
        self.style.configure("TButton",
                           background=self.colors["button_bg"],
                           foreground=self.colors["button_fg"],
                           bordercolor=self.colors["border"],
                           focuscolor=self.colors["accent"],
                           padding=(10, 5))
        
        self.style.map("TButton",
                      background=[("active", self.colors["accent_hover"]),
                                ("disabled", self.colors["disabled"])])
        
        self.style.configure("Accent.TButton",
                           background=self.colors["accent"],
                           foreground=self.colors["button_fg"],
                           font=("Arial", 10, "bold"))
        
        self.style.map("Accent.TButton",
                      background=[("active", self.colors["accent_hover"])])
        
        # Entry
        self.style.configure("TEntry",
                           fieldbackground=self.colors["input_bg"],
                           foreground=self.colors["input_fg"],
                           bordercolor=self.colors["border"],
                           insertcolor=self.colors["fg"])
        
        # Combobox
        self.style.configure("TCombobox",
                           fieldbackground=self.colors["input_bg"],
                           background=self.colors["bg"],
                           foreground=self.colors["input_fg"],
                           bordercolor=self.colors["border"],
                           arrowcolor=self.colors["fg"],
                           selectbackground=self.colors["selection"],
                           selectforeground=self.colors["button_fg"])
        
        self.style.map("TCombobox",
                      fieldbackground=[("readonly", self.colors["input_bg"])],
                      selectbackground=[("readonly", self.colors["input_bg"])])
        
        # Checkbutton
        self.style.configure("TCheckbutton",
                           background=self.colors["bg"],
                           foreground=self.colors["fg"])
        
        # Radiobutton
        self.style.configure("TRadiobutton",
                           background=self.colors["bg"],
                           foreground=self.colors["fg"])
        
        # LabelFrame
        self.style.configure("TLabelframe",
                           background=self.colors["bg"],
                           bordercolor=self.colors["border"])
        
        self.style.configure("TLabelframe.Label",
                           background=self.colors["bg"],
                           foreground=self.colors["fg"],
                           font=("Arial", 10, "bold"))
        
        # Notebook (Tabs)
        self.style.configure("TNotebook",
                           background=self.colors["bg"],
                           bordercolor=self.colors["border"])
        
        self.style.configure("TNotebook.Tab",
                           background=self.colors["bg_secondary"],
                           foreground=self.colors["fg"],
                           padding=(10, 5))
        
        self.style.map("TNotebook.Tab",
                      background=[("selected", self.colors["bg"])],
                      foreground=[("selected", self.colors["accent"])])
        
        # Progressbar
        self.style.configure("TProgressbar",
                           background=self.colors["accent"],
                           troughcolor=self.colors["bg_secondary"],
                           bordercolor=self.colors["border"],
                           lightcolor=self.colors["accent"],
                           darkcolor=self.colors["accent"])
        
        # Scrollbar
        self.style.configure("TScrollbar",
                           background=self.colors["bg_secondary"],
                           troughcolor=self.colors["bg"],
                           bordercolor=self.colors["border"],
                           arrowcolor=self.colors["fg"])
        
        self.style.map("TScrollbar",
                      background=[("active", self.colors["border"])])
        
        # Separator
        self.style.configure("TSeparator",
                           background=self.colors["border"])
    
    def switch_theme(self, theme_name):
        """Switch to a different theme"""
        self.theme_name = theme_name
        self.colors = self.LIGHT_THEME if theme_name == "light" else self.DARK_THEME
        self._setup_style()
    
    def get_color(self, color_name):
        """Get a specific color from current theme"""
        return self.colors.get(color_name, "#000000")
    
    def apply_to_text_widget(self, text_widget):
        """Apply theme to a Text widget"""
        text_widget.configure(
            bg=self.colors["text_bg"],
            fg=self.colors["text_fg"],
            insertbackground=self.colors["fg"],
            selectbackground=self.colors["selection"],
            selectforeground=self.colors["button_fg"]
        )
    
    def apply_to_listbox(self, listbox):
        """Apply theme to a Listbox widget"""
        listbox.configure(
            bg=self.colors["text_bg"],
            fg=self.colors["text_fg"],
            selectbackground=self.colors["selection"],
            selectforeground=self.colors["button_fg"]
        )
    
    def apply_to_menu(self, menu):
        """Apply theme to a Menu widget"""
        menu.configure(
            bg=self.colors["bg"],
            fg=self.colors["fg"],
            activebackground=self.colors["accent"],
            activeforeground=self.colors["button_fg"],
            borderwidth=1,
            relief=tk.FLAT
        )
    
    def apply_to_window(self, window):
        """Apply current theme to a Toplevel window"""
        window.configure(bg=self.colors["bg"])
        
        # Apply to all child widgets recursively
        self._apply_to_widget_tree(window)
    
    def _apply_to_widget_tree(self, widget):
        """Recursively apply theme to widget and its children"""
        # Apply to current widget
        try:
            if isinstance(widget, (tk.Text, tk.Entry)):
                widget.configure(
                    bg=self.colors["text_bg"],
                    fg=self.colors["text_fg"],
                    insertbackground=self.colors["fg"],
                    selectbackground=self.colors["selection"],
                    selectforeground=self.colors["button_fg"]
                )
            elif isinstance(widget, tk.Frame):
                widget.configure(bg=self.colors["bg"])
            elif isinstance(widget, tk.Label):
                widget.configure(
                    bg=self.colors["bg"],
                    fg=self.colors["fg"]
                )
            elif isinstance(widget, tk.Toplevel):
                widget.configure(bg=self.colors["bg"])
        except:
            pass  # Some widgets don't support all options
        
        # Recursively apply to children
        try:
            for child in widget.winfo_children():
                self._apply_to_widget_tree(child)
        except:
            pass


if __name__ == "__main__":
    # Test the theme manager
    root = tk.Tk()
    root.title("Theme Manager Test")
    root.geometry("400x300")
    
    theme_manager = ThemeManager(root, "dark")
    
    frame = ttk.Frame(root, padding="20")
    frame.pack(fill=tk.BOTH, expand=True)
    
    ttk.Label(frame, text="Theme Manager Test", style="Heading.TLabel").pack(pady=5)
    ttk.Label(frame, text="This is a test label").pack(pady=5)
    ttk.Button(frame, text="Test Button").pack(pady=5)
    ttk.Entry(frame).pack(pady=5)
    
    def toggle_theme():
        current = theme_manager.theme_name
        new_theme = "light" if current == "dark" else "dark"
        theme_manager.switch_theme(new_theme)
    
    ttk.Button(frame, text="Toggle Theme", command=toggle_theme).pack(pady=10)
    
    root.mainloop()