UglifyJS/lib/ast.js

function DEFNODE(type, props, methods, base) {
    if (arguments.length < 4) base = AST_Node;
    if (!props) props = [];
    else props = props.split(/\s+/);
    if (base && base.PROPS)
        props = props.concat(base.PROPS);
    var code = "return function AST_" + type + "(props){ if (props) { ";
    for (var i = props.length; --i >= 0;) {
        code += "this." + props[i] + " = props." + props[i] + ";";
    }
    if (methods && methods.initialize)
        code += "this.initialize();"
    code += " } }";
    var ctor = new Function(code)();
    if (base) {
        ctor.prototype = new base;
    }
    ctor.prototype.CTOR = ctor;
    ctor.PROPS = props || null;
    if (type) {
        ctor.prototype.TYPE = ctor.TYPE = type;
    }
    if (methods) for (i in methods) if (HOP(methods, i)) {
        ctor.prototype[i] = methods[i];
    }
    return ctor;
};

var AST_Token = DEFNODE("Token", "type value line col pos endpos nlb comments_before", {

}, null);

var AST_Node = DEFNODE("Node", "start end", {
    clone: function() {
        return new this.CTOR(this);
    },
    // XXX: what was this for?
    // renew: function(args) {
    //     var ctor = this.CTOR, props = ctor.props;
    //     for (var i in props) if (!HOP(args, i)) args[i] = this[i];
    //     return new ctor(args);
    // },
    walk: function(w) {
        w._visit(this);
    }
}, null);

var AST_Directive = DEFNODE("Directive", "value", {
    print: function(output) {
        output.string(this.value);
    }
});

var AST_Debugger = DEFNODE("Debugger", null, {
    print: function(output) {
        output.print("debugger");
    }
});

var AST_Parenthesized = DEFNODE("Parenthesized", "expression", {
    $documentation: "Represents an expression which is always parenthesized.  Used for the \
conditions in IF/WHILE/DO and expression in SWITCH/WITH.",
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
        });
    }
});

var AST_Bracketed = DEFNODE("Bracketed", "body", {
    $documentation: "Represents a block of statements that are always included in brackets. \
Used for bodies of FUNCTION/TRY/CATCH/THROW/SWITCH.",
    walk: function(w) {
        w._visit(this, function(){
            this.body.forEach(function(stat){
                stat.walk(w);
            });
        });
    }
});

/* -----[ loops ]----- */

var AST_Statement = DEFNODE("Statement", "label body", {
    walk: function(w) {
        w._visit(this, function(){
            if (this.label) this.label.walk(w);
            if (this.body) {
                if (this.body instanceof AST_Node)
                    this.body.walk(w);
                else
                    this.walk_array(w);
            }
        });
    },
    walk_array: AST_Bracketed.prototype.walk
});

var AST_SimpleStatement = DEFNODE("SimpleStatement", null, {

}, AST_Statement);

var AST_BlockStatement = DEFNODE("BlockStatement", null, {

}, AST_Statement);

var AST_EmptyStatement = DEFNODE("EmptyStatement", null, {

}, AST_Statement);

var AST_Do = DEFNODE("Do", "condition", {
    walk: function(w) {
        w._visit(this, function(){
            this.condition.walk(w);
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_Statement);

var AST_While = DEFNODE("While", "condition", {
    walk: function(w) {
        w._visit(this, function(){
            this.condition.walk(w);
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_Statement);

var AST_For = DEFNODE("For", "init condition step", {
    walk: function(w) {
        w._visit(this, function(){
            if (this.init) this.init.walk(w);
            if (this.condition) this.condition.walk(w);
            if (this.step) this.step.walk(w);
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_Statement);

var AST_ForIn = DEFNODE("ForIn", "init name object", {
    walk: function(w) {
        w._visit(this, function(){
            if (this.init) this.init.walk(w);
            this.object.walk(w);
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_Statement);

var AST_With = DEFNODE("With", "expression", {
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_Statement);

/* -----[ functions ]----- */

var AST_Scope = DEFNODE("Scope", "identifiers", {
    walk: function(w) {
        w._visit(this, function(){
            if (this.identifiers) this.identifiers.forEach(function(el){
                el.walk(w);
            });
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_Statement);

var AST_Toplevel = DEFNODE("Toplevel", null, {

}, AST_Scope);

var AST_Lambda = DEFNODE("Lambda", "name argnames", {
    walk: function(w) {
        w._visit(this, function(){
            if (this.name) this.name.walk(w);
            this.argnames.forEach(function(el){
                el.walk(w);
            });
            AST_Scope.prototype.walk.call(this, w);
        });
    }
}, AST_Scope);

var AST_Function = DEFNODE("Function", null, {

}, AST_Lambda);

var AST_Defun = DEFNODE("Defun", null, {

}, AST_Function);

/* -----[ JUMPS ]----- */

var AST_Jump = DEFNODE("Jump", null, {

});

var AST_Exit = DEFNODE("Exit", "value", {
    walk: function(w) {
        w._visit(this, function(){
            if (this.value) this.value.walk(w);
        });
    }
}, AST_Jump);

var AST_Return = DEFNODE("Return", null, {

}, AST_Exit);

var AST_Throw = DEFNODE("Throw", null, {

}, AST_Exit);

var AST_LoopControl = DEFNODE("LoopControl", "label", {
    walk: function(w) {
        w._visit(this, function(){
            if (this.label) this.label.walk(w);
        });
    }
}, AST_Jump);

var AST_Break = DEFNODE("Break", null, {

}, AST_LoopControl);

var AST_Continue = DEFNODE("Continue", null, {

}, AST_LoopControl);

/* -----[ IF ]----- */

var AST_If = DEFNODE("If", "condition consequent alternative", {
    walk: function(w) {
        w._visit(this, function(){
            this.condition.walk(w);
            this.consequent.walk(w);
            if (this.alternative) this.alternative.walk(w);
        });
    }
});

/* -----[ SWITCH ]----- */

var AST_Switch = DEFNODE("Switch", "expression", {
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_Statement);

var AST_SwitchBlock = DEFNODE("SwitchBlock", null, {
    walk       : AST_Statement.prototype.walk,
    walk_array : AST_Bracketed.prototype.walk
}, AST_Bracketed);

var AST_SwitchBranch = DEFNODE("SwitchBranch", "body", {
    walk       : AST_Statement.prototype.walk,
    walk_array : AST_Bracketed.prototype.walk
});

var AST_Default = DEFNODE("Default", null, {

}, AST_SwitchBranch);

var AST_Case = DEFNODE("Case", "expression", {
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
            AST_Statement.prototype.walk.call(this, w);
        });
    }
}, AST_SwitchBranch);

/* -----[ EXCEPTIONS ]----- */

var AST_Try = DEFNODE("Try", "btry bcatch bfinally", {
    walk: function(w) {
        w._visit(this, function(){
            this.btry.walk(w);
            if (this.bcatch) this.bcatch.walk(w);
            if (this.bfinally) this.bfinally.walk(w);
        });
    }
});

var AST_Catch = DEFNODE("Catch", "argname body", {
    walk: function(w) {
        w._visit(this, function(){
            this.argname.walk(w);
            this.body.walk(w);
        });
    }
});

var AST_Finally = DEFNODE("Finally", null, {

}, AST_Bracketed);

/* -----[ VAR/CONST ]----- */

var AST_Definitions = DEFNODE("Definitions", "definitions", {
    walk: function(w) {
        w._visit(this, function(){
            this.definitions.forEach(function(el){
                el.walk(w);
            });
        });
    }
});

var AST_Var = DEFNODE("Var", null, {

}, AST_Definitions);

var AST_Const = DEFNODE("Const", null, {

}, AST_Definitions);

var AST_VarDef = DEFNODE("VarDef", "name value", {
    walk: function(w) {
        w._visit(this, function(){
            this.name.walk(w);
            if (this.value) this.value.walk(w);
        });
    }
});

/* -----[ OTHER ]----- */

var AST_Call = DEFNODE("Call", "expression args", {
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
            this.args.forEach(function(el){
                el.walk(w);
            });
        });
    }
});

var AST_New = DEFNODE("New", null, {

}, AST_Call);

var AST_Seq = DEFNODE("Seq", "first second", {
    walk: function(w) {
        w._visit(this, function(){
            this.first.walk(w);
            this.second.walk(w);
        });
    }
});

var AST_PropAccess = DEFNODE("PropAccess", "expression property", {

});

var AST_Dot = DEFNODE("Dot", null, {
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
        });
    }
}, AST_PropAccess);

var AST_Sub = DEFNODE("Sub", null, {
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
            this.property.walk(w);
        });
    }
}, AST_PropAccess);

var AST_Unary = DEFNODE("Unary", "operator expression", {
    walk: function(w) {
        w._visit(this, function(){
            this.expression.walk(w);
        });
    }
});

var AST_UnaryPrefix = DEFNODE("UnaryPrefix", null, {

}, AST_Unary);

var AST_UnaryPostfix = DEFNODE("UnaryPostfix", null, {

}, AST_Unary);

var AST_Binary = DEFNODE("Binary", "left operator right", {
    walk: function(w) {
        w._visit(this, function(){
            this.left.walk(w);
            this.right.walk(w);
        });
    }
});

var AST_Conditional = DEFNODE("Conditional", "condition consequent alternative", {
    walk: function(w) {
        w._visit(this, function(){
            this.condition.walk(w);
            this.consequent.walk(w);
            this.alternative.walk(w);
        });
    }
});

var AST_Assign = DEFNODE("Assign", null, {

}, AST_Binary);

/* -----[ LITERALS ]----- */

var AST_Array = DEFNODE("Array", "elements", {
    walk: function(w) {
        w._visit(this, function(){
            this.elements.forEach(function(el){
                el.walk(w);
            });
        });
    }
});

var AST_Object = DEFNODE("Object", "properties", {
    walk: function(w) {
        w._visit(this, function(){
            this.properties.forEach(function(prop){
                prop.walk(w);
            });
        });
    }
});

var AST_ObjectProperty = DEFNODE("ObjectProperty");

var AST_ObjectKeyVal = DEFNODE("ObjectKeyval", "key value", {
    walk: function(w) {
        w._visit(this, function(){
            this.value.walk(w);
        });
    }
}, AST_ObjectProperty);

var AST_ObjectSetter = DEFNODE("ObjectSetter", "name func", {
    walk: function(w) {
        w._visit(this, function(){
            this.func.walk(w);
        });
    }
}, AST_ObjectProperty);

var AST_ObjectGetter = DEFNODE("ObjectGetter", "name func", {
    walk: function(w) {
        w._visit(this, function(){
            this.func.walk(w);
        });
    }
}, AST_ObjectProperty);

var AST_Symbol = DEFNODE("Symbol", "name", {
});

var AST_This = DEFNODE("This", null, {

}, AST_Symbol);

var AST_SymbolRef = DEFNODE("SymbolRef", "scope symbol", {

}, AST_Symbol);

var AST_Label = DEFNODE("Label", null, {

}, AST_SymbolRef);

var AST_Constant = DEFNODE("Constant", null, {
    getValue: function() {
        return this.value;
    }
});

var AST_String = DEFNODE("String", "value", {

}, AST_Constant);

var AST_Number = DEFNODE("Number", "value", {

}, AST_Constant);

var AST_RegExp = DEFNODE("Regexp", "pattern mods", {
    initialize: function() {
        this.value = new RegExp(this.pattern, this.mods);
    }
}, AST_Constant);

var AST_Atom = DEFNODE("Atom", null, {

}, AST_Constant);

var AST_Null = DEFNODE("Null", null, {
    value: null
}, AST_Atom);

var AST_Undefined = DEFNODE("Undefined", null, {
    value: (function(){}())
}, AST_Atom);

var AST_False = DEFNODE("False", null, {
    value: false
}, AST_Atom);

var AST_True = DEFNODE("True", null, {
    value: true
}, AST_Atom);

/* -----[ Walker ]----- */

function TreeWalker(visitor) {
    this.stack = [];
    if (visitor) this.visit = visitor;
};

TreeWalker.prototype = {
    visit: function(node){},
    parent: function(n) {
        if (n == null) n = 1;
        return this.stack[this.stack.length - n];
    },
    find_parent: function(type) {
        for (var a = this.stack, i = a.length; --i >= 0;)
            if (a[i] instanceof type) return a[i];
        return null;
    },
    _visit: function(node, descend) {
        this.visit(node);
        if (descend) {
            this.stack.push(node);
            descend.call(node);
            this.stack.pop();
        }
    }
};