feat: make document of updateDOM configurable (#7)

vueuse · Jan 19, 2021 · b48809a · b48809a
1 parent b6efb49
commit b48809a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -45,7 +45,7 @@ export type Head = {
 
   removeHeadTags: (tags: HeadTag[]) => void
 
-  updateDOM: () => void
+  updateDOM: (document?: Document) => void
 }
 
 export interface HTMLResult {
@@ -126,7 +126,7 @@ const setAttrs = (el: Element, attrs: HeadAttrs) => {
   el.setAttribute(HEAD_ATTRS_KEY, keys.join(','))
 }
 
-const insertTags = (tags: HeadTag[]) => {
+const insertTags = (tags: HeadTag[], document = window.document) => {
   const head = document.head
   let headCountEl = head.querySelector(`meta[name="${HEAD_COUNT_KEY}"]`)
   const headCount = headCountEl
@@ -234,8 +234,8 @@ export const createHead = () => {
       })
     },
 
-    updateDOM() {
-      insertTags(headTags)
+    updateDOM(document) {
+      insertTags(headTags, document)
     },
   }
   return head