/*--------------------------------------------------------------------------------------------- * Copyright (c) Microsoft Corporation. All rights reserved. * Licensed under the Source EULA. See License.txt in the project root for license information. *--------------------------------------------------------------------------------------------*/ import * as azdata from 'azdata'; import * as vscode from 'vscode'; import { MigrationStateModel } from '../models/stateMachine'; import { SourceConfigurationPage } from './sourceConfigurationPage'; import { WIZARD_TITLE } from '../models/strings'; import { MigrationWizardPage } from '../models/migrationWizardPage'; import { SKURecommendationPage } from './skuRecommendationPage'; import { SubscriptionSelectionPage } from './subscriptionSelectionPage'; export class WizardController { constructor(private readonly extensionContext: vscode.ExtensionContext) { } public async openWizard(profile: azdata.connection.Connection): Promise { const stateModel = new MigrationStateModel(profile); this.extensionContext.subscriptions.push(stateModel); this.createWizard(stateModel); } private async createWizard(stateModel: MigrationStateModel): Promise { const wizard = azdata.window.createWizard(WIZARD_TITLE, 'wide'); wizard.generateScriptButton.enabled = false; const sourceConfigurationPage = new SourceConfigurationPage(wizard, stateModel); const skuRecommendationPage = new SKURecommendationPage(wizard, stateModel); const subscriptionSelectionPage = new SubscriptionSelectionPage(wizard, stateModel); const pages: MigrationWizardPage[] = [sourceConfigurationPage, skuRecommendationPage, subscriptionSelectionPage]; wizard.pages = pages.map(p => p.getwizardPage()); const wizardSetupPromises: Thenable[] = []; wizardSetupPromises.push(...pages.map(p => p.registerWizardContent())); wizardSetupPromises.push(wizard.open()); wizard.onPageChanged(async (pageChangeInfo: azdata.window.WizardPageChangeInfo) => { const newPage = pageChangeInfo.newPage; const lastPage = pageChangeInfo.lastPage; await pages[lastPage]?.onPageLeave(); await pages[newPage]?.onPageEnter(); }); wizard.registerNavigationValidator(async validator => { const lastPage = validator.lastPage; const canLeave = await pages[lastPage]?.canLeave() ?? true; const canEnter = await pages[lastPage]?.canEnter() ?? true; return canEnter && canLeave; // return true; }); await Promise.all(wizardSetupPromises); await pages[0].onPageEnter(); } }