Create new queries in selected folder of queries panel

This will change the behavior of the "Create new query" command to
create the new query in the same folder as the first selected item in
the queries panel. If no items are selected, the behavior is the same
as before.

I've used events to communicate the selection from the queries panel to
the local queries module. This is some more code and some extra
complexity, but it ensures that we don't have a dependency from the
local queries module to the queries panel module. This makes testing
easier.

Author: koesie10

extensions/ql-vscode/src/extension.ts

@@ -795,7 +795,11 @@ async function activateWithInstalledDistribution(
   );
   ctx.subscriptions.push(databaseUI);
 
-  QueriesModule.initialize(app, languageContext, cliServer);
+  const queriesModule = QueriesModule.initialize(
+    app,
+    languageContext,
+    cliServer,
+  );
 
   void extLogger.log("Initializing evaluator log viewer.");
   const evalLogViewer = new EvalLogViewer();
@@ -934,6 +938,10 @@ async function activateWithInstalledDistribution(
   );
   ctx.subscriptions.push(localQueries);
 
+  queriesModule.onDidChangeSelection((event) =>
+    localQueries.setSelectedQueryTreeViewItems(event.selection),
+  );
+
   void extLogger.log("Initializing debugger factory.");
   ctx.subscriptions.push(
     new QLDebugAdapterDescriptorFactory(queryStorageDir, qs, localQueries),

extensions/ql-vscode/src/local-queries/local-queries.ts

@@ -63,6 +63,8 @@ export enum QuickEvalType {
 }
 
 export class LocalQueries extends DisposableObject {
+  private selectedQueryTreeViewItems: readonly QueryTreeViewItem[] = [];
+
   public constructor(
     private readonly app: App,
     private readonly queryRunner: QueryRunner,
@@ -77,6 +79,12 @@ export class LocalQueries extends DisposableObject {
     super();
   }
 
+  public setSelectedQueryTreeViewItems(
+    selection: readonly QueryTreeViewItem[],
+  ) {
+    this.selectedQueryTreeViewItems = selection;
+  }
+
   public getCommands(): LocalQueryCommands {
     return {
       "codeQL.runQuery": this.runQuery.bind(this),
@@ -333,6 +341,7 @@ export class LocalQueries extends DisposableObject {
           this.app.logger,
           this.databaseManager,
           contextStoragePath,
+          this.selectedQueryTreeViewItems,
           language,
         );
         await skeletonQueryWizard.execute();

extensions/ql-vscode/src/local-queries/skeleton-query-wizard.ts

@@ -1,5 +1,5 @@
-import { join } from "path";
-import { Uri, workspace, window as Window } from "vscode";
+import { dirname, join } from "path";
+import { Uri, window as Window, workspace } from "vscode";
 import { CodeQLCliServer } from "../codeql-cli/cli";
 import { BaseLogger } from "../common/logging";
 import { Credentials } from "../common/authentication";
@@ -24,8 +24,9 @@ import {
   isCodespacesTemplate,
   setQlPackLocation,
 } from "../config";
-import { existsSync } from "fs-extra";
+import { lstat, pathExists } from "fs-extra";
 import { askForLanguage } from "../codeql-cli/query-language";
+import { QueryTreeViewItem } from "../queries-panel/query-tree-view-item";
 
 type QueryLanguagesToDatabaseMap = Record<string, string>;
 
@@ -51,6 +52,7 @@ export class SkeletonQueryWizard {
     private readonly logger: BaseLogger,
     private readonly databaseManager: DatabaseManager,
     private readonly databaseStoragePath: string | undefined,
+    private readonly selectedItems: readonly QueryTreeViewItem[],
     private language: QueryLanguage | undefined = undefined,
   ) {}
 
@@ -71,7 +73,7 @@ export class SkeletonQueryWizard {
     this.qlPackStoragePath = await this.determineStoragePath();
 
     const skeletonPackAlreadyExists =
-      existsSync(join(this.qlPackStoragePath, this.folderName)) ||
+      (await pathExists(join(this.qlPackStoragePath, this.folderName))) ||
       isFolderAlreadyInWorkspace(this.folderName);
 
     if (skeletonPackAlreadyExists) {
@@ -109,7 +111,29 @@ export class SkeletonQueryWizard {
     });
   }
 
-  public async determineStoragePath() {
+  public async determineStoragePath(): Promise<string> {
+    if (this.selectedItems.length === 0) {
+      return this.determineRootStoragePath();
+    }
+
+    // Just like VS Code's "New File" command, if the user has selected multiple files/folders in the queries panel,
+    // we will create the new file in the same folder as the first selected item.
+    // See https://github.com/microsoft/vscode/blob/a8b7239d0311d4915b57c837972baf4b01394491/src/vs/workbench/contrib/files/browser/fileActions.ts#L893-L900
+    const selectedItem = this.selectedItems[0];
+
+    const path = selectedItem.path;
+
+    // We use stat to protect against outdated query tree items
+    const fileStat = await lstat(path);
+
+    if (fileStat.isDirectory()) {
+      return path;
+    }
+
+    return dirname(path);
+  }
+
+  public async determineRootStoragePath() {
     const firstStorageFolder = getFirstWorkspaceFolder();
 
     if (isCodespacesTemplate()) {
@@ -118,7 +142,7 @@ export class SkeletonQueryWizard {
 
     let storageFolder = getQlPackLocation();
 
-    if (storageFolder === undefined || !existsSync(storageFolder)) {
+    if (storageFolder === undefined || !(await pathExists(storageFolder))) {
       storageFolder = await Window.showInputBox({
         title:
           "Please choose a folder in which to create your new query pack. You can change this in the extension settings.",
@@ -131,7 +155,7 @@ export class SkeletonQueryWizard {
       throw new UserCancellationException("No storage folder entered.");
     }
 
-    if (!existsSync(storageFolder)) {
+    if (!(await pathExists(storageFolder))) {
       throw new UserCancellationException(
         "Invalid folder. Must be a folder that already exists.",
       );

extensions/ql-vscode/src/queries-panel/queries-module.ts

@@ -7,9 +7,18 @@ import { QueriesPanel } from "./queries-panel";
 import { QueryDiscovery } from "./query-discovery";
 import { QueryPackDiscovery } from "./query-pack-discovery";
 import { LanguageContextStore } from "../language-context-store";
+import { TreeViewSelectionChangeEvent } from "vscode";
+import { QueryTreeViewItem } from "./query-tree-view-item";
 
 export class QueriesModule extends DisposableObject {
   private queriesPanel: QueriesPanel | undefined;
+  private readonly onDidChangeSelectionEmitter = this.push(
+    this.app.createEventEmitter<
+      TreeViewSelectionChangeEvent<QueryTreeViewItem>
+    >(),
+  );
+
+  public readonly onDidChangeSelection = this.onDidChangeSelectionEmitter.event;
 
   private constructor(readonly app: App) {
     super();
@@ -52,6 +61,9 @@ export class QueriesModule extends DisposableObject {
     void queryDiscovery.initialRefresh();
 
     this.queriesPanel = new QueriesPanel(queryDiscovery, app);
+    this.queriesPanel.onDidChangeSelection((event) =>
+      this.onDidChangeSelectionEmitter.fire(event),
+    );
     this.push(this.queriesPanel);
   }
 }

extensions/ql-vscode/src/queries-panel/queries-panel.ts

@@ -1,7 +1,13 @@
 import { DisposableObject } from "../common/disposable-object";
 import { QueryTreeDataProvider } from "./query-tree-data-provider";
 import { QueryDiscovery } from "./query-discovery";
-import { TextEditor, TreeView, window } from "vscode";
+import {
+  Event,
+  TextEditor,
+  TreeView,
+  TreeViewSelectionChangeEvent,
+  window,
+} from "vscode";
 import { App } from "../common/app";
 import { QueryTreeViewItem } from "./query-tree-view-item";
 
@@ -16,6 +22,7 @@ export class QueriesPanel extends DisposableObject {
     super();
 
     this.dataProvider = new QueryTreeDataProvider(queryDiscovery, app);
+    this.push(this.dataProvider);
 
     this.treeView = window.createTreeView("codeQLQueries", {
       treeDataProvider: this.dataProvider,
@@ -25,6 +32,12 @@ export class QueriesPanel extends DisposableObject {
     this.subscribeToTreeSelectionEvents();
   }
 
+  public get onDidChangeSelection(): Event<
+    TreeViewSelectionChangeEvent<QueryTreeViewItem>
+  > {
+    return this.treeView.onDidChangeSelection;
+  }
+
   private subscribeToTreeSelectionEvents(): void {
     // Keep track of whether the user has changed their text editor while
     // the tree view was not visible. If so, we will focus the text editor