feat: Demo of codelens for Jake

packages/extension/src/LegendCodelensProvider.ts

@@ -27,7 +27,7 @@
 } 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 @@
     );
   }
 
+  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 @@
       ) {
         this.addQueryBuilderCodeLens(entity);
       }
+      if (entity.classifierPath === CLASSIFIER_PATH.RELATION) {
+        this.addDatabaseCodeLens(entity.location);
+      }
     });
     return this.codeLenses;
   }

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,
+        );
+  }
 }

packages/extension/src/extension.ts

@@ -52,7 +52,7 @@
 } from '@finos/legend-vscode-extension-dependencies';
 import { LegendTreeDataProvider } from './utils/LegendTreeProvider';
 import { LanguageClientProgressResult } from './model/LanguageClientProgressResult';
 import {
   ACTIVATE_FUNCTION_ID,
   CLASSIFIER_PATH,
   DIAGRAM_RENDERER,
@@ -77,6 +77,7 @@
   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 @@
   );
   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,
     () => {