123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378 |
- /***********************************************************************
- 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";
- /* global global, self */
- import {
- defaults,
- push_uniq,
- } from "./utils/index.js";
- import { base54 } from "./scope.js";
- import {
- AST_Binary,
- AST_Call,
- AST_ClassPrivateProperty,
- AST_Conditional,
- AST_Dot,
- AST_DotHash,
- AST_ObjectKeyVal,
- AST_ObjectProperty,
- AST_PrivateMethod,
- AST_PrivateGetter,
- AST_PrivateSetter,
- AST_PrivateIn,
- AST_Sequence,
- AST_String,
- AST_Sub,
- TreeTransformer,
- TreeWalker,
- } from "./ast.js";
- import { domprops } from "../tools/domprops.js";
- function find_builtins(reserved) {
- domprops.forEach(add);
- // Compatibility fix for some standard defined globals not defined on every js environment
- var new_globals = ["Symbol", "Map", "Promise", "Proxy", "Reflect", "Set", "WeakMap", "WeakSet"];
- var objects = {};
- var global_ref = typeof global === "object" ? global : self;
- new_globals.forEach(function (new_global) {
- objects[new_global] = global_ref[new_global] || function() {};
- });
- [
- "null",
- "true",
- "false",
- "NaN",
- "Infinity",
- "-Infinity",
- "undefined",
- ].forEach(add);
- [ Object, Array, Function, Number,
- String, Boolean, Error, Math,
- Date, RegExp, objects.Symbol, ArrayBuffer,
- DataView, decodeURI, decodeURIComponent,
- encodeURI, encodeURIComponent, eval, EvalError,
- Float32Array, Float64Array, Int8Array, Int16Array,
- Int32Array, isFinite, isNaN, JSON, objects.Map, parseFloat,
- parseInt, objects.Promise, objects.Proxy, RangeError, ReferenceError,
- objects.Reflect, objects.Set, SyntaxError, TypeError, Uint8Array,
- Uint8ClampedArray, Uint16Array, Uint32Array, URIError,
- objects.WeakMap, objects.WeakSet
- ].forEach(function(ctor) {
- Object.getOwnPropertyNames(ctor).map(add);
- if (ctor.prototype) {
- Object.getOwnPropertyNames(ctor.prototype).map(add);
- }
- });
- function add(name) {
- reserved.add(name);
- }
- }
- function reserve_quoted_keys(ast, reserved) {
- function add(name) {
- push_uniq(reserved, name);
- }
- ast.walk(new TreeWalker(function(node) {
- if (node instanceof AST_ObjectKeyVal && node.quote) {
- add(node.key);
- } else if (node instanceof AST_ObjectProperty && node.quote) {
- add(node.key.name);
- } else if (node instanceof AST_Sub) {
- addStrings(node.property, add);
- }
- }));
- }
- function addStrings(node, add) {
- node.walk(new TreeWalker(function(node) {
- if (node instanceof AST_Sequence) {
- addStrings(node.tail_node(), add);
- } else if (node instanceof AST_String) {
- add(node.value);
- } else if (node instanceof AST_Conditional) {
- addStrings(node.consequent, add);
- addStrings(node.alternative, add);
- }
- return true;
- }));
- }
- function mangle_private_properties(ast, options) {
- var cprivate = -1;
- var private_cache = new Map();
- var nth_identifier = options.nth_identifier || base54;
- ast = ast.transform(new TreeTransformer(function(node) {
- if (
- node instanceof AST_ClassPrivateProperty
- || node instanceof AST_PrivateMethod
- || node instanceof AST_PrivateGetter
- || node instanceof AST_PrivateSetter
- || node instanceof AST_PrivateIn
- ) {
- node.key.name = mangle_private(node.key.name);
- } else if (node instanceof AST_DotHash) {
- node.property = mangle_private(node.property);
- }
- }));
- return ast;
- function mangle_private(name) {
- let mangled = private_cache.get(name);
- if (!mangled) {
- mangled = nth_identifier.get(++cprivate);
- private_cache.set(name, mangled);
- }
- return mangled;
- }
- }
- function mangle_properties(ast, options) {
- options = defaults(options, {
- builtins: false,
- cache: null,
- debug: false,
- keep_quoted: false,
- nth_identifier: base54,
- only_cache: false,
- regex: null,
- reserved: null,
- undeclared: false,
- }, true);
- var nth_identifier = options.nth_identifier;
- var reserved_option = options.reserved;
- if (!Array.isArray(reserved_option)) reserved_option = [reserved_option];
- var reserved = new Set(reserved_option);
- if (!options.builtins) find_builtins(reserved);
- var cname = -1;
- var cache;
- if (options.cache) {
- cache = options.cache.props;
- } else {
- cache = new Map();
- }
- var regex = options.regex && new RegExp(options.regex);
- // note debug is either false (disabled), or a string of the debug suffix to use (enabled).
- // note debug may be enabled as an empty string, which is falsey. Also treat passing 'true'
- // the same as passing an empty string.
- var debug = options.debug !== false;
- var debug_name_suffix;
- if (debug) {
- debug_name_suffix = (options.debug === true ? "" : options.debug);
- }
- var names_to_mangle = new Set();
- var unmangleable = new Set();
- // Track each already-mangled name to prevent nth_identifier from generating
- // the same name.
- cache.forEach((mangled_name) => unmangleable.add(mangled_name));
- var keep_quoted = !!options.keep_quoted;
- // step 1: find candidates to mangle
- ast.walk(new TreeWalker(function(node) {
- if (
- node instanceof AST_ClassPrivateProperty
- || node instanceof AST_PrivateMethod
- || node instanceof AST_PrivateGetter
- || node instanceof AST_PrivateSetter
- || node instanceof AST_DotHash
- ) {
- // handled by mangle_private_properties
- } else if (node instanceof AST_ObjectKeyVal) {
- if (typeof node.key == "string" && (!keep_quoted || !node.quote)) {
- add(node.key);
- }
- } else if (node instanceof AST_ObjectProperty) {
- // setter or getter, since KeyVal is handled above
- if (!keep_quoted || !node.quote) {
- add(node.key.name);
- }
- } else if (node instanceof AST_Dot) {
- var declared = !!options.undeclared;
- if (!declared) {
- var root = node;
- while (root.expression) {
- root = root.expression;
- }
- declared = !(root.thedef && root.thedef.undeclared);
- }
- if (declared &&
- (!keep_quoted || !node.quote)) {
- add(node.property);
- }
- } else if (node instanceof AST_Sub) {
- if (!keep_quoted) {
- addStrings(node.property, add);
- }
- } else if (node instanceof AST_Call
- && node.expression.print_to_string() == "Object.defineProperty") {
- addStrings(node.args[1], add);
- } else if (node instanceof AST_Binary && node.operator === "in") {
- addStrings(node.left, add);
- }
- }));
- // step 2: transform the tree, renaming properties
- return ast.transform(new TreeTransformer(function(node) {
- if (
- node instanceof AST_ClassPrivateProperty
- || node instanceof AST_PrivateMethod
- || node instanceof AST_PrivateGetter
- || node instanceof AST_PrivateSetter
- || node instanceof AST_DotHash
- ) {
- // handled by mangle_private_properties
- } else if (node instanceof AST_ObjectKeyVal) {
- if (typeof node.key == "string" && (!keep_quoted || !node.quote)) {
- node.key = mangle(node.key);
- }
- } else if (node instanceof AST_ObjectProperty) {
- // setter, getter, method or class field
- if (!keep_quoted || !node.quote) {
- node.key.name = mangle(node.key.name);
- }
- } else if (node instanceof AST_Dot) {
- if (!keep_quoted || !node.quote) {
- node.property = mangle(node.property);
- }
- } else if (!keep_quoted && node instanceof AST_Sub) {
- node.property = mangleStrings(node.property);
- } else if (node instanceof AST_Call
- && node.expression.print_to_string() == "Object.defineProperty") {
- node.args[1] = mangleStrings(node.args[1]);
- } else if (node instanceof AST_Binary && node.operator === "in") {
- node.left = mangleStrings(node.left);
- }
- }));
- // only function declarations after this line
- function can_mangle(name) {
- if (unmangleable.has(name)) return false;
- if (reserved.has(name)) return false;
- if (options.only_cache) {
- return cache.has(name);
- }
- if (/^-?[0-9]+(\.[0-9]+)?(e[+-][0-9]+)?$/.test(name)) return false;
- return true;
- }
- function should_mangle(name) {
- if (regex && !regex.test(name)) return false;
- if (reserved.has(name)) return false;
- return cache.has(name)
- || names_to_mangle.has(name);
- }
- function add(name) {
- if (can_mangle(name))
- names_to_mangle.add(name);
- if (!should_mangle(name)) {
- unmangleable.add(name);
- }
- }
- function mangle(name) {
- if (!should_mangle(name)) {
- return name;
- }
- var mangled = cache.get(name);
- if (!mangled) {
- if (debug) {
- // debug mode: use a prefix and suffix to preserve readability, e.g. o.foo -> o._$foo$NNN_.
- var debug_mangled = "_$" + name + "$" + debug_name_suffix + "_";
- if (can_mangle(debug_mangled)) {
- mangled = debug_mangled;
- }
- }
- // either debug mode is off, or it is on and we could not use the mangled name
- if (!mangled) {
- do {
- mangled = nth_identifier.get(++cname);
- } while (!can_mangle(mangled));
- }
- cache.set(name, mangled);
- }
- return mangled;
- }
- function mangleStrings(node) {
- return node.transform(new TreeTransformer(function(node) {
- if (node instanceof AST_Sequence) {
- var last = node.expressions.length - 1;
- node.expressions[last] = mangleStrings(node.expressions[last]);
- } else if (node instanceof AST_String) {
- node.value = mangle(node.value);
- } else if (node instanceof AST_Conditional) {
- node.consequent = mangleStrings(node.consequent);
- node.alternative = mangleStrings(node.alternative);
- }
- return node;
- }));
- }
- }
- export {
- reserve_quoted_keys,
- mangle_properties,
- mangle_private_properties,
- };
|