Merge pull request #2892 from github/starcke/local-query-lang-dto

Add language to local query history items

Author: starcke

extensions/ql-vscode/CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## [UNRELEASED]
 
+- It is now possible to show the language of query history items using the `%l` specifier in the `codeQL.queryHistory.format` setting. Note that this only works for queries run after this upgrade, and older items will show `unknown` as a language. [#2892](https://github.com/github/vscode-codeql/pull/2892)
 - Increase the required version of VS Code to 1.82.0. [#2877](https://github.com/github/vscode-codeql/pull/2877)
 - Fix a bug where the query server was restarted twice after configuration changes. [#2884](https://github.com/github/vscode-codeql/pull/2884).
 

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

@@ -21,6 +21,7 @@ import { Method, Usage } from "../model-editor/method";
 import { ModeledMethod } from "../model-editor/modeled-method";
 import { ModelEditorViewState } from "../model-editor/shared/view-state";
 import { Mode } from "../model-editor/shared/mode";
+import { QueryLanguage } from "./query-language";
 
 /**
  * This module contains types and code that are shared between
@@ -51,6 +52,7 @@ export const RAW_RESULTS_LIMIT = 10000;
 export interface DatabaseInfo {
   name: string;
   databaseUri: string;
+  language?: QueryLanguage;
 }
 
 /** Arbitrary query metadata */

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

@@ -49,6 +49,7 @@ import { LocalQueryRun } from "./local-query-run";
 import { createMultiSelectionCommand } from "../common/vscode/selection-commands";
 import { findLanguage } from "../codeql-cli/query-language";
 import type { QueryTreeViewItem } from "../queries-panel/query-tree-view-item";
+import { tryGetQueryLanguage } from "../common/query-language";
 
 interface DatabaseQuickPickItem extends QuickPickItem {
   databaseItem: DatabaseItem;
@@ -364,6 +365,7 @@ export class LocalQueries extends DisposableObject {
     const initialInfo = await createInitialQueryInfo(selectedQuery, {
       databaseUri: dbItem.databaseUri.toString(),
       name: dbItem.name,
+      language: tryGetQueryLanguage(dbItem.language),
     });
 
     // When cancellation is requested from the query history view, we just stop the debug session.

extensions/ql-vscode/src/query-history/history-item-label-provider.ts

@@ -4,6 +4,7 @@ import { QueryHistoryConfig } from "../config";
 import { LocalQueryInfo } from "../query-results";
 import {
   buildRepoLabel,
+  getLanguage,
   getRawQueryName,
   QueryHistoryInfo,
 } from "./query-history-info";
@@ -19,6 +20,7 @@ interface InterpolateReplacements {
   r: string; // Result count/Empty
   s: string; // Status
   f: string; // Query file name
+  l: string; // Query language
   "%": "%"; // Percent sign
 }
 
@@ -84,6 +86,7 @@ export class HistoryItemLabelProvider {
       r: `(${resultCount} results)`,
       s: statusString,
       f: item.getQueryFileName(),
+      l: this.getLanguageLabel(item),
       "%": "%",
     };
   }
@@ -103,7 +106,13 @@ export class HistoryItemLabelProvider {
       r: resultCount,
       s: humanizeQueryStatus(item.status),
       f: basename(item.variantAnalysis.query.filePath),
+      l: this.getLanguageLabel(item),
       "%": "%",
     };
   }
+
+  private getLanguageLabel(item: QueryHistoryInfo): string {
+    const language = getLanguage(item);
+    return language === undefined ? "unknown" : `${language}`;
+  }
 }

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

@@ -6,6 +6,7 @@ import {
   hasRepoScanCompleted,
   getActionsWorkflowRunUrl as getVariantAnalysisActionsWorkflowRunUrl,
 } from "../variant-analysis/shared/variant-analysis";
+import { QueryLanguage } from "../common/query-language";
 
 export type QueryHistoryInfo = LocalQueryInfo | VariantAnalysisHistoryItem;
 
@@ -49,6 +50,17 @@ export function getQueryText(item: QueryHistoryInfo): string {
   }
 }
 
+export function getLanguage(item: QueryHistoryInfo): QueryLanguage | undefined {
+  switch (item.t) {
+    case "local":
+      return item.initialInfo.databaseInfo.language;
+    case "variant-analysis":
+      return item.variantAnalysis.query.language;
+    default:
+      assertNever(item);
+  }
+}
+
 export function buildRepoLabel(item: VariantAnalysisHistoryItem): string {
   const totalScannedRepositoryCount =
     item.variantAnalysis.scannedRepos?.length ?? 0;

extensions/ql-vscode/src/query-history/store/query-history-domain-mapper.ts

@@ -1,8 +1,9 @@
 import { assertNever } from "../../common/helpers-pure";
 import { QueryHistoryInfo } from "../query-history-info";
 import { mapLocalQueryInfoToDto } from "./query-history-local-query-domain-mapper";
-import { QueryHistoryItemDto } from "./query-history-dto";
+import { QueryHistoryItemDto, QueryLanguageDto } from "./query-history-dto";
 import { mapQueryHistoryVariantAnalysisToDto } from "./query-history-variant-analysis-domain-mapper";
+import { QueryLanguage } from "../../common/query-language";
 
 export function mapQueryHistoryToDto(
   queries: QueryHistoryInfo[],
@@ -17,3 +18,28 @@ export function mapQueryHistoryToDto(
     }
   });
 }
+
+export function mapQueryLanguageToDto(
+  language: QueryLanguage,
+): QueryLanguageDto {
+  switch (language) {
+    case QueryLanguage.CSharp:
+      return QueryLanguageDto.CSharp;
+    case QueryLanguage.Cpp:
+      return QueryLanguageDto.Cpp;
+    case QueryLanguage.Go:
+      return QueryLanguageDto.Go;
+    case QueryLanguage.Java:
+      return QueryLanguageDto.Java;
+    case QueryLanguage.Javascript:
+      return QueryLanguageDto.Javascript;
+    case QueryLanguage.Python:
+      return QueryLanguageDto.Python;
+    case QueryLanguage.Ruby:
+      return QueryLanguageDto.Ruby;
+    case QueryLanguage.Swift:
+      return QueryLanguageDto.Swift;
+    default:
+      assertNever(language);
+  }
+}

extensions/ql-vscode/src/query-history/store/query-history-dto-mapper.ts

@@ -1,7 +1,9 @@
 import { QueryHistoryInfo } from "../query-history-info";
-import { QueryHistoryItemDto } from "./query-history-dto";
+import { QueryHistoryItemDto, QueryLanguageDto } from "./query-history-dto";
 import { mapQueryHistoryVariantAnalysisToDomainModel } from "./query-history-variant-analysis-dto-mapper";
 import { mapLocalQueryItemToDomainModel } from "./query-history-local-query-dto-mapper";
+import { QueryLanguage } from "../../common/query-language";
+import { assertNever } from "../../common/helpers-pure";
 
 export function mapQueryHistoryToDomainModel(
   queries: QueryHistoryItemDto[],
@@ -20,3 +22,28 @@ export function mapQueryHistoryToDomainModel(
     );
   });
 }
+
+export function mapQueryLanguageToDomainModel(
+  language: QueryLanguageDto,
+): QueryLanguage {
+  switch (language) {
+    case QueryLanguageDto.CSharp:
+      return QueryLanguage.CSharp;
+    case QueryLanguageDto.Cpp:
+      return QueryLanguage.Cpp;
+    case QueryLanguageDto.Go:
+      return QueryLanguage.Go;
+    case QueryLanguageDto.Java:
+      return QueryLanguage.Java;
+    case QueryLanguageDto.Javascript:
+      return QueryLanguage.Javascript;
+    case QueryLanguageDto.Python:
+      return QueryLanguage.Python;
+    case QueryLanguageDto.Ruby:
+      return QueryLanguage.Ruby;
+    case QueryLanguageDto.Swift:
+      return QueryLanguage.Swift;
+    default:
+      assertNever(language);
+  }
+}

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

@@ -12,3 +12,14 @@ export interface QueryHistoryDto {
 export type QueryHistoryItemDto =
   | QueryHistoryLocalQueryDto
   | QueryHistoryVariantAnalysisDto;
+
+export enum QueryLanguageDto {
+  CSharp = "csharp",
+  Cpp = "cpp",
+  Go = "go",
+  Java = "java",
+  Javascript = "javascript",
+  Python = "python",
+  Ruby = "ruby",
+  Swift = "swift",
+}

extensions/ql-vscode/src/query-history/store/query-history-local-query-domain-mapper.ts

@@ -17,6 +17,7 @@ import {
   SortDirection,
   SortedResultSetInfo,
 } from "../../common/interface-types";
+import { mapQueryLanguageToDto } from "./query-history-domain-mapper";
 
 export function mapLocalQueryInfoToDto(
   query: LocalQueryInfo,
@@ -101,6 +102,10 @@ function mapInitialQueryInfoToDto(
     databaseInfo: {
       databaseUri: localQueryInitialInfo.databaseInfo.databaseUri,
       name: localQueryInitialInfo.databaseInfo.name,
+      language:
+        localQueryInitialInfo.databaseInfo.language === undefined
+          ? undefined
+          : mapQueryLanguageToDto(localQueryInitialInfo.databaseInfo.language),
     },
     start: localQueryInitialInfo.start,
     id: localQueryInitialInfo.id,

extensions/ql-vscode/src/query-history/store/query-history-local-query-dto-mapper.ts

@@ -20,6 +20,7 @@ import {
   SortDirection,
   SortedResultSetInfo,
 } from "../../common/interface-types";
+import { mapQueryLanguageToDomainModel } from "./query-history-dto-mapper";
 
 export function mapLocalQueryItemToDomainModel(
   localQuery: QueryHistoryLocalQueryDto,
@@ -82,6 +83,10 @@ function mapInitialQueryInfoToDomainModel(
     databaseInfo: {
       databaseUri: initialInfo.databaseInfo.databaseUri,
       name: initialInfo.databaseInfo.name,
+      language:
+        initialInfo.databaseInfo.language === undefined
+          ? undefined
+          : mapQueryLanguageToDomainModel(initialInfo.databaseInfo.language),
     },
     start: new Date(initialInfo.start),
     id: initialInfo.id,