From a8f5b766f845ede335b24166b9e6965170f6153c Mon Sep 17 00:00:00 2001
From: Kuitos <kuitos.lau@gmail.com>
Date: Fri, 12 May 2023 19:47:41 +0800
Subject: [PATCH] =?UTF-8?q?=E2=9C=A8port=20document.querySelector('head')?=
 =?UTF-8?q?=20to=20qiankun=20head=20in=20micro=20app=20(#2499)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../dynamicAppend/forStrictSandbox.ts         | 54 +++++++++++++------
 1 file changed, 38 insertions(+), 16 deletions(-)

diff --git a/src/sandbox/patchers/dynamicAppend/forStrictSandbox.ts b/src/sandbox/patchers/dynamicAppend/forStrictSandbox.ts
index 5c0cb58..922cf1c 100644
--- a/src/sandbox/patchers/dynamicAppend/forStrictSandbox.ts
+++ b/src/sandbox/patchers/dynamicAppend/forStrictSandbox.ts
@@ -4,7 +4,7 @@
  */
 
 import type { Freer, SandBox } from '../../../interfaces';
-import { isBoundedFunction, nativeDocument, nativeGlobal, isCallable } from '../../../utils';
+import { isBoundedFunction, isCallable, nativeDocument, nativeGlobal } from '../../../utils';
 import { getCurrentRunningApp } from '../../common';
 import type { ContainerConfig } from './common';
 import {
@@ -67,24 +67,46 @@ function patchDocument(cfg: { sandbox: SandBox; speedy: boolean }) {
         return true;
       },
       get: (target, p, receiver) => {
-        if (p === 'createElement') {
-          // Must store the original createElement function to avoid error in nested sandbox
-          const targetCreateElement = target.createElement;
-          return function createElement(...args: Parameters<typeof document.createElement>) {
-            if (!nativeGlobal.__currentLockingSandbox__) {
-              nativeGlobal.__currentLockingSandbox__ = sandbox.name;
-            }
+        switch (p) {
+          case 'createElement': {
+            // Must store the original createElement function to avoid error in nested sandbox
+            const targetCreateElement = target.createElement;
+            return function createElement(...args: Parameters<typeof document.createElement>) {
+              if (!nativeGlobal.__currentLockingSandbox__) {
+                nativeGlobal.__currentLockingSandbox__ = sandbox.name;
+              }
 
-            const element = targetCreateElement.call(target, ...args);
+              const element = targetCreateElement.call(target, ...args);
 
-            // only record the element which is created by the current sandbox, thus we can avoid the element created by nested sandboxes
-            if (nativeGlobal.__currentLockingSandbox__ === sandbox.name) {
-              attachElementToProxy(element, sandbox.proxy);
-              delete nativeGlobal.__currentLockingSandbox__;
-            }
+              // only record the element which is created by the current sandbox, thus we can avoid the element created by nested sandboxes
+              if (nativeGlobal.__currentLockingSandbox__ === sandbox.name) {
+                attachElementToProxy(element, sandbox.proxy);
+                delete nativeGlobal.__currentLockingSandbox__;
+              }
 
-            return element;
-          };
+              return element;
+            };
+          }
+
+          case 'querySelector': {
+            const targetQuerySelector = target.querySelector;
+            return function querySelector(...args: Parameters<typeof document.querySelector>) {
+              const selector = args[0];
+              switch (selector) {
+                case 'head': {
+                  const containerConfig = proxyAttachContainerConfigMap.get(sandbox.proxy);
+                  if (containerConfig) {
+                    return getAppWrapperHeadElement(containerConfig.appWrapperGetter());
+                  }
+                  break;
+                }
+              }
+
+              return targetQuerySelector.call(target, ...args);
+            };
+          }
+          default:
+            break;
         }
 
         const value = (<any>target)[p];