feat: add Servers class

dzgui/views/pages/servers.py

@@ -0,0 +1,157 @@
+import logging
+
+from typing import Self, TYPE_CHECKING
+
+from dzgui.const.enum import ContextMenuGroup
+from dzgui.views.trees.tree_servers import ServerTreeView
+from dzgui.util.strings import server_labels
+
+import gi
+gi.require_version("Gtk", "3.0")
+from gi.repository import Gtk, Gdk, GLib  # noqa E402
+
+logger = logging.getLogger(__name__)
+
+if TYPE_CHECKING:
+    from dzgui.controllers.mc import Controller
+
+class ServerNotebook(Gtk.ScrolledWindow):
+    def __init__(self, controller: "Controller"):
+        super().__init__()
+
+        self.tab_cache = ""
+        self.controller = controller
+        self.controller.register_widget("servers", self)
+        self.notebook = Gtk.Notebook(show_tabs=True)
+
+        self.browser = ServerTreeView(controller)
+        self.saved = ServerTreeView(controller)
+        self.recent = ServerTreeView(controller)
+        self.lan = ServerTreeView(controller)
+
+        self.lan.set_query_func(self.query_test)
+        # TODO: set context menu on init
+        self.browser.set_context_menu(ContextMenuGroup.SERVER_BROWSER)
+        self.saved.set_context_menu(ContextMenuGroup.SAVED)
+        self.recent.set_context_menu(ContextMenuGroup.RECENT)
+        self.lan.set_context_menu(ContextMenuGroup.SCAN_LAN)
+
+        # TODO: set model manager for each tab on init
+        # TODO: add set_ and get_model_manager() methods
+
+        tabs = [
+            (self.browser, server_labels.browser, self.controller.get_server_store),
+            (self.saved, server_labels.saved, self.controller.get_saved_store),
+            (self.recent, server_labels.recent, self.controller.get_recent_store),
+            (self.lan, server_labels.lan, self.controller.get_lan_store),
+            ]
+
+        for tree, label, func in tabs:
+            store = func()
+            if label == "LAN":
+                pass
+                #store.append(["BAR", "a", "a", "a", 1, 1, 1, "1:1", 0, 0, "a", False])
+            else:
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                store.append(["BAR", "a", "a", "a", 0, 0, 0, "1:1", 0, 0, "a", False])
+                tree.loaded = True
+            # TODO: set model on init of servertreeview
+            tree.set_model(store)
+
+            scrolled = Gtk.ScrolledWindow()
+            scrolled.add(tree)
+            self.notebook.append_page(scrolled, Gtk.Label(label=label))
+
+        self.add(self.notebook)
+        self.notebook.connect_after("switch-page", self._on_page_changed)
+        self.connect("key-press-event", self._on_keypress)
+
+    def _on_keypress(self, widget: Self, event: Gdk.EventKey) -> None:
+
+        match event.keyval:
+            case Gdk.KEY_n:
+                self.notebook.next_page()
+            case Gdk.KEY_p:
+                self.notebook.prev_page()
+            case _:
+                return
+        self.get_active_treeview().grab_focus()
+
+
+    def _on_page_changed(self, notebook: Gtk.Notebook, child: Gtk.Widget, index: int) -> None:
+        # TODO: abstract
+        # TODO: load on first run
+        label = self.notebook.get_tab_label_text(child)
+        if label is None:
+            return
+        # TODO: strings
+        text = label.strip("*")
+        self.notebook.set_tab_label_text(child, text)
+
+        # TODO: strings
+        string = f"Servers > {text}"
+        self.controller.set_crumbs(string)
+        self.tab_cache = string
+
+        child.grab_focus()
+
+        # FIXME: doesn't fire on first run
+        self.controller.update_server_status()
+        self.controller.populate_model()
+
+    # TODO: put in controller
+    def query_test(self) -> None:
+        data = (["BAR", "a", "a", "a", 1, 1, 1, "1:1", 0, 0, "a", False])
+        return data
+
+    def get_cached_label(self) -> str:
+        return self.tab_cache
+
+    def get_active_treeview(self) -> ServerTreeView:
+        index = self.notebook.get_current_page()
+        scrollable = self.notebook.get_nth_page(index)
+        treeview = scrollable.get_children()[0]
+        return treeview
+
+    def add_notification(self) -> None:
+        saved = self.notebook.get_nth_page(1)
+        if saved is None:
+            return
+        text = self.notebook.get_tab_label_text(saved)
+        if text is None:
+            return
+        # TODO: strings
+        if "*" in text:
+            return
+        text += "*"
+        self.notebook.set_tab_label_text(saved, text)
+
+    def update_tab_widths(self, col: Gtk.TreeViewColumn) -> None:
+        # TODO: may cause pixel offsets when application is maximized
+        width = col.get_width()
+        title = col.get_title()
+        for tab in (self.browser, self.saved, self.recent, self.lan):
+            if tab == self.get_active_treeview():
+                continue
+            for col in tab.get_columns():
+                if col.get_title() == title:
+                    self.controller.suppress_signal(tab, col, "_on_col_width_changed", True)
+                    col.set_fixed_width(width)
+                    self.controller.suppress_signal(tab, col, "_on_col_width_changed", False)
+
+    def get_tabs(self) -> tuple:
+        return (self.browser, self.saved, self.recent, self.lan)