Merge remote-tracking branch 'origin/main' into koesie10/export-results

Author: koesie10

extensions/ql-vscode/src/common/app.ts

@@ -0,0 +1,5 @@
+import { AppEventEmitter } from './events';
+
+export interface App {
+  createEventEmitter<T>(): AppEventEmitter<T>;
+}

extensions/ql-vscode/src/common/events.ts

@@ -0,0 +1,10 @@
+import { Disposable } from '../pure/disposable-object';
+
+export interface AppEvent<T> {
+  (listener: (event: T) => void): Disposable;
+}
+
+export interface AppEventEmitter<T> {
+  event: AppEvent<T>;
+  fire(data: T): void;
+}

extensions/ql-vscode/src/common/vscode/events.ts

@@ -0,0 +1,7 @@
+import * as vscode from 'vscode';
+import { AppEventEmitter } from '../events';
+
+export class VSCodeAppEventEmitter<T>
+  extends vscode.EventEmitter<T>
+  implements AppEventEmitter<T> {
+}

extensions/ql-vscode/src/common/vscode/vscode-app.ts

@@ -0,0 +1,9 @@
+import { App } from '../app';
+import { AppEventEmitter } from '../events';
+import { VSCodeAppEventEmitter } from './events';
+
+export class ExtensionApp implements App {
+  public createEventEmitter<T>(): AppEventEmitter<T> {
+    return new VSCodeAppEventEmitter<T>();
+  }
+}

extensions/ql-vscode/src/extension.ts

@@ -948,6 +948,12 @@ async function activateWithInstalledDistribution(
     })
   );
 
+  ctx.subscriptions.push(
+    commandRunner('codeQL.copyVariantAnalysisRepoList', async (variantAnalysisId: number) => {
+      await variantAnalysisManager.copyRepoListToClipboard(variantAnalysisId);
+    })
+  );
+
   ctx.subscriptions.push(
     commandRunner('codeQL.monitorVariantAnalysis', async (
       variantAnalysis: VariantAnalysis,

extensions/ql-vscode/src/pure/disposable-object.ts

@@ -1,7 +1,7 @@
 
 // Avoid explicitly referencing Disposable type in vscode.
 // This file cannot have dependencies on the vscode API.
-interface Disposable {
+export interface Disposable {
   dispose(): any;
 }
 

extensions/ql-vscode/src/pure/interface-types.ts

@@ -472,6 +472,10 @@ export interface OpenQueryTextMessage {
   t: 'openQueryText';
 }
 
+export interface CopyRepositoryListMessage {
+  t: 'copyRepositoryList';
+}
+
 export interface OpenLogsMessage {
   t: 'openLogs';
 }
@@ -490,5 +494,6 @@ export type FromVariantAnalysisMessage =
   | RequestRepositoryResultsMessage
   | OpenQueryFileMessage
   | OpenQueryTextMessage
+  | CopyRepositoryListMessage
   | OpenLogsMessage
   | CancelVariantAnalysisMessage;

extensions/ql-vscode/src/query-history.ts

@@ -1256,11 +1256,15 @@ export class QueryHistoryManager extends DisposableObject {
     const { finalSingleItem, finalMultiSelect } = this.determineSelection(singleItem, multiSelect);
 
     // Remote queries only
-    if (!this.assertSingleQuery(finalMultiSelect) || !finalSingleItem || finalSingleItem.t !== 'remote') {
+    if (!this.assertSingleQuery(finalMultiSelect) || !finalSingleItem) {
       return;
     }
 
-    await commands.executeCommand('codeQL.copyRepoList', finalSingleItem.queryId);
+    if (finalSingleItem.t === 'remote') {
+      await commands.executeCommand('codeQL.copyRepoList', finalSingleItem.queryId);
+    } else if (finalSingleItem.t === 'variant-analysis') {
+      await commands.executeCommand('codeQL.copyVariantAnalysisRepoList', finalSingleItem.variantAnalysis.id);
+    }
   }
 
   async handleExportResults(

extensions/ql-vscode/src/remote-queries/variant-analysis-manager.ts

@@ -1,7 +1,7 @@
 import * as path from 'path';
 
 import * as ghApiClient from './gh-api/gh-api-client';
-import { CancellationToken, commands, EventEmitter, ExtensionContext, Uri, window } from 'vscode';
+import { CancellationToken, commands, env, EventEmitter, ExtensionContext, Uri, window } from 'vscode';
 import { DisposableObject } from '../pure/disposable-object';
 import { Credentials } from '../authentication';
 import { VariantAnalysisMonitor } from './variant-analysis-monitor';
@@ -28,6 +28,7 @@ import {
 import PQueue from 'p-queue';
 import { createTimestampFile, showAndLogErrorMessage, showAndLogInformationMessage } from '../helpers';
 import * as fs from 'fs-extra';
+import * as os from 'os';
 import { cancelVariantAnalysis } from './gh-api/gh-actions-api-client';
 import { ProgressCallback, UserCancellationException } from '../commandRunner';
 import { CodeQLCliServer } from '../cli';
@@ -388,6 +389,27 @@ export class VariantAnalysisManager extends DisposableObject implements VariantA
     await cancelVariantAnalysis(credentials, variantAnalysis);
   }
 
+  public async copyRepoListToClipboard(variantAnalysisId: number) {
+    const variantAnalysis = this.variantAnalyses.get(variantAnalysisId);
+    if (!variantAnalysis) {
+      throw new Error(`No variant analysis with id: ${variantAnalysisId}`);
+    }
+
+    const fullNames = variantAnalysis.scannedRepos?.filter(a => a.resultCount && a.resultCount > 0).map(a => a.repository.fullName);
+    if (!fullNames || fullNames.length === 0) {
+      return;
+    }
+
+    const text = [
+      '"new-repo-list": [',
+      ...fullNames.slice(0, -1).map(repo => `    "${repo}",`),
+      `    "${fullNames[fullNames.length - 1]}"`,
+      ']'
+    ];
+
+    await env.clipboard.writeText(text.join(os.EOL));
+  }
+
   private getRepoStatesStoragePath(variantAnalysisId: number): string {
     return path.join(
       this.getVariantAnalysisStorageLocation(variantAnalysisId),

extensions/ql-vscode/src/remote-queries/variant-analysis-view.ts

@@ -103,6 +103,9 @@ export class VariantAnalysisView extends AbstractWebview<ToVariantAnalysisMessag
       case 'openQueryText':
         await this.openQueryText();
         break;
+      case 'copyRepositoryList':
+        void commands.executeCommand('codeQL.copyVariantAnalysisRepoList', this.variantAnalysisId);
+        break;
       case 'openLogs':
         await this.openLogs();
         break;