Allow selecting mongo version when creating database server (#2223)

* Allow selecting mongo version * Remove noisy pick options * Localize wizard string
microsoft · Jan 19, 2024 · 3944a7c · 3944a7c
1 parent 9497ddb
commit 3944a7c
Show file tree

Hide file tree

Showing 4 changed files with 33 additions and 2 deletions.
diff --git a/src/tree/AzureDBAPIStep.ts b/src/tree/AzureDBAPIStep.ts
@@ -20,6 +20,7 @@ import { CosmosDBAccountCapacityStep } from './CosmosDBAccountWizard/CosmosDBAcc
 import { CosmosDBAccountCreateStep } from './CosmosDBAccountWizard/CosmosDBAccountCreateStep';
 import { CosmosDBAccountNameStep } from './CosmosDBAccountWizard/CosmosDBAccountNameStep';
 import { ICosmosDBWizardContext } from './CosmosDBAccountWizard/ICosmosDBWizardContext';
+import { MongoVersionStep } from './CosmosDBAccountWizard/MongoDBVersionStep';
 import { IAzureDBWizardContext } from './IAzureDBWizardContext';
 
 export class AzureDBAPIStep extends AzureWizardPromptStep<IPostgresServerWizardContext | ICosmosDBWizardContext> {
@@ -61,7 +62,8 @@ export class AzureDBAPIStep extends AzureWizardPromptStep<IPostgresServerWizardC
             promptSteps = [
                 new CosmosDBAccountNameStep(),
                 new CosmosDBAccountCapacityStep(),
-            ];
+                context.defaultExperience?.api === API.MongoDB ? new MongoVersionStep() : undefined
+            ].filter((step): step is AzureWizardPromptStep<ICosmosDBWizardContext> => step !== undefined);
             executeSteps = [
                 new CosmosDBAccountCreateStep(),
                 new VerifyProvidersStep(['Microsoft.DocumentDB'])

diff --git a/src/tree/CosmosDBAccountWizard/CosmosDBAccountCreateStep.ts b/src/tree/CosmosDBAccountWizard/CosmosDBAccountCreateStep.ts
@@ -40,7 +40,9 @@ export class CosmosDBAccountCreateStep extends AzureWizardExecuteStep<ICosmosDBW
         };
 
         if (defaultExperience?.api === 'MongoDB') {
-            options.apiProperties = { serverVersion: '3.6' };
+            if (context.mongoVersion !== undefined) {
+                options.apiProperties = { serverVersion: context.mongoVersion };
+            }
         }
 
         if (defaultExperience.capability) {

diff --git a/src/tree/CosmosDBAccountWizard/ICosmosDBWizardContext.ts b/src/tree/CosmosDBAccountWizard/ICosmosDBWizardContext.ts
@@ -16,4 +16,5 @@ export interface ICosmosDBWizardContext extends IAzureDBWizardContext, ExecuteAc
     databaseAccount?: DatabaseAccountGetResults;
     isServerless?: boolean;
 
+    mongoVersion?: string;
 }
diff --git a/src/tree/CosmosDBAccountWizard/MongoDBVersionStep.ts b/src/tree/CosmosDBAccountWizard/MongoDBVersionStep.ts
@@ -0,0 +1,26 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+
+import { AzureWizardPromptStep } from '@microsoft/vscode-azext-utils';
+import { localize } from "../../utils/localize";
+import { ICosmosDBWizardContext } from './ICosmosDBWizardContext';
+
+export class MongoVersionStep extends AzureWizardPromptStep<ICosmosDBWizardContext> {
+
+    public async prompt(context: ICosmosDBWizardContext): Promise<void> {
+        const mongoVersionOption = await context.ui.showQuickPick([
+            { label: "v4.0", detail: "4.0" },
+            { label: "v3.6", detail: "3.6" },
+            { label: "v3.2", detail: "3.2" },
+        ], {
+            placeHolder: localize("selectMongoVersion", "Select MongoDB version")
+        });
+        context.mongoVersion = mongoVersionOption.detail;
+    }
+
+    public shouldPrompt(_context: ICosmosDBWizardContext): boolean {
+        return true;
+    }
+}