From 89aa8bf13730d6d941b2e16c6d035fdba5f60efe Mon Sep 17 00:00:00 2001
From: Matt Bierner <matb@microsoft.com>
Date: Fri, 28 Jun 2019 16:04:45 -0700
Subject: [PATCH] Moving some electron specific webview js back into electron

---
 .../contrib/webview/browser/pre/main.js       | 62 ++++---------------
 .../electron-browser/pre/electron-index.js    | 46 +++++++++++++-
 2 files changed, 57 insertions(+), 51 deletions(-)

diff --git a/src/vs/workbench/contrib/webview/browser/pre/main.js b/src/vs/workbench/contrib/webview/browser/pre/main.js
index 5017faee033..fc5964ba3f2 100644
--- a/src/vs/workbench/contrib/webview/browser/pre/main.js
+++ b/src/vs/workbench/contrib/webview/browser/pre/main.js
@@ -3,6 +3,17 @@
  *  Licensed under the MIT License. See License.txt in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 // @ts-check
+
+/**
+ * @typedef {{
+ *   postMessage: (channel: string, data?: any) => void,
+ *   onMessage: (channel: string, handler: any) => void,
+ *   focusIframeOnCreate?: boolean,
+ *   ready?: Promise<void>,
+ *   onIframeLoaded: (iframe: HTMLIFrameElement) => void
+ * }} WebviewHost
+ */
+
 (function () {
 	'use strict';
 
@@ -134,23 +145,13 @@
 	}
 
 	/**
-	 * @typedef {{
-	 *   postMessage: (channel: string, data?: any) => void,
-	 *   onMessage: (channel: string, handler: any) => void,
-	 *   focusIframeOnCreate?: boolean,
-	 *   ready?: Promise<void>
-	 * }} HostCommunications
-	 */
-
-	/**
-	 * @param {HostCommunications} host
+	 * @param {WebviewHost} host
 	 */
 	function createWebviewManager(host) {
 		// state
 		let firstLoad = true;
 		let loadTimeout;
 		let pendingMessages = [];
-		let isInDevelopmentMode = false;
 
 		const initData = {
 			initialScrollProgress: undefined
@@ -442,44 +443,10 @@
 						}
 					});
 
-					if (!FAKE_LOAD) {
-						newFrame.contentWindow.onbeforeunload = () => {
-							if (isInDevelopmentMode) { // Allow reloads while developing a webview
-								host.postMessage('do-reload');
-								return false;
-							}
-
-							// Block navigation when not in development mode
-							console.log('prevented webview navigation');
-							return false;
-						};
-					}
-
 					// Bubble out link clicks
 					newFrame.contentWindow.addEventListener('click', handleInnerClick);
 
-					// Electron 4 eats mouseup events from inside webviews
-					// https://github.com/microsoft/vscode/issues/75090
-					// Try to fix this by rebroadcasting mouse moves and mouseups so that we can
-					// emulate these on the main window
-					if (!FAKE_LOAD) {
-						let isMouseDown = false;
-
-						newFrame.contentWindow.addEventListener('mousedown', () => {
-							isMouseDown = true;
-						});
-
-						const tryDispatchSyntheticMouseEvent = (e) => {
-							if (!isMouseDown) {
-								host.postMessage('synthetic-mouse-event', { type: e.type, screenX: e.screenX, screenY: e.screenY, clientX: e.clientX, clientY: e.clientY });
-							}
-						};
-						newFrame.contentWindow.addEventListener('mouseup', e => {
-							tryDispatchSyntheticMouseEvent(e);
-							isMouseDown = false;
-						});
-						newFrame.contentWindow.addEventListener('mousemove', tryDispatchSyntheticMouseEvent);
-					}
+					host.onIframeLoaded(newFrame);
 				}
 
 				if (!FAKE_LOAD) {
@@ -511,9 +478,6 @@
 				initData.initialScrollProgress = progress;
 			});
 
-			host.onMessage('devtools-opened', () => {
-				isInDevelopmentMode = true;
-			});
 
 			trackFocus({
 				onFocus: () => host.postMessage('did-focus'),
diff --git a/src/vs/workbench/contrib/webview/electron-browser/pre/electron-index.js b/src/vs/workbench/contrib/webview/electron-browser/pre/electron-index.js
index a5f4d3b1809..6992535ea97 100644
--- a/src/vs/workbench/contrib/webview/electron-browser/pre/electron-index.js
+++ b/src/vs/workbench/contrib/webview/electron-browser/pre/electron-index.js
@@ -28,14 +28,54 @@
 	// @ts-ignore
 	const ipcRenderer = require('electron').ipcRenderer;
 
-	require('../../browser/pre/main')({
+	let isInDevelopmentMode = false;
+
+	/**
+	 * @type {import('../../browser/pre/main').WebviewHost}
+	 */
+	const host = {
 		postMessage: (channel, data) => {
 			ipcRenderer.sendToHost(channel, data);
 		},
 		onMessage: (channel, handler) => {
 			ipcRenderer.on(channel, handler);
 		},
-		focusIframeOnCreate: true
+		focusIframeOnCreate: true,
+		onIframeLoaded: (newFrame) => {
+			newFrame.contentWindow.onbeforeunload = () => {
+				if (isInDevelopmentMode) { // Allow reloads while developing a webview
+					host.postMessage('do-reload');
+					return false;
+				}
+				// Block navigation when not in development mode
+				console.log('prevented webview navigation');
+				return false;
+			};
+
+			// Electron 4 eats mouseup events from inside webviews
+			// https://github.com/microsoft/vscode/issues/75090
+			// Try to fix this by rebroadcasting mouse moves and mouseups so that we can
+			// emulate these on the main window
+			let isMouseDown = false;
+			newFrame.contentWindow.addEventListener('mousedown', () => {
+				isMouseDown = true;
+			});
+
+			const tryDispatchSyntheticMouseEvent = (e) => {
+				if (!isMouseDown) {
+					host.postMessage('synthetic-mouse-event', { type: e.type, screenX: e.screenX, screenY: e.screenY, clientX: e.clientX, clientY: e.clientY });
+				}
+			};
+			newFrame.contentWindow.addEventListener('mouseup', e => {
+				tryDispatchSyntheticMouseEvent(e);
+				isMouseDown = false;
+			});
+			newFrame.contentWindow.addEventListener('mousemove', tryDispatchSyntheticMouseEvent);
+		}
+	};
+
+	host.onMessage('devtools-opened', () => {
+		isInDevelopmentMode = true;
 	});
 
 	document.addEventListener('DOMContentLoaded', () => {
@@ -46,4 +86,6 @@
 			ipcRenderer.sendToHost(message.data.command, message.data.data);
 		};
 	});
+
+	require('../../browser/pre/main')(host);
 }());
\ No newline at end of file