Move extension pack resolution into LocalQueries

This way the debug configuration will be able to override it.

Author: Dave Bartolomeo

extensions/ql-vscode/src/contextual/queryResolver.ts

@@ -178,6 +178,7 @@ export async function runContextualQuery(
     { queryPath: query, quickEvalPosition: undefined },
     false,
     getOnDiskWorkspaceFolders(),
+    undefined,
     queryStorageDir,
     undefined,
     templates,

extensions/ql-vscode/src/legacy-query-server/legacyRunner.ts

@@ -61,6 +61,7 @@ export class LegacyQueryRunner extends QueryRunner {
     dbPath: string,
     query: CoreQueryTarget,
     additionalPacks: string[],
+    extensionPacks: string[] | undefined,
     generateEvalLog: boolean,
     outputDir: QueryOutputDir,
     progress: ProgressCallback,
@@ -74,6 +75,7 @@ export class LegacyQueryRunner extends QueryRunner {
       query,
       generateEvalLog,
       additionalPacks,
+      extensionPacks,
       outputDir,
       progress,
       token,

extensions/ql-vscode/src/legacy-query-server/run-queries.ts

@@ -308,12 +308,17 @@ export async function compileAndRunQueryAgainstDatabaseCore(
   query: CoreQueryTarget,
   generateEvalLog: boolean,
   additionalPacks: string[],
+  extensionPacks: string[] | undefined,
   outputDir: QueryOutputDir,
   progress: ProgressCallback,
   token: CancellationToken,
   templates: Record<string, string> | undefined,
   logger: Logger,
 ): Promise<CoreQueryResults> {
+  if (extensionPacks !== undefined && extensionPacks.length > 0) {
+    throw new Error("Legacy query server does not support extension packs.");
+  }
+
   const dbContents = await DatabaseResolver.resolveDatabaseContents(
     Uri.file(dbPath),
   );

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

@@ -455,14 +455,20 @@ export class LocalQueries extends DisposableObject {
       throw new Error("Can't run query without a selected database");
     }
 
+    const additionalPacks = getOnDiskWorkspaceFolders();
+    const extensionPacks = (await this.cliServer.useExtensionPacks())
+      ? Object.keys(await this.cliServer.resolveQlpacks(additionalPacks, true))
+      : undefined;
+
     const coreQueryRun = this.queryRunner.createQueryRun(
       databaseItem.databaseUri.fsPath,
       {
         queryPath: selectedQuery.queryPath,
         quickEvalPosition: selectedQuery.quickEvalPosition,
       },
       true,
-      getOnDiskWorkspaceFolders(),
+      additionalPacks,
+      extensionPacks,
       this.queryStorageDir,
       undefined,
       undefined,

extensions/ql-vscode/src/query-server/query-runner.ts

@@ -72,6 +72,7 @@ export class NewQueryRunner extends QueryRunner {
     dbPath: string,
     query: CoreQueryTarget,
     additionalPacks: string[],
+    extensionPacks: string[] | undefined,
     generateEvalLog: boolean,
     outputDir: QueryOutputDir,
     progress: ProgressCallback,
@@ -85,6 +86,7 @@ export class NewQueryRunner extends QueryRunner {
       query,
       generateEvalLog,
       additionalPacks,
+      extensionPacks,
       outputDir,
       progress,
       token,

extensions/ql-vscode/src/query-server/run-queries.ts

@@ -26,6 +26,7 @@ export async function compileAndRunQueryAgainstDatabaseCore(
   query: CoreQueryTarget,
   generateEvalLog: boolean,
   additionalPacks: string[],
+  extensionPacks: string[] | undefined,
   outputDir: QueryOutputDir,
   progress: ProgressCallback,
   token: CancellationToken,
@@ -39,10 +40,6 @@ export async function compileAndRunQueryAgainstDatabaseCore(
         }
       : { query: {} };
 
-  const extensionPacks = (await qs.cliServer.useExtensionPacks())
-    ? Object.keys(await qs.cliServer.resolveQlpacks(additionalPacks, true))
-    : undefined;
-
   const evalLogPath = generateEvalLog ? outputDir.evalLogPath : undefined;
   const queryToRun: messages.RunQueryParams = {
     db: dbPath,

extensions/ql-vscode/src/queryRunner.ts

@@ -70,6 +70,7 @@ export abstract class QueryRunner {
     dbPath: string,
     query: CoreQueryTarget,
     additionalPacks: string[],
+    extensionPacks: string[] | undefined,
     generateEvalLog: boolean,
     outputDir: QueryOutputDir,
     progress: ProgressCallback,
@@ -109,6 +110,7 @@ export abstract class QueryRunner {
     query: CoreQueryTarget,
     generateEvalLog: boolean,
     additionalPacks: string[],
+    extensionPacks: string[] | undefined,
     queryStorageDir: string,
     id: string | undefined,
     templates: Record<string, string> | undefined,
@@ -135,6 +137,7 @@ export abstract class QueryRunner {
             dbPath,
             query,
             additionalPacks,
+            extensionPacks,
             generateEvalLog,
             outputDir,
             progress,