@blueking/bk-weweb 0.0.21 → 0.0.22

package/dist/collect-source.js

@@ -191,6 +191,7 @@ const isJsonpUrl = (url) => {
  */
 const { document: document$1 } = window;
 const { createElement, getElementById, getElementsByClassName, getElementsByName, getElementsByTagName, querySelector, querySelectorAll, } = Document.prototype;
+const { querySelector: bodyQuerySelector } = HTMLBodyElement.prototype;
 const SPECIAL_ELEMENT_TAG = ['body', 'html', 'head'];
 let hasRewrite$1 = false;
 function rewriteDocumentPrototypeMethods() {
@@ -220,17 +221,23 @@ function rewriteDocumentPrototypeMethods() {
      */
     function querySelectorNew(selectors) {
         const app = getCurrentRunningApp();
+        if (selectors.includes('data-bk-mask-uid'))
+            debugger;
         // 如果选择器是特殊元素标签
         if (SPECIAL_ELEMENT_TAG.includes(selectors)) {
             // 如果当前应用程序容器是 ShadowRoot 类型
             if (app?.container instanceof ShadowRoot) {
                 return app?.container;
             }
+            if (this instanceof HTMLBodyElement)
+                return bodyQuerySelector.call(this, selectors);
             // 否则调用原始的 querySelector 方法
             return querySelector.call(this, selectors);
         }
         // 如果没有当前应用程序或选择器为空或文档不是当前文档
-        if (!app || !selectors || document$1 !== this) {
+        if (!app || !selectors || ![document$1, document$1.body].includes(this)) {
+            if (this instanceof HTMLBodyElement)
+                return bodyQuerySelector.call(this, selectors);
             // 调用原始的 querySelector 方法
             return querySelector.call(this, selectors);
         }
@@ -260,6 +267,7 @@ function rewriteDocumentPrototypeMethods() {
     }
     Document.prototype.querySelector = querySelectorNew;
     Document.prototype.querySelectorAll = querySelectorAllNew;
+    HTMLBodyElement.prototype.querySelector = querySelectorNew;
     Document.prototype.getElementById = function getElementByIdNew(key) {
         return getCurrentRunningApp() ? querySelectorNew.call(this, `#${key}`) : getElementById.call(this, key);
     };
@@ -281,6 +289,7 @@ function rewriteDocumentPrototypeMethods() {
 function resetDocumentPrototypeMethods() {
     Document.prototype.createElement = createElement;
     Document.prototype.querySelector = querySelector;
+    HTMLBodyElement.prototype.querySelector = bodyQuerySelector;
     Document.prototype.querySelectorAll = querySelectorAll;
     Document.prototype.getElementById = getElementById;
     Document.prototype.getElementsByClassName = getElementsByClassName;