i3ifiedesk.py

#!/usr/bin/python
import os
import re
import dbus
from dbus.mainloop.glib import DBusGMainLoop
from gi.repository import GLib


class DBusObject:
    def __init__(self, bus, service, path, interface):
        self.bus = bus
        self.service = service
        self.path = path
        self.interface = interface

    def get_obj(self):
        return self.bus.get_object(self.service, self.path)

    def get_prop(self, prop):
        return self.get_obj().Get(self.interface, prop, dbus_interface=dbus.PROPERTIES_IFACE)

    def call(self, method_name, *args):
        iface = dbus.Interface(self.get_obj(), self.interface)
        method = getattr(iface, method_name)
        return method(*args)


class i3ifiedesk:
    def __init__(self):
        DBusGMainLoop(set_as_default=True)
        self.bus = dbus.SessionBus()
        self.bus.add_signal_receiver(
            self.on_shortcut_pressed,
            dbus_interface="org.kde.kglobalaccel.Component",
            signal_name="globalShortcutPressed"
        )
        self.desktop_manager = DBusObject(
            self.bus, 'org.kde.KWin', '/VirtualDesktopManager', 'org.kde.KWin.VirtualDesktopManager')
        self.kwin_manager = DBusObject(
            self.bus, 'org.kde.KWin', '/KWin', 'org.kde.KWin')
        self.kwin_script_manager = DBusObject(
            self.bus, 'org.kde.KWin', '/Scripting', 'org.kde.kwin.Scripting')
        self.shortcuts_manager = DBusObject(
            self.bus, 'org.kde.kglobalaccel', '/component/kwin', 'org.kde.kglobalaccel.Component')
        # self.activity_manager = DBusObject(self.bus, 'org.kde.ActivityManager', '/Activities', 'org.kde.ActivityManager.Activities')

        self.script_name = "rm_desk"
        script_path = os.path.dirname(os.path.realpath(
            __file__)) + '/'+self.script_name+'.js'
        self.script_id = self.kwin_script_manager.call(
            'loadScript', script_path)
        self.script_manager = DBusObject(
            self.bus, 'org.kde.KWin', f'/Scripting/Script{self.script_id}', 'org.kde.kwin.Script')
        self.kwin_manager.call('reconfigure')

        self.loop = GLib.MainLoop()
        try:
            self.loop.run()
        except KeyboardInterrupt:
            self.kwin_script_manager.call("unloadScript", self.script_name)
            self.script_manager.call("stop")
            self.loop.quit()

    def handle_desktop_switch(self, action):
        pattern = r"Switch to Desktop (\d+)"
        match = re.search(pattern, str(action))
        if match:
            target = int(match.group(1))
            desktop_count = self.desktop_manager.get_prop('count')
            if target > desktop_count:
                self.desktop_manager.call(
                    "createDesktop", target, f"Desktop {target}")
                self.kwin_manager.call("setCurrentDesktop", target)

    def handle_window_move(self, action):
        pattern = r"Window to Desktop (\d+)"
        match = re.search(pattern, str(action))
        if match:
            target = int(match.group(1))
            desktop_count = self.desktop_manager.get_prop('count')
            if target > desktop_count:
                self.desktop_manager.call(
                    "createDesktop", target, f"Desktop {target}")
                self.shortcuts_manager.call('invokeShortcut', action)

    def handle_close_desktops(self, action):
        pattern = r"Close Empty Desktops"
        match = re.search(pattern, str(action))
        if not match:
            self.shortcuts_manager.call(
                'invokeShortcut', 'Close Empty Desktops')

    def on_shortcut_pressed(self, component, action, shortcuts):
        self.handle_desktop_switch(action)
        self.handle_window_move(action)
        self.handle_close_desktops(action)


if __name__ == "__main__":
    # make systemd happy
    bus_address = os.getenv('DBUS_SESSION_BUS_ADDRESS')
    if not bus_address:
        with open(f"/run/user/{os.getuid()}/bus") as f:
            bus_address = f.read().strip()
    os.environ['DBUS_SESSION_BUS_ADDRESS'] = bus_address

    i3ifiedesk()