fix various corner cases (#3123)

This commit is contained in:
Alex Lam S.L
2018-05-05 13:17:50 +08:00
committed by GitHub
parent d835c72c80
commit f37b91879f
3 changed files with 84 additions and 20 deletions

View File

@@ -355,20 +355,18 @@ merge(Compressor.prototype, {
} else {
def.fixed = false;
}
if (def.init instanceof AST_Defun && !all(def.references, same_defun_scope)) {
if (def.init instanceof AST_Defun && !all(def.references, function(ref) {
var scope = ref.scope;
do {
if (def.scope === scope) return true;
} while (scope instanceof AST_Function && (scope = scope.parent_scope));
})) {
tw.defun_ids[def.id] = undefined;
}
def.recursive_refs = 0;
def.references = [];
def.should_replace = undefined;
def.single_use = undefined;
function same_defun_scope(ref) {
var scope = ref.scope;
do {
if (def.scope === scope) return true;
} while (scope instanceof AST_Function && (scope = scope.parent_scope));
}
}
function reset_variables(tw, compressor, scope) {
@@ -384,11 +382,11 @@ merge(Compressor.prototype, {
});
}
function walk_defun(tw, def) {
if (def.id in tw.defun_ids) return;
function mark_defun(tw, def) {
if (def.id in tw.defun_ids) return def.fixed;
if (!tw.in_loop) {
tw.defun_ids[def.id] = true;
def.fixed.walk(tw);
tw.defun_ids[def.id] = tw.safe_ids;
return def.fixed;
} else if (tw.defun_ids[def.id] !== false) {
tw.defun_ids[def.id] = undefined;
}
@@ -397,7 +395,7 @@ merge(Compressor.prototype, {
function walk_defuns(tw, scope) {
scope.functions.each(function(def) {
if (def.init instanceof AST_Defun && tw.defun_ids[def.id] === undefined) {
tw.defun_ids[def.id] = true;
tw.defun_ids[def.id] = tw.safe_ids;
def.init.walk(tw);
}
});
@@ -536,11 +534,14 @@ merge(Compressor.prototype, {
return true;
});
def(AST_Call, function(tw, descend) {
descend();
var exp = this.expression;
if (exp instanceof AST_SymbolRef && exp.fixed_value() instanceof AST_Defun) {
walk_defun(tw, exp.definition());
}
if (!(exp instanceof AST_SymbolRef)) return;
var def = exp.definition();
if (!(def.fixed instanceof AST_Defun)) return;
var defun = mark_defun(tw, def);
if (!defun) return;
descend();
defun.walk(tw);
return true;
});
def(AST_Case, function(tw) {
@@ -570,7 +571,7 @@ merge(Compressor.prototype, {
});
def(AST_Defun, function(tw, descend, compressor) {
var id = this.name.definition().id;
if (!tw.defun_ids[id]) return true;
if (tw.defun_ids[id] !== tw.safe_ids) return true;
tw.defun_ids[id] = false;
this.inlined = false;
push(tw);
@@ -707,7 +708,8 @@ merge(Compressor.prototype, {
var parent;
if (d.fixed instanceof AST_Defun
&& !((parent = tw.parent()) instanceof AST_Call && parent.expression === this)) {
walk_defun(tw, d);
var defun = mark_defun(tw, d);
if (defun) defun.walk(tw);
}
});
def(AST_Toplevel, function(tw, descend, compressor) {
@@ -1121,7 +1123,7 @@ merge(Compressor.prototype, {
if (is_lhs(node, multi_replacer.parent())) return node;
def.replaced++;
value_def.replaced--;
return candidate.value;
return candidate.value.clone();
}
// Skip (non-executed) functions and (leading) default case in switch statements
if (node instanceof AST_Default || node instanceof AST_Scope) return node;