keyd-application-mapper: Refactor wayland code

Clean up Wayland() and allow binding to multiple protocol objects simultaneously

scripts/keyd-application-mapper

@@ -176,7 +176,7 @@ class KDE():
 # so this only works for wlroots based compositors :(.
 
 class Wayland():
-    def __init__(self, interface_name):
+    def __init__(self, *interfaces):
         path = os.getenv("WAYLAND_DISPLAY")
         if path == None:
             raise Exception("WAYLAND_DISPLAY not set (is wayland running?)")
@@ -186,7 +186,8 @@ class Wayland():
 
         self.sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
         self.sock.connect(path)
-        self._bind_interface(interface_name)
+
+        self._bind_interfaces(interfaces)
 
     def send_msg(self, object_id, opcode, payload):
         opcode |= (len(payload)+8) << 16
@@ -208,24 +209,31 @@ class Wayland():
         return b[4:4+struct.unpack('I', b[:4])[0]-1].decode('utf8')
 
     # Will bind the requested interface to object id 4
-    def _bind_interface(self, name):
+    def _bind_interfaces(self, interfaces):
         # bind the registry object to id 2
         self.send_msg(1, 1, b'\x02\x00\x00\x00')
         # solicit a sync message to bookend the registry events
         self.send_msg(1, 0, b'\x03\x00\x00\x00')
+
+        interface_object_number = 4
         while True:
             (obj, event, payload) = self.recv_msg()
             if obj == 2 and event == 0:  # registry.global event
                 wl_name = struct.unpack('I', payload[0:4])[0]
-
                 wl_interface = self.read_string(payload[4:])
 
-                if wl_interface == name:
-                    self.send_msg(2, 0, payload+b'\x04\x00\x00\x00')
-                    return
+                for interface in interfaces:
+                    if wl_interface == interface:
+                        bind_payload = payload + struct.pack('I', interface_object_number)
+                        self.send_msg(2, 0, bind_payload)
+                        setattr(self, interface, interface_object_number)
+                        interface_object_number += 1
 
             if obj == 3:  # sync message
-                raise Exception(f"Could not find interface {name}")
+                for interface in interfaces:
+                    if not hasattr(self, interface):
+                        raise Exception(f"Could not find interface {interface}")
+                return
 
 
 class Wlroots():
@@ -246,14 +254,11 @@ class Wlroots():
                 continue
 
             (obj, event, payload) = msg
-            if obj == 4 and event == 0:
+            if obj == self.wl.zwlr_foreign_toplevel_manager_v1 and event == 0:
                 # zwlr_foreign_toplevel_manager_v1::toplevel event
                 windows[struct.unpack('I', payload)[0]] = {'title': '', 'appid': ''}
                 continue
-
-            if obj not in windows:
-                continue
-
+            elif obj in windows:
                 if event == 0:
                     # zwlr_foreign_toplevel_handle_v1::title event
                     windows[obj]['title'] = self.wl.read_string(payload)