46 lines
1.3 KiB
JavaScript
46 lines
1.3 KiB
JavaScript
/**
|
|
* @author Toru Nagashima <https://github.com/mysticatea>
|
|
* See LICENSE file in root directory for full license.
|
|
*/
|
|
"use strict"
|
|
|
|
/**
|
|
* Merge two visitors.
|
|
* This function modifies `visitor1` directly to merge.
|
|
* @param {Visitor} visitor1 The visitor which is assigned.
|
|
* @param {Visitor} visitor2 The visitor which is assigning.
|
|
* @returns {Visitor} `visitor1`.
|
|
*/
|
|
module.exports = function mergeVisitorsInPlace(visitor1, visitor2) {
|
|
for (const key of Object.keys(visitor2)) {
|
|
const handler1 = visitor1[key]
|
|
const handler2 = visitor2[key]
|
|
|
|
if (typeof handler1 === "function") {
|
|
if (handler1._handlers) {
|
|
handler1._handlers.push(handler2)
|
|
} else {
|
|
const handlers = [handler1, handler2]
|
|
visitor1[key] = Object.assign(dispatch.bind(null, handlers), {
|
|
_handlers: handlers,
|
|
})
|
|
}
|
|
} else {
|
|
visitor1[key] = handler2
|
|
}
|
|
}
|
|
|
|
return visitor1
|
|
}
|
|
|
|
/**
|
|
* Dispatch all given functions with a node.
|
|
* @param {function[]} handlers The function list to call.
|
|
* @param {Node} node The AST node to be handled.
|
|
* @returns {void}
|
|
*/
|
|
function dispatch(handlers, node) {
|
|
for (const h of handlers) {
|
|
h(node)
|
|
}
|
|
}
|