"""
Sidebar — server list with search, add/edit/delete buttons, context menu.
"""

import tkinter as tk
import customtkinter as ctk
from core.i18n import t
from core.icons import (
    icon_text, TYPE_COLORS, TYPE_LABELS, CTX_ICONS, icon,
)
from gui.widgets.status_badge import StatusBadge


# Context menu: type → list of (i18n_key, tab_key_or_None)
_CONTEXT_ACTIONS = {
    "ssh":        [("ctx_open_terminal", "terminal"), ("ctx_browse_files", "files"), ("ctx_install_key", "keys")],
    "telnet":     [("ctx_open_terminal", "terminal")],
    "winrm":      [("ctx_open_powershell", "powershell")],
    "mariadb":    [("ctx_open_query", "query")],
    "mssql":      [("ctx_open_query", "query")],
    "postgresql":  [("ctx_open_query", "query")],
    "redis":      [("ctx_open_console", "console")],
    "grafana":    [("ctx_open_browser", None)],
    "prometheus": [("ctx_open_browser", None)],
    "rdp":        [("ctx_connect", "launch")],
    "vnc":        [("ctx_connect", "launch")],
}


class Sidebar(ctk.CTkFrame):
    def __init__(self, master, store, on_select=None, session_pool=None):
        super().__init__(master, width=250, corner_radius=0)
        self.store = store
        self.on_select = on_select
        self.session_pool = session_pool
        self._selected_alias: str | None = None
        self._server_frames: dict[str, ctk.CTkFrame] = {}
        self._badges: dict[str, StatusBadge] = {}
        self._session_indicators: dict[str, ctk.CTkLabel] = {}

        self.pack_propagate(False)

        # Title
        self.title_label = ctk.CTkLabel(self, text=t("servers"), font=ctk.CTkFont(size=18, weight="bold"))
        self.title_label.pack(padx=15, pady=(15, 5))

        # Search
        self.search_var = ctk.StringVar()
        self.search_var.trace_add("write", lambda *_: self._refresh_list())
        self.search_entry = ctk.CTkEntry(self, placeholder_text=t("search"), textvariable=self.search_var)
        self.search_entry.pack(fill="x", padx=10, pady=(5, 10))

        # Server list
        self.list_frame = ctk.CTkScrollableFrame(self, fg_color="transparent")
        self.list_frame.pack(fill="both", expand=True, padx=5, pady=0)

        # Active sessions label
        self._sessions_label = ctk.CTkLabel(
            self, text="", font=ctk.CTkFont(size=10),
            text_color="#6b7280", anchor="w"
        )
        self._sessions_label.pack(fill="x", padx=15, pady=(0, 2))

        # Buttons
        btn_frame = ctk.CTkFrame(self, fg_color="transparent")
        btn_frame.pack(fill="x", padx=10, pady=10)
        self.add_btn = ctk.CTkButton(btn_frame, text=icon_text("add", t("add")), width=70, height=30, command=self._on_add)
        self.add_btn.pack(side="left", padx=(0, 3))
        self.edit_btn = ctk.CTkButton(btn_frame, text=icon_text("edit", t("edit")), width=70, height=30, fg_color="#6b7280", command=self._on_edit)
        self.edit_btn.pack(side="left", padx=3)
        self.del_btn = ctk.CTkButton(btn_frame, text=icon_text("delete", t("delete")), width=70, height=30, fg_color="#ef4444", hover_color="#dc2626", command=self._on_delete)
        self.del_btn.pack(side="right", padx=(3, 0))

        # Callbacks — set by app.py
        self.add_callback = None
        self.edit_callback = None
        self.delete_callback = None
        self.open_tab_callback = None       # (alias, tab_key) → select server + switch tab
        self.check_status_callback = None   # (alias) → check single server
        self.open_browser_callback = None   # (alias) → open server URL in browser

        # Subscribe to store changes
        self.store.subscribe(self._refresh_list)
        self._refresh_list()

    def update_language(self):
        self.title_label.configure(text=t("servers"))
        self.search_entry.configure(placeholder_text=t("search"))
        self.add_btn.configure(text=icon_text("add", t("add")))
        self.edit_btn.configure(text=icon_text("edit", t("edit")))
        self.del_btn.configure(text=icon_text("delete", t("delete")))
        self._update_sessions_label()

    def _refresh_list(self):
        # Clear
        for widget in self.list_frame.winfo_children():
            widget.destroy()
        self._server_frames.clear()
        self._badges.clear()
        self._session_indicators.clear()

        # Get active sessions from pool
        active_aliases = set()
        if self.session_pool:
            active_aliases = set(self.session_pool.get_active_sessions())

        search = self.search_var.get().lower()
        servers = self.store.get_all()

        for server in servers:
            alias = server["alias"]
            ip = server["ip"]
            stype = server.get("type", "ssh")

            if search and search not in alias.lower() and search not in ip.lower():
                continue

            frame = ctk.CTkFrame(self.list_frame, cursor="hand2", height=45)
            frame.pack(fill="x", padx=2, pady=2)
            frame.pack_propagate(False)

            # Status badge
            badge = StatusBadge(frame, status=self.store.get_status(alias))
            badge.pack(side="left", padx=(10, 5), pady=10)
            self._badges[alias] = badge

            # Type badge (colored short label)
            type_color = TYPE_COLORS.get(stype, "#6b7280")
            type_label_text = TYPE_LABELS.get(stype, stype.upper()[:3])
            type_badge = ctk.CTkLabel(
                frame, text=type_label_text,
                font=ctk.CTkFont(size=9, weight="bold"),
                text_color=type_color,
                width=30
            )
            type_badge.pack(side="left", padx=(0, 2), pady=10)

            # Active session indicator (right side)
            session_ind = ctk.CTkLabel(
                frame, text="", width=12, height=12,
                font=ctk.CTkFont(size=8)
            )
            session_ind.pack(side="right", padx=(0, 8), pady=10)
            if alias in active_aliases:
                session_ind.configure(text="\u25cf", text_color="#22c55e")  # green dot
            self._session_indicators[alias] = session_ind

            # Info
            info = ctk.CTkFrame(frame, fg_color="transparent")
            info.pack(side="left", fill="both", expand=True, padx=5)

            name_label = ctk.CTkLabel(info, text=alias, font=ctk.CTkFont(size=13, weight="bold"), anchor="w")
            name_label.pack(fill="x")
            detail_label = ctk.CTkLabel(info, text=ip, font=ctk.CTkFont(size=10), text_color="#9ca3af", anchor="w")
            detail_label.pack(fill="x")

            # Click handlers
            for widget in [frame, info, name_label, detail_label, badge, type_badge, session_ind]:
                widget.bind("<Button-1>", lambda e, a=alias: self._select(a))
                widget.bind("<Button-3>", lambda e, a=alias: self._show_context_menu(e, a))

            self._server_frames[alias] = frame

        self._highlight_selected()
        self._update_sessions_label()

    def _select(self, alias: str):
        self._selected_alias = alias
        self._highlight_selected()
        if self.on_select:
            self.on_select(alias)

    def _highlight_selected(self):
        for alias, frame in self._server_frames.items():
            if alias == self._selected_alias:
                frame.configure(fg_color=("#3b82f6", "#1d4ed8"))
            else:
                frame.configure(fg_color=("gray85", "gray20"))

    def get_selected(self) -> str | None:
        return self._selected_alias

    def update_statuses(self):
        for alias, badge in self._badges.items():
            badge.set_status(self.store.get_status(alias))

    def update_session_indicators(self):
        """Update active session indicators from session pool."""
        if not self.session_pool:
            return
        active_aliases = set(self.session_pool.get_active_sessions())
        for alias, ind in self._session_indicators.items():
            if alias in active_aliases:
                ind.configure(text="\u25cf", text_color="#22c55e")
            else:
                ind.configure(text="")
        self._update_sessions_label()

    def _update_sessions_label(self):
        """Update the active sessions count label."""
        if self.session_pool:
            count = len(self.session_pool.get_active_sessions())
            if count > 0:
                self._sessions_label.configure(text=t("active_sessions").format(count=count))
            else:
                self._sessions_label.configure(text="")
        else:
            self._sessions_label.configure(text="")

    def _on_add(self):
        if self.add_callback:
            self.add_callback()

    def _on_edit(self):
        if self.edit_callback and self._selected_alias:
            self.edit_callback(self._selected_alias)

    def _on_delete(self):
        if self.delete_callback and self._selected_alias:
            self.delete_callback(self._selected_alias)

    def _show_context_menu(self, event, alias: str):
        """Show right-click context menu for a server item."""
        self._select(alias)

        server = self.store.get_server(alias)
        if not server:
            return
        stype = server.get("type", "ssh")

        menu = tk.Menu(self, tearoff=0, bg="#2d2d44", fg="#d3d7cf",
                       activebackground="#44447a", activeforeground="#ffffff",
                       font=("Segoe UI", 10))

        # Type-specific actions
        actions = _CONTEXT_ACTIONS.get(stype, [])
        for label_key, tab_key in actions:
            ctx_icon = icon(CTX_ICONS.get(label_key, ""))
            label_text = f"{ctx_icon} {t(label_key)}" if ctx_icon else t(label_key)
            if tab_key:
                menu.add_command(
                    label=label_text,
                    command=lambda a=alias, tk=tab_key: (
                        self.open_tab_callback(a, tk) if self.open_tab_callback else None
                    ),
                )
            else:
                menu.add_command(
                    label=label_text,
                    command=lambda a=alias: (
                        self.open_browser_callback(a) if self.open_browser_callback else None
                    ),
                )

        if actions:
            menu.add_separator()

        # Universal actions
        menu.add_command(
            label=icon_text("status_check", t("ctx_check_status")),
            command=lambda: (
                self.check_status_callback(alias) if self.check_status_callback else None
            ),
        )
        menu.add_command(
            label=icon_text("copy", t("ctx_copy_alias")),
            command=lambda: self._copy_alias(alias),
        )

        menu.add_separator()

        # Management
        menu.add_command(
            label=icon_text("edit", t("edit")),
            command=lambda: self.edit_callback(alias) if self.edit_callback else None,
        )
        menu.add_command(
            label=icon_text("delete", t("delete")),
            command=lambda: self.delete_callback(alias) if self.delete_callback else None,
            foreground="#ef4444",
        )

        try:
            menu.tk_popup(event.x_root, event.y_root)
        finally:
            menu.grab_release()

    def _copy_alias(self, alias: str):
        """Copy server alias to clipboard."""
        self.clipboard_clear()
        self.clipboard_append(alias)