/*********************************************************************** A JavaScript tokenizer / parser / beautifier / compressor. https://github.com/mishoo/UglifyJS2 -------------------------------- (C) --------------------------------- Author: Mihai Bazon http://mihai.bazon.net/blog Distributed under the BSD license: Copyright 2012 (c) Mihai Bazon 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. ***********************************************************************/ // The layout of the compressor follows the code generator (see // output.js). Basically each node will have a "squeeze" method // that will apply all known compression rules for that node, and // return a new node (or the original node if there was no // compression). We can't quite use the TreeWalker for this // because it's too simplistic. // The Compressor object is for storing the options and for // maintaining various internal state that might be useful for // squeezing nodes. function Compressor(options, false_by_default) { options = defaults(options, { sequences : !false_by_default, properties : !false_by_default, dead_code : !false_by_default, drop_debugger : !false_by_default, unsafe : !false_by_default, conditionals : !false_by_default, comparations : !false_by_default, evaluate : !false_by_default, booleans : !false_by_default, dwloops : !false_by_default, hoist : !false_by_default, warnings : true }); var stack = []; function in_boolean_context() { var i = stack.length, self = stack[--i]; while (i > 0) { var p = stack[--i]; if ((p instanceof AST_If && p.condition === self) || (p instanceof AST_Conditional && p.condition === self) || (p instanceof AST_DWLoop && p.condition === self) || (p instanceof AST_For && p.condition === self) || (p instanceof AST_UnaryPrefix && p.operator == "!" && p.expression === self)) { return true; } if (!(p instanceof AST_Binary && (p.operator == "&&" || p.operator == "||"))) return false; self = p; } }; return { option : function(key) { return options[key] }, push_node : function(node) { stack.push(node) }, pop_node : function() { return stack.pop() }, stack : function() { return stack }, parent : function(n) { return stack[stack.length - 2 - (n || 0)]; }, warn : function() { if (options.warnings) AST_Node.warn.apply(AST_Node, arguments); }, in_boolean_context: in_boolean_context }; }; (function(){ AST_Node.DEFMETHOD("squeeze", function(){ return this; }); AST_Node.DEFMETHOD("optimize", function(){ return this; }); function make_node(ctor, orig, props) { if (!props) props = {}; if (!props.start) props.start = orig.start; if (!props.end) props.end = orig.end; return new ctor(props); }; function SQUEEZE(nodetype, squeeze) { nodetype.DEFMETHOD("squeeze", function(compressor){ compressor.push_node(this); var new_node = squeeze(this, compressor); compressor.pop_node(); return new_node || this; }); }; function do_list(array, compressor) { return MAP(array, function(node){ return node.squeeze(compressor); }); }; function tighten_body(statements, compressor) { statements = do_list(statements, compressor); statements = eliminate_spurious_blocks(statements); if (compressor.option("dead_code")) { statements = eliminate_dead_code(statements, compressor); } if (compressor.option("sequences")) { statements = sequencesize(statements, compressor); } return statements; }; function eliminate_spurious_blocks(statements) { return statements.reduce(function(a, stat){ if (stat.TYPE == "BlockStatement") { // XXX: no instanceof here because we would catch // AST_Lambda-s and other blocks too. perhaps we // should refine the hierarchy. a.push.apply(a, stat.body); } else { a.push(stat); } return a; }, []); }; function eliminate_dead_code(statements, compressor) { var has_quit = false; return statements.reduce(function(a, stat){ if (has_quit) { if (stat instanceof AST_Defun) { a.push(stat); } else { stat.walk(new TreeWalker(function(node){ if (node instanceof AST_Definitions || node instanceof AST_Defun) { compressor.warn("Declarations in unreachable code! [{line},{col}]", node.start); if (node instanceof AST_Definitions) { node = node.clone(); node.remove_initializers(); a.push(node); } else if (node instanceof AST_Defun) { a.push(node); } return true; // no point to descend } if (node instanceof AST_Scope) { // also don't descend any other nested scopes return true; } })); }; } else { a.push(stat); if (stat instanceof AST_Jump) { has_quit = true; } } return a; }, []); }; // XXX: this is destructive -- it modifies tree nodes. function sequencesize(statements) { var prev = null, last = statements.length - 1; if (last) statements = statements.reduce(function(a, cur, i){ if (prev instanceof AST_SimpleStatement && cur instanceof AST_SimpleStatement) { var seq = make_node(AST_Seq, prev, { first: prev.body, second: cur.body }); prev.body = seq; } else if (i == last && cur instanceof AST_Exit && cur.value && a.length == 1 && prev instanceof AST_SimpleStatement) { // it only makes sense to do this transformation // if the AST gets to a single statement. var seq = make_node(AST_Seq, prev, { first: prev.body, second: cur.value }); cur.value = seq; return [ cur ]; } else { a.push(cur); prev = cur; } return a; }, []); return statements; }; /* -----[ boolean/negation helpers ]----- */ // methods to determine whether an expression has a boolean result type (function (def){ var unary_bool = [ "!", "delete" ]; var binary_bool = [ "in", "instanceof", "==", "!=", "===", "!==", "<", "<=", ">=", ">" ]; def(AST_Node, function(){ return false }); def(AST_UnaryPrefix, function(){ return member(this.operator, unary_bool); }); def(AST_Binary, function(){ return member(this.operator, binary_bool) || ( (this.operator == "&&" || this.operator == "||") && this.left.is_boolean() && this.right.is_boolean() ); }); def(AST_Conditional, function(){ return this.consequent.is_boolean() && this.alternative.is_boolean(); }); def(AST_Assign, function(){ return this.operator == "=" && this.right.is_boolean(); }); def(AST_Seq, function(){ return this.second.is_boolean(); }); def(AST_True, function(){ return true }); def(AST_False, function(){ return true }); })(function(node, func){ node.DEFMETHOD("is_boolean", func); }); // methods to determine if an expression has a string result type (function (def){ def(AST_Node, function(){ return false }); def(AST_String, function(){ return true }); def(AST_UnaryPrefix, function(){ return this.operator == "typeof"; }); def(AST_Binary, function(){ return this.operator == "+" && (this.left.is_string() || this.right.is_string()); }); def(AST_Assign, function(){ return this.operator == "=" && this.right.is_string(); }); })(function(node, func){ node.DEFMETHOD("is_string", func); }); function best_of(ast1, ast2) { return ast1.print_to_string({ beautify: false }).length > ast2.print_to_string({ beautify: false }).length ? ast2 : ast1; }; // methods to evaluate a constant expression (function (def){ // The evaluate method returns an array with one or two // elements. If the node has been successfully reduced to a // constant, then the second element tells us the value; // otherwise the second element is missing. The first element // of the array is always an AST_Node descendant; when // evaluation was successful it's a node that represents the // constant; otherwise it's the original node. AST_Node.DEFMETHOD("evaluate", function(compressor){ if (!compressor.option("evaluate")) return [ this ]; try { var val = this._eval(), ast; switch (typeof val) { case "string": ast = make_node(AST_String, this, { value: val }); break; case "number": ast = make_node(AST_Number, this, { value: val }); break; case "boolean": ast = make_node(val ? AST_True : AST_False, this); break; case "undefined": ast = make_node(AST_Undefined, this); break; default: if (val === null) { ast = make_node(AST_Null, this); break; } throw new Error(string_template("Can't handle constant of type: {type}", { type: typeof val })); } return [ ast, val ]; } catch(ex) { if (ex !== def) throw ex; return [ this ]; } }); def(AST_Statement, function(){ throw new Error("Cannot evaluate a statement"); }); function ev(node) { return node._eval(); }; def(AST_Node, function(){ throw def; // not constant }); def(AST_Constant, function(){ return this.getValue(); }); def(AST_UnaryPrefix, function(){ var e = this.expression; switch (this.operator) { case "!": return !ev(e); case "typeof": return typeof ev(e); case "~": return ~ev(e); case "-": return -ev(e); case "+": return +ev(e); } throw def; }); def(AST_Binary, function(){ var left = this.left, right = this.right; switch (this.operator) { case "&&" : return ev(left) && ev(right); case "||" : return ev(left) || ev(right); case "|" : return ev(left) | ev(right); case "&" : return ev(left) & ev(right); case "^" : return ev(left) ^ ev(right); case "+" : return ev(left) + ev(right); case "*" : return ev(left) * ev(right); case "/" : return ev(left) / ev(right); case "%" : return ev(left) % ev(right); case "-" : return ev(left) - ev(right); case "<<" : return ev(left) << ev(right); case ">>" : return ev(left) >> ev(right); case ">>>" : return ev(left) >>> ev(right); case "==" : return ev(left) == ev(right); case "===" : return ev(left) === ev(right); case "!=" : return ev(left) != ev(right); case "!==" : return ev(left) !== ev(right); case "<" : return ev(left) < ev(right); case "<=" : return ev(left) <= ev(right); case ">" : return ev(left) > ev(right); case ">=" : return ev(left) >= ev(right); case "in" : return ev(left) in ev(right); case "instanceof" : return ev(left) instanceof ev(right); } throw def; }); def(AST_Conditional, function(){ return ev(this.condition) ? ev(this.consequent) : ev(this.alternative); }); })(function(node, func){ node.DEFMETHOD("_eval", func); }); // method to negate an expression (function(def){ function basic_negation(exp) { return make_node(AST_UnaryPrefix, exp, { operator: "!", expression: exp }); }; def(AST_Node, function(){ return basic_negation(this); }); def(AST_Statement, function(){ throw new Error("Cannot negate a statement"); }); def(AST_UnaryPrefix, function(){ if (this.operator == "!") return this.expression; return basic_negation(this); }); def(AST_Seq, function(compressor){ var self = this.clone(); self.second = self.second.negate(compressor); return self; }); def(AST_Conditional, function(){ var self = this.clone(); self.consequent = self.consequent.negate(compressor); self.alternative = self.alternative.negate(compressor); //return best_of(basic_negation(this), self); return self; }); def(AST_Binary, function(compressor){ var self = this.clone(), op = this.operator; if (compressor.option("comparations")) switch (op) { case "<=" : self.operator = ">" ; return self; case "<" : self.operator = ">=" ; return self; case ">=" : self.operator = "<" ; return self; case ">" : self.operator = "<=" ; return self; } switch (op) { case "==" : self.operator = "!="; return self; case "!=" : self.operator = "=="; return self; case "===": self.operator = "!=="; return self; case "!==": self.operator = "==="; return self; case "&&": self.operator = "||"; self.left = self.left.negate(compressor); self.right = self.right.negate(compressor); //return best_of(basic_negation(this), self); return self; case "||": self.operator = "&&"; self.left = self.left.negate(compressor); self.right = self.right.negate(compressor); //return best_of(basic_negation(this), self); return self; } return basic_negation(this); }); })(function(node, func){ node.DEFMETHOD("negate", func); }); /* -----[ node squeezers ]----- */ SQUEEZE(AST_Debugger, function(self, compressor){ if (compressor.option("drop_debugger")) return new AST_EmptyStatement(self); }); SQUEEZE(AST_LabeledStatement, function(self, compressor){ self = self.clone(); self.body = self.body.squeeze(compressor); return self.label.references.length == 0 ? self.body : self; }); SQUEEZE(AST_Statement, function(self, compressor){ self = self.clone(); self.body = self.body.squeeze(compressor); return self; }); SQUEEZE(AST_BlockStatement, function(self, compressor){ self = self.clone(); self.body = tighten_body(self.body, compressor); if (self.body.length == 1 && !self.required) return self.body[0]; return self; }); SQUEEZE(AST_Scope, function(self, compressor){ self = self.clone(); self.hoist_declarations(compressor); self.body = tighten_body(self.body, compressor); return self; }); AST_Scope.DEFMETHOD("hoist_declarations", function(compressor){ if (compressor.option("hoist")) { } }); SQUEEZE(AST_EmptyStatement, function(self, compressor){ return self; }); SQUEEZE(AST_DWLoop, function(self, compressor){ self = self.clone(); self.condition = self.condition.squeeze(compressor); self.body = self.body.squeeze(compressor); return self.optimize(compressor); }); AST_DWLoop.DEFMETHOD("optimize", function(compressor){ var self = this; if (!compressor.option("dwloops")) return self; var cond = self.condition.evaluate(compressor); if (cond.length == 2) { if (cond[1]) { return make_node(AST_For, self, { body: self.body }); } else if (self instanceof AST_While) { AST_Node.warn("Unreachable code [{line},{col}]", self.start); return make_node(AST_EmptyStatement, self); } } return self; }); SQUEEZE(AST_For, function(self, compressor){ self = self.clone(); if (self.init) self.init = self.init.squeeze(compressor); if (self.condition) self.condition = self.condition.squeeze(compressor); if (self.step) self.step = self.step.squeeze(compressor); self.body = self.body.squeeze(compressor); return self; }); SQUEEZE(AST_ForIn, function(self, compressor){ self = self.clone(); self.init = self.init.squeeze(compressor); self.object = self.object.squeeze(compressor); self.body = self.body.squeeze(compressor); return self; }); SQUEEZE(AST_With, function(self, compressor){ self = self.clone(); self.expression = self.expression.squeeze(compressor); self.body = self.body.squeeze(compressor); }); SQUEEZE(AST_Exit, function(self, compressor){ self = self.clone(); if (self.value) self.value = self.value.squeeze(compressor); return self; }); SQUEEZE(AST_LoopControl, function(self, compressor){ self = self.clone(); if (self.label) self.label = self.label.squeeze(compressor); return self; }); SQUEEZE(AST_If, function(self, compressor){ self = self.clone(); self.condition = self.condition.squeeze(compressor); self.body = self.body.squeeze(compressor); if (self.alternative) self.alternative = self.alternative.squeeze(compressor); return self.optimize(compressor); }); AST_If.DEFMETHOD("optimize", function(compressor){ var self = this; if (!compressor.option("conditionals")) return self; // if condition can be statically determined, warn and drop // one of the blocks. note, statically determined implies // “has no side effects”; also it doesn't work for cases like // `x && true`, though it probably should. var cond = self.condition.evaluate(compressor); self.condition = cond[0]; if (cond.length == 2) { if (cond[1]) { AST_Node.warn("Condition always true [{line},{col}]", self.condition.start); return self.body; } else { AST_Node.warn("Condition always false [{line},{col}]", self.condition.start); return self.alternative || make_node(AST_EmptyStatement, self); } } if (self.condition instanceof AST_UnaryPrefix && self.condition.operator == "!") { self.condition = self.condition.expression; var tmp = self.body; self.body = self.alternative || make_node(AST_EmptyStatement, self); self.alternative = tmp; } if (self.body instanceof AST_SimpleStatement && self.alternative instanceof AST_SimpleStatement) { return make_node(AST_SimpleStatement, self, { body: make_node(AST_Conditional, self, { condition : self.condition, consequent : self.body.body, alternative : self.alternative.body }).optimize(compressor) }); } if ((!self.alternative || self.alternative instanceof AST_EmptyStatement) && self.body instanceof AST_SimpleStatement) { return make_node(AST_SimpleStatement, self, { body: make_node(AST_Binary, self, { operator : "&&", left : self.condition, right : self.body.body }).optimize(compressor) }); } if (self.body instanceof AST_EmptyStatement && self.alternative && self.alternative instanceof AST_SimpleStatement) { return make_node(AST_SimpleStatement, self, { body: make_node(AST_Binary, self, { operator : "||", left : self.condition, right : self.alternative.body }).optimize(compressor) }); } if (self.body instanceof AST_Exit && self.alternative instanceof AST_Exit && self.body.TYPE == self.alternative.TYPE) { return make_node(self.body.CTOR, self, { value: make_node(AST_Conditional, self, { condition : self.condition, consequent : self.body.value, alternative : self.alternative.value }).optimize(compressor) }); } return self; }); SQUEEZE(AST_Switch, function(self, compressor){ self = self.clone(); self.expression = self.expression.squeeze(compressor); self.body = self.body.squeeze(compressor); return self; }); SQUEEZE(AST_Case, function(self, compressor){ self = self.clone(); self.expression = self.expression.squeeze(compressor); self.body = tighten_body(self.body, compressor); return self; }); SQUEEZE(AST_Try, function(self, compressor){ self = self.clone(); self.btry = self.btry.squeeze(compressor); if (self.bcatch) self.bcatch = self.bcatch.squeeze(compressor); if (self.bfinally) self.bfinally = self.bfinally.squeeze(compressor); return self; }); AST_Definitions.DEFMETHOD("remove_initializers", function(){ this.definitions = this.definitions.map(function(def){ var def = def.clone(); def.value = null; return def; }); }); SQUEEZE(AST_Definitions, function(self, compressor){ self = self.clone(); self.definitions = do_list(self.definitions, compressor); return self; }); SQUEEZE(AST_VarDef, function(self, compressor){ self = self.clone(); if (self.value) self.value = self.value.squeeze(compressor); return self; }); SQUEEZE(AST_Lambda, function(self, compressor){ self = self.clone(); self.hoist_declarations(compressor); if (self.name) self.name = self.name.squeeze(compressor); self.argnames = do_list(self.argnames, compressor); self.body = self.body.squeeze(compressor); return self; }); SQUEEZE(AST_Call, function(self, compressor){ self = self.clone(); self.expression = self.expression.squeeze(compressor); self.args = do_list(self.args, compressor); return self; }); SQUEEZE(AST_Seq, function(self, compressor){ self = self.clone(); self.first = self.first.squeeze(compressor); self.second = self.second.squeeze(compressor); return self; }); SQUEEZE(AST_Dot, function(self, compressor){ self = self.clone(); self.expression = self.expression.squeeze(compressor); return self; }); SQUEEZE(AST_Sub, function(self, compressor){ self = self.clone(); self.expression = self.expression.squeeze(compressor); var prop = self.property = self.property.squeeze(compressor); if (prop instanceof AST_String && compressor.option("properties")) { prop = prop.getValue(); if (is_identifier(prop)) { self = new AST_Dot(self); self.property = prop; } } return self; }); SQUEEZE(AST_Unary, function(self, compressor){ self = self.clone(); self.expression = self.expression.squeeze(compressor); return self; }); SQUEEZE(AST_UnaryPrefix, function(self, compressor){ // need to determine the context before cloning the node var bool = compressor.in_boolean_context(); self = self.clone(); var e = self.expression = self.expression.squeeze(compressor); if (compressor.option("booleans") && bool) { switch (self.operator) { case "!": if (e instanceof AST_UnaryPrefix && e.operator == "!") { // !!foo ==> foo, if we're in boolean context return e.expression; } break; case "typeof": // typeof always returns a non-empty string, thus it's // always true in booleans AST_Node.warn("Boolean expression always true [{line},{col}]", self.start); return make_node(AST_True, self).optimize(compressor); } } return self.evaluate(compressor)[0]; }); SQUEEZE(AST_Binary, function(self, compressor){ self = self.clone(); self.left = self.left.squeeze(compressor); self.right = self.right.squeeze(compressor); return self.optimize(compressor); }); AST_Binary.DEFMETHOD("optimize", function(compressor){ if (compressor.option("booleans") && compressor.in_boolean_context()) switch (this.operator) { case "&&": var ll = this.left.evaluate(compressor), left = ll[0]; var rr = this.right.evaluate(compressor), right = rr[0]; if ((ll.length == 2 && !ll[1]) || (rr.length == 2 && !rr[1])) { AST_Node.warn("Boolean && always false [{line},{col}]", this.start); return make_node(AST_False, this).optimize(compressor); } if (ll.length == 2 && ll[1]) { return rr[0]; } if (rr.length == 2 && rr[1]) { return ll[0]; } break; case "||": var ll = this.left.evaluate(compressor), left = ll[0]; var rr = this.right.evaluate(compressor), right = rr[0]; if ((ll.length == 2 && ll[1]) || (rr.length == 2 && rr[1])) { AST_Node.warn("Boolean || always true [{line},{col}]", this.start); return make_node(AST_True, this).optimize(compressor); } if (ll.length == 2 && !ll[1]) { return rr[0]; } if (rr.length == 2 && !rr[1]) { return ll[0]; } break; case "+": var ll = this.left.evaluate(compressor), left = ll[0]; var rr = this.right.evaluate(compressor), right = rr[0]; if ((ll.length == 2 && ll[0] instanceof AST_String && ll[1]) || (rr.length == 2 && rr[0] instanceof AST_String && rr[1])) { AST_Node.warn("+ in boolean context always true [{line},{col}]", this.start); return make_node(AST_True, this).optimize(compressor); } break; } return this.evaluate(compressor)[0]; }); SQUEEZE(AST_Assign, function(self, compressor){ self = self.clone(); self.left = self.left.squeeze(compressor); self.right = self.right.squeeze(compressor); return self; }); SQUEEZE(AST_Conditional, function(self, compressor){ self = self.clone(); self.condition = self.condition.squeeze(compressor); self.consequent = self.consequent.squeeze(compressor); self.alternative = self.alternative.squeeze(compressor); return self.optimize(compressor); }); AST_Conditional.DEFMETHOD("optimize", function(compressor){ var self = this; if (!compressor.option("conditionals")) return self; var cond = self.condition.evaluate(compressor); if (cond.length == 2) { if (cond[1]) { AST_Node.warn("Condition always true [{line},{col}]", self.start); return self.consequent; } else { AST_Node.warn("Condition always false [{line},{col}]", self.start); return self.alternative; } } var rev = self.clone(); rev.condition = cond[0].negate(compressor); var tmp = rev.consequent; rev.consequent = rev.alternative; rev.alternative = tmp; return best_of(self, rev); }); SQUEEZE(AST_Array, function(self, compressor){ self = self.clone(); self.elements = do_list(self.elements, compressor); return self; }); SQUEEZE(AST_Object, function(self, compressor){ self = self.clone(); self.properties = do_list(self.properties, compressor); return self; }); SQUEEZE(AST_ObjectProperty, function(self, compressor){ self = self.clone(); self.value = self.value.squeeze(compressor); return self; }); SQUEEZE(AST_True, function(self, compressor){ return self.optimize(compressor); }); AST_True.DEFMETHOD("optimize", function(compressor){ if (compressor.option("booleans")) return make_node(AST_UnaryPrefix, this, { operator: "!", expression: make_node(AST_Number, this, { value: 0 }) }); return this; }); SQUEEZE(AST_False, function(self, compressor){ return self.optimize(compressor); }); AST_False.DEFMETHOD("optimize", function(compressor){ if (compressor.option("booleans")) return make_node(AST_UnaryPrefix, this, { operator: "!", expression: make_node(AST_Number, this, { value: 1 }) }); return this; }); })();