Add Debug Console to Qui Domains systray widget

fixes: QubesOS/qubes-issues#9788

Author: alimirjamali

qui/tray/domains.py

@@ -316,6 +316,34 @@ async def perform_action(self):
             )
 
 
+class RunDebugConsoleItem(VMActionMenuItem):
+    """Run Debug Console menu Item. When activated runs a qvm-console-dispvm."""
+
+    def __init__(self, vm, icon_cache):
+        img = Gtk.Image.new_from_file(
+            "/usr/share/icons/HighContrast/16x16/apps/logviewer.png"
+        )
+        super().__init__(
+            vm,
+            label=_("Debug Console"),
+            img=img,
+        )
+        self.visible = False
+        self.connect("show", self.on_show_event)
+
+    def on_show_event(self, widget):
+        if self.visible:
+            widget.show()
+        else:
+            widget.hide()
+
+    async def perform_action(self):
+        # pylint: disable=consider-using-with
+        await asyncio.create_subprocess_exec(
+            "qvm-console-dispvm", self.vm.name, stderr=subprocess.PIPE
+        )
+
+
 class OpenFileManagerItem(VMActionMenuItem):
     """Attempts to open a file manager in the VM. If fails, displays an
     error message."""
@@ -368,15 +396,37 @@ def __init__(self, vm, app, icon_cache):
 
         self.add(OpenFileManagerItem(self.vm, icon_cache))
         self.add(RunTerminalItem(self.vm, icon_cache))
+
+        # Debug console for developers, troubleshooting, headless qubes
+        self.debug_console = RunDebugConsoleItem(self.vm, icon_cache)
+        self.add(self.debug_console)
+
         self.add(PreferencesItem(self.vm, icon_cache))
         self.add(PauseItem(self.vm, icon_cache))
         self.add(ShutdownItem(self.vm, icon_cache))
         if self.vm.klass != "DispVM" or not self.vm.auto_cleanup:
             self.add(RestartItem(self.vm, icon_cache))
 
         self.set_reserve_toggle_size(False)
+        self.debug_console_update()
         self.show_all()
 
+    def debug_console_update(self, *_args, **_kwargs):
+        # Debug console is shown only if debug property is set, no GUIVM is set
+        # ... or with `wizard-mode` feature per qube or per entire GUIVM.
+        if (
+            self.app.wizard_mode
+            or getattr(self.vm, "debug")
+            or not getattr(self.vm, "guivm")
+            or not self.vm.features.check_with_template("gui", False)
+            or self.vm.features.get("wizard-mode", False)
+        ):
+            self.debug_console.visible = True
+            self.debug_console.show()
+        else:
+            self.debug_console.visible = False
+            self.debug_console.hide()
+
 
 class PausedMenu(Gtk.Menu):
     """The sub-menu for a paused domain"""
@@ -668,6 +718,11 @@ def __init__(self, app_name, qapp, dispatcher, stats_dispatcher):
         self.set_application_id(app_name)
         self.register()  # register Gtk Application
 
+        # to display debug console for all qubes
+        self.wizard_mode = self.qapp.domains[self.qapp.local_name].features.get(
+            "wizard-mode", False
+        )
+
     def register_events(self):
         self.dispatcher.add_handler("connection-established", self.refresh_all)
         self.dispatcher.add_handler("domain-pre-start", self.update_domain_item)
@@ -717,6 +772,32 @@ def register_events(self):
 
         self.stats_dispatcher.add_handler("vm-stats", self.update_stats)
 
+        self.dispatcher.add_handler("property-set:debug", self.debug_change)
+        self.dispatcher.add_handler("property-set:guivm", self.debug_change)
+        self.dispatcher.add_handler("domain-feature-set:gui", self.debug_change)
+        self.dispatcher.add_handler(
+            "domain-feature-delete:gui", self.debug_change
+        )
+        self.dispatcher.add_handler(
+            "domain-feature-set:wizard-mode", self.debug_change
+        )
+        self.dispatcher.add_handler(
+            "domain-feature-delete:wizard-mode", self.debug_change
+        )
+
+    def debug_change(self, vm, *_args, **_kwargs):
+        if vm == self.qapp.local_name:
+            self.wizard_mode = self.qapp.domains[
+                self.qapp.local_name
+            ].features.get("wizard-mode", False)
+            vms = self.menu_items
+        else:
+            vms = {vm}
+        for menu in vms:
+            submenu = self.menu_items[menu].get_submenu()
+            if isinstance(submenu, StartedMenu):
+                submenu.debug_console_update()
+
     def show_menu(self, _unused, event):
         self.tray_menu.popup_at_pointer(event)  # None means current event
 
@@ -1069,6 +1150,21 @@ def _disconnect_signals(self, _event):
 
         self.stats_dispatcher.remove_handler("vm-stats", self.update_stats)
 
+        self.dispatcher.remove_handler("property-set:debug", self.debug_change)
+        self.dispatcher.remove_handler("property-set:guivm", self.debug_change)
+        self.dispatcher.remove_handler(
+            "domain-feature-set:gui", self.debug_change
+        )
+        self.dispatcher.remove_handler(
+            "domain-feature-delete:gui", self.debug_change
+        )
+        self.dispatcher.remove_handler(
+            "domain-feature-set:wizard-mode", self.debug_change
+        )
+        self.dispatcher.remove_handler(
+            "domain-feature-delete:wizard-mode", self.debug_change
+        )
+
 
 def main():
     """main function"""