STREAMS-576: do not show MDB version for SPI

mongodb-js · Sep 20, 2023 · 911035c · 911035c
1 parent e099bc0
commit 911035c
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 8 deletions.
diff --git a/packages/cli-repl/src/mongosh-repl.spec.ts b/packages/cli-repl/src/mongosh-repl.spec.ts
@@ -1309,4 +1309,28 @@ describe('MongoshNodeRepl', function () {
       expect(output).to.not.match(/Using MongoDB/);
     });
   });
+
+  context('with is_stream: true', function () {
+    beforeEach(async function () {
+      sp.getConnectionInfo.resolves({
+        extraInfo: {
+          uri: 'mongodb://localhost:27017/test',
+          is_localhost: true,
+          is_stream: true,
+        },
+        buildInfo: {
+          version: '4.4.1',
+        },
+      });
+      mongoshReplOptions.shellCliOptions = {
+        nodb: false,
+      };
+      mongoshRepl = new MongoshNodeRepl(mongoshReplOptions);
+      await mongoshRepl.initialize(serviceProvider);
+    });
+
+    it('shows Atlas Stream Processing', function () {
+      expect(output).to.match(/Using MongoDB:\t\tAtlas Stream Processing/);
+    });
+  });
 });
diff --git a/packages/cli-repl/src/mongosh-repl.ts b/packages/cli-repl/src/mongosh-repl.ts
@@ -173,12 +173,17 @@ class MongoshNodeRepl implements EvaluationListener {
     instanceState.setEvaluationListener(this);
     await instanceState.fetchConnectionInfo();
 
-    let mongodVersion = instanceState.connectionInfo.buildInfo?.version;
-    const apiVersion = serviceProvider.getRawClient()?.serverApi?.version;
-    if (apiVersion) {
-      mongodVersion =
-        (mongodVersion ? mongodVersion + ' ' : '') +
-        `(API Version ${apiVersion})`;
+    const { buildInfo, extraInfo } = instanceState.connectionInfo;
+    let mongodVersion = extraInfo?.is_stream
+      ? 'Atlas Stream Processing'
+      : buildInfo?.version;
+    if (!extraInfo?.is_stream) {
+      const apiVersion = serviceProvider.getRawClient()?.serverApi?.version;
+      if (apiVersion) {
+        mongodVersion =
+          (mongodVersion ? mongodVersion + ' ' : '') +
+          `(API Version ${apiVersion})`;
+      }
     }
     await this.greet(mongodVersion, moreRecentMongoshVersion);
     await this.printBasicConnectivityWarning(instanceState);
@@ -456,15 +461,15 @@ class MongoshNodeRepl implements EvaluationListener {
    * The greeting for the shell, showing server and shell version.
    */
   async greet(
-    mongodVersion: string,
+    mongodVersion?: string,
     moreRecentMongoshVersion?: string | null
   ): Promise<void> {
     if (this.shellCliOptions.quiet) {
       return;
     }
     const { version } = require('../package.json');
     let text = '';
-    if (!this.shellCliOptions.nodb) {
+    if (mongodVersion && !this.shellCliOptions.nodb) {
       text += `Using MongoDB:\t\t${mongodVersion}\n`;
     }
     text += `${this.clr(