W-17260440 - feat: gather context

packages/salesforcedx-vscode-apex/src/commands/metadataOrchestrator.ts

@@ -4,7 +4,7 @@
  * Licensed under the BSD 3-Clause license.
  * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
  */
 import { AiApiClient, CommandSource, ServiceProvider, ServiceType } from '@salesforce/vscode-service-provider';
 import * as path from 'path';
 import * as vscode from 'vscode';
 import { languageClientUtils } from '../languageUtils';
@@ -98,6 +98,26 @@
     return response;
   };
 
+  public gatherContext = async (
+    sourceUri: vscode.Uri | vscode.Uri[]
+  ): Promise<ApexClassOASEligibleResponses | undefined> => {
+    const telemetryService = await getTelemetryService();
+    let response;
+    const languageClient = languageClientUtils.getClientInstance();
+    if (languageClient) {
+      try {
+        response = (await languageClient?.sendRequest(
+          'apexoas/gatherContext',
+          sourceUri?.toString() ?? vscode.window.activeTextEditor?.document.uri.toString()
+        )) as ApexClassOASEligibleResponses;
+      } catch (error) {
+        // fallback TBD after we understand it better
+        throw new Error(nls.localize('cannot_gather_context'));
+      }
+    }
+    return response;
+  };
+
   public validateEligibility = async (
     sourceUri: vscode.Uri | vscode.Uri[],
     isMethodSelected: boolean = false
@@ -140,6 +160,7 @@
       requests.push(request);
     }
 
+    const response = this.gatherContext(sourceUri);
     const responses = await this.eligibilityDelegate({ payload: requests });
     return responses;
   };
@@ -203,6 +224,6 @@
   };
 
   getAiApiClient = async (): Promise<AiApiClient> => {
     return ServiceProvider.getService(ServiceType.AiApiClient);
   };
 }

packages/salesforcedx-vscode-apex/src/messages/i18n.ja.ts

@@ -63,6 +63,7 @@ export const messages = {
     'Method %s is not eligible for Apex Action creation. It is not annotated with @AuraEnabled or has wrong access modifiers.',
   unknown: 'Unknown',
   invalid_active_text_editor: 'The active text editor is missing or is an invalid file.',
+  cannot_gather_context: 'An error occurred while gathering context for the Apex class.',
   sobjects_no_refresh_if_already_active_error_text:
     'sObject 定義の更新が既に実行中です。プロセスを再起動する必要がある場合は、実行中のタスクをキャンセルしてください。',
   test_view_loading_message: 'Apex テストを読み込んでいます...',

packages/salesforcedx-vscode-apex/src/messages/i18n.ts

@@ -106,5 +106,6 @@ export const messages = {
   orphan_process_advice:
     "The list of processes below are Apex Language Server instances that didn't properly shutdown. These\nprocesses can be stopped from the warning message that brought you here, or you can handle this\ntask yourself. If you choose to terminate these processes yourself, refer to relevant documentation\nto stop these processes.",
   unknown: 'Unknown',
-  invalid_active_text_editor: 'The active text editor is missing or is an invalid file.'
+  invalid_active_text_editor: 'The active text editor is missing or is an invalid file.',
+  cannot_gather_context: 'An error occurred while gathering context for the Apex class.'
 };