You can not select more than 25 topics
			Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
		
		
		
		
		
			
		
			
				
					
					
						
							379 lines
						
					
					
						
							12 KiB
						
					
					
				
			
		
		
	
	
							379 lines
						
					
					
						
							12 KiB
						
					
					
				/***********************************************************************
 | 
						|
 | 
						|
  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,
 | 
						|
};
 |