From 88bee159f60078b4ed6783a825e2867bcc0681f4 Mon Sep 17 00:00:00 2001
From: jaywcjlove <398188662@qq.com>
Date: Sat, 2 Oct 2021 16:44:33 +0800
Subject: [PATCH] type: Fix type errors.
---
src/__tests__/index.test.ts | 2 +-
src/index.ts | 9 ++++-----
2 files changed, 5 insertions(+), 6 deletions(-)
diff --git a/src/__tests__/index.test.ts b/src/__tests__/index.test.ts
index 55b6dd9..aaddddf 100644
--- a/src/__tests__/index.test.ts
+++ b/src/__tests__/index.test.ts
@@ -30,7 +30,7 @@ describe('rehype-rewrite test case', () => {
const html = `header
`;
const expected = `header
`
const htmlStr = rehype()
- .use(rehypeRewrite, (node: any) => {
+ .use(rehypeRewrite, (node) => {
if (node.type == 'element' && node.tagName == 'body') {
node.properties = { ...node.properties, style: 'color:red;'}
}
diff --git a/src/index.ts b/src/index.ts
index 514c5d6..2cb117d 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -1,15 +1,14 @@
import { Plugin } from 'unified';
-import { Root, Element } from 'hast';
-import { Node, Data } from 'unist';
+import { Root, Element, ElementContent } from 'hast';
import { visit } from 'unist-util-visit';
-export type RehypeRewriteOptions = (node: Node, index: number | null, parent: Root | Element | null) => void;
+export type RehypeRewriteOptions = (node: ElementContent, index: number | null, parent: Root | Element | null) => void;
const remarkRewrite: Plugin<[RehypeRewriteOptions?], Root> = (handle) => {
- return (tree) => {
+ return (tree: Root) => {
visit(tree, (node, index, parent) => {
if (handle && typeof handle === 'function') {
- handle(node, index, parent);
+ handle(node as ElementContent, index, parent);
}
});
}