ui_utils.py

"""
UI Utilities Module

Helper utilities for improved user experience including tooltips and notifications.
"""

import customtkinter as ctk
from typing import Optional
import tkinter as tk


class ToolTip:
    """
    Create tooltip for a widget.
    """
    
    def __init__(self, widget, text: str, delay: int = 500):
        """
        Initialize tooltip.
        
        Args:
            widget: Widget to attach tooltip to
            text: Tooltip text
            delay: Delay in milliseconds before showing
        """
        self.widget = widget
        self.text = text
        self.delay = delay
        self.tooltip_window = None
        self.id = None
        
        # Bind events
        self.widget.bind('<Enter>', self.on_enter)
        self.widget.bind('<Leave>', self.on_leave)
        self.widget.bind('<Button>', self.on_leave)
    
    def on_enter(self, event=None):
        """Schedule tooltip display."""
        self.schedule()
    
    def on_leave(self, event=None):
        """Hide and unschedule tooltip."""
        self.unschedule()
        self.hide()
    
    def schedule(self):
        """Schedule tooltip to appear."""
        self.unschedule()
        self.id = self.widget.after(self.delay, self.show)
    
    def unschedule(self):
        """Cancel scheduled tooltip."""
        if self.id:
            self.widget.after_cancel(self.id)
            self.id = None
    
    def show(self):
        """Display the tooltip."""
        if self.tooltip_window:
            return
        
        # Get widget position
        x = self.widget.winfo_rootx() + 20
        y = self.widget.winfo_rooty() + self.widget.winfo_height() + 5
        
        # Create tooltip window
        self.tooltip_window = tk.Toplevel(self.widget)
        self.tooltip_window.wm_overrideredirect(True)
        self.tooltip_window.wm_geometry(f"+{x}+{y}")
        
        # Create label with text
        label = tk.Label(
            self.tooltip_window,
            text=self.text,
            background="#2b2b2b",
            foreground="white",
            relief="solid",
            borderwidth=1,
            font=("Arial", 9),
            padx=8,
            pady=4
        )
        label.pack()
    
    def hide(self):
        """Hide the tooltip."""
        if self.tooltip_window:
            self.tooltip_window.destroy()
            self.tooltip_window = None


class StatusBar(ctk.CTkFrame):
    """
    Professional status bar for the application.
    """
    
    def __init__(self, parent):
        """
        Initialize status bar.
        
        Args:
            parent: Parent widget
        """
        super().__init__(parent, height=30)
        
        self.grid_columnconfigure(0, weight=1)
        
        # Status message label
        self.status_label = ctk.CTkLabel(
            self,
            text="Ready",
            font=ctk.CTkFont(size=10),
            anchor="w"
        )
        self.status_label.grid(row=0, column=0, padx=10, sticky="w")
        
        # Connection indicator
        self.connection_label = ctk.CTkLabel(
            self,
            text="⚫ Disconnected",
            font=ctk.CTkFont(size=10),
            text_color="gray"
        )
        self.connection_label.grid(row=0, column=1, padx=10)
        
        # Activity indicator
        self.activity_label = ctk.CTkLabel(
            self,
            text="",
            font=ctk.CTkFont(size=10)
        )
        self.activity_label.grid(row=0, column=2, padx=10)
    
    def set_status(self, message: str, type: str = "info"):
        """
        Set status message.
        
        Args:
            message: Status message
            type: Message type ('info', 'success', 'warning', 'error')
        """
        colors = {
            'info': 'white',
            'success': 'green',
            'warning': 'orange',
            'error': 'red'
        }
        
        self.status_label.configure(
            text=message,
            text_color=colors.get(type, 'white')
        )
    
    def set_connection(self, connected: bool, mode: str = ""):
        """
        Update connection status.
        
        Args:
            connected: Connection state
            mode: Trading mode (PAPER/LIVE)
        """
        if connected:
            color = "orange" if mode == "PAPER" else "green"
            symbol = "🟢" if mode == "LIVE" else "🟠"
            self.connection_label.configure(
                text=f"{symbol} Connected ({mode})",
                text_color=color
            )
        else:
            self.connection_label.configure(
                text="⚫ Disconnected",
                text_color="gray"
            )
    
    def show_activity(self, message: str = ""):
        """
        Show activity indicator.
        
        Args:
            message: Activity message
        """
        if message:
            self.activity_label.configure(text=f"⏳ {message}")
        else:
            self.activity_label.configure(text="")


def create_labeled_entry(parent, label: str, row: int, tooltip: Optional[str] = None, **kwargs):
    """
    Create labeled entry with optional tooltip.
    
    Args:
        parent: Parent widget
        label: Label text
        row: Grid row
        tooltip: Optional tooltip text
        **kwargs: Additional entry arguments
    
    Returns:
        Entry widget
    """
    # Label
    lbl = ctk.CTkLabel(parent, text=label + ":")
    lbl.grid(row=row, column=0, padx=10, pady=5, sticky="e")
    
    if tooltip:
        ToolTip(lbl, tooltip)
    
    # Entry
    entry = ctk.CTkEntry(parent, **kwargs)
    entry.grid(row=row, column=1, padx=10, pady=5, sticky="ew")
    
    return entry


def show_confirmation(parent, title: str, message: str, on_confirm=None):
    """
    Show confirmation dialog.
    
    Args:
        parent: Parent widget
        title: Dialog title
        message: Confirmation message
        on_confirm: Callback function if confirmed
    """
    dialog = ctk.CTkToplevel(parent)
    dialog.title(title)
    dialog.geometry("400x200")
    dialog.transient(parent)
    dialog.grab_set()
    
    # Center dialog
    dialog.update_idletasks()
    x = parent.winfo_x() + (parent.winfo_width() // 2) - (dialog.winfo_width() // 2)
    y = parent.winfo_y() + (parent.winfo_height() // 2) - (dialog.winfo_height() // 2)
    dialog.geometry(f"+{x}+{y}")
    
    # Message
    ctk.CTkLabel(
        dialog,
        text=message,
        font=ctk.CTkFont(size=12),
        wraplength=350
    ).pack(pady=30, padx=20)
    
    # Buttons
    button_frame = ctk.CTkFrame(dialog, fg_color="transparent")
    button_frame.pack(pady=10)
    
    def on_yes():
        dialog.destroy()
        if on_confirm:
            on_confirm()
    
    def on_no():
        dialog.destroy()
    
    ctk.CTkButton(
        button_frame,
        text="Yes",
        command=on_yes,
        fg_color="green",
        hover_color="darkgreen",
        width=100
    ).pack(side="left", padx=10)
    
    ctk.CTkButton(
        button_frame,
        text="No",
        command=on_no,
        width=100
    ).pack(side="left", padx=10)