From ac162ea5c4e8f555fb116a4f9ed20d86f6f5c6cc Mon Sep 17 00:00:00 2001
From: wfjsw <wfjsw@users.noreply.github.com>
Date: Tue, 16 Jan 2024 23:16:28 -0600
Subject: [PATCH] fix e2e

---
 .gitignore                                    |  3 +-
 .../src/observers/general/GeneralObserver.ts  | 12 ++--
 .../web/src/observers/general/NodeHandler.ts  | 56 +++++++++++++------
 3 files changed, 47 insertions(+), 24 deletions(-)

diff --git a/.gitignore b/.gitignore
index 685af258cf..91667b8511 100644
--- a/.gitignore
+++ b/.gitignore
@@ -6,5 +6,6 @@ node_modules
 .gradle
 .idea
 **/.turbo
+**/'.turbo'
 .pnpm-store
-.VERSION
\ No newline at end of file
+.VERSION
diff --git a/packages/web/src/observers/general/GeneralObserver.ts b/packages/web/src/observers/general/GeneralObserver.ts
index 4b756611db..b5ebf0d108 100644
--- a/packages/web/src/observers/general/GeneralObserver.ts
+++ b/packages/web/src/observers/general/GeneralObserver.ts
@@ -172,9 +172,11 @@ export function GeneralObserver() {
 
           case 'childList':
             handleKeyAttribute(mutation.target, true);
-            nodeHandler
-              .handleChildList(mutation.target)
-              .forEach((t) => result.add(t));
+            if (mutation.addedNodes.length > 0) {
+              nodeHandler
+                .handleChildList(Array.from(mutation.addedNodes))
+                .forEach((t) => result.add(t));
+            }
             break;
 
           case 'attributes':
@@ -182,7 +184,7 @@ export function GeneralObserver() {
               handleKeyAttribute(mutation.target, false);
             }
             nodeHandler
-              .handleAttributes(mutation.target)
+              .handleAttributes(mutation.target, false)
               .forEach((t) => result.add(t));
             break;
         }
@@ -196,7 +198,7 @@ export function GeneralObserver() {
 
     // initially go through all elements
     handleKeyAttribute(targetElement, true);
-    handleNodes(nodeHandler.handleChildList(targetElement));
+    handleNodes(nodeHandler.handleChildList([targetElement]));
 
     // then observe for changes
     observer.observe(targetElement, {
diff --git a/packages/web/src/observers/general/NodeHandler.ts b/packages/web/src/observers/general/NodeHandler.ts
index 430dde18ff..f249f8135a 100644
--- a/packages/web/src/observers/general/NodeHandler.ts
+++ b/packages/web/src/observers/general/NodeHandler.ts
@@ -5,7 +5,7 @@ export function NodeHandler(
   wrapper: WrapperMiddleware
 ) {
   const self = Object.freeze({
-    handleAttributes(node: Node) {
+    handleAttributes(node: Node, includeChild = true) {
       const result: Attr[] = [];
 
       const tagAttributes = Object.fromEntries(
@@ -15,18 +15,8 @@ export function NodeHandler(
         ])
       ) as Record<string, string[]>;
 
-      const walker = document.createTreeWalker(
-        node,
-        NodeFilter.SHOW_ELEMENT,
-        (f) =>
-          tagAttributes[(f as Element).tagName.toUpperCase()]?.some((t) =>
-            (f as Element).hasAttribute(t)
-          ) || tagAttributes['*']?.some((t) => (f as Element).hasAttribute(t))
-            ? NodeFilter.FILTER_ACCEPT
-            : NodeFilter.FILTER_SKIP
-      );
-      while (walker.nextNode()) {
-        const element = walker.currentNode as Element;
+      if (node.nodeType === Node.ELEMENT_NODE) {
+        const element = node as Element;
         let attributes = tagAttributes[element.tagName.toUpperCase()] ?? [];
         if ('*' in tagAttributes) {
           attributes = attributes.concat(tagAttributes['*']);
@@ -41,13 +31,41 @@ export function NodeHandler(
         );
       }
 
+      if (includeChild) {
+        const walker = document.createTreeWalker(
+          node,
+          NodeFilter.SHOW_ELEMENT,
+          (f) =>
+            tagAttributes[(f as Element).tagName.toUpperCase()]?.some((t) =>
+              (f as Element).hasAttribute(t)
+            ) || tagAttributes['*']?.some((t) => (f as Element).hasAttribute(t))
+              ? NodeFilter.FILTER_ACCEPT
+              : NodeFilter.FILTER_SKIP
+        );
+        while (walker.nextNode()) {
+          const element = walker.currentNode as Element;
+          let attributes = tagAttributes[element.tagName.toUpperCase()] ?? [];
+          if ('*' in tagAttributes) {
+            attributes = attributes.concat(tagAttributes['*']);
+          }
+          result.push(
+            ...(attributes
+              .filter((attrName) => element.hasAttribute(attrName))
+              .map((attrName) => element.getAttributeNode(attrName))
+              .filter((attrNode) =>
+                wrapper.testAttribute(attrNode as Attr)
+              ) as Attr[])
+          );
+        }
+      }
+
       return result;
     },
 
-    handleChildList(node: Node) {
-      let result: (Attr | Text)[] = [];
-      result = result.concat(self.handleAttributes(node));
-      result = result.concat(self.handleText(node));
+    handleChildList(node: Node[]) {
+      const result: (Attr | Text)[] = [];
+      result.push(...node.flatMap((n) => self.handleAttributes(n, true)));
+      result.push(...node.flatMap((n) => self.handleText(n)));
       // wrappedHandler(node);
       return result;
     },
@@ -57,6 +75,8 @@ export function NodeHandler(
         return wrapper.testTextNode(node as Text) ? [node as Text] : [];
       }
 
+      const nodes = [];
+
       const walker = document.createTreeWalker(
         node,
         NodeFilter.SHOW_TEXT,
@@ -65,10 +85,10 @@ export function NodeHandler(
             ? NodeFilter.FILTER_ACCEPT
             : NodeFilter.FILTER_SKIP
       );
-      const nodes = [];
       while (walker.nextNode()) {
         nodes.push(walker.currentNode);
       }
+
       return nodes as Text[];
     },
   });