Pass a root node instead of doc

iansan5653 · web-flow · commit 53763514dd79 · 2023-09-19T14:58:59.000Z
diff --git a/src/paste-markdown-html.ts b/src/paste-markdown-html.ts
@@ -136,8 +136,8 @@ function isChildOfSupportedMarkdownNode({parentNode}: Node): boolean {
 }
 
 /** NodeIterator is not iterable, so this wrapper makes it usable in `for...of` loops. */
-function* iterateNodes(document: Document, whatToShow?: number, filter?: NodeFilter | null) {
-  const iterator = document.createNodeIterator(document.body, whatToShow, filter)
+function* iterateNodes(root: Node, whatToShow?: number, filter?: NodeFilter | null) {
+  const iterator = (root.ownerDocument ?? document).createNodeIterator(root, whatToShow, filter)
 
   let node
   while ((node = iterator.nextNode())) yield node

Original file line number	Diff line number	Diff line change
`@@ -136,8 +136,8 @@ function isChildOfSupportedMarkdownNode({parentNode}: Node): boolean {`
`136`	`136`	`}`
`137`	`137`
`138`	`138`	/** NodeIterator is not iterable, so this wrapper makes it usable in `for...of` loops. */
`139`		`-function* iterateNodes(document: Document, whatToShow?: number, filter?: NodeFilter \| null) {`
`140`		`- const iterator = document.createNodeIterator(document.body, whatToShow, filter)`
	`139`	`+function* iterateNodes(root: Node, whatToShow?: number, filter?: NodeFilter \| null) {`
	`140`	`+ const iterator = (root.ownerDocument ?? document).createNodeIterator(root, whatToShow, filter)`
`141`	`141`
`142`	`142`	`let node`
`143`	`143`	`while ((node = iterator.nextNode())) yield node`