feat: Demo of codelens for Jake

travisstebbins · Jan 28, 2025 · 71fe98a · 71fe98a
1 parent b50fc48
commit 71fe98a
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 1 deletion.
diff --git a/packages/extension/src/LegendCodelensProvider.ts b/packages/extension/src/LegendCodelensProvider.ts
@@ -27,7 +27,7 @@ import {
 } from './LegendLanguageClient';
 import { TextDocumentIdentifier } from 'vscode-languageclient';
 import { type LegendEntity } from './model/LegendEntity';
-import { CLASSIFIER_PATH } from '@finos/legend-engine-ide-client-vscode-shared';
+import { CLASSIFIER_PATH, TextLocation } from '@finos/legend-engine-ide-client-vscode-shared';
 
 export class LegendCodelensProvider implements CodeLensProvider {
   private client: LegendLanguageClient;
@@ -55,6 +55,24 @@ export class LegendCodelensProvider implements CodeLensProvider {
     );
   }
 
+  private addDatabaseCodeLens(entityLocation: TextLocation): void {
+    this.codeLenses.push(
+      new CodeLens(
+        new Range(
+          entityLocation.textInterval.start.line,
+          entityLocation.textInterval.start.column,
+          entityLocation.textInterval.end.line,
+          entityLocation.textInterval.end.column,
+        ),
+        {
+          title: 'Edit/Execute in QueryBuilder',
+          command: LEGEND_GENERATE_MODELS,
+          arguments: [entityLocation],
+        },
+      ),
+    );
+  }
+
   public async provideCodeLenses(
     document: TextDocument,
     _token: CancellationToken,
@@ -74,6 +92,9 @@ export class LegendCodelensProvider implements CodeLensProvider {
       ) {
         this.addQueryBuilderCodeLens(entity);
       }
+      if (entity.classifierPath === CLASSIFIER_PATH.RELATION) {
+        this.addDatabaseCodeLens(entity.location);
+      }
     });
     return this.codeLenses;
   }

diff --git a/packages/extension/src/LegendLanguageClient.ts b/packages/extension/src/LegendLanguageClient.ts
@@ -548,4 +548,24 @@ export class LegendLanguageClient extends LanguageClient {
           executableArgs,
         );
   }
+
+  async generateModels(
+    entityDetails:
+      | TextLocation
+      | { documentUri: string; sectionIndex: number; entityId: string },
+  ): Promise<LegendExecutionResult[]> {
+    return entityDetails instanceof TextLocation
+      ? commands.executeCommand(
+          LEGEND_COMMAND,
+          entityDetails,
+          GET_QUERY_TYPEAHEAD_COMMAND_ID,
+        )
+      : commands.executeCommand(
+          LEGEND_COMMAND,
+          entityDetails.documentUri,
+          entityDetails.sectionIndex,
+          entityDetails.entityId,
+          GET_QUERY_TYPEAHEAD_COMMAND_ID,
+        );
+  }
 }
diff --git a/packages/extension/src/extension.ts b/packages/extension/src/extension.ts
@@ -77,6 +77,7 @@ import {
   SERVICE_QUERY_EDITOR,
   SHOW_RESULTS_COMMAND_ID,
   TDSLegendExecutionResult,
+  TextLocation,
 } from '@finos/legend-engine-ide-client-vscode-shared';
 import { LegendWebViewProvider } from './utils/LegendWebViewProvider';
 import {
@@ -501,6 +502,15 @@ export function registerCommands(context: ExtensionContext): void {
   );
   context.subscriptions.push(editInQueryBuilder);
 
+  const generateModels = commands.registerCommand(
+    LEGEND_GENERATE_MODELS, // TODO: add as const
+    async (...args: unknown[]) => {
+      const path = args[0] as TextLocation;
+      await client.generateModels(path);
+    }
+
+  )
+
   const oneEntityPerFileRefactor = commands.registerCommand(
     ONE_ENTITY_PER_FILE_COMMAND_ID,
     () => {