[79a0317] | 1 | /***********************************************************************
|
---|
| 2 |
|
---|
| 3 | A JavaScript tokenizer / parser / beautifier / compressor.
|
---|
| 4 | https://github.com/mishoo/UglifyJS2
|
---|
| 5 |
|
---|
| 6 | -------------------------------- (C) ---------------------------------
|
---|
| 7 |
|
---|
| 8 | Author: Mihai Bazon
|
---|
| 9 | <mihai.bazon@gmail.com>
|
---|
| 10 | http://mihai.bazon.net/blog
|
---|
| 11 |
|
---|
| 12 | Distributed under the BSD license:
|
---|
| 13 |
|
---|
| 14 | Copyright 2012 (c) Mihai Bazon <mihai.bazon@gmail.com>
|
---|
| 15 |
|
---|
| 16 | Redistribution and use in source and binary forms, with or without
|
---|
| 17 | modification, are permitted provided that the following conditions
|
---|
| 18 | are met:
|
---|
| 19 |
|
---|
| 20 | * Redistributions of source code must retain the above
|
---|
| 21 | copyright notice, this list of conditions and the following
|
---|
| 22 | disclaimer.
|
---|
| 23 |
|
---|
| 24 | * Redistributions in binary form must reproduce the above
|
---|
| 25 | copyright notice, this list of conditions and the following
|
---|
| 26 | disclaimer in the documentation and/or other materials
|
---|
| 27 | provided with the distribution.
|
---|
| 28 |
|
---|
| 29 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDER “AS IS” AND ANY
|
---|
| 30 | EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
---|
| 31 | IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
---|
| 32 | PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER BE
|
---|
| 33 | LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
|
---|
| 34 | OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
---|
| 35 | PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
---|
| 36 | PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
---|
| 37 | THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR
|
---|
| 38 | TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF
|
---|
| 39 | THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
|
---|
| 40 | SUCH DAMAGE.
|
---|
| 41 |
|
---|
| 42 | ***********************************************************************/
|
---|
| 43 |
|
---|
| 44 | import {
|
---|
| 45 | AST_Array,
|
---|
| 46 | AST_Assign,
|
---|
| 47 | AST_Block,
|
---|
| 48 | AST_Call,
|
---|
| 49 | AST_Catch,
|
---|
| 50 | AST_Class,
|
---|
| 51 | AST_ClassExpression,
|
---|
| 52 | AST_DefaultAssign,
|
---|
| 53 | AST_DefClass,
|
---|
| 54 | AST_Defun,
|
---|
| 55 | AST_Destructuring,
|
---|
| 56 | AST_EmptyStatement,
|
---|
| 57 | AST_Expansion,
|
---|
| 58 | AST_Export,
|
---|
| 59 | AST_Function,
|
---|
| 60 | AST_IterationStatement,
|
---|
| 61 | AST_Lambda,
|
---|
| 62 | AST_Node,
|
---|
| 63 | AST_Number,
|
---|
| 64 | AST_Object,
|
---|
| 65 | AST_ObjectKeyVal,
|
---|
| 66 | AST_PropAccess,
|
---|
| 67 | AST_Return,
|
---|
| 68 | AST_Scope,
|
---|
| 69 | AST_SimpleStatement,
|
---|
| 70 | AST_Statement,
|
---|
| 71 | AST_SymbolDefun,
|
---|
| 72 | AST_SymbolFunarg,
|
---|
| 73 | AST_SymbolLambda,
|
---|
| 74 | AST_SymbolRef,
|
---|
| 75 | AST_SymbolVar,
|
---|
| 76 | AST_This,
|
---|
| 77 | AST_Toplevel,
|
---|
| 78 | AST_UnaryPrefix,
|
---|
| 79 | AST_Undefined,
|
---|
| 80 | AST_Var,
|
---|
| 81 | AST_VarDef,
|
---|
| 82 |
|
---|
| 83 | walk,
|
---|
| 84 |
|
---|
| 85 | _INLINE,
|
---|
| 86 | _NOINLINE,
|
---|
| 87 | _PURE,
|
---|
| 88 | } from "../ast.js";
|
---|
| 89 | import { make_node, has_annotation } from "../utils/index.js";
|
---|
| 90 | import "../size.js";
|
---|
| 91 |
|
---|
| 92 | import "./evaluate.js";
|
---|
| 93 | import "./drop-side-effect-free.js";
|
---|
| 94 | import "./reduce-vars.js";
|
---|
| 95 | import {
|
---|
| 96 | SQUEEZED,
|
---|
| 97 | INLINED,
|
---|
| 98 | UNUSED,
|
---|
| 99 |
|
---|
| 100 | has_flag,
|
---|
| 101 | set_flag,
|
---|
| 102 | } from "./compressor-flags.js";
|
---|
| 103 | import {
|
---|
| 104 | make_sequence,
|
---|
| 105 | best_of,
|
---|
| 106 | make_node_from_constant,
|
---|
| 107 | identifier_atom,
|
---|
| 108 | is_empty,
|
---|
| 109 | is_func_expr,
|
---|
| 110 | is_iife_call,
|
---|
| 111 | is_reachable,
|
---|
| 112 | is_recursive_ref,
|
---|
| 113 | retain_top_func,
|
---|
| 114 | } from "./common.js";
|
---|
| 115 |
|
---|
| 116 | /**
|
---|
| 117 | * Module that contains the inlining logic.
|
---|
| 118 | *
|
---|
| 119 | * @module
|
---|
| 120 | *
|
---|
| 121 | * The stars of the show are `inline_into_symbolref` and `inline_into_call`.
|
---|
| 122 | */
|
---|
| 123 |
|
---|
| 124 | function within_array_or_object_literal(compressor) {
|
---|
| 125 | var node, level = 0;
|
---|
| 126 | while (node = compressor.parent(level++)) {
|
---|
| 127 | if (node instanceof AST_Statement) return false;
|
---|
| 128 | if (node instanceof AST_Array
|
---|
| 129 | || node instanceof AST_ObjectKeyVal
|
---|
| 130 | || node instanceof AST_Object) {
|
---|
| 131 | return true;
|
---|
| 132 | }
|
---|
| 133 | }
|
---|
| 134 | return false;
|
---|
| 135 | }
|
---|
| 136 |
|
---|
| 137 | function scope_encloses_variables_in_this_scope(scope, pulled_scope) {
|
---|
| 138 | for (const enclosed of pulled_scope.enclosed) {
|
---|
| 139 | if (pulled_scope.variables.has(enclosed.name)) {
|
---|
| 140 | continue;
|
---|
| 141 | }
|
---|
| 142 | const looked_up = scope.find_variable(enclosed.name);
|
---|
| 143 | if (looked_up) {
|
---|
| 144 | if (looked_up === enclosed) continue;
|
---|
| 145 | return true;
|
---|
| 146 | }
|
---|
| 147 | }
|
---|
| 148 | return false;
|
---|
| 149 | }
|
---|
| 150 |
|
---|
| 151 | /**
|
---|
| 152 | * An extra check function for `top_retain` option, compare the length of const identifier
|
---|
| 153 | * and init value length and return true if init value is longer than identifier. for example:
|
---|
| 154 | * ```
|
---|
| 155 | * // top_retain: ["example"]
|
---|
| 156 | * const example = 100
|
---|
| 157 | * ```
|
---|
| 158 | * it will return false because length of "100" is short than identifier "example".
|
---|
| 159 | */
|
---|
| 160 | function is_const_symbol_short_than_init_value(def, fixed_value) {
|
---|
| 161 | if (def.orig.length === 1 && fixed_value) {
|
---|
| 162 | const init_value_length = fixed_value.size();
|
---|
| 163 | const identifer_length = def.name.length;
|
---|
| 164 | return init_value_length > identifer_length;
|
---|
| 165 | }
|
---|
| 166 | return true;
|
---|
| 167 | }
|
---|
| 168 |
|
---|
| 169 | export function inline_into_symbolref(self, compressor) {
|
---|
| 170 | if (compressor.in_computed_key()) return self;
|
---|
| 171 |
|
---|
| 172 | const parent = compressor.parent();
|
---|
| 173 | const def = self.definition();
|
---|
| 174 | const nearest_scope = compressor.find_scope();
|
---|
| 175 | let fixed = self.fixed_value();
|
---|
| 176 | if (
|
---|
| 177 | compressor.top_retain &&
|
---|
| 178 | def.global &&
|
---|
| 179 | compressor.top_retain(def) &&
|
---|
| 180 | // when identifier is in top_retain option dose not mean we can always inline it.
|
---|
| 181 | // if identifier name is longer then init value, we can replace it.
|
---|
| 182 | is_const_symbol_short_than_init_value(def, fixed)
|
---|
| 183 | ) {
|
---|
| 184 | // keep it
|
---|
| 185 | def.fixed = false;
|
---|
| 186 | def.single_use = false;
|
---|
| 187 | return self;
|
---|
| 188 | }
|
---|
| 189 |
|
---|
| 190 | let single_use = def.single_use
|
---|
| 191 | && !(parent instanceof AST_Call
|
---|
| 192 | && (parent.is_callee_pure(compressor))
|
---|
| 193 | || has_annotation(parent, _NOINLINE))
|
---|
| 194 | && !(parent instanceof AST_Export
|
---|
| 195 | && fixed instanceof AST_Lambda
|
---|
| 196 | && fixed.name);
|
---|
| 197 |
|
---|
| 198 | if (single_use && fixed instanceof AST_Node) {
|
---|
| 199 | single_use =
|
---|
| 200 | !fixed.has_side_effects(compressor)
|
---|
| 201 | && !fixed.may_throw(compressor);
|
---|
| 202 | }
|
---|
| 203 |
|
---|
| 204 | if (fixed instanceof AST_Class && def.scope !== self.scope) {
|
---|
| 205 | return self;
|
---|
| 206 | }
|
---|
| 207 |
|
---|
| 208 | if (single_use && (fixed instanceof AST_Lambda || fixed instanceof AST_Class)) {
|
---|
| 209 | if (retain_top_func(fixed, compressor)) {
|
---|
| 210 | single_use = false;
|
---|
| 211 | } else if (def.scope !== self.scope
|
---|
| 212 | && (def.escaped == 1
|
---|
| 213 | || has_flag(fixed, INLINED)
|
---|
| 214 | || within_array_or_object_literal(compressor)
|
---|
| 215 | || !compressor.option("reduce_funcs"))) {
|
---|
| 216 | single_use = false;
|
---|
| 217 | } else if (is_recursive_ref(compressor, def)) {
|
---|
| 218 | single_use = false;
|
---|
| 219 | } else if (def.scope !== self.scope || def.orig[0] instanceof AST_SymbolFunarg) {
|
---|
| 220 | single_use = fixed.is_constant_expression(self.scope);
|
---|
| 221 | if (single_use == "f") {
|
---|
| 222 | var scope = self.scope;
|
---|
| 223 | do {
|
---|
| 224 | if (scope instanceof AST_Defun || is_func_expr(scope)) {
|
---|
| 225 | set_flag(scope, INLINED);
|
---|
| 226 | }
|
---|
| 227 | } while (scope = scope.parent_scope);
|
---|
| 228 | }
|
---|
| 229 | }
|
---|
| 230 | }
|
---|
| 231 |
|
---|
| 232 | if (single_use && (fixed instanceof AST_Lambda || fixed instanceof AST_Class)) {
|
---|
| 233 | single_use =
|
---|
| 234 | def.scope === self.scope
|
---|
| 235 | && !scope_encloses_variables_in_this_scope(nearest_scope, fixed)
|
---|
| 236 | || parent instanceof AST_Call
|
---|
| 237 | && parent.expression === self
|
---|
| 238 | && !scope_encloses_variables_in_this_scope(nearest_scope, fixed)
|
---|
| 239 | && !(fixed.name && fixed.name.definition().recursive_refs > 0);
|
---|
| 240 | }
|
---|
| 241 |
|
---|
| 242 | if (single_use && fixed) {
|
---|
| 243 | if (fixed instanceof AST_DefClass) {
|
---|
| 244 | set_flag(fixed, SQUEEZED);
|
---|
| 245 | fixed = make_node(AST_ClassExpression, fixed, fixed);
|
---|
| 246 | }
|
---|
| 247 | if (fixed instanceof AST_Defun) {
|
---|
| 248 | set_flag(fixed, SQUEEZED);
|
---|
| 249 | fixed = make_node(AST_Function, fixed, fixed);
|
---|
| 250 | }
|
---|
| 251 | if (def.recursive_refs > 0 && fixed.name instanceof AST_SymbolDefun) {
|
---|
| 252 | const defun_def = fixed.name.definition();
|
---|
| 253 | let lambda_def = fixed.variables.get(fixed.name.name);
|
---|
| 254 | let name = lambda_def && lambda_def.orig[0];
|
---|
| 255 | if (!(name instanceof AST_SymbolLambda)) {
|
---|
| 256 | name = make_node(AST_SymbolLambda, fixed.name, fixed.name);
|
---|
| 257 | name.scope = fixed;
|
---|
| 258 | fixed.name = name;
|
---|
| 259 | lambda_def = fixed.def_function(name);
|
---|
| 260 | }
|
---|
| 261 | walk(fixed, node => {
|
---|
| 262 | if (node instanceof AST_SymbolRef && node.definition() === defun_def) {
|
---|
| 263 | node.thedef = lambda_def;
|
---|
| 264 | lambda_def.references.push(node);
|
---|
| 265 | }
|
---|
| 266 | });
|
---|
| 267 | }
|
---|
| 268 | if (
|
---|
| 269 | (fixed instanceof AST_Lambda || fixed instanceof AST_Class)
|
---|
| 270 | && fixed.parent_scope !== nearest_scope
|
---|
| 271 | ) {
|
---|
| 272 | fixed = fixed.clone(true, compressor.get_toplevel());
|
---|
| 273 |
|
---|
| 274 | nearest_scope.add_child_scope(fixed);
|
---|
| 275 | }
|
---|
| 276 | return fixed.optimize(compressor);
|
---|
| 277 | }
|
---|
| 278 |
|
---|
| 279 | // multiple uses
|
---|
| 280 | if (fixed) {
|
---|
| 281 | let replace;
|
---|
| 282 |
|
---|
| 283 | if (fixed instanceof AST_This) {
|
---|
| 284 | if (!(def.orig[0] instanceof AST_SymbolFunarg)
|
---|
| 285 | && def.references.every((ref) =>
|
---|
| 286 | def.scope === ref.scope
|
---|
| 287 | )) {
|
---|
| 288 | replace = fixed;
|
---|
| 289 | }
|
---|
| 290 | } else {
|
---|
| 291 | var ev = fixed.evaluate(compressor);
|
---|
| 292 | if (
|
---|
| 293 | ev !== fixed
|
---|
| 294 | && (compressor.option("unsafe_regexp") || !(ev instanceof RegExp))
|
---|
| 295 | ) {
|
---|
| 296 | replace = make_node_from_constant(ev, fixed);
|
---|
| 297 | }
|
---|
| 298 | }
|
---|
| 299 |
|
---|
| 300 | if (replace) {
|
---|
| 301 | const name_length = self.size(compressor);
|
---|
| 302 | const replace_size = replace.size(compressor);
|
---|
| 303 |
|
---|
| 304 | let overhead = 0;
|
---|
| 305 | if (compressor.option("unused") && !compressor.exposed(def)) {
|
---|
| 306 | overhead =
|
---|
| 307 | (name_length + 2 + fixed.size(compressor)) /
|
---|
| 308 | (def.references.length - def.assignments);
|
---|
| 309 | }
|
---|
| 310 |
|
---|
| 311 | if (replace_size <= name_length + overhead) {
|
---|
| 312 | return replace;
|
---|
| 313 | }
|
---|
| 314 | }
|
---|
| 315 | }
|
---|
| 316 |
|
---|
| 317 | return self;
|
---|
| 318 | }
|
---|
| 319 |
|
---|
| 320 | export function inline_into_call(self, compressor) {
|
---|
| 321 | if (compressor.in_computed_key()) return self;
|
---|
| 322 |
|
---|
| 323 | var exp = self.expression;
|
---|
| 324 | var fn = exp;
|
---|
| 325 | var simple_args = self.args.every((arg) => !(arg instanceof AST_Expansion));
|
---|
| 326 |
|
---|
| 327 | if (compressor.option("reduce_vars")
|
---|
| 328 | && fn instanceof AST_SymbolRef
|
---|
| 329 | && !has_annotation(self, _NOINLINE)
|
---|
| 330 | ) {
|
---|
| 331 | const fixed = fn.fixed_value();
|
---|
| 332 |
|
---|
| 333 | if (
|
---|
| 334 | retain_top_func(fixed, compressor)
|
---|
| 335 | || !compressor.toplevel.funcs && exp.definition().global
|
---|
| 336 | ) {
|
---|
| 337 | return self;
|
---|
| 338 | }
|
---|
| 339 |
|
---|
| 340 | fn = fixed;
|
---|
| 341 | }
|
---|
| 342 |
|
---|
| 343 | var is_func = fn instanceof AST_Lambda;
|
---|
| 344 |
|
---|
| 345 | var stat = is_func && fn.body[0];
|
---|
| 346 | var is_regular_func = is_func && !fn.is_generator && !fn.async;
|
---|
| 347 | var can_inline = is_regular_func && compressor.option("inline") && !self.is_callee_pure(compressor);
|
---|
| 348 | if (can_inline && stat instanceof AST_Return) {
|
---|
| 349 | let returned = stat.value;
|
---|
| 350 | if (!returned || returned.is_constant_expression()) {
|
---|
| 351 | if (returned) {
|
---|
| 352 | returned = returned.clone(true);
|
---|
| 353 | } else {
|
---|
| 354 | returned = make_node(AST_Undefined, self);
|
---|
| 355 | }
|
---|
| 356 | const args = self.args.concat(returned);
|
---|
| 357 | return make_sequence(self, args).optimize(compressor);
|
---|
| 358 | }
|
---|
| 359 |
|
---|
| 360 | // optimize identity function
|
---|
| 361 | if (
|
---|
| 362 | fn.argnames.length === 1
|
---|
| 363 | && (fn.argnames[0] instanceof AST_SymbolFunarg)
|
---|
| 364 | && self.args.length < 2
|
---|
| 365 | && !(self.args[0] instanceof AST_Expansion)
|
---|
| 366 | && returned instanceof AST_SymbolRef
|
---|
| 367 | && returned.name === fn.argnames[0].name
|
---|
| 368 | ) {
|
---|
| 369 | const replacement =
|
---|
| 370 | (self.args[0] || make_node(AST_Undefined)).optimize(compressor);
|
---|
| 371 |
|
---|
| 372 | let parent;
|
---|
| 373 | if (
|
---|
| 374 | replacement instanceof AST_PropAccess
|
---|
| 375 | && (parent = compressor.parent()) instanceof AST_Call
|
---|
| 376 | && parent.expression === self
|
---|
| 377 | ) {
|
---|
| 378 | // identity function was being used to remove `this`, like in
|
---|
| 379 | //
|
---|
| 380 | // id(bag.no_this)(...)
|
---|
| 381 | //
|
---|
| 382 | // Replace with a larger but more effish (0, bag.no_this) wrapper.
|
---|
| 383 |
|
---|
| 384 | return make_sequence(self, [
|
---|
| 385 | make_node(AST_Number, self, { value: 0 }),
|
---|
| 386 | replacement
|
---|
| 387 | ]);
|
---|
| 388 | }
|
---|
| 389 | // replace call with first argument or undefined if none passed
|
---|
| 390 | return replacement;
|
---|
| 391 | }
|
---|
| 392 | }
|
---|
| 393 |
|
---|
| 394 | if (can_inline) {
|
---|
| 395 | var scope, in_loop, level = -1;
|
---|
| 396 | let def;
|
---|
| 397 | let returned_value;
|
---|
| 398 | let nearest_scope;
|
---|
| 399 | if (simple_args
|
---|
| 400 | && !fn.uses_arguments
|
---|
| 401 | && !(compressor.parent() instanceof AST_Class)
|
---|
| 402 | && !(fn.name && fn instanceof AST_Function)
|
---|
| 403 | && (returned_value = can_flatten_body(stat))
|
---|
| 404 | && (exp === fn
|
---|
| 405 | || has_annotation(self, _INLINE)
|
---|
| 406 | || compressor.option("unused")
|
---|
| 407 | && (def = exp.definition()).references.length == 1
|
---|
| 408 | && !is_recursive_ref(compressor, def)
|
---|
| 409 | && fn.is_constant_expression(exp.scope))
|
---|
| 410 | && !has_annotation(self, _PURE | _NOINLINE)
|
---|
| 411 | && !fn.contains_this()
|
---|
| 412 | && can_inject_symbols()
|
---|
| 413 | && (nearest_scope = compressor.find_scope())
|
---|
| 414 | && !scope_encloses_variables_in_this_scope(nearest_scope, fn)
|
---|
| 415 | && !(function in_default_assign() {
|
---|
| 416 | // Due to the fact function parameters have their own scope
|
---|
| 417 | // which can't use `var something` in the function body within,
|
---|
| 418 | // we simply don't inline into DefaultAssign.
|
---|
| 419 | let i = 0;
|
---|
| 420 | let p;
|
---|
| 421 | while ((p = compressor.parent(i++))) {
|
---|
| 422 | if (p instanceof AST_DefaultAssign) return true;
|
---|
| 423 | if (p instanceof AST_Block) break;
|
---|
| 424 | }
|
---|
| 425 | return false;
|
---|
| 426 | })()
|
---|
| 427 | && !(scope instanceof AST_Class)
|
---|
| 428 | ) {
|
---|
| 429 | set_flag(fn, SQUEEZED);
|
---|
| 430 | nearest_scope.add_child_scope(fn);
|
---|
| 431 | return make_sequence(self, flatten_fn(returned_value)).optimize(compressor);
|
---|
| 432 | }
|
---|
| 433 | }
|
---|
| 434 |
|
---|
| 435 | if (can_inline && has_annotation(self, _INLINE)) {
|
---|
| 436 | set_flag(fn, SQUEEZED);
|
---|
| 437 | fn = make_node(fn.CTOR === AST_Defun ? AST_Function : fn.CTOR, fn, fn);
|
---|
| 438 | fn = fn.clone(true);
|
---|
| 439 | fn.figure_out_scope({}, {
|
---|
| 440 | parent_scope: compressor.find_scope(),
|
---|
| 441 | toplevel: compressor.get_toplevel()
|
---|
| 442 | });
|
---|
| 443 |
|
---|
| 444 | return make_node(AST_Call, self, {
|
---|
| 445 | expression: fn,
|
---|
| 446 | args: self.args,
|
---|
| 447 | }).optimize(compressor);
|
---|
| 448 | }
|
---|
| 449 |
|
---|
| 450 | const can_drop_this_call = is_regular_func && compressor.option("side_effects") && fn.body.every(is_empty);
|
---|
| 451 | if (can_drop_this_call) {
|
---|
| 452 | var args = self.args.concat(make_node(AST_Undefined, self));
|
---|
| 453 | return make_sequence(self, args).optimize(compressor);
|
---|
| 454 | }
|
---|
| 455 |
|
---|
| 456 | if (compressor.option("negate_iife")
|
---|
| 457 | && compressor.parent() instanceof AST_SimpleStatement
|
---|
| 458 | && is_iife_call(self)) {
|
---|
| 459 | return self.negate(compressor, true);
|
---|
| 460 | }
|
---|
| 461 |
|
---|
| 462 | var ev = self.evaluate(compressor);
|
---|
| 463 | if (ev !== self) {
|
---|
| 464 | ev = make_node_from_constant(ev, self).optimize(compressor);
|
---|
| 465 | return best_of(compressor, ev, self);
|
---|
| 466 | }
|
---|
| 467 |
|
---|
| 468 | return self;
|
---|
| 469 |
|
---|
| 470 | function return_value(stat) {
|
---|
| 471 | if (!stat) return make_node(AST_Undefined, self);
|
---|
| 472 | if (stat instanceof AST_Return) {
|
---|
| 473 | if (!stat.value) return make_node(AST_Undefined, self);
|
---|
| 474 | return stat.value.clone(true);
|
---|
| 475 | }
|
---|
| 476 | if (stat instanceof AST_SimpleStatement) {
|
---|
| 477 | return make_node(AST_UnaryPrefix, stat, {
|
---|
| 478 | operator: "void",
|
---|
| 479 | expression: stat.body.clone(true)
|
---|
| 480 | });
|
---|
| 481 | }
|
---|
| 482 | }
|
---|
| 483 |
|
---|
| 484 | function can_flatten_body(stat) {
|
---|
| 485 | var body = fn.body;
|
---|
| 486 | var len = body.length;
|
---|
| 487 | if (compressor.option("inline") < 3) {
|
---|
| 488 | return len == 1 && return_value(stat);
|
---|
| 489 | }
|
---|
| 490 | stat = null;
|
---|
| 491 | for (var i = 0; i < len; i++) {
|
---|
| 492 | var line = body[i];
|
---|
| 493 | if (line instanceof AST_Var) {
|
---|
| 494 | if (stat && !line.definitions.every((var_def) =>
|
---|
| 495 | !var_def.value
|
---|
| 496 | )) {
|
---|
| 497 | return false;
|
---|
| 498 | }
|
---|
| 499 | } else if (stat) {
|
---|
| 500 | return false;
|
---|
| 501 | } else if (!(line instanceof AST_EmptyStatement)) {
|
---|
| 502 | stat = line;
|
---|
| 503 | }
|
---|
| 504 | }
|
---|
| 505 | return return_value(stat);
|
---|
| 506 | }
|
---|
| 507 |
|
---|
| 508 | function can_inject_args(block_scoped, safe_to_inject) {
|
---|
| 509 | for (var i = 0, len = fn.argnames.length; i < len; i++) {
|
---|
| 510 | var arg = fn.argnames[i];
|
---|
| 511 | if (arg instanceof AST_DefaultAssign) {
|
---|
| 512 | if (has_flag(arg.left, UNUSED)) continue;
|
---|
| 513 | return false;
|
---|
| 514 | }
|
---|
| 515 | if (arg instanceof AST_Destructuring) return false;
|
---|
| 516 | if (arg instanceof AST_Expansion) {
|
---|
| 517 | if (has_flag(arg.expression, UNUSED)) continue;
|
---|
| 518 | return false;
|
---|
| 519 | }
|
---|
| 520 | if (has_flag(arg, UNUSED)) continue;
|
---|
| 521 | if (!safe_to_inject
|
---|
| 522 | || block_scoped.has(arg.name)
|
---|
| 523 | || identifier_atom.has(arg.name)
|
---|
| 524 | || scope.conflicting_def(arg.name)) {
|
---|
| 525 | return false;
|
---|
| 526 | }
|
---|
| 527 | if (in_loop) in_loop.push(arg.definition());
|
---|
| 528 | }
|
---|
| 529 | return true;
|
---|
| 530 | }
|
---|
| 531 |
|
---|
| 532 | function can_inject_vars(block_scoped, safe_to_inject) {
|
---|
| 533 | var len = fn.body.length;
|
---|
| 534 | for (var i = 0; i < len; i++) {
|
---|
| 535 | var stat = fn.body[i];
|
---|
| 536 | if (!(stat instanceof AST_Var)) continue;
|
---|
| 537 | if (!safe_to_inject) return false;
|
---|
| 538 | for (var j = stat.definitions.length; --j >= 0;) {
|
---|
| 539 | var name = stat.definitions[j].name;
|
---|
| 540 | if (name instanceof AST_Destructuring
|
---|
| 541 | || block_scoped.has(name.name)
|
---|
| 542 | || identifier_atom.has(name.name)
|
---|
| 543 | || scope.conflicting_def(name.name)) {
|
---|
| 544 | return false;
|
---|
| 545 | }
|
---|
| 546 | if (in_loop) in_loop.push(name.definition());
|
---|
| 547 | }
|
---|
| 548 | }
|
---|
| 549 | return true;
|
---|
| 550 | }
|
---|
| 551 |
|
---|
| 552 | function can_inject_symbols() {
|
---|
| 553 | var block_scoped = new Set();
|
---|
| 554 | do {
|
---|
| 555 | scope = compressor.parent(++level);
|
---|
| 556 | if (scope.is_block_scope() && scope.block_scope) {
|
---|
| 557 | // TODO this is sometimes undefined during compression.
|
---|
| 558 | // But it should always have a value!
|
---|
| 559 | scope.block_scope.variables.forEach(function (variable) {
|
---|
| 560 | block_scoped.add(variable.name);
|
---|
| 561 | });
|
---|
| 562 | }
|
---|
| 563 | if (scope instanceof AST_Catch) {
|
---|
| 564 | // TODO can we delete? AST_Catch is a block scope.
|
---|
| 565 | if (scope.argname) {
|
---|
| 566 | block_scoped.add(scope.argname.name);
|
---|
| 567 | }
|
---|
| 568 | } else if (scope instanceof AST_IterationStatement) {
|
---|
| 569 | in_loop = [];
|
---|
| 570 | } else if (scope instanceof AST_SymbolRef) {
|
---|
| 571 | if (scope.fixed_value() instanceof AST_Scope) return false;
|
---|
| 572 | }
|
---|
| 573 | } while (!(scope instanceof AST_Scope));
|
---|
| 574 |
|
---|
| 575 | var safe_to_inject = !(scope instanceof AST_Toplevel) || compressor.toplevel.vars;
|
---|
| 576 | var inline = compressor.option("inline");
|
---|
| 577 | if (!can_inject_vars(block_scoped, inline >= 3 && safe_to_inject)) return false;
|
---|
| 578 | if (!can_inject_args(block_scoped, inline >= 2 && safe_to_inject)) return false;
|
---|
| 579 | return !in_loop || in_loop.length == 0 || !is_reachable(fn, in_loop);
|
---|
| 580 | }
|
---|
| 581 |
|
---|
| 582 | function append_var(decls, expressions, name, value) {
|
---|
| 583 | var def = name.definition();
|
---|
| 584 |
|
---|
| 585 | // Name already exists, only when a function argument had the same name
|
---|
| 586 | const already_appended = scope.variables.has(name.name);
|
---|
| 587 | if (!already_appended) {
|
---|
| 588 | scope.variables.set(name.name, def);
|
---|
| 589 | scope.enclosed.push(def);
|
---|
| 590 | decls.push(make_node(AST_VarDef, name, {
|
---|
| 591 | name: name,
|
---|
| 592 | value: null
|
---|
| 593 | }));
|
---|
| 594 | }
|
---|
| 595 |
|
---|
| 596 | var sym = make_node(AST_SymbolRef, name, name);
|
---|
| 597 | def.references.push(sym);
|
---|
| 598 | if (value) expressions.push(make_node(AST_Assign, self, {
|
---|
| 599 | operator: "=",
|
---|
| 600 | logical: false,
|
---|
| 601 | left: sym,
|
---|
| 602 | right: value.clone()
|
---|
| 603 | }));
|
---|
| 604 | }
|
---|
| 605 |
|
---|
| 606 | function flatten_args(decls, expressions) {
|
---|
| 607 | var len = fn.argnames.length;
|
---|
| 608 | for (var i = self.args.length; --i >= len;) {
|
---|
| 609 | expressions.push(self.args[i]);
|
---|
| 610 | }
|
---|
| 611 | for (i = len; --i >= 0;) {
|
---|
| 612 | var name = fn.argnames[i];
|
---|
| 613 | var value = self.args[i];
|
---|
| 614 | if (has_flag(name, UNUSED) || !name.name || scope.conflicting_def(name.name)) {
|
---|
| 615 | if (value) expressions.push(value);
|
---|
| 616 | } else {
|
---|
| 617 | var symbol = make_node(AST_SymbolVar, name, name);
|
---|
| 618 | name.definition().orig.push(symbol);
|
---|
| 619 | if (!value && in_loop) value = make_node(AST_Undefined, self);
|
---|
| 620 | append_var(decls, expressions, symbol, value);
|
---|
| 621 | }
|
---|
| 622 | }
|
---|
| 623 | decls.reverse();
|
---|
| 624 | expressions.reverse();
|
---|
| 625 | }
|
---|
| 626 |
|
---|
| 627 | function flatten_vars(decls, expressions) {
|
---|
| 628 | var pos = expressions.length;
|
---|
| 629 | for (var i = 0, lines = fn.body.length; i < lines; i++) {
|
---|
| 630 | var stat = fn.body[i];
|
---|
| 631 | if (!(stat instanceof AST_Var)) continue;
|
---|
| 632 | for (var j = 0, defs = stat.definitions.length; j < defs; j++) {
|
---|
| 633 | var var_def = stat.definitions[j];
|
---|
| 634 | var name = var_def.name;
|
---|
| 635 | append_var(decls, expressions, name, var_def.value);
|
---|
| 636 | if (in_loop && fn.argnames.every((argname) =>
|
---|
| 637 | argname.name != name.name
|
---|
| 638 | )) {
|
---|
| 639 | var def = fn.variables.get(name.name);
|
---|
| 640 | var sym = make_node(AST_SymbolRef, name, name);
|
---|
| 641 | def.references.push(sym);
|
---|
| 642 | expressions.splice(pos++, 0, make_node(AST_Assign, var_def, {
|
---|
| 643 | operator: "=",
|
---|
| 644 | logical: false,
|
---|
| 645 | left: sym,
|
---|
| 646 | right: make_node(AST_Undefined, name)
|
---|
| 647 | }));
|
---|
| 648 | }
|
---|
| 649 | }
|
---|
| 650 | }
|
---|
| 651 | }
|
---|
| 652 |
|
---|
| 653 | function flatten_fn(returned_value) {
|
---|
| 654 | var decls = [];
|
---|
| 655 | var expressions = [];
|
---|
| 656 | flatten_args(decls, expressions);
|
---|
| 657 | flatten_vars(decls, expressions);
|
---|
| 658 | expressions.push(returned_value);
|
---|
| 659 |
|
---|
| 660 | if (decls.length) {
|
---|
| 661 | const i = scope.body.indexOf(compressor.parent(level - 1)) + 1;
|
---|
| 662 | scope.body.splice(i, 0, make_node(AST_Var, fn, {
|
---|
| 663 | definitions: decls
|
---|
| 664 | }));
|
---|
| 665 | }
|
---|
| 666 |
|
---|
| 667 | return expressions.map(exp => exp.clone(true));
|
---|
| 668 | }
|
---|
| 669 | }
|
---|