Show selected item in new DB panel (#1824)

Author: shati-patel

extensions/ql-vscode/src/databases/db-module.ts

@@ -1,10 +1,12 @@
+import { window } from "vscode";
 import { App, AppMode } from "../common/app";
 import { isCanary, isNewQueryRunExperienceEnabled } from "../config";
 import { extLogger } from "../common";
 import { DisposableObject } from "../pure/disposable-object";
 import { DbConfigStore } from "./config/db-config-store";
 import { DbManager } from "./db-manager";
 import { DbPanel } from "./ui/db-panel";
+import { DbSelectionDecorationProvider } from "./ui/db-selection-decoration-provider";
 
 export class DbModule extends DisposableObject {
   public async initialize(app: App): Promise<void> {
@@ -30,6 +32,10 @@ export class DbModule extends DisposableObject {
 
     this.push(dbPanel);
     this.push(dbConfigStore);
+
+    const dbSelectionDecorationProvider = new DbSelectionDecorationProvider();
+
+    window.registerFileDecorationProvider(dbSelectionDecorationProvider);
   }
 }
 

extensions/ql-vscode/src/databases/ui/db-selection-decoration-provider.ts

@@ -0,0 +1,22 @@
+import {
+  CancellationToken,
+  FileDecoration,
+  FileDecorationProvider,
+  ProviderResult,
+  Uri,
+} from "vscode";
+
+export class DbSelectionDecorationProvider implements FileDecorationProvider {
+  provideFileDecoration(
+    uri: Uri,
+    _token: CancellationToken,
+  ): ProviderResult<FileDecoration> {
+    if (uri?.query === "selected=true") {
+      return {
+        badge: "✔",
+      };
+    }
+
+    return undefined;
+  }
+}