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

src/vs/workbench/services/viewlet/browser/viewlet.ts

@@ -0,0 +1,51 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { TPromise } from 'vs/base/common/winjs.base';
+import { IViewlet } from 'vs/workbench/common/viewlet';
+import { createDecorator, ServiceIdentifier } from 'vs/platform/instantiation/common/instantiation';
+import Event from 'vs/base/common/event';
+import { ViewletDescriptor } from 'vs/workbench/browser/viewlet';
+import { IProgressService } from 'vs/platform/progress/common/progress';
+
+export const IViewletService = createDecorator<IViewletService>('viewletService');
+
+export interface IViewletService {
+	_serviceBrand: ServiceIdentifier<any>;
+
+	onDidViewletOpen: Event<IViewlet>;
+	onDidViewletClose: Event<IViewlet>;
+
+	/**
+	 * Opens a viewlet with the given identifier and pass keyboard focus to it if specified.
+	 */
+	openViewlet(id: string, focus?: boolean): TPromise<IViewlet>;
+
+	/**
+	 * Returns the current active viewlet or null if none.
+	 */
+	getActiveViewlet(): IViewlet;
+
+	/**
+	 * Returns the id of the default viewlet.
+	 */
+	getDefaultViewletId(): string;
+
+	/**
+	 * Returns the viewlet by id.
+	 */
+	getViewlet(id: string): ViewletDescriptor;
+
+	/**
+	 * Returns all registered viewlets
+	 */
+	getViewlets(): ViewletDescriptor[];
+
+	/**
+	 *
+	 */
+	getProgressIndicator(id: string): IProgressService;
+}

src/vs/workbench/services/viewlet/browser/viewletService.ts

@@ -0,0 +1,107 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { TPromise, ValueCallback } from 'vs/base/common/winjs.base';
+import { IViewlet } from 'vs/workbench/common/viewlet';
+import { IViewletService } from 'vs/workbench/services/viewlet/browser/viewlet';
+import Event from 'vs/base/common/event';
+import { SidebarPart } from 'vs/workbench/browser/parts/sidebar/sidebarPart';
+import { Registry } from 'vs/platform/registry/common/platform';
+import { ViewletDescriptor, ViewletRegistry, Extensions as ViewletExtensions } from 'vs/workbench/browser/viewlet';
+import { IExtensionService } from 'vs/platform/extensions/common/extensions';
+import { IProgressService } from 'vs/platform/progress/common/progress';
+
+export class ViewletService implements IViewletService {
+
+	public _serviceBrand: any;
+
+	private sidebarPart: SidebarPart;
+	private viewletRegistry: ViewletRegistry;
+
+	private extensionViewlets: ViewletDescriptor[];
+	private extensionViewletsLoaded: TPromise<void>;
+	private extensionViewletsLoadedPromiseComplete: ValueCallback;
+
+	public get onDidViewletOpen(): Event<IViewlet> { return this.sidebarPart.onDidViewletOpen; };
+	public get onDidViewletClose(): Event<IViewlet> { return this.sidebarPart.onDidViewletClose; };
+
+	constructor(
+		sidebarPart: SidebarPart,
+		@IExtensionService private extensionService: IExtensionService
+	) {
+		this.sidebarPart = sidebarPart;
+		this.viewletRegistry = Registry.as<ViewletRegistry>(ViewletExtensions.Viewlets);
+
+		this.loadExtensionViewlets();
+	}
+
+	private loadExtensionViewlets(): void {
+		this.extensionViewlets = [];
+
+		this.extensionViewletsLoaded = new TPromise<void>(c => {
+			this.extensionViewletsLoadedPromiseComplete = c;
+		});
+
+		this.extensionService.onReady().then(() => {
+			const viewlets = this.viewletRegistry.getViewlets();
+			viewlets.forEach(v => {
+				if (!!v.extensionId) {
+					this.extensionViewlets.push(v);
+				}
+			});
+
+			this.extensionViewletsLoadedPromiseComplete(void 0);
+		});
+	}
+
+	public openViewlet(id: string, focus?: boolean): TPromise<IViewlet> {
+
+		// Built in viewlets do not need to wait for extensions to be loaded
+		const builtInViewletIds = this.getBuiltInViewlets().map(v => v.id);
+		const isBuiltInViewlet = builtInViewletIds.indexOf(id) !== -1;
+		if (isBuiltInViewlet) {
+			return this.sidebarPart.openViewlet(id, focus);
+		}
+
+		// Extension viewlets need to be loaded first which can take time
+		return this.extensionViewletsLoaded.then(() => {
+			if (this.viewletRegistry.getViewlet(id)) {
+				return this.sidebarPart.openViewlet(id, focus);
+			}
+
+			// Fallback to default viewlet if extension viewlet is still not found (e.g. uninstalled)
+			return this.sidebarPart.openViewlet(this.getDefaultViewletId(), focus);
+		});
+	}
+
+	public getActiveViewlet(): IViewlet {
+		return this.sidebarPart.getActiveViewlet();
+	}
+
+	public getViewlets(): ViewletDescriptor[] {
+		const builtInViewlets = this.getBuiltInViewlets();
+
+		return builtInViewlets.concat(this.extensionViewlets);
+	}
+
+	private getBuiltInViewlets(): ViewletDescriptor[] {
+		return this.viewletRegistry.getViewlets()
+			.filter(viewlet => !viewlet.extensionId)
+			.sort((v1, v2) => v1.order - v2.order);
+	}
+
+	public getDefaultViewletId(): string {
+		return this.viewletRegistry.getDefaultViewletId();
+	}
+
+	public getViewlet(id: string): ViewletDescriptor {
+		return this.getViewlets().filter(viewlet => viewlet.id === id)[0];
+	}
+
+	public getProgressIndicator(id: string): IProgressService {
+		return this.sidebarPart.getProgressIndicator(id);
+	}
+}