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
			| 
											3 years ago
										 | /*********************************************************************** | ||
|  | 
 | ||
|  |   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, | ||
|  | }; |