feat: add runWhenRootExists

Author: PipecraftNet

lib/index.d.ts

@@ -140,6 +140,9 @@ export function getOffsetPosition(
 // eslint-disable-next-line @typescript-eslint/ban-types
 export async function runOnce(key: string, func: Function): Promise<any>
 
+// eslint-disable-next-line @typescript-eslint/ban-types
+export function runWhenRootExists(func: Function): void
+
 // eslint-disable-next-line @typescript-eslint/ban-types
 export function runWhenHeadExists(func: Function): void
 

lib/index.js

@@ -372,6 +372,7 @@ export const parseInt10 = (number, defaultValue) => {
   return Number.isNaN(result) ? defaultValue : result
 }
 
+const rootFuncArray = []
 const headFuncArray = []
 const bodyFuncArray = []
 let headBodyObserver
@@ -386,6 +387,14 @@ const startObserveHeadBodyExists = () => {
       headBodyObserver.disconnect()
     }
 
+    if (doc.documentElement && rootFuncArray.length > 0) {
+      for (const func of rootFuncArray) {
+        func()
+      }
+
+      rootFuncArray.length = 0
+    }
+
     if (doc.head && headFuncArray.length > 0) {
       for (const func of headFuncArray) {
         func()
@@ -409,6 +418,19 @@ const startObserveHeadBodyExists = () => {
   })
 }
 
+/**
+ * Run function when document.documentElement exsits.
+ */
+export const runWhenRootExists = (func) => {
+  if (!doc.documentElement) {
+    rootFuncArray.push(func)
+    startObserveHeadBodyExists()
+    return
+  }
+
+  func()
+}
+
 /**
  * Run function when document.head exsits.
  */