forked from zhurui/management
601 lines
20 KiB
JavaScript
601 lines
20 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 SymbolDef(scope, orig, init) {
|
|
this.name = orig.name;
|
|
this.orig = [ orig ];
|
|
this.init = init;
|
|
this.eliminated = 0;
|
|
this.scope = scope;
|
|
this.references = [];
|
|
this.replaced = 0;
|
|
this.global = false;
|
|
this.mangled_name = null;
|
|
this.undeclared = false;
|
|
this.id = SymbolDef.next_id++;
|
|
}
|
|
|
|
SymbolDef.next_id = 1;
|
|
|
|
SymbolDef.prototype = {
|
|
unmangleable: function(options) {
|
|
if (!options) options = {};
|
|
|
|
return this.global && !options.toplevel
|
|
|| this.undeclared
|
|
|| !options.eval && this.scope.pinned()
|
|
|| options.keep_fnames
|
|
&& (this.orig[0] instanceof AST_SymbolLambda
|
|
|| this.orig[0] instanceof AST_SymbolDefun);
|
|
},
|
|
mangle: function(options) {
|
|
var cache = options.cache && options.cache.props;
|
|
if (this.global && cache && cache.has(this.name)) {
|
|
this.mangled_name = cache.get(this.name);
|
|
} else if (!this.mangled_name && !this.unmangleable(options)) {
|
|
var def;
|
|
if (def = this.redefined()) {
|
|
this.mangled_name = def.mangled_name || def.name;
|
|
} else {
|
|
this.mangled_name = next_mangled_name(this.scope, options, this);
|
|
}
|
|
if (this.global && cache) {
|
|
cache.set(this.name, this.mangled_name);
|
|
}
|
|
}
|
|
},
|
|
redefined: function() {
|
|
return this.defun && this.defun.variables.get(this.name);
|
|
}
|
|
};
|
|
|
|
AST_Toplevel.DEFMETHOD("figure_out_scope", function(options) {
|
|
options = defaults(options, {
|
|
cache: null,
|
|
ie8: false,
|
|
});
|
|
|
|
// pass 1: setup scope chaining and handle definitions
|
|
var self = this;
|
|
var scope = self.parent_scope = null;
|
|
var defun = null;
|
|
var tw = new TreeWalker(function(node, descend) {
|
|
if (node instanceof AST_Catch) {
|
|
var save_scope = scope;
|
|
scope = new AST_Scope(node);
|
|
scope.init_scope_vars(save_scope);
|
|
descend();
|
|
scope = save_scope;
|
|
return true;
|
|
}
|
|
if (node instanceof AST_Scope) {
|
|
node.init_scope_vars(scope);
|
|
var save_scope = scope;
|
|
var save_defun = defun;
|
|
defun = scope = node;
|
|
descend();
|
|
scope = save_scope;
|
|
defun = save_defun;
|
|
return true;
|
|
}
|
|
if (node instanceof AST_With) {
|
|
for (var s = scope; s; s = s.parent_scope) s.uses_with = true;
|
|
return;
|
|
}
|
|
if (node instanceof AST_Symbol) {
|
|
node.scope = scope;
|
|
}
|
|
if (node instanceof AST_Label) {
|
|
node.thedef = node;
|
|
node.references = [];
|
|
}
|
|
if (node instanceof AST_SymbolDefun) {
|
|
// This should be defined in the parent scope, as we encounter the
|
|
// AST_Defun node before getting to its AST_Symbol.
|
|
(node.scope = defun.parent_scope.resolve()).def_function(node, defun);
|
|
} else if (node instanceof AST_SymbolLambda) {
|
|
var def = defun.def_function(node, node.name == "arguments" ? undefined : defun);
|
|
if (options.ie8) def.defun = defun.parent_scope.resolve();
|
|
} else if (node instanceof AST_SymbolVar) {
|
|
defun.def_variable(node, node.TYPE == "SymbolVar" ? null : undefined);
|
|
if (defun !== scope) {
|
|
node.mark_enclosed(options);
|
|
var def = scope.find_variable(node);
|
|
if (node.thedef !== def) {
|
|
node.thedef = def;
|
|
}
|
|
node.reference(options);
|
|
}
|
|
} else if (node instanceof AST_SymbolCatch) {
|
|
scope.def_variable(node).defun = defun;
|
|
}
|
|
});
|
|
self.walk(tw);
|
|
|
|
// pass 2: find back references and eval
|
|
self.globals = new Dictionary();
|
|
var tw = new TreeWalker(function(node) {
|
|
if (node instanceof AST_LoopControl) {
|
|
if (node.label) node.label.thedef.references.push(node);
|
|
return true;
|
|
}
|
|
if (node instanceof AST_SymbolRef) {
|
|
var name = node.name;
|
|
if (name == "eval" && tw.parent() instanceof AST_Call) {
|
|
for (var s = node.scope; s && !s.uses_eval; s = s.parent_scope) {
|
|
s.uses_eval = true;
|
|
}
|
|
}
|
|
var sym = node.scope.find_variable(name);
|
|
if (!sym) {
|
|
sym = self.def_global(node);
|
|
} else if (sym.scope instanceof AST_Lambda && name == "arguments") {
|
|
sym.scope.uses_arguments = true;
|
|
}
|
|
node.thedef = sym;
|
|
node.reference(options);
|
|
return true;
|
|
}
|
|
// ensure mangling works if catch reuses a scope variable
|
|
if (node instanceof AST_SymbolCatch) {
|
|
var def = node.definition().redefined();
|
|
if (def) for (var s = node.scope; s; s = s.parent_scope) {
|
|
push_uniq(s.enclosed, def);
|
|
if (s === def.scope) break;
|
|
}
|
|
return true;
|
|
}
|
|
});
|
|
self.walk(tw);
|
|
|
|
// pass 3: fix up any scoping issue with IE8
|
|
if (options.ie8) self.walk(new TreeWalker(function(node) {
|
|
if (node instanceof AST_SymbolCatch) {
|
|
redefine(node, node.thedef.defun);
|
|
return true;
|
|
}
|
|
if (node instanceof AST_SymbolLambda) {
|
|
var def = node.thedef;
|
|
if (def.orig.length == 1) {
|
|
redefine(node, node.scope.parent_scope);
|
|
node.thedef.init = def.init;
|
|
}
|
|
return true;
|
|
}
|
|
}));
|
|
|
|
function redefine(node, scope) {
|
|
var name = node.name;
|
|
var refs = node.thedef.references;
|
|
var def = scope.find_variable(name) || self.globals.get(name) || scope.def_variable(node);
|
|
refs.forEach(function(ref) {
|
|
ref.thedef = def;
|
|
ref.reference(options);
|
|
});
|
|
node.thedef = def;
|
|
node.reference(options);
|
|
}
|
|
});
|
|
|
|
AST_Toplevel.DEFMETHOD("def_global", function(node) {
|
|
var globals = this.globals, name = node.name;
|
|
if (globals.has(name)) {
|
|
return globals.get(name);
|
|
} else {
|
|
var g = new SymbolDef(this, node);
|
|
g.undeclared = true;
|
|
g.global = true;
|
|
globals.set(name, g);
|
|
return g;
|
|
}
|
|
});
|
|
|
|
AST_Scope.DEFMETHOD("init_scope_vars", function(parent_scope) {
|
|
this.variables = new Dictionary(); // map name to AST_SymbolVar (variables defined in this scope; includes functions)
|
|
this.functions = new Dictionary(); // map name to AST_SymbolDefun (functions defined in this scope)
|
|
this.uses_with = false; // will be set to true if this or some nested scope uses the `with` statement
|
|
this.uses_eval = false; // will be set to true if this or nested scope uses the global `eval`
|
|
this.parent_scope = parent_scope; // the parent scope
|
|
this.enclosed = []; // a list of variables from this or outer scope(s) that are referenced from this or inner scopes
|
|
this.cname = -1; // the current index for mangling functions/variables
|
|
});
|
|
|
|
AST_Lambda.DEFMETHOD("init_scope_vars", function() {
|
|
AST_Scope.prototype.init_scope_vars.apply(this, arguments);
|
|
this.uses_arguments = false;
|
|
this.def_variable(new AST_SymbolFunarg({
|
|
name: "arguments",
|
|
start: this.start,
|
|
end: this.end
|
|
}));
|
|
});
|
|
|
|
AST_Symbol.DEFMETHOD("mark_enclosed", function(options) {
|
|
var def = this.definition();
|
|
for (var s = this.scope; s; s = s.parent_scope) {
|
|
push_uniq(s.enclosed, def);
|
|
if (options.keep_fnames) {
|
|
s.functions.each(function(d) {
|
|
push_uniq(def.scope.enclosed, d);
|
|
});
|
|
}
|
|
if (s === def.scope) break;
|
|
}
|
|
});
|
|
|
|
AST_Symbol.DEFMETHOD("reference", function(options) {
|
|
this.definition().references.push(this);
|
|
this.mark_enclosed(options);
|
|
});
|
|
|
|
AST_Scope.DEFMETHOD("find_variable", function(name) {
|
|
if (name instanceof AST_Symbol) name = name.name;
|
|
return this.variables.get(name)
|
|
|| (this.parent_scope && this.parent_scope.find_variable(name));
|
|
});
|
|
|
|
AST_Scope.DEFMETHOD("def_function", function(symbol, init) {
|
|
var def = this.def_variable(symbol, init);
|
|
if (!def.init || def.init instanceof AST_Defun) def.init = init;
|
|
this.functions.set(symbol.name, def);
|
|
return def;
|
|
});
|
|
|
|
AST_Scope.DEFMETHOD("def_variable", function(symbol, init) {
|
|
var def = this.variables.get(symbol.name);
|
|
if (def) {
|
|
def.orig.push(symbol);
|
|
if (def.init && (def.scope !== symbol.scope || def.init instanceof AST_Function)) {
|
|
def.init = init;
|
|
}
|
|
} else {
|
|
def = new SymbolDef(this, symbol, init);
|
|
this.variables.set(symbol.name, def);
|
|
def.global = !this.parent_scope;
|
|
}
|
|
return symbol.thedef = def;
|
|
});
|
|
|
|
AST_Lambda.DEFMETHOD("resolve", return_this);
|
|
AST_Scope.DEFMETHOD("resolve", function() {
|
|
return this.parent_scope;
|
|
});
|
|
AST_Toplevel.DEFMETHOD("resolve", return_this);
|
|
|
|
function names_in_use(scope, options) {
|
|
var names = scope.names_in_use;
|
|
if (!names) {
|
|
scope.names_in_use = names = Object.create(scope.mangled_names || null);
|
|
scope.cname_holes = [];
|
|
var cache = options.cache && options.cache.props;
|
|
scope.enclosed.forEach(function(def) {
|
|
if (def.unmangleable(options)) names[def.name] = true;
|
|
if (def.global && cache && cache.has(def.name)) {
|
|
names[cache.get(def.name)] = true;
|
|
}
|
|
});
|
|
}
|
|
return names;
|
|
}
|
|
|
|
function next_mangled_name(scope, options, def) {
|
|
var in_use = names_in_use(scope, options);
|
|
var holes = scope.cname_holes;
|
|
var names = Object.create(null);
|
|
var scopes = [ scope ];
|
|
def.references.forEach(function(sym) {
|
|
var scope = sym.scope;
|
|
do {
|
|
if (scopes.indexOf(scope) < 0) {
|
|
for (var name in names_in_use(scope, options)) {
|
|
names[name] = true;
|
|
}
|
|
scopes.push(scope);
|
|
} else break;
|
|
} while (scope = scope.parent_scope);
|
|
});
|
|
var name;
|
|
for (var i = 0, len = holes.length; i < len; i++) {
|
|
name = base54(holes[i]);
|
|
if (names[name]) continue;
|
|
holes.splice(i, 1);
|
|
scope.names_in_use[name] = true;
|
|
return name;
|
|
}
|
|
while (true) {
|
|
name = base54(++scope.cname);
|
|
if (in_use[name] || !is_identifier(name) || options.reserved.has[name]) continue;
|
|
if (!names[name]) break;
|
|
holes.push(scope.cname);
|
|
}
|
|
scope.names_in_use[name] = true;
|
|
return name;
|
|
}
|
|
|
|
AST_Symbol.DEFMETHOD("unmangleable", function(options) {
|
|
var def = this.definition();
|
|
return !def || def.unmangleable(options);
|
|
});
|
|
|
|
// labels are always mangleable
|
|
AST_Label.DEFMETHOD("unmangleable", return_false);
|
|
|
|
AST_Symbol.DEFMETHOD("unreferenced", function() {
|
|
return !this.definition().references.length && !this.scope.pinned();
|
|
});
|
|
|
|
AST_Symbol.DEFMETHOD("definition", function() {
|
|
return this.thedef;
|
|
});
|
|
|
|
AST_Symbol.DEFMETHOD("global", function() {
|
|
return this.definition().global;
|
|
});
|
|
|
|
function _default_mangler_options(options) {
|
|
options = defaults(options, {
|
|
eval : false,
|
|
ie8 : false,
|
|
keep_fnames : false,
|
|
reserved : [],
|
|
toplevel : false,
|
|
});
|
|
if (!Array.isArray(options.reserved)) options.reserved = [];
|
|
// Never mangle arguments
|
|
push_uniq(options.reserved, "arguments");
|
|
options.reserved.has = makePredicate(options.reserved);
|
|
return options;
|
|
}
|
|
|
|
AST_Toplevel.DEFMETHOD("mangle_names", function(options) {
|
|
options = _default_mangler_options(options);
|
|
|
|
// We only need to mangle declaration nodes. Special logic wired
|
|
// into the code generator will display the mangled name if it's
|
|
// present (and for AST_SymbolRef-s it'll use the mangled name of
|
|
// the AST_SymbolDeclaration that it points to).
|
|
var lname = -1;
|
|
|
|
if (options.cache && options.cache.props) {
|
|
var mangled_names = this.mangled_names = Object.create(null);
|
|
options.cache.props.each(function(mangled_name) {
|
|
mangled_names[mangled_name] = true;
|
|
});
|
|
}
|
|
|
|
var redefined = [];
|
|
var tw = new TreeWalker(function(node, descend) {
|
|
if (node instanceof AST_LabeledStatement) {
|
|
// lname is incremented when we get to the AST_Label
|
|
var save_nesting = lname;
|
|
descend();
|
|
lname = save_nesting;
|
|
return true;
|
|
}
|
|
if (node instanceof AST_Scope) {
|
|
descend();
|
|
if (options.cache && node instanceof AST_Toplevel) {
|
|
node.globals.each(mangle);
|
|
}
|
|
node.variables.each(mangle);
|
|
return true;
|
|
}
|
|
if (node instanceof AST_Label) {
|
|
var name;
|
|
do {
|
|
name = base54(++lname);
|
|
} while (!is_identifier(name));
|
|
node.mangled_name = name;
|
|
return true;
|
|
}
|
|
if (!options.ie8 && node instanceof AST_Catch) {
|
|
var def = node.argname.definition();
|
|
var redef = def.redefined();
|
|
if (redef) {
|
|
redefined.push(def);
|
|
reference(node.argname);
|
|
def.references.forEach(reference);
|
|
}
|
|
descend();
|
|
if (!redef) mangle(def);
|
|
return true;
|
|
}
|
|
|
|
function reference(sym) {
|
|
sym.thedef = redef;
|
|
sym.reference(options);
|
|
sym.thedef = def;
|
|
}
|
|
});
|
|
this.walk(tw);
|
|
redefined.forEach(mangle);
|
|
|
|
function mangle(def) {
|
|
if (options.reserved.has[def.name]) return;
|
|
def.mangle(options);
|
|
}
|
|
});
|
|
|
|
AST_Toplevel.DEFMETHOD("find_colliding_names", function(options) {
|
|
var cache = options.cache && options.cache.props;
|
|
var avoid = Object.create(null);
|
|
options.reserved.forEach(to_avoid);
|
|
this.globals.each(add_def);
|
|
this.walk(new TreeWalker(function(node) {
|
|
if (node instanceof AST_Scope) node.variables.each(add_def);
|
|
if (node instanceof AST_SymbolCatch) add_def(node.definition());
|
|
}));
|
|
return avoid;
|
|
|
|
function to_avoid(name) {
|
|
avoid[name] = true;
|
|
}
|
|
|
|
function add_def(def) {
|
|
var name = def.name;
|
|
if (def.global && cache && cache.has(name)) name = cache.get(name);
|
|
else if (!def.unmangleable(options)) return;
|
|
to_avoid(name);
|
|
}
|
|
});
|
|
|
|
AST_Toplevel.DEFMETHOD("expand_names", function(options) {
|
|
base54.reset();
|
|
base54.sort();
|
|
options = _default_mangler_options(options);
|
|
var avoid = this.find_colliding_names(options);
|
|
var cname = 0;
|
|
this.globals.each(rename);
|
|
this.walk(new TreeWalker(function(node) {
|
|
if (node instanceof AST_Scope) node.variables.each(rename);
|
|
if (node instanceof AST_SymbolCatch) rename(node.definition());
|
|
}));
|
|
|
|
function next_name() {
|
|
var name;
|
|
do {
|
|
name = base54(cname++);
|
|
} while (avoid[name] || !is_identifier(name));
|
|
return name;
|
|
}
|
|
|
|
function rename(def) {
|
|
if (def.global && options.cache) return;
|
|
if (def.unmangleable(options)) return;
|
|
if (options.reserved.has[def.name]) return;
|
|
var d = def.redefined();
|
|
def.name = d ? d.name : next_name();
|
|
def.orig.forEach(function(sym) {
|
|
sym.name = def.name;
|
|
});
|
|
def.references.forEach(function(sym) {
|
|
sym.name = def.name;
|
|
});
|
|
}
|
|
});
|
|
|
|
AST_Node.DEFMETHOD("tail_node", return_this);
|
|
AST_Sequence.DEFMETHOD("tail_node", function() {
|
|
return this.expressions[this.expressions.length - 1];
|
|
});
|
|
|
|
AST_Toplevel.DEFMETHOD("compute_char_frequency", function(options) {
|
|
options = _default_mangler_options(options);
|
|
base54.reset();
|
|
try {
|
|
AST_Node.prototype.print = function(stream, force_parens) {
|
|
this._print(stream, force_parens);
|
|
if (this instanceof AST_Symbol && !this.unmangleable(options)) {
|
|
base54.consider(this.name, -1);
|
|
} else if (options.properties) {
|
|
if (this instanceof AST_Dot) {
|
|
base54.consider(this.property, -1);
|
|
} else if (this instanceof AST_Sub) {
|
|
skip_string(this.property);
|
|
}
|
|
}
|
|
};
|
|
base54.consider(this.print_to_string(), 1);
|
|
} finally {
|
|
AST_Node.prototype.print = AST_Node.prototype._print;
|
|
}
|
|
base54.sort();
|
|
|
|
function skip_string(node) {
|
|
if (node instanceof AST_String) {
|
|
base54.consider(node.value, -1);
|
|
} else if (node instanceof AST_Conditional) {
|
|
skip_string(node.consequent);
|
|
skip_string(node.alternative);
|
|
} else if (node instanceof AST_Sequence) {
|
|
skip_string(node.tail_node());
|
|
}
|
|
}
|
|
});
|
|
|
|
var base54 = (function() {
|
|
var freq = Object.create(null);
|
|
function init(chars) {
|
|
var array = [];
|
|
for (var i = 0, len = chars.length; i < len; i++) {
|
|
var ch = chars[i];
|
|
array.push(ch);
|
|
freq[ch] = -1e-2 * i;
|
|
}
|
|
return array;
|
|
}
|
|
var digits = init("0123456789");
|
|
var leading = init("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ$_");
|
|
var chars, frequency;
|
|
function reset() {
|
|
frequency = Object.create(freq);
|
|
}
|
|
base54.consider = function(str, delta) {
|
|
for (var i = str.length; --i >= 0;) {
|
|
frequency[str[i]] += delta;
|
|
}
|
|
};
|
|
function compare(a, b) {
|
|
return frequency[b] - frequency[a];
|
|
}
|
|
base54.sort = function() {
|
|
chars = leading.sort(compare).concat(digits.sort(compare));
|
|
};
|
|
base54.reset = reset;
|
|
reset();
|
|
function base54(num) {
|
|
var ret = "", base = 54;
|
|
num++;
|
|
do {
|
|
num--;
|
|
ret += chars[num % base];
|
|
num = Math.floor(num / base);
|
|
base = 64;
|
|
} while (num > 0);
|
|
return ret;
|
|
}
|
|
return base54;
|
|
})();
|