Merge pull request #1 from RealOrangeOne/feature/support-alternative-vscode-installs

Support alternative VSCode installs

krunner_vscode/__main__.py

@@ -15,6 +15,13 @@ objpath = "/vscode"
 iface = "org.kde.krunner1"
 
 
+VSCODE_DIRS = [
+    "Code",
+    "Code - OSS",
+    "VSCodium"
+]
+
+
 class Match(NamedTuple):
     data: str
     display_text: str
@@ -24,25 +31,31 @@ class Match(NamedTuple):
     properties: dict
 
 
-# TODO Check VSCode or Code - OSS or VSCodium
-
 # Read path_list from database
 def get_path_list():
-    con = sqlite3.connect(
-        os.path.join(
-            os.environ["HOME"], ".config", "Code - OSS/User/globalStorage/state.vscdb"
-        )
+    paths = {}
+
+    for vscode_dir in VSCODE_DIRS:
+        state_file = os.path.join(
+            os.environ["HOME"], ".config", vscode_dir, "User/globalStorage/state.vscdb"
         )
+
+        if not os.path.exists(state_file):
+            continue
+
+        con = sqlite3.connect(state_file)
         cur = con.cursor()
         rows = cur.execute(
             "SELECT value FROM ItemTable WHERE key = 'history.recentlyOpenedPathsList'"
         )
         data = json.loads(rows.fetchone()[0])
         con.close()
-    return [
+        paths.update(
+            {
                 "~" + path[len(os.environ["HOME"]) :] if os.environ["HOME"] in path else path
                 for path in [i["folderUri"][7:] for i in data["entries"] if "folderUri" in i]
-    ]
+            }
+        )
 
 
 class Runner(dbus.service.Object):