Save And Close Functionality (#17000)

* save and close

* wip

* working save and close

* cleanup

* pr changes

* pr changes

* fix capitalization

* fix build

* pr fix

extensions/sql-migration/src/dialog/assessmentResults/assessmentResultsDialog.ts

@@ -32,6 +32,9 @@ export class AssessmentResultsDialog {
 
 	constructor(public ownerUri: string, public model: MigrationStateModel, public title: string, private _skuRecommendationPage: SKURecommendationPage, private _targetType: MigrationTargetType) {
 		this._model = model;
+		if (this._model.resumeAssessment && this._model.savedInfo.closedPage >= 2) {
+			this._model._databaseAssessment = <string[]>this._model.savedInfo.databaseAssessment;
+		}
 		this._tree = new SqlDatabaseTree(this._model, this._targetType);
 	}
 

extensions/sql-migration/src/dialog/assessmentResults/savedAssessmentDialog.ts

@@ -0,0 +1,144 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 * as constants from '../../constants/strings';
+import { MigrationStateModel } from '../../models/stateMachine';
+import { WizardController } from '../../wizard/wizardController';
+
+export class SavedAssessmentDialog {
+
+	private static readonly OkButtonText: string = constants.NEXT_LABEL;
+	private static readonly CancelButtonText: string = constants.CANCEL_LABEL;
+
+	private _isOpen: boolean = false;
+	private dialog: azdata.window.Dialog | undefined;
+	private _rootContainer!: azdata.FlexContainer;
+	private stateModel: MigrationStateModel;
+	private context: vscode.ExtensionContext;
+	private _disposables: vscode.Disposable[] = [];
+
+	constructor(context: vscode.ExtensionContext, stateModel: MigrationStateModel) {
+		this.stateModel = stateModel;
+		this.context = context;
+	}
+
+	private async initializeDialog(dialog: azdata.window.Dialog): Promise<void> {
+		return new Promise<void>((resolve, reject) => {
+			dialog.registerContent(async (view) => {
+				try {
+					this._rootContainer = this.initializePageContent(view);
+					await view.initializeModel(this._rootContainer);
+					this._disposables.push(dialog.okButton.onClick(async e => {
+						await this.execute();
+					}));
+					this._disposables.push(dialog.cancelButton.onClick(e => {
+						this.cancel();
+					}));
+
+					this._disposables.push(view.onClosed(e => {
+						this._disposables.forEach(
+							d => { try { d.dispose(); } catch { } }
+						);
+					}));
+					resolve();
+				} catch (ex) {
+					reject(ex);
+				}
+			});
+		});
+	}
+
+	public async openDialog(dialogName?: string) {
+		if (!this._isOpen) {
+			this._isOpen = true;
+			this.dialog = azdata.window.createModelViewDialog(constants.SAVED_ASSESSMENT_RESULT, constants.SAVED_ASSESSMENT_RESULT, '60%');
+			this.dialog.okButton.label = SavedAssessmentDialog.OkButtonText;
+			this.dialog.cancelButton.label = SavedAssessmentDialog.CancelButtonText;
+			const dialogSetupPromises: Thenable<void>[] = [];
+			dialogSetupPromises.push(this.initializeDialog(this.dialog));
+			azdata.window.openDialog(this.dialog);
+			await Promise.all(dialogSetupPromises);
+		}
+	}
+
+	protected async execute() {
+		if (this.stateModel.resumeAssessment) {
+			const wizardController = new WizardController(this.context, this.stateModel);
+			await wizardController.openWizard(this.stateModel.sourceConnectionId);
+			console.log(this.stateModel.savedInfo.selectedDatabases);
+		} else {
+			// normal flow
+			const wizardController = new WizardController(this.context, this.stateModel);
+			await wizardController.openWizard(this.stateModel.sourceConnectionId);
+		}
+		this._isOpen = false;
+	}
+
+	protected cancel() {
+		this._isOpen = false;
+	}
+
+	public get isOpen(): boolean {
+		return this._isOpen;
+	}
+
+	public initializePageContent(view: azdata.ModelView): azdata.FlexContainer {
+		const buttonGroup = 'resumeMigration';
+
+		const text = view.modelBuilder.text().withProps({
+			CSSStyles: {
+				'font-size': '18px',
+				'font-weight': 'bold',
+				'margin': '100px 8px 0px 36px'
+			},
+			value: constants.RESUME_TITLE
+		}).component();
+
+		const radioStart = view.modelBuilder.radioButton().withProps({
+			label: constants.START_MIGRATION,
+			name: buttonGroup,
+			CSSStyles: {
+				'font-size': '14px',
+				'margin': '40px 8px 0px 36px'
+			},
+			checked: true
+		}).component();
+
+		radioStart.onDidChangeCheckedState((e) => {
+			if (e) {
+				this.stateModel.resumeAssessment = false;
+			}
+		});
+		const radioContinue = view.modelBuilder.radioButton().withProps({
+			label: constants.CONTINUE_MIGRATION,
+			name: buttonGroup,
+			CSSStyles: {
+				'font-size': '14px',
+				'margin': '10px 8px 0px 36px'
+			},
+			checked: false
+		}).component();
+
+		radioContinue.onDidChangeCheckedState((e) => {
+			if (e) {
+				this.stateModel.resumeAssessment = true;
+			}
+		});
+
+		const flex = view.modelBuilder.flexContainer().withLayout({
+			flexFlow: 'column',
+			height: '100%',
+			width: '100%'
+		}).component();
+		flex.addItem(text, { flex: '0 0 auto' });
+		flex.addItem(radioStart, { flex: '0 0 auto' });
+		flex.addItem(radioContinue, { flex: '0 0 auto' });
+
+		return flex;
+	}
+
+}

extensions/sql-migration/src/dialog/assessmentResults/sqlDatabasesTree.ts

@@ -187,6 +187,7 @@ export class SqlDatabaseTree {
 			await this._databaseCount.updateProperties({
 				'value': constants.DATABASES(this.selectedDbs().length, this._model._databaseAssessment.length)
 			});
+			this._model._databaseSelection = <azdata.DeclarativeTableCellValue[][]>this._databaseTable.dataValues;
 		}));
 
 		this._disposables.push(this._databaseTable.onRowSelected(async (e) => {
@@ -947,7 +948,11 @@ export class SqlDatabaseTree {
 			});
 		}
 		await this._instanceTable.setDataValues(instanceTableValues);
-		await this._databaseTable.setDataValues(this._databaseTableValues);
+		if (this._model.resumeAssessment && this._model.savedInfo.closedPage >= 2) {
+			await this._databaseTable.setDataValues(this._model.savedInfo.migrationDatabases);
+		} else {
+			await this._databaseTable.setDataValues(this._databaseTableValues);
+		}
 	}
 
 	// undo when bug #16445 is fixed