azuredatastudio/extensions/sql-migration/src/dialog/retryMigration/retryMigrationDialog.ts

/*---------------------------------------------------------------------------------------------
 *  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 mssql from 'mssql';
import { azureResource } from 'azurecore';
import { getLocations, getResourceGroupFromId, getBlobContainerId, getFullResourceGroupFromId, getResourceName, DatabaseMigration, getMigrationTargetInstance } from '../../api/azure';
import { MigrationMode, MigrationStateModel, NetworkContainerType, SavedInfo } from '../../models/stateMachine';
import { MigrationServiceContext } from '../../models/migrationLocalStorage';
import { WizardController } from '../../wizard/wizardController';
import { getMigrationModeEnum, getMigrationTargetTypeEnum } from '../../constants/helper';
import * as constants from '../../constants/strings';

export class RetryMigrationDialog {

	constructor(
		private readonly _context: vscode.ExtensionContext,
		private readonly _serviceContext: MigrationServiceContext,
		private readonly _migration: DatabaseMigration,
		private readonly _onClosedCallback: () => Promise<void>) {
	}

	private async createMigrationStateModel(serviceContext: MigrationServiceContext, migration: DatabaseMigration, connectionId: string, serverName: string, api: mssql.IExtension, location: azureResource.AzureLocation): Promise<MigrationStateModel> {
		let stateModel = new MigrationStateModel(this._context, connectionId, api.sqlMigration);

		const sourceDatabaseName = migration.properties.sourceDatabaseName;
		let savedInfo: SavedInfo;
		savedInfo = {
			closedPage: 0,

			// DatabaseSelector
			databaseAssessment: [sourceDatabaseName],

			// SKURecommendation
			databaseList: [sourceDatabaseName],
			serverAssessment: null,
			skuRecommendation: null,
			migrationTargetType: getMigrationTargetTypeEnum(migration)!,

			// TargetSelection
			azureAccount: serviceContext.azureAccount!,
			azureTenant: serviceContext.azureAccount!.properties.tenants[0]!,
			subscription: serviceContext.subscription!,
			location: location,
			resourceGroup: {
				id: getFullResourceGroupFromId(migration.id),
				name: getResourceGroupFromId(migration.id),
				subscription: serviceContext.subscription!,
			},
			targetServerInstance: await getMigrationTargetInstance(
				serviceContext.azureAccount!,
				serviceContext.subscription!,
				migration),

			// MigrationMode
			migrationMode: getMigrationModeEnum(migration),

			// DatabaseBackup
			targetDatabaseNames: [migration.name],
			networkContainerType: null,
			networkShares: [],
			blobs: [],

			// Integration Runtime
			sqlMigrationService: serviceContext.migrationService,
		};

		const getStorageAccountResourceGroup = (storageAccountResourceId: string): azureResource.AzureResourceResourceGroup => {
			return {
				id: getFullResourceGroupFromId(storageAccountResourceId!),
				name: getResourceGroupFromId(storageAccountResourceId!),
				subscription: this._serviceContext.subscription!
			};
		};
		const getStorageAccount = (storageAccountResourceId: string): azureResource.AzureGraphResource => {
			const storageAccountName = getResourceName(storageAccountResourceId);
			return {
				type: 'microsoft.storage/storageaccounts',
				id: storageAccountResourceId!,
				tenantId: savedInfo.azureTenant?.id!,
				subscriptionId: this._serviceContext.subscription?.id!,
				name: storageAccountName,
				location: savedInfo.location!.name,
			};
		};

		const sourceLocation = migration.properties.backupConfiguration?.sourceLocation;
		if (sourceLocation?.fileShare) {
			savedInfo.networkContainerType = NetworkContainerType.NETWORK_SHARE;
			const storageAccountResourceId = migration.properties.backupConfiguration?.targetLocation?.storageAccountResourceId!;
			savedInfo.networkShares = [
				{
					password: '',
					networkShareLocation: sourceLocation?.fileShare?.path!,
					windowsUser: sourceLocation?.fileShare?.username!,
					storageAccount: getStorageAccount(storageAccountResourceId!),
					resourceGroup: getStorageAccountResourceGroup(storageAccountResourceId!),
					storageKey: ''
				}
			];
		} else if (sourceLocation?.azureBlob) {
			savedInfo.networkContainerType = NetworkContainerType.BLOB_CONTAINER;
			const storageAccountResourceId = sourceLocation?.azureBlob?.storageAccountResourceId!;
			savedInfo.blobs = [
				{
					blobContainer: {
						id: getBlobContainerId(getFullResourceGroupFromId(storageAccountResourceId!), getResourceName(storageAccountResourceId!), sourceLocation?.azureBlob.blobContainerName),
						name: sourceLocation?.azureBlob.blobContainerName,
						subscription: this._serviceContext.subscription!
					},
					lastBackupFile: getMigrationModeEnum(migration) === MigrationMode.OFFLINE ? migration.properties.offlineConfiguration?.lastBackupName! : undefined,
					storageAccount: getStorageAccount(storageAccountResourceId!),
					resourceGroup: getStorageAccountResourceGroup(storageAccountResourceId!),
					storageKey: ''
				}
			];
		}

		stateModel.retryMigration = true;
		stateModel.savedInfo = savedInfo;
		stateModel.serverName = serverName;
		return stateModel;
	}

	public async openDialog(dialogName?: string) {
		const locations = await getLocations(
			this._serviceContext.azureAccount!,
			this._serviceContext.subscription!);

		const targetInstance = await getMigrationTargetInstance(
			this._serviceContext.azureAccount!,
			this._serviceContext.subscription!,
			this._migration);

		let location: azureResource.AzureLocation;
		locations.forEach(azureLocation => {
			if (azureLocation.name === targetInstance.location) {
				location = azureLocation;
			}
		});

		let activeConnection = await azdata.connection.getCurrentConnection();
		let connectionId: string = '';
		let serverName: string = '';
		if (!activeConnection) {
			const connection = await azdata.connection.openConnectionDialog();
			if (connection) {
				connectionId = connection.connectionId;
				serverName = connection.options.server;
			}
		} else {
			connectionId = activeConnection.connectionId;
			serverName = activeConnection.serverName;
		}

		const api = (await vscode.extensions.getExtension(mssql.extension.name)?.activate()) as mssql.IExtension;
		const stateModel = await this.createMigrationStateModel(this._serviceContext, this._migration, connectionId, serverName, api, location!);

		if (await stateModel.loadSavedInfo()) {
			const wizardController = new WizardController(
				this._context,
				stateModel,
				this._onClosedCallback);
			await wizardController.openWizard(stateModel.sourceConnectionId);
		} else {
			void vscode.window.showInformationMessage(constants.MIGRATION_CANNOT_RETRY);
		}
	}
}