refactor: use helper to check node type

juggle · Jul 31, 2021 · 365a6bd · 365a6bd
1 parent 1efb026
commit 365a6bd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/utils/register.ts b/src/utils/register.ts
@@ -1,5 +1,6 @@
 import { scheduler } from "./scheduler";
 import { documentEvents } from './events';
+import { isDocument } from "./element";
 
 /**
  * Global count of all observed elements.
@@ -42,7 +43,7 @@ const mo = new MutationObserver(handleEvent);
  */
 const getDocument = (node: Node): Document | ShadowRoot | null => {
   const root = node.getRootNode ? node.getRootNode({ composed: node.nodeType === 11 }) : node.ownerDocument;
-  if (root && root !== node && !(/9|11/).test(root.nodeType.toString())) {
+  if (root && root !== node && !isDocument(root)) {
     return getDocument(root);
   }
   return root as Document | ShadowRoot | null;