azuredatastudio/extensions/agent/src/dialogs/pickScheduleDialog.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 nls from 'vscode-nls';
import * as azdata from 'azdata';
import * as vscode from 'vscode';
import { PickScheduleData } from '../data/pickScheduleData';

const localize = nls.loadMessageBundle();

export class PickScheduleDialog {

	// TODO: localize
	// Top level
	private readonly DialogTitle: string = localize('pickSchedule.jobSchedules', "Job Schedules");
	private readonly OkButtonText: string = localize('pickSchedule.ok', "OK");
	private readonly CancelButtonText: string = localize('pickSchedule.cancel', "Cancel");
	private readonly SchedulesLabelText: string = localize('pickSchedule.availableSchedules', "Available Schedules:");
	public static readonly ScheduleNameLabelText: string = localize('pickSchedule.scheduleName', "Name");
	public static readonly SchedulesIDText: string = localize('pickSchedule.scheduleID', "ID");
	public static readonly ScheduleDescription: string = localize('pickSchedule.description', "Description");


	// UI Components
	private dialog: azdata.window.Dialog;
	private schedulesTable: azdata.TableComponent;
	private loadingComponent: azdata.LoadingComponent;

	private model: PickScheduleData;

	private _onSuccess: vscode.EventEmitter<PickScheduleData> = new vscode.EventEmitter<PickScheduleData>();
	public readonly onSuccess: vscode.Event<PickScheduleData> = this._onSuccess.event;

	constructor(ownerUri: string, jobName: string) {
		this.model = new PickScheduleData(ownerUri, jobName);
	}

	public async showDialog() {
		this.model.initialize().then((result) => {
			if (this.loadingComponent) {
				this.loadingComponent.loading = false;
			}
			if (this.model.schedules) {
				let data: any[][] = [];
				for (let i = 0; i < this.model.schedules.length; ++i) {
					let schedule = this.model.schedules[i];
					data[i] = [schedule.id, schedule.name, schedule.description];
				}
				this.schedulesTable.data = data;
			}
		});
		this.dialog = azdata.window.createModelViewDialog(this.DialogTitle);
		this.initializeContent();
		this.dialog.okButton.onClick(async () => await this.execute());
		this.dialog.cancelButton.onClick(async () => await this.cancel());
		this.dialog.okButton.label = this.OkButtonText;
		this.dialog.cancelButton.label = this.CancelButtonText;
		azdata.window.openDialog(this.dialog);
	}

	private initializeContent() {
		this.dialog.registerContent(async view => {
			this.schedulesTable = view.modelBuilder.table()
				.withProperties({
					columns: [
						PickScheduleDialog.SchedulesIDText,
						PickScheduleDialog.ScheduleNameLabelText,
						PickScheduleDialog.ScheduleDescription
					],
					data: [],
					height: 750,
					width: 430
				}).component();

			let formModel = view.modelBuilder.formContainer()
				.withFormItems([{
					component: this.schedulesTable,
					title: this.SchedulesLabelText
				}]).withLayout({ width: '100%' }).component();

			this.loadingComponent = view.modelBuilder.loadingComponent().withItem(formModel).component();
			this.loadingComponent.loading = true;
			this.model.initialize().then((result) => {
				this.loadingComponent.loading = false;
				if (this.model.schedules) {
					let data: any[][] = [];
					for (let i = 0; i < this.model.schedules.length; ++i) {
						let schedule = this.model.schedules[i];
						data[i] = [schedule.id, schedule.name, schedule.description];
					}
					this.schedulesTable.data = data;
				}
			});
			this.loadingComponent.loading = !this.model.isInitialized();
			await view.initializeModel(this.loadingComponent);
		});

	}

	private async execute() {
		this.updateModel();
		await this.model.save();
		this._onSuccess.fire(this.model);
	}

	private async cancel() {
	}

	private updateModel() {
		let selectedRows = this.schedulesTable.selectedRows;
		if (selectedRows && selectedRows.length > 0) {
			let selectedRow = selectedRows[0];
			this.model.selectedSchedule = this.model.schedules[selectedRow];
		}
	}
}