mirror of
https://github.com/j1g5awi/krunner-vscode.git
synced 2024-12-05 02:07:39 +00:00
90 lines
2.2 KiB
Python
90 lines
2.2 KiB
Python
|
import json
|
||
|
|
import os
|
||
|
|
import pathlib
|
||
|
|
import sqlite3
|
||
|
|
from typing import NamedTuple
|
||
|
|
|
||
|
|
import dbus.service
|
||
|
|
from dbus.mainloop.glib import DBusGMainLoop
|
||
|
|
from gi.repository import GLib
|
||
|
|
|
||
|
|
DBusGMainLoop(set_as_default=True)
|
||
|
|
|
||
|
|
objpath = "/vscode"
|
||
|
|
|
||
|
|
iface = "org.kde.krunner1"
|
||
|
|
|
||
|
|
|
||
|
|
class Match(NamedTuple):
|
||
|
|
data: str
|
||
|
|
display_text: str
|
||
|
|
icon: str
|
||
|
|
type: int
|
||
|
|
relevance: float
|
||
|
|
properties: dict
|
||
|
|
|
||
|
|
|
||
|
|
# TODO Check VSCode or Code - OSS or VSCodium
|
||
|
|
|
||
|
|
# Read path_list from database
|
||
|
|
con = sqlite3.connect(
|
||
|
|
os.path.join(
|
||
|
|
os.environ["HOME"], ".config", "Code - OSS/User/globalStorage/state.vscdb"
|
||
|
|
)
|
||
|
|
)
|
||
|
|
cur = con.cursor()
|
||
|
|
rows = cur.execute(
|
||
|
|
"SELECT value FROM ItemTable WHERE key = 'history.recentlyOpenedPathsList'"
|
||
|
|
)
|
||
|
|
data = json.loads(rows.fetchone()[0])
|
||
|
|
con.close()
|
||
|
|
path_list = [
|
||
|
|
"~" + path[len(os.environ["HOME"]) :]
|
||
|
|
for path in [i["folderUri"][7:] for i in data["entries"] if "folderUri" in i]
|
||
|
|
if os.environ["HOME"] in path
|
||
|
|
]
|
||
|
|
|
||
|
|
|
||
|
|
class Runner(dbus.service.Object):
|
||
|
|
def __init__(self):
|
||
|
|
dbus.service.Object.__init__(
|
||
|
|
self,
|
||
|
|
dbus.service.BusName(
|
||
|
|
"com.github.j1g5awi.vscode", dbus.SessionBus() # type:ignore
|
||
|
|
),
|
||
|
|
objpath,
|
||
|
|
)
|
||
|
|
|
||
|
|
@dbus.service.method(iface, in_signature="s", out_signature="a(sssida{sv})")
|
||
|
|
def Match(self, query: str):
|
||
|
|
# data, display text, icon, type (Plasma::QueryType), relevance (0-1), properties (subtext, category and urls)
|
||
|
|
return [
|
||
|
|
Match(
|
||
|
|
path,
|
||
|
|
pathlib.Path(path).name,
|
||
|
|
"com.visualstudio.code.oss",
|
||
|
|
100,
|
||
|
|
1.0,
|
||
|
|
{"subtext": path},
|
||
|
|
)
|
||
|
|
for path in path_list
|
||
|
|
if query in path
|
||
|
|
]
|
||
|
|
|
||
|
|
@dbus.service.method(iface, out_signature="a(sss)")
|
||
|
|
def Actions(self):
|
||
|
|
# id, text, icon
|
||
|
|
return [("id", "Open Folder", "document-open-folder")]
|
||
|
|
|
||
|
|
@dbus.service.method(iface, in_signature="ss")
|
||
|
|
def Run(self, data: str, action_id: str):
|
||
|
|
if not action_id:
|
||
|
|
os.system("code " + data)
|
||
|
|
else:
|
||
|
|
os.system("xdg-open " + data)
|
||
|
|
|
||
|
|
|
||
|
|
runner = Runner()
|
||
|
|
loop = GLib.MainLoop()
|
||
|
|
loop.run()
|