You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
48 lines
1.2 KiB
48 lines
1.2 KiB
const Node = require('./node'); |
|
|
|
class Container extends Node { |
|
walk(callback) { |
|
return this.each((child, i) => { |
|
let result = callback(child, i); |
|
if (result !== false && child.walk) { |
|
result = child.walk(callback); |
|
} |
|
return result; |
|
}); |
|
} |
|
|
|
walkType(type, callback) { |
|
if (!type || !callback) { |
|
throw new Error('Parameters {type} and {callback} are required.'); |
|
} |
|
|
|
// allow users to pass a constructor, or node type string; eg. Word. |
|
const isTypeCallable = typeof type === 'function'; |
|
|
|
return this.walk((node, index) => { |
|
if ((isTypeCallable && node instanceof type) || (!isTypeCallable && node.type === type)) { |
|
return callback.call(this, node, index); |
|
} |
|
}); |
|
} |
|
} |
|
|
|
Container.registerWalker = (constructor) => { |
|
let walkerName = `walk${constructor.name}`; |
|
|
|
// plural sugar |
|
if (walkerName.lastIndexOf('s') !== walkerName.length - 1) { |
|
walkerName += 's'; |
|
} |
|
|
|
if (Container.prototype[walkerName]) { |
|
return; |
|
} |
|
|
|
// we need access to `this` so we can't use an arrow function |
|
Container.prototype[walkerName] = function(callback) { |
|
return this.walkType(constructor, callback); |
|
}; |
|
}; |
|
|
|
module.exports = Container;
|
|
|