chore: rebuild code.

package.json

@@ -48,11 +48,11 @@
     "node": "^12.20.0 || ^14.13.1 || >=16.0.0"
   },
   "dependencies": {
-    "unified": "10.1.0"
+    "unified": "10.1.0",
+    "unist-util-visit": "4.1.0",
+    "unist-util-visit-parents": "5.1.0"
   },
   "devDependencies": {
-    "@types/jest": "27.0.2",
-    "jest": "27.2.0",
     "rehype": "12.0.0",
     "rehype-stringify": "9.0.2",
     "rehype-raw": "6.1.0",

src/__tests__/index.test.ts

@@ -15,6 +15,7 @@ describe('rehype-rewrite test case', () => {
       .data('settings', { fragment: true })
       .use(rehypeRewrite, (node) => {
         if(node.type == 'text' && node.value == 'header') {
+          console.log('node.value:', node.value)
           node.value = ''
         }
       })
@@ -46,7 +47,7 @@ describe('rehype-rewrite test case', () => {
     const expected = `<h1>hello<span> world</span></h1>`
     const htmlStr = rehype()
       .data('settings', { fragment: true })
-      .use(rehypeRewrite, (node: any) => {
+      .use(rehypeRewrite, (node) => {
         if (node.type == 'element' && node.tagName == 'h1') {
           node.children = [ ...node.children, {
             type: 'element',
@@ -72,7 +73,7 @@ describe('rehype-rewrite test case', () => {
       .use(remarkParse)
       .use(remark2rehype, { allowDangerousHtml: true })
       .use(rehypeRaw)
-      .use(rehypeRewrite, (node: any) => {
+      .use(rehypeRewrite, (node) => {
         if (node.type == 'element' && node.tagName == 'p') {
           node.properties = { ...node.properties, style: 'color:red;' }
         }

src/index.ts

@@ -1,20 +1,18 @@
-import { Plugin, Transformer } from 'unified';
-import { Parent, NodeData, Node } from 'unist';
-import { visit } from './visit';
+import { Plugin } from 'unified';
+import { Root, Element } from 'hast';
+import { Node, Data } from 'unist';
+import { visit } from 'unist-util-visit';
 
-export * from './visit';
+export type RehypeRewriteOptions = (node: Node<Data>, index: number | null, parent: Root | Element | null) => void;
 
-export type RehypeRewriteOptions = (node: NodeData<Parent>, index: number, parent: NodeData<Parent>) => void
-
-const remarkRewrite: Plugin<[RehypeRewriteOptions?]> = (handle): Transformer => {
-  return transformer;
-  function transformer(tree: Node<NodeData<Parent>>): void {
-    visit(tree as any, (node: NodeData<Parent>, index: number, parent: NodeData<Parent>) => {
+const remarkRewrite: Plugin<[RehypeRewriteOptions?], Root> = (handle) => {
+  return (tree) => {
+    visit(tree, (node, index, parent) => {
       if (handle && typeof handle === 'function') {
-        handle(node, index, parent)
+        handle(node, index, parent);
       }
-    })
+    });
   }
 }
 
-export default remarkRewrite
+export default remarkRewrite;