59 lines
1.3 KiB
JavaScript
59 lines
1.3 KiB
JavaScript
|
/**
|
||
|
|
* @typedef {import('unist').Parent} Parent
|
||
|
|
*/
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @template {Parent} Kind
|
||
|
|
* Node type.
|
||
|
|
* @callback Visitor
|
||
|
|
* Callback called for each `child` in `parent` later given to `visit`.
|
||
|
|
* @param {Kind['children'][number]} child
|
||
|
|
* Child of parent.
|
||
|
|
* @param {number} index
|
||
|
|
* Position of `child` in parent.
|
||
|
|
* @param {Kind} parent
|
||
|
|
* Parent node.
|
||
|
|
* @returns {undefined}
|
||
|
|
* Nothing.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @template {Parent} Kind
|
||
|
|
* Node type.
|
||
|
|
* @callback Visit
|
||
|
|
* Function to call the bound `visitor` for each child in `parent`.
|
||
|
|
* @param {Kind} node
|
||
|
|
* Parent node.
|
||
|
|
* @returns {undefined}
|
||
|
|
* Nothing.
|
||
|
|
*/
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Wrap `visitor` to be called for each child in the nodes later given to
|
||
|
|
* `visit`.
|
||
|
|
*
|
||
|
|
* @template {Parent} Kind
|
||
|
|
* Node type.
|
||
|
|
* @param {Visitor<Kind>} visitor
|
||
|
|
* Callback called for each `child` in `parent` later given to `visit`.
|
||
|
|
* @returns {Visit<Kind>}
|
||
|
|
* Function to call the bound `visitor` for each child in `parent`.
|
||
|
|
*/
|
||
|
|
export function visitChildren(visitor) {
|
||
|
|
return visit
|
||
|
|
|
||
|
|
/** @type {Visit<Kind>} */
|
||
|
|
function visit(parent) {
|
||
|
|
const children = parent && parent.children
|
||
|
|
let index = -1
|
||
|
|
|
||
|
|
if (!children) {
|
||
|
|
throw new Error('Missing children in `parent` for `visit`')
|
||
|
|
}
|
||
|
|
|
||
|
|
while (++index in children) {
|
||
|
|
visitor(children[index], index, parent)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|