Move webview editor and input to own part

src/vs/workbench/parts/webview/electron-browser/webviewEditor.ts

@@ -0,0 +1,214 @@
+/*---------------------------------------------------------------------------------------------
+ *  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 { TPromise } from 'vs/base/common/winjs.base';
+import { IDisposable, } from 'vs/base/common/lifecycle';
+import { EditorOptions } from 'vs/workbench/common/editor';
+import { Position } from 'vs/platform/editor/common/editor';
+import { WebviewEditor as BaseWebviewEditor, KEYBINDING_CONTEXT_WEBVIEWEDITOR_FOCUS, KEYBINDING_CONTEXT_WEBVIEWEDITOR_FIND_WIDGET_INPUT_FOCUSED, KEYBINDING_CONTEXT_WEBVIEW_FIND_WIDGET_VISIBLE } from 'vs/workbench/parts/html/browser/webviewEditor';
+import { Builder, Dimension } from 'vs/base/browser/builder';
+import { Webview } from 'vs/workbench/parts/html/browser/webview';
+import { IContextKeyService } from 'vs/platform/contextkey/common/contextkey';
+import { ITelemetryService } from 'vs/platform/telemetry/common/telemetry';
+import { IThemeService } from 'vs/platform/theme/common/themeService';
+import { IPartService, Parts } from 'vs/workbench/services/part/common/partService';
+import { IStorageService } from 'vs/platform/storage/common/storage';
+import { IContextViewService } from 'vs/platform/contextview/browser/contextView';
+import { IEnvironmentService } from 'vs/platform/environment/common/environment';
+import { IWorkspaceContextService } from 'vs/platform/workspace/common/workspace';
+import DOM = require('vs/base/browser/dom');
+import Event, { Emitter } from 'vs/base/common/event';
+import { WebviewInput } from 'vs/workbench/parts/webview/electron-browser/webviewInput';
+
+export class WebviewEditor extends BaseWebviewEditor {
+
+	public static readonly ID = 'WebviewEditor';
+
+	private editorFrame: HTMLElement;
+	private webviewContent: HTMLElement;
+	private _onDidFocusWebview: Emitter<void>;
+	private _webviewFocusTracker?: DOM.IFocusTracker;
+	private _webviewFocusListenerDisposable?: IDisposable;
+
+	constructor(
+		@ITelemetryService telemetryService: ITelemetryService,
+		@IStorageService storageService: IStorageService,
+		@IThemeService themeService: IThemeService,
+		@IContextKeyService private _contextKeyService: IContextKeyService,
+		@IPartService private readonly _partService: IPartService,
+		@IContextViewService private readonly _contextViewService: IContextViewService,
+		@IEnvironmentService private readonly _environmentService: IEnvironmentService,
+		@IWorkspaceContextService private readonly _contextService: IWorkspaceContextService
+	) {
+		super(WebviewEditor.ID, telemetryService, themeService, storageService, _contextKeyService);
+
+		this._onDidFocusWebview = new Emitter<void>();
+	}
+
+	protected createEditor(parent: Builder): void {
+		this.editorFrame = parent.getHTMLElement();
+		this.content = document.createElement('div');
+		parent.append(this.content);
+	}
+
+	private doUpdateContainer() {
+		const webviewContainer = this.input && (this.input as WebviewInput).container;
+		if (webviewContainer) {
+			const frameRect = this.editorFrame.getBoundingClientRect();
+			const containerRect = webviewContainer.parentElement.getBoundingClientRect();
+
+			webviewContainer.style.position = 'absolute';
+			webviewContainer.style.top = `${frameRect.top - containerRect.top}px`;
+			webviewContainer.style.left = `${frameRect.left - containerRect.left}px`;
+			webviewContainer.style.width = `${frameRect.width}px`;
+			webviewContainer.style.height = `${frameRect.height}px`;
+		}
+	}
+
+	public layout(dimension: Dimension): void {
+		if (this._webview) {
+			this.doUpdateContainer();
+		}
+		super.layout(dimension);
+	}
+
+	public dispose(): void {
+		// Let the editor input dispose of the webview.
+		this._webview = undefined;
+		this.webviewContent = undefined;
+
+		this._onDidFocusWebview.dispose();
+
+		if (this._webviewFocusTracker) {
+			this._webviewFocusTracker.dispose();
+		}
+
+		if (this._webviewFocusListenerDisposable) {
+			this._webviewFocusListenerDisposable.dispose();
+		}
+
+		super.dispose();
+	}
+
+	public sendMessage(data: any): void {
+		if (this._webview) {
+			this._webview.sendMessage(data);
+		}
+	}
+
+	public get onDidFocus(): Event<any> {
+		return this._onDidFocusWebview.event;
+	}
+
+	protected setEditorVisible(visible: boolean, position?: Position): void {
+		if (this.input && this.input instanceof WebviewInput) {
+			if (visible) {
+				this.input.claimWebview(this);
+			} else {
+				this.input.releaseWebview(this);
+			}
+
+			this.updateWebview(this.input as WebviewInput);
+		}
+
+		if (this.webviewContent) {
+			if (visible) {
+				this.webviewContent.style.visibility = 'visible';
+				this.doUpdateContainer();
+			} else {
+				this.webviewContent.style.visibility = 'hidden';
+			}
+		}
+
+		super.setEditorVisible(visible, position);
+	}
+
+	public clearInput() {
+		if (this.input && this.input instanceof WebviewInput) {
+			this.input.releaseWebview(this);
+		}
+
+		this._webview = undefined;
+		this.webviewContent = undefined;
+
+		super.clearInput();
+	}
+
+	async setInput(input: WebviewInput, options: EditorOptions): TPromise<void> {
+		if (this.input && this.input.matches(input)) {
+			return undefined;
+		}
+
+		if (this.input && this.input.getResource().fsPath !== input.getResource().fsPath) {
+			(this.input as WebviewInput).releaseWebview(this);
+			this._webview = undefined;
+			this.webviewContent = undefined;
+		}
+
+		await super.setInput(input, options);
+
+		input.onDidChangePosition(this.position);
+		this.updateWebview(input);
+	}
+
+	private updateWebview(input: WebviewInput) {
+		const webview = this.getWebview(input);
+		input.claimWebview(this);
+		webview.options = {
+			allowScripts: input.options.enableScripts,
+			enableWrappedPostMessage: true,
+			useSameOriginForRoot: false,
+			localResourceRoots: (input && input.options.localResourceRoots) || this._contextService.getWorkspace().folders.map(x => x.uri)
+		};
+		input.setHtml(input.html);
+		this.webviewContent.style.visibility = 'visible';
+		this.doUpdateContainer();
+	}
+
+	private getWebview(input: WebviewInput): Webview {
+		if (this._webview) {
+			return this._webview;
+		}
+
+		this.webviewContent = input.container;
+		const existing = input.webview;
+		if (existing) {
+			this._webview = existing;
+			return existing;
+		}
+
+		this._webviewFocusTracker = DOM.trackFocus(this.webviewContent);
+		this._webviewFocusListenerDisposable = this._webviewFocusTracker.onDidFocus(() => {
+			this._onDidFocusWebview.fire();
+		});
+
+		this._contextKeyService = this._contextKeyService.createScoped(this.webviewContent);
+		this.contextKey = KEYBINDING_CONTEXT_WEBVIEWEDITOR_FOCUS.bindTo(this._contextKeyService);
+		this.findInputFocusContextKey = KEYBINDING_CONTEXT_WEBVIEWEDITOR_FIND_WIDGET_INPUT_FOCUSED.bindTo(this._contextKeyService);
+		this.findWidgetVisible = KEYBINDING_CONTEXT_WEBVIEW_FIND_WIDGET_VISIBLE.bindTo(this._contextKeyService);
+
+		this._webview = new Webview(
+			this.webviewContent,
+			this._partService.getContainer(Parts.EDITOR_PART),
+			this.themeService,
+			this._environmentService,
+			this._contextViewService,
+			this.contextKey,
+			this.findInputFocusContextKey,
+			{
+				enableWrappedPostMessage: true,
+				useSameOriginForRoot: false
+			});
+		input.webview = this._webview;
+
+		this.content.setAttribute('aria-flowto', this.webviewContent.id);
+
+		this.doUpdateContainer();
+		return this._webview;
+	}
+}
+