From 364d89eb34ebf284684eb021a9673657d5ca599a Mon Sep 17 00:00:00 2001
From: Matt Bierner <matb@microsoft.com>
Date: Fri, 13 Sep 2019 22:50:43 -0700
Subject: [PATCH] Re-layout webview editor when input changes

---
 src/vs/workbench/contrib/webview/browser/webviewEditor.ts | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/vs/workbench/contrib/webview/browser/webviewEditor.ts b/src/vs/workbench/contrib/webview/browser/webviewEditor.ts
index dd9cc51550c7..3223d496d8bb 100644
--- a/src/vs/workbench/contrib/webview/browser/webviewEditor.ts
+++ b/src/vs/workbench/contrib/webview/browser/webviewEditor.ts
@@ -27,6 +27,7 @@ export class WebviewEditor extends BaseEditor {
 	private _findWidgetVisible: IContextKey<boolean>;
 	private _editorFrame?: HTMLElement;
 	private _content?: HTMLElement;
+	private _dimension?: DOM.Dimension;
 
 	private readonly _webviewFocusTrackerDisposables = this._register(new DisposableStore());
 	private readonly _onFocusWindowHandler = this._register(new MutableDisposable());
@@ -89,6 +90,7 @@ export class WebviewEditor extends BaseEditor {
 	}
 
 	public layout(dimension: DOM.Dimension): void {
+		this._dimension = dimension;
 		if (this.input && this.input instanceof WebviewInput) {
 			this.synchronizeWebviewContainerDimensions(this.input.webview, dimension);
 			this.input.webview.layout();
@@ -157,6 +159,9 @@ export class WebviewEditor extends BaseEditor {
 			}
 
 			this.claimWebview(input);
+			if (this._dimension) {
+				this.layout(this._dimension);
+			}
 		}
 	}