fix: update near-membrane-dom to use ShadowRealm

Author: rwaldron

packages/near-membrane-dom/src/browser-realm.ts

@@ -146,6 +146,49 @@ interface BrowserEnvironmentOptions {
 
 const createHooksCallback = createMembraneMarshall();
 
+interface RealmControls {
+    cleanup: () => {};
+    evaluator: typeof eval;
+    needsManualCleanUp: boolean;
+}
+// @ts-ignore: TypeScript doesn't know what a ShadowRealm is yet.
+const { ShadowRealm: ShadowRealmCtor } = globalThis;
+function createRealmControls(options: BrowserEnvironmentOptions): RealmControls {
+    let cleanup = () => {};
+    let evaluator: typeof eval;
+    let needsManualCleanUp = false;
+
+    if (typeof ShadowRealmCtor === 'function') {
+        // @ts-ignore: TypeScript doesn't know what a ShadowRealm is yet.
+        const shadowRealm = new ShadowRealm();
+        evaluator = shadowRealm.evaluate.bind(shadowRealm);
+    } else {
+        const { keepAlive } = options;
+        const iframe = createDetachableIframe();
+        const redWindow = HTMLIFrameElementContentWindowGetter(iframe)!.window;
+        const { document: redDocument } = redWindow;
+        cleanup = () => {
+            // Once we get the iframe info ready, and all mapped, we can proceed
+            // to detach the iframe only if the keepAlive option isn't true.
+            if (keepAlive !== true) {
+                removeIframe(iframe);
+            } else {
+                // TODO: Temporary hack to preserve the document reference in Firefox.
+                // https://bugzilla.mozilla.org/show_bug.cgi?id=543435
+                DocumentOpen(redDocument);
+                DocumentClose(redDocument);
+            }
+        };
+        evaluator = redWindow.eval;
+        needsManualCleanUp = true;
+    }
+    return {
+        cleanup,
+        evaluator,
+        needsManualCleanUp,
+    } as RealmControls;
+}
+
 export default function createVirtualEnvironment(
     globalObjectShape: object,
     globalObjectVirtualizationTarget: WindowProxy & typeof globalThis,
@@ -164,12 +207,10 @@ export default function createVirtualEnvironment(
         providedOptions
     );
     // eslint-disable-next-line prefer-object-spread
-    const { distortionCallback, endowments = {}, keepAlive, support, instrumentation } = options;
-    const iframe = createDetachableIframe();
-    const redWindow = HTMLIFrameElementContentWindowGetter(iframe)!.window;
-    const { document: redDocument } = redWindow;
+    const { distortionCallback, endowments = {}, support, instrumentation } = options;
+    const realmControls = createRealmControls(options);
     const blueConnector = createHooksCallback;
-    const redConnector = createConnector(redWindow.eval);
+    const redConnector = createConnector(realmControls.evaluator);
     // Extract the global references and descriptors before any interference.
     const blueRefs = getCachedBlueReferences(globalObjectVirtualizationTarget);
     // Create a new environment.
@@ -184,15 +225,9 @@ export default function createVirtualEnvironment(
     linkIntrinsics(env, globalObjectVirtualizationTarget);
     linkUnforgeables(env, globalObjectVirtualizationTarget);
     tameDOM(env, blueRefs, getResolvedShapeDescriptors(globalObjectShape, endowments));
-    // Once we get the iframe info ready, and all mapped, we can proceed
-    // to detach the iframe only if the keepAlive option isn't true.
-    if (keepAlive !== true) {
-        removeIframe(iframe);
-    } else {
-        // TODO: Temporary hack to preserve the document reference in Firefox.
-        // https://bugzilla.mozilla.org/show_bug.cgi?id=543435
-        DocumentOpen(redDocument);
-        DocumentClose(redDocument);
+
+    if (realmControls.needsManualCleanUp) {
+        realmControls.cleanup();
     }
     return env;
 }