Split up ext host <-> main thread communication to separate files

2026-04-25 02:58:56 +01:00 · 2016-06-27 14:58:03 +02:00
parent fa9d70718c
commit d285fe05e1
43 changed files with 2370 additions and 2196 deletions
--- a/src/vs/workbench/api/node/mainThreadExtensionService.ts
+++ b/src/vs/workbench/api/node/mainThreadExtensionService.ts
@@ -0,0 +1,143 @@
+/*---------------------------------------------------------------------------------------------
+ *  Copyright (c) Microsoft Corporation. All rights reserved.
+ *  Licensed under the MIT License. See License.txt in the project root for license information.
+ *--------------------------------------------------------------------------------------------*/
+'use strict';
+
+import Severity from 'vs/base/common/severity';
+import {TPromise} from 'vs/base/common/winjs.base';
+import {AbstractExtensionService, ActivatedExtension} from 'vs/platform/extensions/common/abstractExtensionService';
+import {IMessage, IExtensionDescription, IExtensionsStatus} from 'vs/platform/extensions/common/extensions';
+import {ExtensionsRegistry} from 'vs/platform/extensions/common/extensionsRegistry';
+import {IMessageService} from 'vs/platform/message/common/message';
+import {IThreadService} from 'vs/workbench/services/thread/common/threadService';
+import {IWorkspaceContextService} from 'vs/platform/workspace/common/workspace';
+import {ExtHostContext} from './extHostProtocol';
+import {ExtHostExtensionService} from './extHostExtensionService';
+
+/**
+ * Represents a failed extension in the ext host.
+ */
+class MainProcessFailedExtension extends ActivatedExtension {
+	constructor() {
+		super(true);
+	}
+}
+
+/**
+ * Represents an extension that was successfully loaded or an
+ * empty extension in the ext host.
+ */
+class MainProcessSuccessExtension extends ActivatedExtension {
+	constructor() {
+		super(false);
+	}
+}
+
+function messageWithSource(msg:IMessage): string {
+	return (msg.source ? '[' + msg.source + ']: ' : '') + msg.message;
+}
+
+export class MainProcessExtensionService extends AbstractExtensionService<ActivatedExtension> {
+
+	private _threadService: IThreadService;
+	private _messageService: IMessageService;
+	private _proxy: ExtHostExtensionService;
+	private _isDev: boolean;
+	private _extensionsStatus: { [id: string]: IExtensionsStatus };
+
+	/**
+	 * This class is constructed manually because it is a service, so it doesn't use any ctor injection
+	 */
+	constructor(
+		@IWorkspaceContextService contextService: IWorkspaceContextService,
+		@IThreadService threadService: IThreadService,
+		@IMessageService messageService: IMessageService
+	) {
+		super(false);
+		let config = contextService.getConfiguration();
+		this._isDev = !config.env.isBuilt || !!config.env.extensionDevelopmentPath;
+
+		this._messageService = messageService;
+		this._threadService = threadService;
+		this._proxy = this._threadService.get(ExtHostContext.ExtHostExtensionService);
+		this._extensionsStatus = {};
+
+		ExtensionsRegistry.handleExtensionPoints((msg) => this._handleMessage(msg));
+	}
+
+	private _handleMessage(msg: IMessage) {
+		this._showMessage(msg.type, messageWithSource(msg));
+
+		if (!this._extensionsStatus[msg.source]) {
+			this._extensionsStatus[msg.source] = { messages: [] };
+		}
+		this._extensionsStatus[msg.source].messages.push(msg);
+	}
+
+	public $localShowMessage(severity: Severity, msg: string): void {
+		let messageShown = false;
+		if (severity === Severity.Error || severity === Severity.Warning) {
+			if (this._isDev) {
+				// Only show nasty intrusive messages if doing extension development.
+				this._messageService.show(severity, msg);
+				messageShown = true;
+			}
+		}
+
+		if (!messageShown) {
+			switch (severity) {
+				case Severity.Error:
+					console.error(msg);
+					break;
+				case Severity.Warning:
+					console.warn(msg);
+					break;
+				default:
+					console.log(msg);
+			}
+		}
+	}
+
+	// -- overwriting AbstractExtensionService
+
+	public getExtensionsStatus(): { [id: string]: IExtensionsStatus } {
+		return this._extensionsStatus;
+	}
+
+	protected _showMessage(severity: Severity, msg: string): void {
+		this._proxy.$localShowMessage(severity, msg);
+		this.$localShowMessage(severity, msg);
+	}
+
+	protected _createFailedExtension(): ActivatedExtension {
+		return new MainProcessFailedExtension();
+	}
+
+	protected _actualActivateExtension(extensionDescription: IExtensionDescription): TPromise<ActivatedExtension> {
+
+		// redirect extension activation to the extension host
+		return this._proxy.$activateExtension(extensionDescription).then(_ => {
+
+			// the extension host calls $onExtensionActivated, where we write to `_activatedExtensions`
+			return this._activatedExtensions[extensionDescription.id];
+		});
+	}
+
+	// -- called by extension host
+
+	public $onExtensionHostReady(extensionDescriptions: IExtensionDescription[], messages: IMessage[]): TPromise<void> {
+		ExtensionsRegistry.registerExtensions(extensionDescriptions);
+		messages.forEach((entry) => this._handleMessage(entry));
+		this._triggerOnReady();
+		return;
+	}
+
+	public $onExtensionActivated(extensionId: string): void {
+		this._activatedExtensions[extensionId] = new MainProcessSuccessExtension();
+	}
+
+	public $onExtensionActivationFailed(extensionId: string): void {
+		this._activatedExtensions[extensionId] = new MainProcessFailedExtension();
+	}
+}