Add win32 support for listing windows

Author: clemux

snapstudysensei/windows_list/__init__.py

@@ -0,0 +1,6 @@
+import sys
+
+if sys.platform.startswith("win"):
+    from .windows_list_win import WindowsList
+elif sys.platform.startswith("linux"):
+    from .windows_list_x11 import WindowsList

snapstudysensei/windows_list/windows_list_win.py

@@ -0,0 +1,31 @@
+import ctypes
+from ctypes import wintypes
+
+user32 = ctypes.windll.user32
+
+
+class WindowsList:
+    def __init__(self):
+        self.windows = {}
+
+    def __call__(self) -> dict[int, tuple[str, bool]]:
+        WNDENUMPROC = ctypes.WINFUNCTYPE(
+            wintypes.BOOL,
+            wintypes.HWND,
+            wintypes.LPARAM
+        )
+        cb_worker = WNDENUMPROC(self._enum_windows_cb)
+        user32.EnumWindows(cb_worker, None)
+        return self.windows
+
+    def _enum_windows_cb(self, hwnd: wintypes.HWND, _: wintypes.LPARAM):
+        """Callback for EnumWindows.
+
+        See https://learn.microsoft.com/en-us/previous-versions/windows/desktop/legacy/ms633498(v=vs.85)
+        """
+        length = user32.GetWindowTextLengthW(hwnd) + 1
+        buffer = ctypes.create_unicode_buffer(length)
+        user32.GetWindowTextW(hwnd, buffer, length)
+        if user32.IsWindowVisible(hwnd) and len(buffer.value) > 0:
+            self.windows[hwnd] = (buffer.value, True)
+        return True