Reduce nesting in runRemoteQuery

Now that we do not have a dry run mode, we can create and clean up the
temporary directory in the same function. This allows us to remove the
complete try..finally block inside `runRemoteQuery` and move it to a
much more local spot.

Author: koesie10

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

@@ -181,7 +181,6 @@ export async function prepareRemoteQueryRun(
   cliServer: cli.CodeQLCliServer,
   credentials: Credentials,
   uri: Uri | undefined,
-  queryPackDir: string,
   progress: ProgressCallback,
   token: CancellationToken,
 ) {
@@ -225,7 +224,17 @@ export async function prepareRemoteQueryRun(
     throw new UserCancellationException('Cancelled');
   }
 
-  const { base64Pack, language } = await generateQueryPack(cliServer, queryFile, queryPackDir);
+  const { remoteQueryDir, queryPackDir } = await createRemoteQueriesTempDirectory();
+
+  let pack: Awaited<ReturnType<typeof generateQueryPack>>;
+
+  try {
+    pack = await generateQueryPack(cliServer, queryFile, queryPackDir);
+  } finally {
+    await remoteQueryDir.cleanup();
+  }
+
+  const { base64Pack, language } = pack;
 
   if (token.isCancellationRequested) {
     throw new UserCancellationException('Cancelled');
@@ -266,87 +275,81 @@ export async function runRemoteQuery(
       } or later.`);
   }
 
-  const { remoteQueryDir, queryPackDir } = await createRemoteQueriesTempDirectory();
-  try {
-    const {
-      actionBranch,
-      base64Pack,
-      repoSelection,
-      queryFile,
-      queryMetadata,
-      controllerRepo,
-      queryStartTime,
-      language,
-    } = await prepareRemoteQueryRun(cliServer, credentials, uri, queryPackDir, progress, token);
+  const {
+    actionBranch,
+    base64Pack,
+    repoSelection,
+    queryFile,
+    queryMetadata,
+    controllerRepo,
+    queryStartTime,
+    language,
+  } = await prepareRemoteQueryRun(cliServer, credentials, uri, progress, token);
 
-    if (isVariantAnalysisLiveResultsEnabled()) {
-      const queryName = getQueryName(queryMetadata, queryFile);
-      const variantAnalysisLanguage = parseVariantAnalysisQueryLanguage(language);
-      if (variantAnalysisLanguage === undefined) {
-        throw new UserCancellationException(`Found unsupported language: ${language}`);
-      }
+  if (isVariantAnalysisLiveResultsEnabled()) {
+    const queryName = getQueryName(queryMetadata, queryFile);
+    const variantAnalysisLanguage = parseVariantAnalysisQueryLanguage(language);
+    if (variantAnalysisLanguage === undefined) {
+      throw new UserCancellationException(`Found unsupported language: ${language}`);
+    }
 
-      const queryText = await fs.readFile(queryFile, 'utf8');
-
-      const variantAnalysisSubmission: VariantAnalysisSubmission = {
-        startTime: queryStartTime,
-        actionRepoRef: actionBranch,
-        controllerRepoId: controllerRepo.id,
-        query: {
-          name: queryName,
-          filePath: queryFile,
-          pack: base64Pack,
-          language: variantAnalysisLanguage,
-          text: queryText,
-        },
-        databases: {
-          repositories: repoSelection.repositories,
-          repositoryLists: repoSelection.repositoryLists,
-          repositoryOwners: repoSelection.owners
-        }
-      };
+    const queryText = await fs.readFile(queryFile, 'utf8');
 
-      const variantAnalysisResponse = await ghApiClient.submitVariantAnalysis(
-        credentials,
-        variantAnalysisSubmission
-      );
+    const variantAnalysisSubmission: VariantAnalysisSubmission = {
+      startTime: queryStartTime,
+      actionRepoRef: actionBranch,
+      controllerRepoId: controllerRepo.id,
+      query: {
+        name: queryName,
+        filePath: queryFile,
+        pack: base64Pack,
+        language: variantAnalysisLanguage,
+        text: queryText,
+      },
+      databases: {
+        repositories: repoSelection.repositories,
+        repositoryLists: repoSelection.repositoryLists,
+        repositoryOwners: repoSelection.owners
+      }
+    };
 
-      const processedVariantAnalysis = processVariantAnalysis(variantAnalysisSubmission, variantAnalysisResponse);
+    const variantAnalysisResponse = await ghApiClient.submitVariantAnalysis(
+      credentials,
+      variantAnalysisSubmission
+    );
 
-      await variantAnalysisManager.onVariantAnalysisSubmitted(processedVariantAnalysis);
+    const processedVariantAnalysis = processVariantAnalysis(variantAnalysisSubmission, variantAnalysisResponse);
 
-      void logger.log(`Variant analysis:\n${JSON.stringify(processedVariantAnalysis, null, 2)}`);
+    await variantAnalysisManager.onVariantAnalysisSubmitted(processedVariantAnalysis);
 
-      void showAndLogInformationMessage(`Variant analysis ${processedVariantAnalysis.query.name} submitted for processing`);
+    void logger.log(`Variant analysis:\n${JSON.stringify(processedVariantAnalysis, null, 2)}`);
 
-      void commands.executeCommand('codeQL.openVariantAnalysisView', processedVariantAnalysis.id);
-      void commands.executeCommand('codeQL.monitorVariantAnalysis', processedVariantAnalysis);
+    void showAndLogInformationMessage(`Variant analysis ${processedVariantAnalysis.query.name} submitted for processing`);
 
-      return { variantAnalysis: processedVariantAnalysis };
-    } else {
-      const apiResponse = await runRemoteQueriesApiRequest(credentials, actionBranch, language, repoSelection, controllerRepo, base64Pack);
+    void commands.executeCommand('codeQL.openVariantAnalysisView', processedVariantAnalysis.id);
+    void commands.executeCommand('codeQL.monitorVariantAnalysis', processedVariantAnalysis);
 
-      if (!apiResponse) {
-        return;
-      }
+    return { variantAnalysis: processedVariantAnalysis };
+  } else {
+    const apiResponse = await runRemoteQueriesApiRequest(credentials, actionBranch, language, repoSelection, controllerRepo, base64Pack);
 
-      const workflowRunId = apiResponse.workflow_run_id;
-      const repositoryCount = apiResponse.repositories_queried.length;
-      const remoteQuery = await buildRemoteQueryEntity(
-        queryFile,
-        queryMetadata,
-        controllerRepo,
-        queryStartTime,
-        workflowRunId,
-        language,
-        repositoryCount);
-
-      // don't return the path because it has been deleted
-      return { query: remoteQuery };
+    if (!apiResponse) {
+      return;
     }
 
-  } finally {
-    await remoteQueryDir.cleanup();
+    const workflowRunId = apiResponse.workflow_run_id;
+    const repositoryCount = apiResponse.repositories_queried.length;
+    const remoteQuery = await buildRemoteQueryEntity(
+      queryFile,
+      queryMetadata,
+      controllerRepo,
+      queryStartTime,
+      workflowRunId,
+      language,
+      repositoryCount);
+
+    // don't return the path because it has been deleted
+    return { query: remoteQuery };
   }
 }