989 lines
		
	
	
		
			32 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
			
		
		
	
	
			989 lines
		
	
	
		
			32 KiB
		
	
	
	
		
			JavaScript
		
	
	
	
	
	
| /***********************************************************************
 | |
| 
 | |
|   A JavaScript tokenizer / parser / beautifier / compressor.
 | |
|   https://github.com/mishoo/UglifyJS2
 | |
| 
 | |
|   -------------------------------- (C) ---------------------------------
 | |
| 
 | |
|                            Author: Mihai Bazon
 | |
|                          <mihai.bazon@gmail.com>
 | |
|                        http://mihai.bazon.net/blog
 | |
| 
 | |
|   Distributed under the BSD license:
 | |
| 
 | |
|     Copyright 2012 (c) Mihai Bazon <mihai.bazon@gmail.com>
 | |
| 
 | |
|     Redistribution and use in source and binary forms, with or without
 | |
|     modification, are permitted provided that the following conditions
 | |
|     are met:
 | |
| 
 | |
|         * Redistributions of source code must retain the above
 | |
|           copyright notice, this list of conditions and the following
 | |
|           disclaimer.
 | |
| 
 | |
|         * Redistributions in binary form must reproduce the above
 | |
|           copyright notice, this list of conditions and the following
 | |
|           disclaimer in the documentation and/or other materials
 | |
|           provided with the distribution.
 | |
| 
 | |
|     THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER “AS IS” AND ANY
 | |
|     EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 | |
|     IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 | |
|     PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER BE
 | |
|     LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
 | |
|     OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
 | |
|     PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
 | |
|     PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 | |
|     THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
 | |
|     TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
 | |
|     THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 | |
|     SUCH DAMAGE.
 | |
| 
 | |
|  ***********************************************************************/
 | |
| 
 | |
| "use strict";
 | |
| 
 | |
| function DEFNODE(type, props, methods, base) {
 | |
|     if (typeof base === "undefined") base = AST_Node;
 | |
|     props = props ? props.split(/\s+/) : [];
 | |
|     var self_props = props;
 | |
|     if (base && base.PROPS) props = props.concat(base.PROPS);
 | |
|     var code = [
 | |
|         "return function AST_", type, "(props){",
 | |
|         "if(props){",
 | |
|     ];
 | |
|     props.forEach(function(prop) {
 | |
|         code.push("this.", prop, "=props.", prop, ";");
 | |
|     });
 | |
|     var proto = base && new base;
 | |
|     if (proto && proto.initialize || methods && methods.initialize) code.push("this.initialize();");
 | |
|     code.push("}}");
 | |
|     var ctor = new Function(code.join(""))();
 | |
|     if (proto) {
 | |
|         ctor.prototype = proto;
 | |
|         ctor.BASE = base;
 | |
|     }
 | |
|     if (base) base.SUBCLASSES.push(ctor);
 | |
|     ctor.prototype.CTOR = ctor;
 | |
|     ctor.PROPS = props || null;
 | |
|     ctor.SELF_PROPS = self_props;
 | |
|     ctor.SUBCLASSES = [];
 | |
|     if (type) {
 | |
|         ctor.prototype.TYPE = ctor.TYPE = type;
 | |
|     }
 | |
|     if (methods) for (var name in methods) if (HOP(methods, name)) {
 | |
|         if (/^\$/.test(name)) {
 | |
|             ctor[name.substr(1)] = methods[name];
 | |
|         } else {
 | |
|             ctor.prototype[name] = methods[name];
 | |
|         }
 | |
|     }
 | |
|     ctor.DEFMETHOD = function(name, method) {
 | |
|         this.prototype[name] = method;
 | |
|     };
 | |
|     if (typeof exports !== "undefined") {
 | |
|         exports["AST_" + type] = ctor;
 | |
|     }
 | |
|     return ctor;
 | |
| }
 | |
| 
 | |
| var AST_Token = DEFNODE("Token", "type value line col pos endline endcol endpos nlb comments_before comments_after file raw", {
 | |
| }, null);
 | |
| 
 | |
| var AST_Node = DEFNODE("Node", "start end", {
 | |
|     _clone: function(deep) {
 | |
|         if (deep) {
 | |
|             var self = this.clone();
 | |
|             return self.transform(new TreeTransformer(function(node) {
 | |
|                 if (node !== self) {
 | |
|                     return node.clone(true);
 | |
|                 }
 | |
|             }));
 | |
|         }
 | |
|         return new this.CTOR(this);
 | |
|     },
 | |
|     clone: function(deep) {
 | |
|         return this._clone(deep);
 | |
|     },
 | |
|     $documentation: "Base class of all AST nodes",
 | |
|     $propdoc: {
 | |
|         start: "[AST_Token] The first token of this node",
 | |
|         end: "[AST_Token] The last token of this node"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this);
 | |
|     },
 | |
|     walk: function(visitor) {
 | |
|         return this._walk(visitor); // not sure the indirection will be any help
 | |
|     }
 | |
| }, null);
 | |
| 
 | |
| (AST_Node.log_function = function(fn, verbose) {
 | |
|     var printed = Object.create(null);
 | |
|     if (fn) {
 | |
|         AST_Node.info = verbose ? function(text, props) {
 | |
|             log("INFO: " + string_template(text, props));
 | |
|         } : noop;
 | |
|         AST_Node.warn = function(text, props) {
 | |
|             log("WARN: " + string_template(text, props));
 | |
|         };
 | |
|     } else {
 | |
|         AST_Node.info = AST_Node.warn = noop;
 | |
|     }
 | |
| 
 | |
|     function log(msg) {
 | |
|         if (printed[msg]) return;
 | |
|         printed[msg] = true;
 | |
|         fn(msg);
 | |
|     }
 | |
| })();
 | |
| 
 | |
| /* -----[ statements ]----- */
 | |
| 
 | |
| var AST_Statement = DEFNODE("Statement", null, {
 | |
|     $documentation: "Base class of all statements",
 | |
| });
 | |
| 
 | |
| var AST_Debugger = DEFNODE("Debugger", null, {
 | |
|     $documentation: "Represents a debugger statement",
 | |
| }, AST_Statement);
 | |
| 
 | |
| var AST_Directive = DEFNODE("Directive", "value quote", {
 | |
|     $documentation: "Represents a directive, like \"use strict\";",
 | |
|     $propdoc: {
 | |
|         value: "[string] The value of this directive as a plain string (it's not an AST_String!)",
 | |
|         quote: "[string] the original quote character"
 | |
|     },
 | |
| }, AST_Statement);
 | |
| 
 | |
| var AST_SimpleStatement = DEFNODE("SimpleStatement", "body", {
 | |
|     $documentation: "A statement consisting of an expression, i.e. a = 1 + 2",
 | |
|     $propdoc: {
 | |
|         body: "[AST_Node] an expression node (should not be instanceof AST_Statement)"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.body._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Statement);
 | |
| 
 | |
| function walk_body(node, visitor) {
 | |
|     var body = node.body;
 | |
|     if (body instanceof AST_Statement) {
 | |
|         body._walk(visitor);
 | |
|     } else body.forEach(function(node) {
 | |
|         node._walk(visitor);
 | |
|     });
 | |
| }
 | |
| 
 | |
| var AST_Block = DEFNODE("Block", "body", {
 | |
|     $documentation: "A body of statements (usually braced)",
 | |
|     $propdoc: {
 | |
|         body: "[AST_Statement*] an array of statements"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             walk_body(this, visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Statement);
 | |
| 
 | |
| var AST_BlockStatement = DEFNODE("BlockStatement", null, {
 | |
|     $documentation: "A block statement",
 | |
| }, AST_Block);
 | |
| 
 | |
| var AST_EmptyStatement = DEFNODE("EmptyStatement", null, {
 | |
|     $documentation: "The empty statement (empty block or simply a semicolon)"
 | |
| }, AST_Statement);
 | |
| 
 | |
| var AST_StatementWithBody = DEFNODE("StatementWithBody", "body", {
 | |
|     $documentation: "Base class for all statements that contain one nested body: `For`, `ForIn`, `Do`, `While`, `With`",
 | |
|     $propdoc: {
 | |
|         body: "[AST_Statement] the body; this should always be present, even if it's an AST_EmptyStatement"
 | |
|     }
 | |
| }, AST_Statement);
 | |
| 
 | |
| var AST_LabeledStatement = DEFNODE("LabeledStatement", "label", {
 | |
|     $documentation: "Statement with a label",
 | |
|     $propdoc: {
 | |
|         label: "[AST_Label] a label definition"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.label._walk(visitor);
 | |
|             this.body._walk(visitor);
 | |
|         });
 | |
|     },
 | |
|     clone: function(deep) {
 | |
|         var node = this._clone(deep);
 | |
|         if (deep) {
 | |
|             var label = node.label;
 | |
|             var def = this.label;
 | |
|             node.walk(new TreeWalker(function(node) {
 | |
|                 if (node instanceof AST_LoopControl && node.label && node.label.thedef === def) {
 | |
|                     node.label.thedef = label;
 | |
|                     label.references.push(node);
 | |
|                 }
 | |
|             }));
 | |
|         }
 | |
|         return node;
 | |
|     }
 | |
| }, AST_StatementWithBody);
 | |
| 
 | |
| var AST_IterationStatement = DEFNODE("IterationStatement", null, {
 | |
|     $documentation: "Internal class.  All loops inherit from it."
 | |
| }, AST_StatementWithBody);
 | |
| 
 | |
| var AST_DWLoop = DEFNODE("DWLoop", "condition", {
 | |
|     $documentation: "Base class for do/while statements",
 | |
|     $propdoc: {
 | |
|         condition: "[AST_Node] the loop condition.  Should not be instanceof AST_Statement"
 | |
|     }
 | |
| }, AST_IterationStatement);
 | |
| 
 | |
| var AST_Do = DEFNODE("Do", null, {
 | |
|     $documentation: "A `do` statement",
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.body._walk(visitor);
 | |
|             this.condition._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_DWLoop);
 | |
| 
 | |
| var AST_While = DEFNODE("While", null, {
 | |
|     $documentation: "A `while` statement",
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.condition._walk(visitor);
 | |
|             this.body._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_DWLoop);
 | |
| 
 | |
| var AST_For = DEFNODE("For", "init condition step", {
 | |
|     $documentation: "A `for` statement",
 | |
|     $propdoc: {
 | |
|         init: "[AST_Node?] the `for` initialization code, or null if empty",
 | |
|         condition: "[AST_Node?] the `for` termination clause, or null if empty",
 | |
|         step: "[AST_Node?] the `for` update clause, or null if empty"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             if (this.init) this.init._walk(visitor);
 | |
|             if (this.condition) this.condition._walk(visitor);
 | |
|             if (this.step) this.step._walk(visitor);
 | |
|             this.body._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_IterationStatement);
 | |
| 
 | |
| var AST_ForIn = DEFNODE("ForIn", "init object", {
 | |
|     $documentation: "A `for ... in` statement",
 | |
|     $propdoc: {
 | |
|         init: "[AST_Node] the `for/in` initialization code",
 | |
|         object: "[AST_Node] the object that we're looping through"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.init._walk(visitor);
 | |
|             this.object._walk(visitor);
 | |
|             this.body._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_IterationStatement);
 | |
| 
 | |
| var AST_With = DEFNODE("With", "expression", {
 | |
|     $documentation: "A `with` statement",
 | |
|     $propdoc: {
 | |
|         expression: "[AST_Node] the `with` expression"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expression._walk(visitor);
 | |
|             this.body._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_StatementWithBody);
 | |
| 
 | |
| /* -----[ scope and functions ]----- */
 | |
| 
 | |
| var AST_Scope = DEFNODE("Scope", "variables functions uses_with uses_eval parent_scope enclosed cname", {
 | |
|     $documentation: "Base class for all statements introducing a lexical scope",
 | |
|     $propdoc: {
 | |
|         variables: "[Object/S] a map of name -> SymbolDef for all variables/functions defined in this scope",
 | |
|         functions: "[Object/S] like `variables`, but only lists function declarations",
 | |
|         uses_with: "[boolean/S] tells whether this scope uses the `with` statement",
 | |
|         uses_eval: "[boolean/S] tells whether this scope contains a direct call to the global `eval`",
 | |
|         parent_scope: "[AST_Scope?/S] link to the parent scope",
 | |
|         enclosed: "[SymbolDef*/S] a list of all symbol definitions that are accessed from this scope or any subscopes",
 | |
|         cname: "[integer/S] current index for mangling variables (used internally by the mangler)",
 | |
|     },
 | |
|     clone: function(deep) {
 | |
|         var node = this._clone(deep);
 | |
|         if (this.variables) node.variables = this.variables.clone();
 | |
|         if (this.functions) node.functions = this.functions.clone();
 | |
|         if (this.enclosed) node.enclosed = this.enclosed.slice();
 | |
|         return node;
 | |
|     },
 | |
|     pinned: function() {
 | |
|         return this.uses_eval || this.uses_with;
 | |
|     }
 | |
| }, AST_Block);
 | |
| 
 | |
| var AST_Toplevel = DEFNODE("Toplevel", "globals", {
 | |
|     $documentation: "The toplevel scope",
 | |
|     $propdoc: {
 | |
|         globals: "[Object/S] a map of name -> SymbolDef for all undeclared names",
 | |
|     },
 | |
|     wrap: function(name) {
 | |
|         var body = this.body;
 | |
|         return parse([
 | |
|             "(function(exports){'$ORIG';})(typeof ",
 | |
|             name,
 | |
|             "=='undefined'?(",
 | |
|             name,
 | |
|             "={}):",
 | |
|             name,
 | |
|             ");"
 | |
|         ].join(""), {
 | |
|             filename: "wrap=" + JSON.stringify(name)
 | |
|         }).transform(new TreeTransformer(function(node) {
 | |
|             if (node instanceof AST_Directive && node.value == "$ORIG") {
 | |
|                 return MAP.splice(body);
 | |
|             }
 | |
|         }));
 | |
|     },
 | |
|     enclose: function(args_values) {
 | |
|         if (typeof args_values != "string") args_values = "";
 | |
|         var index = args_values.indexOf(":");
 | |
|         if (index < 0) index = args_values.length;
 | |
|         var body = this.body;
 | |
|         return parse([
 | |
|             "(function(",
 | |
|             args_values.slice(0, index),
 | |
|             '){"$ORIG"})(',
 | |
|             args_values.slice(index + 1),
 | |
|             ")"
 | |
|         ].join(""), {
 | |
|             filename: "enclose=" + JSON.stringify(args_values)
 | |
|         }).transform(new TreeTransformer(function(node) {
 | |
|             if (node instanceof AST_Directive && node.value == "$ORIG") {
 | |
|                 return MAP.splice(body);
 | |
|             }
 | |
|         }));
 | |
|     }
 | |
| }, AST_Scope);
 | |
| 
 | |
| var AST_Lambda = DEFNODE("Lambda", "name argnames uses_arguments length_read", {
 | |
|     $documentation: "Base class for functions",
 | |
|     $propdoc: {
 | |
|         name: "[AST_SymbolDeclaration?] the name of this function",
 | |
|         argnames: "[AST_SymbolFunarg*] array of function arguments",
 | |
|         uses_arguments: "[boolean/S] tells whether this function accesses the arguments array"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             if (this.name) this.name._walk(visitor);
 | |
|             this.argnames.forEach(function(argname) {
 | |
|                 argname._walk(visitor);
 | |
|             });
 | |
|             walk_body(this, visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Scope);
 | |
| 
 | |
| var AST_Accessor = DEFNODE("Accessor", null, {
 | |
|     $documentation: "A setter/getter function.  The `name` property is always null."
 | |
| }, AST_Lambda);
 | |
| 
 | |
| var AST_Function = DEFNODE("Function", "inlined", {
 | |
|     $documentation: "A function expression"
 | |
| }, AST_Lambda);
 | |
| 
 | |
| var AST_Defun = DEFNODE("Defun", "inlined", {
 | |
|     $documentation: "A function definition"
 | |
| }, AST_Lambda);
 | |
| 
 | |
| /* -----[ JUMPS ]----- */
 | |
| 
 | |
| var AST_Jump = DEFNODE("Jump", null, {
 | |
|     $documentation: "Base class for “jumps” (for now that's `return`, `throw`, `break` and `continue`)"
 | |
| }, AST_Statement);
 | |
| 
 | |
| var AST_Exit = DEFNODE("Exit", "value", {
 | |
|     $documentation: "Base class for “exits” (`return` and `throw`)",
 | |
|     $propdoc: {
 | |
|         value: "[AST_Node?] the value returned or thrown by this statement; could be null for AST_Return"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, this.value && function() {
 | |
|             this.value._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Jump);
 | |
| 
 | |
| var AST_Return = DEFNODE("Return", null, {
 | |
|     $documentation: "A `return` statement"
 | |
| }, AST_Exit);
 | |
| 
 | |
| var AST_Throw = DEFNODE("Throw", null, {
 | |
|     $documentation: "A `throw` statement"
 | |
| }, AST_Exit);
 | |
| 
 | |
| var AST_LoopControl = DEFNODE("LoopControl", "label", {
 | |
|     $documentation: "Base class for loop control statements (`break` and `continue`)",
 | |
|     $propdoc: {
 | |
|         label: "[AST_LabelRef?] the label, or null if none",
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, this.label && function() {
 | |
|             this.label._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Jump);
 | |
| 
 | |
| var AST_Break = DEFNODE("Break", null, {
 | |
|     $documentation: "A `break` statement"
 | |
| }, AST_LoopControl);
 | |
| 
 | |
| var AST_Continue = DEFNODE("Continue", null, {
 | |
|     $documentation: "A `continue` statement"
 | |
| }, AST_LoopControl);
 | |
| 
 | |
| /* -----[ IF ]----- */
 | |
| 
 | |
| var AST_If = DEFNODE("If", "condition alternative", {
 | |
|     $documentation: "A `if` statement",
 | |
|     $propdoc: {
 | |
|         condition: "[AST_Node] the `if` condition",
 | |
|         alternative: "[AST_Statement?] the `else` part, or null if not present"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.condition._walk(visitor);
 | |
|             this.body._walk(visitor);
 | |
|             if (this.alternative) this.alternative._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_StatementWithBody);
 | |
| 
 | |
| /* -----[ SWITCH ]----- */
 | |
| 
 | |
| var AST_Switch = DEFNODE("Switch", "expression", {
 | |
|     $documentation: "A `switch` statement",
 | |
|     $propdoc: {
 | |
|         expression: "[AST_Node] the `switch` “discriminant”"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expression._walk(visitor);
 | |
|             walk_body(this, visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Block);
 | |
| 
 | |
| var AST_SwitchBranch = DEFNODE("SwitchBranch", null, {
 | |
|     $documentation: "Base class for `switch` branches",
 | |
| }, AST_Block);
 | |
| 
 | |
| var AST_Default = DEFNODE("Default", null, {
 | |
|     $documentation: "A `default` switch branch",
 | |
| }, AST_SwitchBranch);
 | |
| 
 | |
| var AST_Case = DEFNODE("Case", "expression", {
 | |
|     $documentation: "A `case` switch branch",
 | |
|     $propdoc: {
 | |
|         expression: "[AST_Node] the `case` expression"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expression._walk(visitor);
 | |
|             walk_body(this, visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_SwitchBranch);
 | |
| 
 | |
| /* -----[ EXCEPTIONS ]----- */
 | |
| 
 | |
| var AST_Try = DEFNODE("Try", "bcatch bfinally", {
 | |
|     $documentation: "A `try` statement",
 | |
|     $propdoc: {
 | |
|         bcatch: "[AST_Catch?] the catch block, or null if not present",
 | |
|         bfinally: "[AST_Finally?] the finally block, or null if not present"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             walk_body(this, visitor);
 | |
|             if (this.bcatch) this.bcatch._walk(visitor);
 | |
|             if (this.bfinally) this.bfinally._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Block);
 | |
| 
 | |
| var AST_Catch = DEFNODE("Catch", "argname", {
 | |
|     $documentation: "A `catch` node; only makes sense as part of a `try` statement",
 | |
|     $propdoc: {
 | |
|         argname: "[AST_SymbolCatch] symbol for the exception"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.argname._walk(visitor);
 | |
|             walk_body(this, visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_Block);
 | |
| 
 | |
| var AST_Finally = DEFNODE("Finally", null, {
 | |
|     $documentation: "A `finally` node; only makes sense as part of a `try` statement"
 | |
| }, AST_Block);
 | |
| 
 | |
| /* -----[ VAR ]----- */
 | |
| 
 | |
| var AST_Definitions = DEFNODE("Definitions", "definitions", {
 | |
|     $documentation: "Base class for `var` nodes (variable declarations/initializations)",
 | |
|     $propdoc: {
 | |
|         definitions: "[AST_VarDef*] array of variable definitions"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.definitions.forEach(function(defn) {
 | |
|                 defn._walk(visitor);
 | |
|             });
 | |
|         });
 | |
|     }
 | |
| }, AST_Statement);
 | |
| 
 | |
| var AST_Var = DEFNODE("Var", null, {
 | |
|     $documentation: "A `var` statement"
 | |
| }, AST_Definitions);
 | |
| 
 | |
| var AST_VarDef = DEFNODE("VarDef", "name value", {
 | |
|     $documentation: "A variable declaration; only appears in a AST_Definitions node",
 | |
|     $propdoc: {
 | |
|         name: "[AST_SymbolVar] name of the variable",
 | |
|         value: "[AST_Node?] initializer, or null of there's no initializer"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.name._walk(visitor);
 | |
|             if (this.value) this.value._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| /* -----[ OTHER ]----- */
 | |
| 
 | |
| var AST_Call = DEFNODE("Call", "expression args", {
 | |
|     $documentation: "A function call expression",
 | |
|     $propdoc: {
 | |
|         expression: "[AST_Node] expression to invoke as function",
 | |
|         args: "[AST_Node*] array of arguments"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expression._walk(visitor);
 | |
|             this.args.forEach(function(node) {
 | |
|                 node._walk(visitor);
 | |
|             });
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_New = DEFNODE("New", null, {
 | |
|     $documentation: "An object instantiation.  Derives from a function call since it has exactly the same properties"
 | |
| }, AST_Call);
 | |
| 
 | |
| var AST_Sequence = DEFNODE("Sequence", "expressions", {
 | |
|     $documentation: "A sequence expression (comma-separated expressions)",
 | |
|     $propdoc: {
 | |
|         expressions: "[AST_Node*] array of expressions (at least two)"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expressions.forEach(function(node) {
 | |
|                 node._walk(visitor);
 | |
|             });
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_PropAccess = DEFNODE("PropAccess", "expression property", {
 | |
|     $documentation: "Base class for property access expressions, i.e. `a.foo` or `a[\"foo\"]`",
 | |
|     $propdoc: {
 | |
|         expression: "[AST_Node] the “container” expression",
 | |
|         property: "[AST_Node|string] the property to access.  For AST_Dot this is always a plain string, while for AST_Sub it's an arbitrary AST_Node"
 | |
|     },
 | |
|     getProperty: function() {
 | |
|         var p = this.property;
 | |
|         if (p instanceof AST_Constant) {
 | |
|             return p.getValue();
 | |
|         }
 | |
|         if (p instanceof AST_UnaryPrefix
 | |
|             && p.operator == "void"
 | |
|             && p.expression instanceof AST_Constant) {
 | |
|             return;
 | |
|         }
 | |
|         return p;
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_Dot = DEFNODE("Dot", null, {
 | |
|     $documentation: "A dotted property access expression",
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expression._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_PropAccess);
 | |
| 
 | |
| var AST_Sub = DEFNODE("Sub", null, {
 | |
|     $documentation: "Index-style property access, i.e. `a[\"foo\"]`",
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expression._walk(visitor);
 | |
|             this.property._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| }, AST_PropAccess);
 | |
| 
 | |
| var AST_Unary = DEFNODE("Unary", "operator expression", {
 | |
|     $documentation: "Base class for unary expressions",
 | |
|     $propdoc: {
 | |
|         operator: "[string] the operator",
 | |
|         expression: "[AST_Node] expression that this unary operator applies to"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.expression._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_UnaryPrefix = DEFNODE("UnaryPrefix", null, {
 | |
|     $documentation: "Unary prefix expression, i.e. `typeof i` or `++i`"
 | |
| }, AST_Unary);
 | |
| 
 | |
| var AST_UnaryPostfix = DEFNODE("UnaryPostfix", null, {
 | |
|     $documentation: "Unary postfix expression, i.e. `i++`"
 | |
| }, AST_Unary);
 | |
| 
 | |
| var AST_Binary = DEFNODE("Binary", "operator left right", {
 | |
|     $documentation: "Binary expression, i.e. `a + b`",
 | |
|     $propdoc: {
 | |
|         left: "[AST_Node] left-hand side expression",
 | |
|         operator: "[string] the operator",
 | |
|         right: "[AST_Node] right-hand side expression"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.left._walk(visitor);
 | |
|             this.right._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_Conditional = DEFNODE("Conditional", "condition consequent alternative", {
 | |
|     $documentation: "Conditional expression using the ternary operator, i.e. `a ? b : c`",
 | |
|     $propdoc: {
 | |
|         condition: "[AST_Node]",
 | |
|         consequent: "[AST_Node]",
 | |
|         alternative: "[AST_Node]"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.condition._walk(visitor);
 | |
|             this.consequent._walk(visitor);
 | |
|             this.alternative._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_Assign = DEFNODE("Assign", null, {
 | |
|     $documentation: "An assignment expression — `a = b + 5`",
 | |
| }, AST_Binary);
 | |
| 
 | |
| /* -----[ LITERALS ]----- */
 | |
| 
 | |
| var AST_Array = DEFNODE("Array", "elements", {
 | |
|     $documentation: "An array literal",
 | |
|     $propdoc: {
 | |
|         elements: "[AST_Node*] array of elements"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.elements.forEach(function(element) {
 | |
|                 element._walk(visitor);
 | |
|             });
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_Object = DEFNODE("Object", "properties", {
 | |
|     $documentation: "An object literal",
 | |
|     $propdoc: {
 | |
|         properties: "[AST_ObjectProperty*] array of properties"
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.properties.forEach(function(prop) {
 | |
|                 prop._walk(visitor);
 | |
|             });
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_ObjectProperty = DEFNODE("ObjectProperty", "key value", {
 | |
|     $documentation: "Base class for literal object properties",
 | |
|     $propdoc: {
 | |
|         key: "[string|AST_SymbolAccessor] property name. For ObjectKeyVal this is a string. For getters and setters this is an AST_SymbolAccessor.",
 | |
|         value: "[AST_Node] property value.  For getters and setters this is an AST_Accessor."
 | |
|     },
 | |
|     _walk: function(visitor) {
 | |
|         return visitor._visit(this, function() {
 | |
|             this.value._walk(visitor);
 | |
|         });
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_ObjectKeyVal = DEFNODE("ObjectKeyVal", "quote", {
 | |
|     $documentation: "A key: value object property",
 | |
|     $propdoc: {
 | |
|         quote: "[string] the original quote character"
 | |
|     }
 | |
| }, AST_ObjectProperty);
 | |
| 
 | |
| var AST_ObjectSetter = DEFNODE("ObjectSetter", null, {
 | |
|     $documentation: "An object setter property",
 | |
| }, AST_ObjectProperty);
 | |
| 
 | |
| var AST_ObjectGetter = DEFNODE("ObjectGetter", null, {
 | |
|     $documentation: "An object getter property",
 | |
| }, AST_ObjectProperty);
 | |
| 
 | |
| var AST_Symbol = DEFNODE("Symbol", "scope name thedef", {
 | |
|     $propdoc: {
 | |
|         name: "[string] name of this symbol",
 | |
|         scope: "[AST_Scope/S] the current scope (not necessarily the definition scope)",
 | |
|         thedef: "[SymbolDef/S] the definition of this symbol"
 | |
|     },
 | |
|     $documentation: "Base class for all symbols",
 | |
| });
 | |
| 
 | |
| var AST_SymbolAccessor = DEFNODE("SymbolAccessor", null, {
 | |
|     $documentation: "The name of a property accessor (setter/getter function)"
 | |
| }, AST_Symbol);
 | |
| 
 | |
| var AST_SymbolDeclaration = DEFNODE("SymbolDeclaration", "init", {
 | |
|     $documentation: "A declaration symbol (symbol in var, function name or argument, symbol in catch)",
 | |
| }, AST_Symbol);
 | |
| 
 | |
| var AST_SymbolVar = DEFNODE("SymbolVar", null, {
 | |
|     $documentation: "Symbol defining a variable",
 | |
| }, AST_SymbolDeclaration);
 | |
| 
 | |
| var AST_SymbolFunarg = DEFNODE("SymbolFunarg", null, {
 | |
|     $documentation: "Symbol naming a function argument",
 | |
| }, AST_SymbolVar);
 | |
| 
 | |
| var AST_SymbolDefun = DEFNODE("SymbolDefun", null, {
 | |
|     $documentation: "Symbol defining a function",
 | |
| }, AST_SymbolDeclaration);
 | |
| 
 | |
| var AST_SymbolLambda = DEFNODE("SymbolLambda", null, {
 | |
|     $documentation: "Symbol naming a function expression",
 | |
| }, AST_SymbolDeclaration);
 | |
| 
 | |
| var AST_SymbolCatch = DEFNODE("SymbolCatch", null, {
 | |
|     $documentation: "Symbol naming the exception in catch",
 | |
| }, AST_SymbolDeclaration);
 | |
| 
 | |
| var AST_Label = DEFNODE("Label", "references", {
 | |
|     $documentation: "Symbol naming a label (declaration)",
 | |
|     $propdoc: {
 | |
|         references: "[AST_LoopControl*] a list of nodes referring to this label"
 | |
|     },
 | |
|     initialize: function() {
 | |
|         this.references = [];
 | |
|         this.thedef = this;
 | |
|     }
 | |
| }, AST_Symbol);
 | |
| 
 | |
| var AST_SymbolRef = DEFNODE("SymbolRef", "fixed", {
 | |
|     $documentation: "Reference to some symbol (not definition/declaration)",
 | |
| }, AST_Symbol);
 | |
| 
 | |
| var AST_LabelRef = DEFNODE("LabelRef", null, {
 | |
|     $documentation: "Reference to a label symbol",
 | |
| }, AST_Symbol);
 | |
| 
 | |
| var AST_This = DEFNODE("This", null, {
 | |
|     $documentation: "The `this` symbol",
 | |
| }, AST_Symbol);
 | |
| 
 | |
| var AST_Constant = DEFNODE("Constant", null, {
 | |
|     $documentation: "Base class for all constants",
 | |
|     getValue: function() {
 | |
|         return this.value;
 | |
|     }
 | |
| });
 | |
| 
 | |
| var AST_String = DEFNODE("String", "value quote", {
 | |
|     $documentation: "A string literal",
 | |
|     $propdoc: {
 | |
|         value: "[string] the contents of this string",
 | |
|         quote: "[string] the original quote character"
 | |
|     }
 | |
| }, AST_Constant);
 | |
| 
 | |
| var AST_Number = DEFNODE("Number", "value literal", {
 | |
|     $documentation: "A number literal",
 | |
|     $propdoc: {
 | |
|         value: "[number] the numeric value",
 | |
|         literal: "[string] numeric value as string (optional)"
 | |
|     }
 | |
| }, AST_Constant);
 | |
| 
 | |
| var AST_RegExp = DEFNODE("RegExp", "value", {
 | |
|     $documentation: "A regexp literal",
 | |
|     $propdoc: {
 | |
|         value: "[RegExp] the actual regexp"
 | |
|     }
 | |
| }, AST_Constant);
 | |
| 
 | |
| var AST_Atom = DEFNODE("Atom", null, {
 | |
|     $documentation: "Base class for atoms",
 | |
| }, AST_Constant);
 | |
| 
 | |
| var AST_Null = DEFNODE("Null", null, {
 | |
|     $documentation: "The `null` atom",
 | |
|     value: null
 | |
| }, AST_Atom);
 | |
| 
 | |
| var AST_NaN = DEFNODE("NaN", null, {
 | |
|     $documentation: "The impossible value",
 | |
|     value: 0/0
 | |
| }, AST_Atom);
 | |
| 
 | |
| var AST_Undefined = DEFNODE("Undefined", null, {
 | |
|     $documentation: "The `undefined` value",
 | |
|     value: function(){}()
 | |
| }, AST_Atom);
 | |
| 
 | |
| var AST_Hole = DEFNODE("Hole", null, {
 | |
|     $documentation: "A hole in an array",
 | |
|     value: function(){}()
 | |
| }, AST_Atom);
 | |
| 
 | |
| var AST_Infinity = DEFNODE("Infinity", null, {
 | |
|     $documentation: "The `Infinity` value",
 | |
|     value: 1/0
 | |
| }, AST_Atom);
 | |
| 
 | |
| var AST_Boolean = DEFNODE("Boolean", null, {
 | |
|     $documentation: "Base class for booleans",
 | |
| }, AST_Atom);
 | |
| 
 | |
| var AST_False = DEFNODE("False", null, {
 | |
|     $documentation: "The `false` atom",
 | |
|     value: false
 | |
| }, AST_Boolean);
 | |
| 
 | |
| var AST_True = DEFNODE("True", null, {
 | |
|     $documentation: "The `true` atom",
 | |
|     value: true
 | |
| }, AST_Boolean);
 | |
| 
 | |
| /* -----[ TreeWalker ]----- */
 | |
| 
 | |
| function TreeWalker(callback) {
 | |
|     this.visit = callback;
 | |
|     this.stack = [];
 | |
|     this.directives = Object.create(null);
 | |
| }
 | |
| TreeWalker.prototype = {
 | |
|     _visit: function(node, descend) {
 | |
|         this.push(node);
 | |
|         var ret = this.visit(node, descend ? function() {
 | |
|             descend.call(node);
 | |
|         } : noop);
 | |
|         if (!ret && descend) {
 | |
|             descend.call(node);
 | |
|         }
 | |
|         this.pop();
 | |
|         return ret;
 | |
|     },
 | |
|     parent: function(n) {
 | |
|         return this.stack[this.stack.length - 2 - (n || 0)];
 | |
|     },
 | |
|     push: function(node) {
 | |
|         if (node instanceof AST_Lambda) {
 | |
|             this.directives = Object.create(this.directives);
 | |
|         } else if (node instanceof AST_Directive && !this.directives[node.value]) {
 | |
|             this.directives[node.value] = node;
 | |
|         }
 | |
|         this.stack.push(node);
 | |
|     },
 | |
|     pop: function() {
 | |
|         if (this.stack.pop() instanceof AST_Lambda) {
 | |
|             this.directives = Object.getPrototypeOf(this.directives);
 | |
|         }
 | |
|     },
 | |
|     self: function() {
 | |
|         return this.stack[this.stack.length - 1];
 | |
|     },
 | |
|     find_parent: function(type) {
 | |
|         var stack = this.stack;
 | |
|         for (var i = stack.length; --i >= 0;) {
 | |
|             var x = stack[i];
 | |
|             if (x instanceof type) return x;
 | |
|         }
 | |
|     },
 | |
|     has_directive: function(type) {
 | |
|         var dir = this.directives[type];
 | |
|         if (dir) return dir;
 | |
|         var node = this.stack[this.stack.length - 1];
 | |
|         if (node instanceof AST_Scope) {
 | |
|             for (var i = 0; i < node.body.length; ++i) {
 | |
|                 var st = node.body[i];
 | |
|                 if (!(st instanceof AST_Directive)) break;
 | |
|                 if (st.value == type) return st;
 | |
|             }
 | |
|         }
 | |
|     },
 | |
|     loopcontrol_target: function(node) {
 | |
|         var stack = this.stack;
 | |
|         if (node.label) for (var i = stack.length; --i >= 0;) {
 | |
|             var x = stack[i];
 | |
|             if (x instanceof AST_LabeledStatement && x.label.name == node.label.name)
 | |
|                 return x.body;
 | |
|         } else for (var i = stack.length; --i >= 0;) {
 | |
|             var x = stack[i];
 | |
|             if (x instanceof AST_IterationStatement
 | |
|                 || node instanceof AST_Break && x instanceof AST_Switch)
 | |
|                 return x;
 | |
|         }
 | |
|     },
 | |
|     in_boolean_context: function() {
 | |
|         var self = this.self();
 | |
|         for (var i = 0, p; p = this.parent(i); i++) {
 | |
|             if (p instanceof AST_SimpleStatement
 | |
|                 || p instanceof AST_Conditional && p.condition === self
 | |
|                 || p instanceof AST_DWLoop && p.condition === self
 | |
|                 || p instanceof AST_For && p.condition === self
 | |
|                 || p instanceof AST_If && p.condition === self
 | |
|                 || p instanceof AST_UnaryPrefix && p.operator == "!" && p.expression === self) {
 | |
|                 return true;
 | |
|             }
 | |
|             if (p instanceof AST_Binary && (p.operator == "&&" || p.operator == "||")
 | |
|                 || p instanceof AST_Conditional
 | |
|                 || p.tail_node() === self) {
 | |
|                 self = p;
 | |
|             } else {
 | |
|                 return false;
 | |
|             }
 | |
|         }
 | |
|     }
 | |
| };
 |