initial changes to add config services to coretk gui

daemon/core/gui/dialogs/nodeconfigservice.py

@@ -0,0 +1,169 @@
+"""
+core node services
+"""
+import tkinter as tk
+from tkinter import messagebox, ttk
+from typing import TYPE_CHECKING, Any, Set
+
+from core.gui.dialogs.configserviceconfig import ConfigServiceConfigDialog
+from core.gui.dialogs.dialog import Dialog
+from core.gui.themes import FRAME_PAD, PADX, PADY
+from core.gui.widgets import CheckboxList, ListboxScroll
+
+if TYPE_CHECKING:
+    from core.gui.app import Application
+    from core.gui.graph.node import CanvasNode
+
+
+class NodeConfigServiceDialog(Dialog):
+    def __init__(
+        self,
+        master: Any,
+        app: "Application",
+        canvas_node: "CanvasNode",
+        services: Set[str] = None,
+    ):
+        title = f"{canvas_node.core_node.name} Config Services"
+        super().__init__(master, app, title, modal=True)
+        self.app = app
+        self.canvas_node = canvas_node
+        self.node_id = canvas_node.core_node.id
+        self.groups = None
+        self.services = None
+        self.current = None
+        if services is None:
+            services = set(canvas_node.core_node.config_services)
+        self.current_services = services
+        self.draw()
+
+    def draw(self):
+        self.top.columnconfigure(0, weight=1)
+        self.top.rowconfigure(0, weight=1)
+
+        frame = ttk.Frame(self.top)
+        frame.grid(stick="nsew", pady=PADY)
+        frame.rowconfigure(0, weight=1)
+        for i in range(3):
+            frame.columnconfigure(i, weight=1)
+        label_frame = ttk.LabelFrame(frame, text="Groups", padding=FRAME_PAD)
+        label_frame.grid(row=0, column=0, sticky="nsew")
+        label_frame.rowconfigure(0, weight=1)
+        label_frame.columnconfigure(0, weight=1)
+        self.groups = ListboxScroll(label_frame)
+        self.groups.grid(sticky="nsew")
+        for group in sorted(self.app.core.config_services):
+            self.groups.listbox.insert(tk.END, group)
+        self.groups.listbox.bind("<<ListboxSelect>>", self.handle_group_change)
+        self.groups.listbox.selection_set(0)
+
+        label_frame = ttk.LabelFrame(frame, text="Services")
+        label_frame.grid(row=0, column=1, sticky="nsew")
+        label_frame.columnconfigure(0, weight=1)
+        label_frame.rowconfigure(0, weight=1)
+        self.services = CheckboxList(
+            label_frame, self.app, clicked=self.service_clicked, padding=FRAME_PAD
+        )
+        self.services.grid(sticky="nsew")
+
+        label_frame = ttk.LabelFrame(frame, text="Selected", padding=FRAME_PAD)
+        label_frame.grid(row=0, column=2, sticky="nsew")
+        label_frame.rowconfigure(0, weight=1)
+        label_frame.columnconfigure(0, weight=1)
+        self.current = ListboxScroll(label_frame)
+        self.current.grid(sticky="nsew")
+        for service in sorted(self.current_services):
+            self.current.listbox.insert(tk.END, service)
+            if self.is_custom_service(service):
+                self.current.listbox.itemconfig(tk.END, bg="green")
+
+        frame = ttk.Frame(self.top)
+        frame.grid(stick="ew")
+        for i in range(4):
+            frame.columnconfigure(i, weight=1)
+        button = ttk.Button(frame, text="Configure", command=self.click_configure)
+        button.grid(row=0, column=0, sticky="ew", padx=PADX)
+        button = ttk.Button(frame, text="Save", command=self.click_save)
+        button.grid(row=0, column=1, sticky="ew", padx=PADX)
+        button = ttk.Button(frame, text="Remove", command=self.click_remove)
+        button.grid(row=0, column=2, sticky="ew", padx=PADX)
+        button = ttk.Button(frame, text="Cancel", command=self.click_cancel)
+        button.grid(row=0, column=3, sticky="ew")
+
+        # trigger group change
+        self.groups.listbox.event_generate("<<ListboxSelect>>")
+
+    def handle_group_change(self, event: tk.Event = None):
+        selection = self.groups.listbox.curselection()
+        if selection:
+            index = selection[0]
+            group = self.groups.listbox.get(index)
+            self.services.clear()
+            for name in sorted(self.app.core.config_services[group]):
+                checked = name in self.current_services
+                self.services.add(name, checked)
+
+    def service_clicked(self, name: str, var: tk.IntVar):
+        if var.get() and name not in self.current_services:
+            self.current_services.add(name)
+        elif not var.get() and name in self.current_services:
+            self.current_services.remove(name)
+        self.current.listbox.delete(0, tk.END)
+        for name in sorted(self.current_services):
+            self.current.listbox.insert(tk.END, name)
+            if self.is_custom_service(name):
+                self.current.listbox.itemconfig(tk.END, bg="green")
+        self.canvas_node.core_node.config_services[:] = self.current_services
+
+    def click_configure(self):
+        current_selection = self.current.listbox.curselection()
+        if len(current_selection):
+            dialog = ConfigServiceConfigDialog(
+                master=self,
+                app=self.app,
+                service_name=self.current.listbox.get(current_selection[0]),
+                node_id=self.node_id,
+            )
+            dialog.show()
+        else:
+            messagebox.showinfo(
+                "Node service configuration", "Select a service to configure"
+            )
+
+    def click_save(self):
+        if (
+            self.current_services
+            != self.app.core.default_services[self.canvas_node.core_node.model]
+        ):
+            self.canvas_node.core_node.config_services[:] = self.current_services
+        else:
+            if len(self.canvas_node.core_node.config_services) > 0:
+                self.canvas_node.core_node.config_services[:] = []
+        self.destroy()
+
+    def click_cancel(self):
+        self.current_services = None
+        self.destroy()
+
+    def click_remove(self):
+        cur = self.current.listbox.curselection()
+        if cur:
+            service = self.current.listbox.get(cur[0])
+            self.current.listbox.delete(cur[0])
+            self.current_services.remove(service)
+            for checkbutton in self.services.frame.winfo_children():
+                if checkbutton["text"] == service:
+                    checkbutton.invoke()
+                    return
+
+    def is_custom_service(self, service: str) -> bool:
+        service_configs = self.app.core.service_configs
+        file_configs = self.app.core.file_configs
+        if self.node_id in service_configs and service in service_configs[self.node_id]:
+            return True
+        if (
+            self.node_id in file_configs
+            and service in file_configs[self.node_id]
+            and file_configs[self.node_id][service]
+        ):
+            return True
+        return False