search_provider.py

import os
import subprocess
import functools

from dbus.mainloop.glib import DBusGMainLoop
DBusGMainLoop(set_as_default=True)
import dbus.service

from gi.repository import GLib

sbn = dict(dbus_interface='org.gnome.Shell.SearchProvider2')

class SearchService(dbus.service.Object):
    bus_name = "org.gnome.Command.SearchProvider"
    object_path = "/" + bus_name.replace(".", "/")

    def __init__(self):
        self.session_bus = dbus.SessionBus()
        self.is_with_argument=False
        bus_name = dbus.service.BusName(self.bus_name, bus=self.session_bus)
        dbus.service.Object.__init__(self, bus_name, self.object_path)

    @dbus.service.method(in_signature='sasu', **sbn)
    def ActivateResult(self, id, terms, timestamp):
        command = id
        os.system(f"{command} &")

    @dbus.service.method(in_signature='as', out_signature='as', **sbn)
    def GetInitialResultSet(self, terms):
        commands=[line.decode("utf-8") for line in subprocess.check_output(['dmenu_path']).splitlines()]
        return self.GetSubsearchResultSet(commands,terms)

    @dbus.service.method(in_signature='as', out_signature='aa{sv}', **sbn)
    def GetResultMetas(self, ids):
        return [dict(id=id, name=id, description=subprocess.check_output(['which',id.split(' ')[0]])) for id in ids]

    @dbus.service.method(in_signature='asas', out_signature='as', **sbn)
    def GetSubsearchResultSet(self, previous_results, new_terms):
        search = " ".join(new_terms)
        results=''
        def useSort(element1,element2):
            searchLen=len(search)
            e1Len=len(element1)
            e2Len=len(element2)

            e1i=0
            while(e1i<e1Len and e1i<searchLen and element1[e1i]==search[e1i]):
                e1i+=1
            e2i=0
            while(e2i<e2Len and e2i<searchLen and element2[e2i]==search[e2i]):
                e2i+=1
            return e2i-e1i

        self.is_with_argument=len(search.split(' '))>1
        if self.is_with_argument:
            results=[search]
        else:
            results = [command for command in previous_results if search in command]
            results.sort(key=functools.cmp_to_key(useSort))

        return results
                
    @dbus.service.method(in_signature='asu', terms='as', timestamp='u', **sbn)
    def LaunchSearch(self, terms, timestamp):
        pass

if __name__ == "__main__":
    SearchService()
    GLib.MainLoop().run()