199 lines
5.7 KiB
TypeScript
199 lines
5.7 KiB
TypeScript
export function visit<Tree extends import('unist').Node, Check extends Test>(
|
||
tree: Tree,
|
||
check: Check,
|
||
visitor: BuildVisitor<Tree, Check>,
|
||
reverse?: boolean | null | undefined
|
||
): undefined
|
||
export function visit<Tree extends import('unist').Node, Check extends Test>(
|
||
tree: Tree,
|
||
visitor: BuildVisitor<Tree, Test>,
|
||
reverse?: boolean | null | undefined
|
||
): undefined
|
||
export type UnistNode = import('unist').Node
|
||
export type UnistParent = import('unist').Parent
|
||
export type VisitorResult = import('unist-util-visit-parents').VisitorResult
|
||
/**
|
||
* Test from `unist-util-is`.
|
||
*
|
||
* Note: we have remove and add `undefined`, because otherwise when generating
|
||
* automatic `.d.ts` files, TS tries to flatten paths from a local perspective,
|
||
* which doesn’t work when publishing on npm.
|
||
*/
|
||
export type Test = Exclude<import('unist-util-is').Test, undefined> | undefined
|
||
/**
|
||
* Get the value of a type guard `Fn`.
|
||
*/
|
||
export type Predicate<Fn, Fallback> = Fn extends (
|
||
value: any
|
||
) => value is infer Thing
|
||
? Thing
|
||
: Fallback
|
||
/**
|
||
* Check whether a node matches a primitive check in the type system.
|
||
*/
|
||
export type MatchesOne<Value, Check> = Check extends null | undefined
|
||
? Value
|
||
: Value extends {
|
||
type: Check
|
||
}
|
||
? Value
|
||
: Value extends Check
|
||
? Value
|
||
: Check extends Function
|
||
? Predicate<Check, Value> extends Value
|
||
? Predicate<Check, Value>
|
||
: never
|
||
: never
|
||
/**
|
||
* Check whether a node matches a check in the type system.
|
||
*/
|
||
export type Matches<Value, Check> = Check extends Array<any>
|
||
? MatchesOne<Value, Check[keyof Check]>
|
||
: MatchesOne<Value, Check>
|
||
/**
|
||
* Number; capped reasonably.
|
||
*/
|
||
export type Uint = 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10
|
||
/**
|
||
* Increment a number in the type system.
|
||
*/
|
||
export type Increment<I extends Uint = 0> = I extends 0
|
||
? 1
|
||
: I extends 1
|
||
? 2
|
||
: I extends 2
|
||
? 3
|
||
: I extends 3
|
||
? 4
|
||
: I extends 4
|
||
? 5
|
||
: I extends 5
|
||
? 6
|
||
: I extends 6
|
||
? 7
|
||
: I extends 7
|
||
? 8
|
||
: I extends 8
|
||
? 9
|
||
: 10
|
||
/**
|
||
* Collect nodes that can be parents of `Child`.
|
||
*/
|
||
export type InternalParent<
|
||
Node extends import('unist').Node,
|
||
Child extends import('unist').Node
|
||
> = Node extends import('unist').Parent
|
||
? Node extends {
|
||
children: (infer Children)[]
|
||
}
|
||
? Child extends Children
|
||
? Node
|
||
: never
|
||
: never
|
||
: never
|
||
/**
|
||
* Collect nodes in `Tree` that can be parents of `Child`.
|
||
*/
|
||
export type Parent<
|
||
Tree extends import('unist').Node,
|
||
Child extends import('unist').Node
|
||
> = InternalParent<InclusiveDescendant<Tree>, Child>
|
||
/**
|
||
* Collect nodes in `Tree` that can be ancestors of `Child`.
|
||
*/
|
||
export type InternalAncestor<
|
||
Node extends import('unist').Node,
|
||
Child extends import('unist').Node,
|
||
Max extends Uint = 10,
|
||
Depth extends Uint = 0
|
||
> = Depth extends Max
|
||
? never
|
||
:
|
||
| InternalParent<Node, Child>
|
||
| InternalAncestor<
|
||
Node,
|
||
InternalParent<Node, Child>,
|
||
Max,
|
||
Increment<Depth>
|
||
>
|
||
/**
|
||
* Collect all (inclusive) descendants of `Tree`.
|
||
*
|
||
* > 👉 **Note**: for performance reasons, this seems to be the fastest way to
|
||
* > recurse without actually running into an infinite loop, which the
|
||
* > previous version did.
|
||
* >
|
||
* > Practically, a max of `2` is typically enough assuming a `Root` is
|
||
* > passed, but it doesn’t improve performance.
|
||
* > It gets higher with `List > ListItem > Table > TableRow > TableCell`.
|
||
* > Using up to `10` doesn’t hurt or help either.
|
||
*/
|
||
export type InclusiveDescendant<
|
||
Tree extends import('unist').Node,
|
||
Max extends Uint = 10,
|
||
Depth extends Uint = 0
|
||
> = Tree extends UnistParent
|
||
? Depth extends Max
|
||
? Tree
|
||
:
|
||
| Tree
|
||
| InclusiveDescendant<Tree['children'][number], Max, Increment<Depth>>
|
||
: Tree
|
||
/**
|
||
* Handle a node (matching `test`, if given).
|
||
*
|
||
* Visitors are free to transform `node`.
|
||
* They can also transform `parent`.
|
||
*
|
||
* Replacing `node` itself, if `SKIP` is not returned, still causes its
|
||
* descendants to be walked (which is a bug).
|
||
*
|
||
* When adding or removing previous siblings of `node` (or next siblings, in
|
||
* case of reverse), the `Visitor` should return a new `Index` to specify the
|
||
* sibling to traverse after `node` is traversed.
|
||
* Adding or removing next siblings of `node` (or previous siblings, in case
|
||
* of reverse) is handled as expected without needing to return a new `Index`.
|
||
*
|
||
* Removing the children property of `parent` still results in them being
|
||
* traversed.
|
||
*/
|
||
export type Visitor<
|
||
Visited extends import('unist').Node = import('unist').Node,
|
||
Ancestor extends import('unist').Parent = import('unist').Parent
|
||
> = (
|
||
node: Visited,
|
||
index: Visited extends UnistNode ? number | undefined : never,
|
||
parent: Ancestor extends UnistParent ? Ancestor | undefined : never
|
||
) => VisitorResult
|
||
/**
|
||
* Build a typed `Visitor` function from a node and all possible parents.
|
||
*
|
||
* It will infer which values are passed as `node` and which as `parent`.
|
||
*/
|
||
export type BuildVisitorFromMatch<
|
||
Visited extends import('unist').Node,
|
||
Ancestor extends import('unist').Parent
|
||
> = Visitor<Visited, Parent<Ancestor, Visited>>
|
||
/**
|
||
* Build a typed `Visitor` function from a list of descendants and a test.
|
||
*
|
||
* It will infer which values are passed as `node` and which as `parent`.
|
||
*/
|
||
export type BuildVisitorFromDescendants<
|
||
Descendant extends import('unist').Node,
|
||
Check extends Test
|
||
> = BuildVisitorFromMatch<
|
||
Matches<Descendant, Check>,
|
||
Extract<Descendant, UnistParent>
|
||
>
|
||
/**
|
||
* Build a typed `Visitor` function from a tree and a test.
|
||
*
|
||
* It will infer which values are passed as `node` and which as `parent`.
|
||
*/
|
||
export type BuildVisitor<
|
||
Tree extends import('unist').Node = import('unist').Node,
|
||
Check extends Test = Test
|
||
> = BuildVisitorFromDescendants<InclusiveDescendant<Tree>, Check>
|
||
export {CONTINUE, EXIT, SKIP} from 'unist-util-visit-parents'
|