feat: show MongoDB completion items before other JS completion VSCODE-382, VSCODE-385 (#490)

* feat: show MongoDB completion items before other JS completion VSCODE-382

* docs: add comment.

* test: context in one line

* fix: close connections

* refactor: use worker only for evaluate

* refactor: types clean up

* refactor: close connection in finally

* refactor: hmm ignore mongosh/browser-runtime-electron

* test: wait 1000

* test: fake getTranspiledContent

* docs: add completion comments

* test: rename suite name

* refactor: return default connection details

* refactor: address pr comments

* fix: send error to formatError

* test: try to mock info as well

* test: try to not count stages

* refactor: remove ignored dep

Author: alenakhineika

.github/PULL_REQUEST_TEMPLATE.md

@@ -1,11 +1,10 @@
 <!-- Ticket number and a general summary of your changes in the Title above -->
 <!-- e.g. VSCODE-1111: updates ace editor width in agg pipeline view -->
-
 <!--- The following fields are not obligatory. Use your best judgement on what you think is applicable to the work you've done -->
-
 ## Description
 <!--- Describe your changes in detail -->
 <!--- If applicable, describe (or illustrate) architecture flow -->
+
 ### Checklist
 - [ ] New tests and/or benchmarks are included
 - [ ] Documentation is changed or added

package-lock.json

@@ -1005,6 +1005,7 @@
   },
   "devDependencies": {
     "@mongodb-js/prettier-config-compass": "^1.0.0",
+    "@mongosh/service-provider-core": "^1.8.0",
     "@types/analytics-node": "^3.1.9",
     "@types/babel__core": "^7.1.19",
     "@types/babel__traverse": "^7.18.1",

package.json

@@ -24,7 +24,7 @@ import playgroundCloneDocumentTemplate from '../templates/playgroundCloneDocumen
 import playgroundInsertDocumentTemplate from '../templates/playgroundInsertDocumentTemplate';
 import {
   PlaygroundResult,
-  ShellExecuteAllResult,
+  ShellEvaluateResult,
   ExportToLanguageAddons,
   ExportToLanguageNamespace,
   ExportToLanguageMode,
@@ -378,7 +378,7 @@ export default class PlaygroundController {
     return this._createPlaygroundFileWithContent(content);
   }
 
-  async _evaluate(codeToEvaluate: string): Promise<ShellExecuteAllResult> {
+  async _evaluate(codeToEvaluate: string): Promise<ShellEvaluateResult> {
     const connectionId = this._connectionController.getActiveConnectionId();
 
     if (!connectionId) {
@@ -391,8 +391,8 @@ export default class PlaygroundController {
 
     try {
       // Send a request to the language server to execute scripts from a playground.
-      const result: ShellExecuteAllResult =
-        await this._languageServerController.executeAll({
+      const result: ShellEvaluateResult =
+        await this._languageServerController.evaluate({
           codeToEvaluate,
           connectionId,
         });
@@ -429,7 +429,7 @@ export default class PlaygroundController {
     return this._activeTextEditor?.document.getText(selection) || '';
   }
 
-  async _evaluateWithCancelModal(): Promise<ShellExecuteAllResult> {
+  async _evaluateWithCancelModal(): Promise<ShellEvaluateResult> {
     if (!this._connectionController.isCurrentlyConnected()) {
       throw new Error(
         'Please connect to a database before running a playground.'
@@ -452,7 +452,7 @@ export default class PlaygroundController {
           });
 
           // Run all playground scripts.
-          const result: ShellExecuteAllResult = await this._evaluate(
+          const result: ShellEvaluateResult = await this._evaluate(
             this._codeToEvaluate
           );
 
@@ -550,7 +550,7 @@ export default class PlaygroundController {
 
     this._outputChannel.clear();
 
-    const evaluateResponse: ShellExecuteAllResult =
+    const evaluateResponse: ShellEvaluateResult =
       await this._evaluateWithCancelModal();
 
     if (evaluateResponse?.outputLines?.length) {

src/editors/playgroundController.ts

@@ -12,8 +12,8 @@ import { workspace, ExtensionContext } from 'vscode';
 
 import { createLogger } from '../logging';
 import {
-  PlaygroundExecuteParameters,
-  ShellExecuteAllResult,
+  PlaygroundEvaluateParams,
+  ShellEvaluateResult,
   ExportToLanguageMode,
   ExportToLanguageNamespace,
   PlaygroundTextAndSelection,
@@ -120,9 +120,9 @@ export default class LanguageServerController {
     void this._client.stop();
   }
 
-  async executeAll(
-    playgroundExecuteParameters: PlaygroundExecuteParameters
-  ): Promise<ShellExecuteAllResult> {
+  async evaluate(
+    playgroundExecuteParameters: PlaygroundEvaluateParams
+  ): Promise<ShellEvaluateResult> {
     this._isExecutingInProgress = true;
 
     // Instantiate a new CancellationTokenSource object
@@ -132,8 +132,8 @@ export default class LanguageServerController {
     // Send a request with a cancellation token
     // to the language server server to execute scripts from a playground
     // and return results to the playground controller when ready
-    const result: ShellExecuteAllResult = await this._client.sendRequest(
-      ServerCommands.EXECUTE_ALL_FROM_PLAYGROUND,
+    const result: ShellEvaluateResult = await this._client.sendRequest(
+      ServerCommands.EXECUTE_CODE_FROM_PLAYGROUND,
       playgroundExecuteParameters,
       this._source.token
     );
@@ -188,4 +188,11 @@ export default class LanguageServerController {
       this._isExecutingInProgress = false;
     }
   }
+
+  async updateCurrentSessionFields(params): Promise<void> {
+    await this._client.sendRequest(
+      ServerCommands.UPDATE_CURRENT_SESSION_FIELDS,
+      params
+    );
+  }
 }