Pass sourceLocationPrefix down through all the functions

Author: shati-patel

extensions/ql-vscode/src/remote-queries/analyses-results-manager.ts

@@ -148,7 +148,7 @@ export class AnalysesResultsManager {
         status: 'Completed'
       };
     } else if (fileExtension === '.bqrs') {
-      const queryResults = await this.readBqrsResults(artifactPath, fileLinkPrefix);
+      const queryResults = await this.readBqrsResults(artifactPath, fileLinkPrefix, analysis.sourceLocationPrefix);
       newAnaysisResults = {
         ...analysisResults,
         rawResults: queryResults,
@@ -180,8 +180,8 @@ export class AnalysesResultsManager {
     return await fs.pathExists(createDownloadPath(this.storagePath, analysis.downloadLink));
   }
 
-  private async readBqrsResults(filePath: string, fileLinkPrefix: string): Promise<AnalysisRawResults> {
-    return await extractRawResults(this.cliServer, this.logger, filePath, fileLinkPrefix);
+  private async readBqrsResults(filePath: string, fileLinkPrefix: string, sourceLocationPrefix: string): Promise<AnalysisRawResults> {
+    return await extractRawResults(this.cliServer, this.logger, filePath, fileLinkPrefix, sourceLocationPrefix);
   }
 
   private async readSarifResults(filePath: string, fileLinkPrefix: string): Promise<AnalysisAlert[]> {

extensions/ql-vscode/src/remote-queries/bqrs-processing.ts

@@ -9,6 +9,7 @@ export async function extractRawResults(
   logger: Logger,
   filePath: string,
   fileLinkPrefix: string,
+  sourceLocationPrefix: string
 ): Promise<AnalysisRawResults> {
   const bqrsInfo = await cliServer.bqrsInfo(filePath);
   const resultSets = bqrsInfo['result-sets'];
@@ -31,5 +32,5 @@ export async function extractRawResults(
 
   const capped = !!chunk.next;
 
-  return { schema, resultSet, fileLinkPrefix, capped };
+  return { schema, resultSet, fileLinkPrefix, sourceLocationPrefix, capped };
 }

extensions/ql-vscode/src/remote-queries/gh-actions-api-client.ts

@@ -17,6 +17,7 @@ interface ApiSuccessIndexItem {
   results_count: number;
   bqrs_file_size: number;
   sarif_file_size?: number;
+  source_location_prefix: string;
 }
 
 interface ApiFailureIndexItem {
@@ -59,7 +60,8 @@ export async function getRemoteQueryIndex(
       sha: item.sha,
       resultCount: item.results_count,
       bqrsFileSize: item.bqrs_file_size,
-      sarifFileSize: item.sarif_file_size
+      sarifFileSize: item.sarif_file_size,
+      sourceLocationPrefix: item.source_location_prefix
     } as RemoteQuerySuccessIndexItem;
   });
 

extensions/ql-vscode/src/remote-queries/remote-queries-interface.ts

@@ -291,6 +291,7 @@ export class RemoteQueriesInterfaceManager {
       databaseSha: analysisResult.databaseSha || 'HEAD',
       resultCount: analysisResult.resultCount,
       downloadLink: analysisResult.downloadLink,
+      sourceLocationPrefix: analysisResult.sourceLocationPrefix,
       fileSize: this.formatFileSize(analysisResult.fileSizeInBytes),
       starCount: analysisResult.starCount,
       lastUpdated: analysisResult.lastUpdated

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

@@ -171,6 +171,7 @@ export class RemoteQueriesManager extends DisposableObject {
         nwo: a.nwo,
         databaseSha: a.databaseSha,
         resultCount: a.resultCount,
+        sourceLocationPrefix: a.sourceLocationPrefix,
         downloadLink: a.downloadLink,
         fileSize: String(a.fileSizeInBytes)
       }));
@@ -191,6 +192,7 @@ export class RemoteQueriesManager extends DisposableObject {
       nwo: item.nwo,
       databaseSha: item.sha || 'HEAD',
       resultCount: item.resultCount,
+      sourceLocationPrefix: item.sourceLocationPrefix,
       fileSizeInBytes: item.sarifFileSize ? item.sarifFileSize : item.bqrsFileSize,
       starCount: metadata[item.nwo]?.starCount,
       lastUpdated: metadata[item.nwo]?.lastUpdated,

extensions/ql-vscode/src/remote-queries/remote-queries-markdown-generation.ts

@@ -238,7 +238,7 @@ function generateMarkdownForRawResults(
 
   for (const row of analysisRawResults.resultSet.rows) {
     const cells = row.map((cell) =>
-      generateMarkdownForRawTableCell(cell, analysisRawResults.fileLinkPrefix)
+      generateMarkdownForRawTableCell(cell, analysisRawResults.fileLinkPrefix, analysisRawResults.sourceLocationPrefix)
     );
     tableRows.push(`| ${cells.join(' | ')} |`);
   }
@@ -247,7 +247,8 @@ function generateMarkdownForRawResults(
 
 function generateMarkdownForRawTableCell(
   value: CellValue,
-  fileLinkPrefix: string
+  fileLinkPrefix: string,
+  sourceLocationPrefix: string
 ) {
   let cellValue: string;
   switch (typeof value) {
@@ -258,7 +259,7 @@ function generateMarkdownForRawTableCell(
       break;
     case 'object':
       {
-        const url = tryGetRemoteLocation(value.url, fileLinkPrefix);
+        const url = tryGetRemoteLocation(value.url, fileLinkPrefix, sourceLocationPrefix);
         cellValue = `[\`${convertNonPrintableChars(value.label)}\`](${url})`;
       }
       break;

extensions/ql-vscode/src/remote-queries/remote-query-result-index.ts

@@ -12,6 +12,7 @@ export interface RemoteQuerySuccessIndexItem {
   resultCount: number;
   bqrsFileSize: number;
   sarifFileSize?: number;
+  sourceLocationPrefix: string;
 }
 
 export interface RemoteQueryFailureIndexItem {

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

@@ -12,6 +12,7 @@ export interface AnalysisSummary {
   nwo: string,
   databaseSha: string,
   resultCount: number,
+  sourceLocationPrefix: string,
   downloadLink: DownloadLink,
   fileSizeInBytes: number,
   starCount?: number,

extensions/ql-vscode/src/remote-queries/shared/analysis-result.ts

@@ -16,6 +16,7 @@ export interface AnalysisRawResults {
   schema: ResultSetSchema;
   resultSet: RawResultSet;
   fileLinkPrefix: string;
+  sourceLocationPrefix: string;
   capped: boolean;
 }
 

extensions/ql-vscode/src/remote-queries/shared/remote-query-result.ts

@@ -21,6 +21,7 @@ export interface AnalysisSummary {
   nwo: string,
   databaseSha: string,
   resultCount: number,
+  sourceLocationPrefix: string,
   downloadLink: DownloadLink,
   fileSize: string,
   starCount?: number,