integrate home-manager config

home-manager/jwm/ptray/i3sway/monitor.py

@@ -0,0 +1,112 @@
+from i3sway.protocol import Connection, MessageType
+
+class Monitor:
+    tracked_classes = None
+    cmd_sock = None
+    subscription_sock = None
+    event_handler = None
+    conn_type = None
+
+    def __enter__(self):
+        return self
+
+    def __exit__(self, *details):
+        self.cmd_sock.close()
+        self.subscription_sock.close()
+
+    def __init__(self, event_handler, tracked_classes):
+        self.event_handler = event_handler
+        self.tracked_classes = tracked_classes
+
+        self.cmd_sock = Connection()
+        self.subscription_sock = Connection()
+
+        tree = self.cmd_sock.request(MessageType.GET_TREE).payload
+        
+        def parse_tree(root):
+            if root.get('window') is not None or root.get('app_id') is not None:
+                # Application window
+                for name, classes in tracked_classes.items():
+                    if self.container_matches_classes(root, classes):
+                        id = str(root['id'])
+                        self.event_handler.on_create(name, id)
+                        if root['visible']:
+                            self.event_handler.on_visible(name, id)
+
+                        if root['type'] == 'floating_con':
+                            self.event_handler.on_float(name, id)
+                        elif root['type'] == 'con':
+                            self.event_handler.on_tile(name, id)
+            else:
+                for node in root['nodes']:
+                    parse_tree(node)
+                for node in root['floating_nodes']:
+                    parse_tree(node)
+
+        parse_tree(tree)
+
+    def container_matches_classes(self, container, classes):
+        normalize = lambda v: v.lower() if v is not None else ''
+        app_id = container.get('app_id')
+        window_props = container.get('window_properties')
+        instance = window_props.get('instance') if window_props is not None else None
+        window_class = window_props.get('class') if window_props is not None else None
+
+        app_id = normalize(app_id)
+        instance = normalize(instance)
+        window_class = normalize(window_class)
+        for keyword in classes:
+            kw = normalize(keyword)
+            if kw in app_id or kw in instance or kw in window_class:
+                return True
+
+        return False
+
+    def listen(self):
+        for event in self.subscription_sock.subscribe(['window']):
+            self.handle_event(event)
+
+    def is_window_in_scratchpad(self, id):
+        def scan_tree(root, is_in_scratch):
+            if root['id'] == id:
+                return is_in_scratch
+            is_scratch = is_in_scratch or (
+                root['type'] == 'workspace' and root['name'] == '__i3_scratch'
+            )
+            for node in root['nodes'] + root['floating_nodes']:
+                result = scan_tree(node, is_scratch)
+                if result is not None:
+                    return result
+
+        tree = self.cmd_sock.request(MessageType.GET_TREE)
+        return scan_tree(tree.payload, False)
+
+    def handle_event(self, event):
+        if event.type == MessageType.window:
+            return self.handle_window_event(event)
+
+    def handle_window_event(self, event):
+        container = event['container']
+
+        for name, classes in self.tracked_classes.items():
+            change = event['change']
+            if self.container_matches_classes(container, classes):
+                id = str(container['id'])
+                if change == 'new':
+                    self.event_handler.on_create(name, id)
+                elif change == 'focus':
+                    self.event_handler.on_visible(name, id)
+                elif change == 'close':
+                    self.event_handler.on_destroy(name, id)
+                elif change == 'move':
+                    if container['visible']:
+                        self.event_handler.on_visible(name, id)
+                    else:
+                        self.event_handler.on_invisible(name, id)
+                elif change == 'floating':
+                    if container['type'] == 'floating_con':
+                        self.event_handler.on_float(name, id)
+                    elif container['type'] == 'con':
+                        self.event_handler.on_tile(name, id)
+
+