type: more type export.

jaywcjlove · May 30, 2021 · d030781 · d030781
1 parent 896d9ab
commit d030781
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git 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
@@ -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)