Allow generateCode function to be provided for a node

This commit is contained in:
Patrick Steele-Idem 2015-12-15 14:56:40 -07:00
parent 43211edb09
commit a8f6737755

View File

@ -116,8 +116,17 @@ class Builder {
return new Literal({value});
}
node(type) {
return new Node(type);
node(type, generateCode) {
if (typeof type === 'function') {
generateCode = arguments[0];
type = 'Node';
}
var node = new Node(type);
if (generateCode) {
node.setCodeGenerator(generateCode);
}
return node;
}
program(body) {