Expose custom dialog extension APIs (#1206)

src/sql/workbench/api/node/mainThreadModelViewDialog.ts

@@ -0,0 +1,131 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the Source EULA. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+'use strict';
+
+import { MainThreadModelViewDialogShape, SqlMainContext, ExtHostModelViewDialogShape, SqlExtHostContext } from 'sql/workbench/api/node/sqlExtHost.protocol';
+import { extHostNamedCustomer } from 'vs/workbench/api/electron-browser/extHostCustomers';
+import { Dialog, DialogTab, DialogButton } from 'sql/platform/dialog/dialogTypes';
+import { IExtHostContext } from 'vs/workbench/api/node/extHost.protocol';
+import { CustomDialogService } from 'sql/platform/dialog/customDialogService';
+import { IInstantiationService } from 'vs/platform/instantiation/common/instantiation';
+import { IModelViewDialogDetails, IModelViewTabDetails, IModelViewButtonDetails } from 'sql/workbench/api/common/sqlExtHostTypes';
+
+@extHostNamedCustomer(SqlMainContext.MainThreadModelViewDialog)
+export class MainThreadModelViewDialog implements MainThreadModelViewDialogShape {
+	private readonly _proxy: ExtHostModelViewDialogShape;
+	private readonly _dialogs = new Map<number, Dialog>();
+	private readonly _tabs = new Map<number, DialogTab>();
+	private readonly _buttons = new Map<number, DialogButton>();
+	private _dialogService: CustomDialogService;
+
+	constructor(
+		context: IExtHostContext,
+		@IInstantiationService instatiationService: IInstantiationService
+	) {
+		this._proxy = context.getProxy(SqlExtHostContext.ExtHostModelViewDialog);
+		this._dialogService = new CustomDialogService(instatiationService);
+	}
+
+	public dispose(): void {
+		throw new Error('Method not implemented.');
+	}
+
+	public $open(handle: number): Thenable<void> {
+		let dialog = this.getDialog(handle);
+		this._dialogService.showDialog(dialog);
+		return Promise.resolve();
+	}
+
+	public $close(handle: number): Thenable<void> {
+		let dialog = this.getDialog(handle);
+		dialog.close();
+		return Promise.resolve();
+	}
+
+	public $setDialogDetails(handle: number, details: IModelViewDialogDetails): Thenable<void> {
+		let dialog = this._dialogs.get(handle);
+		if (!dialog) {
+			dialog = new Dialog(details.title);
+			let okButton = this.getButton(details.okButton);
+			let cancelButton = this.getButton(details.cancelButton);
+			dialog.okButton = okButton;
+			dialog.cancelButton = cancelButton;
+			this._dialogs.set(handle, dialog);
+		}
+
+		dialog.title = details.title;
+		if (details.content && typeof details.content !== 'string') {
+			dialog.content = details.content.map(tabHandle => this.getTab(tabHandle));
+		} else {
+			dialog.content = details.content as string;
+		}
+
+		if (details.customButtons) {
+			dialog.customButtons = details.customButtons.map(buttonHandle => this.getButton(buttonHandle));
+		}
+
+		dialog.updateContent();
+		return Promise.resolve();
+	}
+
+	public $setTabDetails(handle: number, details: IModelViewTabDetails): Thenable<void> {
+		let tab = this._tabs.get(handle);
+		if (!tab) {
+			tab = new DialogTab(details.title);
+			this._tabs.set(handle, tab);
+		}
+
+		tab.title = details.title;
+		tab.content = details.content;
+
+		tab.updateContent();
+		return Promise.resolve();
+	}
+
+	public $setButtonDetails(handle: number, details: IModelViewButtonDetails): Thenable<void> {
+		let button = this._buttons.get(handle);
+		if (!button) {
+			button = new DialogButton(details.label, details.enabled);
+			button.onClick(() => this.onButtonClick(handle));
+			this._buttons.set(handle, button);
+		} else {
+			button.label = details.label;
+			button.enabled = details.enabled;
+		}
+
+		return Promise.resolve();
+	}
+
+	private getDialog(handle: number): Dialog {
+		let dialog = this._dialogs.get(handle);
+		if (!dialog) {
+			throw new Error('No dialog matching the given handle');
+		}
+
+		return dialog;
+	}
+
+	private getTab(handle: number): DialogTab {
+		let tab = this._tabs.get(handle);
+		if (!tab) {
+			throw new Error('No tab matching the given handle');
+		}
+
+		return tab;
+	}
+
+	private getButton(handle: number): DialogButton {
+		let button = this._buttons.get(handle);
+		if (!button) {
+			throw new Error('No button matching the given handle');
+		}
+
+		return button;
+	}
+
+	private onButtonClick(handle: number): void {
+		this._proxy.$onButtonClick(handle);
+	}
+}