type: Fix rewrite type.

jaywcjlove · Oct 4, 2021 · a6d0611 · a6d0611
1 parent 8ab4158
commit a6d0611
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -1,12 +1,12 @@
 import { Plugin } from 'unified';
-import { Root, Element, ElementContent } from 'hast';
+import { Root, Element, RootContent } from 'hast';
 import { visit } from 'unist-util-visit';
 import { selectAll } from 'hast-util-select';
 import { Test } from 'unist-util-is';
 
 export type RehypeRewriteOptions = {
   selector?: string;
-  rewrite(node: ElementContent, index: number | null, parent: Root | Element | null): void;
+  rewrite(node: Root | RootContent, index: number | null, parent: Root | Element | null): void;
 }
 
 const remarkRewrite: Plugin<[RehypeRewriteOptions?], Root> = (options) => {
@@ -23,7 +23,7 @@ const remarkRewrite: Plugin<[RehypeRewriteOptions?], Root> = (options) => {
       return;
     }
 
-    visit(tree, (node: ElementContent, index, parent) => {
+    visit(tree, (node: Root | RootContent, index, parent) => {
       rewrite(node, index, parent);
     });
   }