azuredatastudio/src/vs/workbench/api/browser/mainThreadStatusBar.ts

/*---------------------------------------------------------------------------------------------
 *  Copyright (c) Microsoft Corporation. All rights reserved.
 *  Licensed under the Source EULA. See License.txt in the project root for license information.
 *--------------------------------------------------------------------------------------------*/

import { IStatusbarService, StatusbarAlignment as MainThreadStatusBarAlignment } from 'vs/platform/statusbar/common/statusbar';
import { IDisposable } from 'vs/base/common/lifecycle';
import { MainThreadStatusBarShape, MainContext, IExtHostContext } from '../common/extHost.protocol';
import { ThemeColor } from 'vs/platform/theme/common/themeService';
import { extHostNamedCustomer } from 'vs/workbench/api/common/extHostCustomers';
import { ExtensionIdentifier } from 'vs/platform/extensions/common/extensions';

@extHostNamedCustomer(MainContext.MainThreadStatusBar)
export class MainThreadStatusBar implements MainThreadStatusBarShape {

	private readonly _entries: { [id: number]: IDisposable };

	constructor(
		extHostContext: IExtHostContext,
		@IStatusbarService private readonly _statusbarService: IStatusbarService
	) {
		this._entries = Object.create(null);
	}

	dispose(): void {
		for (const key in this._entries) {
			this._entries[key].dispose();
		}
	}

	$setEntry(id: number, extensionId: ExtensionIdentifier, text: string, tooltip: string, command: string, color: string | ThemeColor, alignment: MainThreadStatusBarAlignment, priority: number): void {

		// Dispose any old
		this.$dispose(id);

		// Add new
		const entry = this._statusbarService.addEntry({ text, tooltip, command, color, extensionId }, alignment, priority);
		this._entries[id] = entry;
	}

	$dispose(id: number) {
		const disposeable = this._entries[id];
		if (disposeable) {
			disposeable.dispose();
		}

		delete this._entries[id];
	}
}