feat: identify a namespace to place in generated query VSCODE-579

package.json

@@ -87,9 +87,11 @@
         "id": "mongodb.participant",
         "name": "MongoDB",
         "description": "Ask anything about MongoDB, from writing queries to questions about your cluster.",
+        "isSticky": true,
         "commands": [
           {
             "name": "query",
+            "isSticky": true,
             "description": "Ask how to write MongoDB queries or pipelines. For example, you can ask: \"Show me all the documents where the address contains the word street\"."
           }
         ]
@@ -161,13 +163,25 @@
       }
     ],
     "commands": [
+      {
+        "command": "mdb.selectDatabaseWithParticipant",
+        "title": "MongoDB: Select Database with Participant"
+      },
+      {
+        "command": "mdb.selectCollectionWithParticipant",
+        "title": "MongoDB: Select Collection with Participant"
+      },
+      {
+        "command": "mdb.connectWithParticipant",
+        "title": "MongoDB: Change Active Connection with Participant"
+      },
       {
         "command": "mdb.runParticipantQuery",
-        "title": "Run Content Generated by the Chat Participant"
+        "title": "Run Content Generated by Participant"
       },
       {
         "command": "mdb.openParticipantQueryInPlayground",
-        "title": "Open Generated by the Chat Participant Content In Playground"
+        "title": "Open Generated by Participant Content In Playground"
       },
       {
         "command": "mdb.connect",
@@ -719,6 +733,22 @@
         }
       ],
       "commandPalette": [
+        {
+          "command": "mdb.selectDatabaseWithParticipant",
+          "when": "false"
+        },
+        {
+          "command": "mdb.selectCollectionWithParticipant",
+          "when": "false"
+        },
+        {
+          "command": "mdb.connectWithParticipant",
+          "when": "false"
+        },
+        {
+          "command": "mdb.runParticipantQuery",
+          "when": "false"
+        },
         {
           "command": "mdb.openParticipantQueryInPlayground",
           "when": "false"

src/commands/index.ts

@@ -79,6 +79,9 @@ enum EXTENSION_COMMANDS {
   // Chat participant.
   OPEN_PARTICIPANT_QUERY_IN_PLAYGROUND = 'mdb.openParticipantQueryInPlayground',
   RUN_PARTICIPANT_QUERY = 'mdb.runParticipantQuery',
+  CONNECT_WITH_PARTICIPANT = 'mdb.connectWithParticipant',
+  SELECT_DATABASE_WITH_PARTICIPANT = 'mdb.selectDatabaseWithParticipant',
+  SELECT_COLLECTION_WITH_PARTICIPANT = 'mdb.selectCollectionWithParticipant',
 }
 
 export default EXTENSION_COMMANDS;

src/connectionController.ts

@@ -419,7 +419,14 @@ export default class ConnectionController {
     }
 
     log.info('Successfully connected', { connectionId });
-    void vscode.window.showInformationMessage('MongoDB connection successful.');
+
+    const message = 'MongoDB connection successful.';
+    this._statusView.showMessage(message);
+    setTimeout(() => {
+      if (this._statusView._statusBarItem.text === message) {
+        this._statusView.hideMessage();
+      }
+    }, 5000);
 
     dataService.addReauthenticationHandler(
       this._reauthenticationHandler.bind(this)
@@ -603,10 +610,16 @@ export default class ConnectionController {
       'mdb.isAtlasStreams',
       false
     );
-    void vscode.window.showInformationMessage('MongoDB disconnected.');
 
     this._disconnecting = false;
-    this._statusView.hideMessage();
+
+    const message = 'MongoDB disconnected.';
+    this._statusView.showMessage(message);
+    setTimeout(() => {
+      if (this._statusView._statusBarItem.text === message) {
+        this._statusView.hideMessage();
+      }
+    }, 5000);
 
     return true;
   }

src/editors/mongoDBDocumentService.ts

@@ -98,13 +98,11 @@ export default class MongoDBDocumentService {
           returnDocument: 'after',
         }
       );
-
-      this._statusView.hideMessage();
       this._telemetryService.trackDocumentUpdated(source, true);
     } catch (error) {
-      this._statusView.hideMessage();
-
       return this._saveDocumentFailed(formatError(error).message);
+    } finally {
+      this._statusView.hideMessage();
     }
   }
 
@@ -141,17 +139,15 @@ export default class MongoDBDocumentService {
         { limit: 1 }
       );
 
-      this._statusView.hideMessage();
-
       if (!documents || documents.length === 0) {
         return;
       }
 
       return getEJSON(documents[0]);
     } catch (error) {
-      this._statusView.hideMessage();
-
       return this._fetchDocumentFailed(formatError(error).message);
+    } finally {
+      this._statusView.hideMessage();
     }
   }
 }

src/mdbExtensionController.ts

@@ -292,7 +292,8 @@ export default class MDBExtensionController implements vscode.Disposable {
       () => {
         return this._playgroundController.createPlaygroundFromParticipantQuery({
           text:
-            this._participantController._chatResult.metadata.queryContent || '',
+            this._participantController._chatResult?.metadata
+              ?.responseContent || '',
         });
       }
     );
@@ -301,10 +302,25 @@ export default class MDBExtensionController implements vscode.Disposable {
       () => {
         return this._playgroundController.evaluateParticipantQuery({
           text:
-            this._participantController._chatResult.metadata.queryContent || '',
+            this._participantController._chatResult?.metadata
+              ?.responseContent || '',
         });
       }
     );
+    this.registerCommand(
+      EXTENSION_COMMANDS.CONNECT_WITH_PARTICIPANT,
+      (id: string) => this._participantController.connectWithParticipant(id)
+    );
+    this.registerCommand(
+      EXTENSION_COMMANDS.SELECT_DATABASE_WITH_PARTICIPANT,
+      (name: string) =>
+        this._participantController.selectDatabaseWithParticipant(name)
+    );
+    this.registerCommand(
+      EXTENSION_COMMANDS.SELECT_COLLECTION_WITH_PARTICIPANT,
+      (name: string) =>
+        this._participantController.selectCollectionWithParticipant(name)
+    );
   };
 
   registerParticipantCommand = (