Implement PR feedback

Author: koesie10

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

@@ -42,7 +42,7 @@ export interface QueryConstraints {
  * @param additionalPacks Additional pack paths to search.
  * @returns The found queries from the first pack in which any matching queries were found.
  */
-async function resolveQueriesFromPacks(
+export async function resolveQueriesFromPacks(
   cli: CodeQLCliServer,
   qlpacks: string[],
   constraints: QueryConstraints,
@@ -99,7 +99,6 @@ export async function resolveQueriesByLanguagePack(
  * @param packsToSearch The list of packs to search.
  * @param name The name of the query to use in error messages.
  * @param constraints Constraints on the queries to search for.
- * @param allowNoQueriesFound If true, will not throw an error if no queries are found.
  * @param additionalPacks Additional pack paths to search.
  * @returns The found queries from the first pack in which any matching queries were found.
  */
@@ -108,7 +107,6 @@ export async function resolveQueries(
   packsToSearch: string[],
   name: string,
   constraints: QueryConstraints,
-  allowNoQueriesFound = false,
   additionalPacks: string[] = [],
 ): Promise<string[]> {
   const queries = await resolveQueriesFromPacks(
@@ -135,10 +133,6 @@ export async function resolveQueries(
     );
   }
 
-  if (allowNoQueriesFound) {
-    return [];
-  }
-
   const joinedPacksToSearch = packsToSearch.join(", ");
   const error = redactableError`No ${name} queries (${humanConstraints.join(
     ", ",

extensions/ql-vscode/src/model-editor/external-api-usage-queries.ts

@@ -105,9 +105,13 @@ export async function runExternalApiQueries(
     mode,
     [syntheticQueryPackName],
     [queryDir],
-    false,
   );
   if (!queryPath) {
+    void showAndLogExceptionWithTelemetry(
+      extLogger,
+      telemetryListener,
+      redactableError`The ${mode} model editor query could not be found. Try re-opening the model editor. If that doesn't work, try upgrading the CodeQL libraries.`,
+    );
     return;
   }
 

extensions/ql-vscode/src/model-editor/model-editor-queries.ts

@@ -6,9 +6,8 @@ import { prepareExternalApiQuery } from "./external-api-usage-queries";
 import { CodeQLCliServer } from "../codeql-cli/cli";
 import { showLlmGeneration } from "../config";
 import { Mode } from "./shared/mode";
-import { resolveQueries } from "../local-queries";
+import { resolveQueriesFromPacks } from "../local-queries";
 import { modeTag } from "./mode-tag";
-import { extLogger } from "../common/logging/vscode";
 
 export const syntheticQueryPackName = "codeql/external-api-usage";
 
@@ -39,18 +38,17 @@ export async function setUpPack(
   // Download the required query packs
   await cliServer.packDownload([`codeql/${language}-queries`]);
 
+  // We'll only check if the application mode query exists in the pack and assume that if it does,
+  // the framework mode query will also exist.
   const applicationModeQuery = await resolveEndpointsQuery(
     cliServer,
     language,
     Mode.Application,
     [],
     [],
-    true,
   );
 
   if (applicationModeQuery) {
-    void extLogger.log("Using application mode queries");
-
     // Set up a synthetic pack so CodeQL doesn't crash later when we try
     // to resolve a query within this directory
     const syntheticQueryPack = {
@@ -62,8 +60,6 @@ export async function setUpPack(
     const qlpackFile = join(queryDir, "codeql-pack.yml");
     await writeFile(qlpackFile, dump(syntheticQueryPack), "utf8");
   } else {
-    void extLogger.log("Writing external API usage queries to disk");
-
     // If we can't resolve the query, we need to write them to desk ourselves.
     const externalApiQuerySuccess = await prepareExternalApiQuery(
       queryDir,
@@ -105,31 +101,27 @@ export async function setUpPack(
  * @param mode The mode to resolve the query for.
  * @param additionalPackNames Additional pack names to search.
  * @param additionalPackPaths Additional pack paths to search.
- * @param allowNoQueriesFound If true, will not throw an error if no queries are found.
  */
 export async function resolveEndpointsQuery(
   cliServer: CodeQLCliServer,
   language: string,
   mode: Mode,
   additionalPackNames: string[] = [],
   additionalPackPaths: string[] = [],
-  allowNoQueriesFound = false,
 ): Promise<string | undefined> {
   const packsToSearch = [`codeql/${language}-queries`, ...additionalPackNames];
 
   // First, resolve the query that we want to run.
   // All queries are tagged like this:
   // internal extract automodel <mode> <queryTag>
   // Example: internal extract automodel framework-mode candidates
-  const queries = await resolveQueries(
+  const queries = await resolveQueriesFromPacks(
     cliServer,
     packsToSearch,
-    `Fetch endpoints query for ${mode}`,
     {
       kind: "table",
       "tags contain all": ["modeleditor", "endpoints", modeTag(mode)],
     },
-    allowNoQueriesFound,
     additionalPackPaths,
   );
   if (queries.length > 1) {