1 files changed, 67 insertions, 0 deletions
diff --git a/remote/test/puppeteer/packages/puppeteer-core/src/injected/util.ts b/remote/test/puppeteer/packages/puppeteer-core/src/injected/util.ts
new file mode 100644
index 0000000000..34fe8f7748
--- /dev/null
+++ b/remote/test/puppeteer/packages/puppeteer-core/src/injected/util.ts
@@ -0,0 +1,67 @@
+const HIDDEN_VISIBILITY_VALUES = ['hidden', 'collapse'];
+
+/**
+ * @internal
+ */
+export const checkVisibility = (
+  node: Node | null,
+  visible?: boolean
+): Node | boolean => {
+  if (!node) {
+    return visible === false;
+  }
+  if (visible === undefined) {
+    return node;
+  }
+  const element = (
+    node.nodeType === Node.TEXT_NODE ? node.parentElement : node
+  ) as Element;
+
+  const style = window.getComputedStyle(element);
+  const isVisible =
+    style &&
+    !HIDDEN_VISIBILITY_VALUES.includes(style.visibility) &&
+    !isBoundingBoxEmpty(element);
+  return visible === isVisible ? node : false;
+};
+
+function isBoundingBoxEmpty(element: Element): boolean {
+  const rect = element.getBoundingClientRect();
+  return rect.width === 0 || rect.height === 0;
+}
+
+const hasShadowRoot = (node: Node): node is Node & {shadowRoot: ShadowRoot} => {
+  return 'shadowRoot' in node && node.shadowRoot instanceof ShadowRoot;
+};
+
+/**
+ * @internal
+ */
+export function* pierce(root: Node): IterableIterator<Node | ShadowRoot> {
+  if (hasShadowRoot(root)) {
+    yield root.shadowRoot;
+  } else {
+    yield root;
+  }
+}
+
+/**
+ * @internal
+ */
+export function* pierceAll(root: Node): IterableIterator<Node | ShadowRoot> {
+  root = pierce(root).next().value;
+  yield root;
+  const walkers = [document.createTreeWalker(root, NodeFilter.SHOW_ELEMENT)];
+  for (const walker of walkers) {
+    let node: Element | null;
+    while ((node = walker.nextNode() as Element | null)) {
+      if (!node.shadowRoot) {
+        continue;
+      }
+      yield node.shadowRoot;
+      walkers.push(
+        document.createTreeWalker(node.shadowRoot, NodeFilter.SHOW_ELEMENT)
+      );
+    }
+  }
+}