azuredatastudio/extensions/sql-migration/src/dashboard/migrationsTab.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 loc from '../constants/strings';
import { AdsMigrationStatus, MigrationDetailsEvent, ServiceContextChangeEvent, TabBase } from './tabBase';
import { MigrationsListTab, MigrationsListTabId } from './migrationsListTab';
import { DatabaseMigration, getMigrationDetails } from '../api/azure';
import { MigrationLocalStorage } from '../models/migrationLocalStorage';
import { FileStorageType } from '../models/stateMachine';
import { MigrationDetailsTabBase } from './migrationDetailsTabBase';
import { MigrationDetailsFileShareTab } from './migrationDetailsFileShareTab';
import { MigrationDetailsBlobContainerTab } from './migrationDetailsBlobContainerTab';
import { MigrationDetailsTableTab } from './migrationDetailsTableTab';
import { DashboardStatusBar } from './DashboardStatusBar';

export const MigrationsTabId = 'MigrationsTab';

export class MigrationsTab extends TabBase<MigrationsTab> {
	private _tab!: azdata.DivContainer;
	private _migrationsListTab!: MigrationsListTab;
	private _migrationDetailsTab!: MigrationDetailsTabBase<any>;
	private _migrationDetailsFileShareTab!: MigrationDetailsTabBase<any>;
	private _migrationDetailsBlobTab!: MigrationDetailsTabBase<any>;
	private _migrationDetailsTableTab!: MigrationDetailsTabBase<any>;
	private _selectedTabId: string | undefined = undefined;
	private _migrationDetailsEvent!: vscode.EventEmitter<MigrationDetailsEvent>;

	constructor() {
		super();
		this.title = loc.DESKTOP_MIGRATIONS_TAB_TITLE;
		this.id = MigrationsTabId;
	}

	public async create(
		context: vscode.ExtensionContext,
		view: azdata.ModelView,
		serviceContextChangedEvent: vscode.EventEmitter<ServiceContextChangeEvent>,
		migrationDetailsEvent: vscode.EventEmitter<MigrationDetailsEvent>,
		statusBar: DashboardStatusBar): Promise<MigrationsTab> {

		this.context = context;
		this.view = view;
		this.serviceContextChangedEvent = serviceContextChangedEvent;
		this._migrationDetailsEvent = migrationDetailsEvent;
		this.statusBar = statusBar;

		await this.initialize(view);
		await this._openTab(this._migrationsListTab);

		return this;
	}

	public async refresh(): Promise<void> {
		switch (this._selectedTabId) {
			case undefined:
			case MigrationsListTabId:
				return this._migrationsListTab.refresh();
			default:
				return this._migrationDetailsTab.refresh();
		}
	}

	protected async initialize(view: azdata.ModelView): Promise<void> {
		this._tab = this.view.modelBuilder.divContainer()
			.withLayout({ height: '100%' })
			.withProps({
				CSSStyles: {
					'margin': '0px',
					'padding': '0px',
					'width': '100%'
				}
			})
			.component();

		this._migrationsListTab = await new MigrationsListTab().create(
			this.context,
			this.view,
			async (migration) => await this.openMigrationDetails(migration),
			this.serviceContextChangedEvent,
			this.statusBar);
		this.disposables.push(this._migrationsListTab);

		const openMigrationsListTab = async (): Promise<void> => {
			await this.statusBar.clearError();
			await this._openTab(this._migrationsListTab);
		};

		this._migrationDetailsBlobTab = await new MigrationDetailsBlobContainerTab().create(
			this.context,
			this.view,
			openMigrationsListTab,
			this.statusBar);
		this.disposables.push(this._migrationDetailsBlobTab);

		this._migrationDetailsFileShareTab = await new MigrationDetailsFileShareTab().create(
			this.context,
			this.view,
			openMigrationsListTab,
			this.statusBar);
		this.disposables.push(this._migrationDetailsFileShareTab);

		this._migrationDetailsTableTab = await new MigrationDetailsTableTab().create(
			this.context,
			this.view,
			openMigrationsListTab,
			this.statusBar);
		this.disposables.push(this._migrationDetailsFileShareTab);

		const connectionProfile = await azdata.connection.getCurrentConnection();
		const connectionId = connectionProfile.connectionId;
		this.disposables.push(
			this._migrationDetailsEvent.event(async e => {
				if (e.connectionId === connectionId) {
					const migration = await this._getMigrationDetails(e.migrationId, e.migrationOperationId);
					if (migration) {
						await this.openMigrationDetails(migration);
					}
				}
			}));

		this.content = this._tab;
	}

	public async setMigrationFilter(filter: AdsMigrationStatus): Promise<void> {
		await this._openTab(this._migrationsListTab);
		await this._migrationsListTab.setMigrationFilter(filter);
	}

	public async openMigrationDetails(migration: DatabaseMigration): Promise<void> {
		switch (migration.properties.backupConfiguration?.sourceLocation?.fileStorageType) {
			case FileStorageType.AzureBlob:
				this._migrationDetailsTab = this._migrationDetailsBlobTab;
				break;
			case FileStorageType.FileShare:
				this._migrationDetailsTab = this._migrationDetailsFileShareTab;
				break;
			case FileStorageType.None:
				this._migrationDetailsTab = this._migrationDetailsTableTab;
				break;
		}

		await this._migrationDetailsTab.setMigrationContext(
			await MigrationLocalStorage.getMigrationServiceContext(),
			migration);

		const promise = this._migrationDetailsTab.refresh();
		await this._openTab(this._migrationDetailsTab);
		await promise;
	}

	private async _getMigrationDetails(migrationId: string, migrationOperationId: string): Promise<DatabaseMigration | undefined> {
		const context = await MigrationLocalStorage.getMigrationServiceContext();
		if (context.azureAccount && context.subscription) {
			return getMigrationDetails(
				context.azureAccount,
				context.subscription,
				migrationId,
				migrationOperationId);
		}
		return undefined;
	}

	private async _openTab(tab: azdata.Tab): Promise<void> {
		if (tab.id === this._selectedTabId) {
			return;
		}

		await this.statusBar.clearError();
		this._tab.clearItems();
		this._tab.addItem(tab.content);
		this._selectedTabId = tab.id;
	}
}