From f120aa8c5100d3926217b5ceee6220294f8d9939 Mon Sep 17 00:00:00 2001
From: James Lyne <jim+github@not-null.co.uk>
Date: Sun, 27 Feb 2022 21:36:46 +0000
Subject: [PATCH] Lazily create sandbox iframe

---
 src/util.ts | 19 ++++++++++++++-----
 1 file changed, 14 insertions(+), 5 deletions(-)

diff --git a/src/util.ts b/src/util.ts
index 85e638a..92767f2 100644
--- a/src/util.ts
+++ b/src/util.ts
@@ -344,9 +344,9 @@ export const getMiddle = (bounds: LiveAtlasBounds): LiveAtlasLocation => {
 /**
  * Creates an "allow-scripts" sandboxed <iframe>
  * @see {@link runSandboxed}
- * @returns {Window} The iframe's contentWindow
+ * @returns {Window} A promise that resolves to the iframe's contentWindow
  */
-const createIframeSandbox = () => {
+const createIframeSandbox = async (): Promise<Window> => {
 	const frame = document.createElement('iframe');
 	frame.hidden = true;
 	frame.sandbox.add('allow-scripts');
@@ -394,11 +394,16 @@ const createIframeSandbox = () => {
 		}
     });
 
-	document.body.appendChild(frame);
-	return frame.contentWindow;
+	return new Promise(resolve => {
+		document.body.appendChild(frame);
+
+		frame.onload = () => {
+			resolve(frame.contentWindow as Window);
+		};
+	});
 }
 
-const sandboxWindow: Window | null = createIframeSandbox();
+let sandboxWindow: Window | null = null;
 const sandboxSuccessCallbacks: Map<number, (result?: any) => void> = new Map();
 const sandboxErrorCallbacks: Map<number, (reason?: any) => void> = new Map();
 
@@ -411,6 +416,10 @@ const sandboxErrorCallbacks: Map<number, (reason?: any) => void> = new Map();
  * or will reject with any Errors that occurred during execution.
  */
 export const runSandboxed = async (code: string) => {
+	if(!sandboxWindow) {
+		sandboxWindow = await createIframeSandbox();
+	}
+
 	return new Promise((resolve, reject) => {
 		const key = Math.random();