feat: add Servers class

This commit is contained in:
aclist 2026-01-07 23:34:25 +09:00
parent 3b7b131cf5
commit 058cbc39b1

View File

@ -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)