SQL Operations Studio Public Preview 1 (0.23) release source code

src/vs/workbench/api/node/extHostStatusBar.ts

@@ -0,0 +1,190 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { StatusbarAlignment as MainThreadStatusBarAlignment } from 'vs/platform/statusbar/common/statusbar';
+import { StatusBarAlignment as ExtHostStatusBarAlignment, Disposable, ThemeColor } from './extHostTypes';
+import { StatusBarItem, StatusBarAlignment } from 'vscode';
+import { MainContext, MainThreadStatusBarShape, IMainContext } from './extHost.protocol';
+
+export class ExtHostStatusBarEntry implements StatusBarItem {
+	private static ID_GEN = 0;
+
+	private _id: number;
+	private _alignment: number;
+	private _priority: number;
+	private _disposed: boolean;
+	private _visible: boolean;
+
+	private _text: string;
+	private _tooltip: string;
+	private _color: string | ThemeColor;
+	private _command: string;
+
+	private _timeoutHandle: number;
+	private _proxy: MainThreadStatusBarShape;
+
+	private _extensionId: string;
+
+	constructor(proxy: MainThreadStatusBarShape, extensionId: string, alignment: ExtHostStatusBarAlignment = ExtHostStatusBarAlignment.Left, priority?: number) {
+		this._id = ExtHostStatusBarEntry.ID_GEN++;
+		this._proxy = proxy;
+		this._alignment = alignment;
+		this._priority = priority;
+		this._extensionId = extensionId;
+	}
+
+	public get id(): number {
+		return this._id;
+	}
+
+	public get alignment(): StatusBarAlignment {
+		return this._alignment;
+	}
+
+	public get priority(): number {
+		return this._priority;
+	}
+
+	public get text(): string {
+		return this._text;
+	}
+
+	public get tooltip(): string {
+		return this._tooltip;
+	}
+
+	public get color(): string | ThemeColor {
+		return this._color;
+	}
+
+	public get command(): string {
+		return this._command;
+	}
+
+	public set text(text: string) {
+		this._text = text;
+		this.update();
+	}
+
+	public set tooltip(tooltip: string) {
+		this._tooltip = tooltip;
+		this.update();
+	}
+
+	public set color(color: string | ThemeColor) {
+		this._color = color;
+		this.update();
+	}
+
+	public set command(command: string) {
+		this._command = command;
+		this.update();
+	}
+
+	public show(): void {
+		this._visible = true;
+		this.update();
+	}
+
+	public hide(): void {
+		clearTimeout(this._timeoutHandle);
+		this._visible = false;
+		this._proxy.$dispose(this.id);
+	}
+
+	private update(): void {
+		if (this._disposed || !this._visible) {
+			return;
+		}
+
+		clearTimeout(this._timeoutHandle);
+
+		// Defer the update so that multiple changes to setters dont cause a redraw each
+		this._timeoutHandle = setTimeout(() => {
+			this._timeoutHandle = undefined;
+
+			// Set to status bar
+			this._proxy.$setEntry(this.id, this._extensionId, this.text, this.tooltip, this.command, this.color,
+				this._alignment === ExtHostStatusBarAlignment.Left ? MainThreadStatusBarAlignment.LEFT : MainThreadStatusBarAlignment.RIGHT,
+				this._priority);
+		}, 0);
+	}
+
+	public dispose(): void {
+		this.hide();
+		this._disposed = true;
+	}
+}
+
+class StatusBarMessage {
+
+	private _item: StatusBarItem;
+	private _messages: { message: string }[] = [];
+
+	constructor(statusBar: ExtHostStatusBar) {
+		this._item = statusBar.createStatusBarEntry(void 0, ExtHostStatusBarAlignment.Left, Number.MIN_VALUE);
+	}
+
+	dispose() {
+		this._messages.length = 0;
+		this._item.dispose();
+	}
+
+	setMessage(message: string): Disposable {
+		const data: { message: string } = { message }; // use object to not confuse equal strings
+		this._messages.unshift(data);
+		this._update();
+
+		return new Disposable(() => {
+			let idx = this._messages.indexOf(data);
+			if (idx >= 0) {
+				this._messages.splice(idx, 1);
+				this._update();
+			}
+		});
+	}
+
+	private _update() {
+		if (this._messages.length > 0) {
+			this._item.text = this._messages[0].message;
+			this._item.show();
+		} else {
+			this._item.hide();
+		}
+	}
+}
+
+export class ExtHostStatusBar {
+
+	private _proxy: MainThreadStatusBarShape;
+	private _statusMessage: StatusBarMessage;
+
+	constructor(mainContext: IMainContext) {
+		this._proxy = mainContext.get(MainContext.MainThreadStatusBar);
+		this._statusMessage = new StatusBarMessage(this);
+	}
+
+	createStatusBarEntry(extensionId: string, alignment?: ExtHostStatusBarAlignment, priority?: number): StatusBarItem {
+		return new ExtHostStatusBarEntry(this._proxy, extensionId, alignment, priority);
+	}
+
+	setStatusBarMessage(text: string, timeoutOrThenable?: number | Thenable<any>): Disposable {
+
+		let d = this._statusMessage.setMessage(text);
+		let handle: number;
+
+		if (typeof timeoutOrThenable === 'number') {
+			handle = setTimeout(() => d.dispose(), timeoutOrThenable);
+		} else if (typeof timeoutOrThenable !== 'undefined') {
+			timeoutOrThenable.then(() => d.dispose(), () => d.dispose());
+		}
+
+		return new Disposable(() => {
+			d.dispose();
+			clearTimeout(handle);
+		});
+	}
+}