1 files changed, 101 insertions, 0 deletions
diff --git a/sway/switch.py b/sway/switch.py
new file mode 100755
index 0000000..479cf74
--- /dev/null
+++ b/sway/switch.py
@@ -0,0 +1,101 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+
+import subprocess
+import json
+import re
+import gi
+gi.require_version('Gtk', '3.0')
+from gi.repository import Gtk
+
+def parse_icon_name(app_name):
+  app_name = app_name.lower()
+  if app_name in icon_names:
+    return icon_names[app_name]
+  if app_name.endswith('telegram_desktop'):
+    return 'telegram'
+  if app_name.startswith('gimp'):
+    return 'gimp'
+  return app_name
+
+def get_icon(app_name):
+  icon_info = icon_theme.lookup_icon(parse_icon_name(app_name), 48, 0)
+  if icon_info == None:
+    print("Failed to get icon for: " + app_name)
+  # return icon_info.get_filename() if icon_info != None else default_icon_file
+  return default_icon_file
+
+def get_cons(tree):
+  if len(tree.get('nodes')) == 0 and len(tree.get('floating_nodes')) == 0 and (tree.get('type') == 'con' or tree.get('type') == 'floating_con'):
+    return [tree]
+  nodes = []
+  for node in tree.get('floating_nodes'):
+    cons = get_cons(node)
+    nodes += cons
+  for node in tree.get('nodes'):
+    cons = get_cons(node)
+    nodes += cons
+  return nodes
+
+def show_wofi(windows):
+  # command = 'wofi -W 600 -L ' + str(len(windows)) + ' -I -d -i -b -k /dev/null -p "Window" 2>/dev/null'
+  command = 'wofi -G -L ' + str(len(windows)) + ' -I -d -i -b -p "Window" 2>/dev/null'
+  process = subprocess.Popen(command, shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE)
+  (result, _) = process.communicate(input='\n'.join(windows).encode())
+  return result
+
+
+def get_tree():
+  sway_tree = subprocess.check_output(['swaymsg', '-rt', 'get_tree'])
+  return json.loads(sway_tree)
+
+def get_windows():
+  cons = get_cons(get_tree())
+  windows = []
+  for con in cons:
+    name = con['name']
+    if len(name) > 70:
+      name = name[:67]+"..."
+    app_id = con.get('app_id')
+    if app_id == None:
+      window_prop = con.get('window_properties')
+      if window_prop != None:
+        app_id = window_prop.get('instance')
+
+    if app_id != None:
+      app_icon = get_icon(app_id)
+    else:
+      app_icon = default_icon_file
+
+    # windows.append("img:id=" + str(con.get('id')) +":img:" + app_icon + ":text:" + name)
+    windows.append("img:id=" + str(con.get('id')) +":img:" + app_icon + ":text:" + app_id + ": " + name)
+  return windows
+
+def parse_selection(selection):
+  id_search = re.search('id=([0-9]*):', selection, re.IGNORECASE)
+  if id_search:
+    return id_search.group(1)
+  return None
+
+def switch_window(id):
+  command="swaymsg [con_id={}] focus".format(id)
+  process = subprocess.Popen(command,shell=True,stdout=subprocess.PIPE)
+  process.communicate()
+
+if __name__ == "__main__":
+  icon_theme = Gtk.IconTheme.get_default()
+  default_icon = icon_theme.lookup_icon("preferences-system-windows", 48, 0)
+  default_icon_file = ""
+  if default_icon:
+    default_icon_file = default_icon.get_filename()
+
+  icon_names = {
+    "code": "visual-studio-code",
+    "qt5ct": "preferences-desktop-theme"
+  }
+
+  windows = get_windows()
+  selection = show_wofi(windows)
+  con_id = parse_selection(str(selection))
+  if con_id:
+    switch_window(con_id)