From d030781b3db18ea4eaafe62f2ab35321441c314f Mon Sep 17 00:00:00 2001
From: jaywcjlove <398188662@qq.com>
Date: Mon, 31 May 2021 01:52:53 +0800
Subject: [PATCH] type: more type export.

---
 src/index.ts | 3 +++
 src/visit.ts | 6 +++---
 2 files changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index 75de896..c402abf 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -3,6 +3,9 @@ import {Plugin} from 'unified'
 import visit from './visit'
 import { propertiesHandle, nextChild, prevChild, getCommentObject } from './utils'
 
+export * from './visit'
+export * from './utils'
+
 export type MdastTransformer = (tree: Root) => void
 
 export type RehypeAttrsOptions = {
diff --git a/src/visit.ts b/src/visit.ts
index 141c148..33d22fc 100644
--- a/src/visit.ts
+++ b/src/visit.ts
@@ -1,8 +1,8 @@
 import { Root, Parent, Content } from 'ts-mdast'
 
-type Callback = (node: Root | Content | Parent, index: number, parent: Parent | Content) => void
+export type VisitCallback = (node: Root | Content | Parent, index: number, parent: Parent | Content) => void
 
-export default function visit(tree: Root, element: string, callback: Callback) {
+export default function visit(tree: Root, element: string, callback: VisitCallback) {
   if (!element || !tree || !callback || typeof callback !== 'function') {
     return
   }
@@ -11,7 +11,7 @@ export default function visit(tree: Root, element: string, callback: Callback) {
   }
 }
 
-function handle(tree: Content[], element: string, parent: Parent | Content, callback: Callback) {
+function handle(tree: Content[], element: string, parent: Parent | Content, callback: VisitCallback) {
   tree.forEach((item, index) => {
     if (item.type === element) {
       callback(item, index, parent)