simplify ast transformer code

This commit is contained in:
Sebastian McKenzie 2014-11-07 20:46:54 +11:00
parent 795e38e4f4
commit 159f5217e3

View File

@ -28,9 +28,13 @@ Transformer.prototype.transform = function (file) {
var transformer = this.transformer;
var ast = file.ast;
if (transformer.ast && transformer.ast.enter) {
transformer.ast.enter(ast, file);
}
var astRun = function (key) {
if (transformer.ast && transformer.ast[key]) {
transformer.ast[key](ast, file);
}
};
astRun("enter");
var build = function (exit) {
return function (node, parent, opts) {
@ -59,9 +63,7 @@ Transformer.prototype.transform = function (file) {
exit: build(true)
});
if (transformer.ast && transformer.ast.exit) {
transformer.ast.exit(ast, file);
}
astRun("exit");
};
Transformer.prototype.canRun = function (file) {