123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250 |
- /***********************************************************************
- A JavaScript tokenizer / parser / beautifier / compressor.
- https://github.com/mishoo/UglifyJS
- -------------------------------- (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";
- function TreeTransformer(before, after) {
- TreeWalker.call(this);
- this.before = before;
- this.after = after;
- }
- TreeTransformer.prototype = new TreeWalker;
- (function(DEF) {
- function do_list(list, tw) {
- return List(list, function(node) {
- return node.transform(tw, true);
- });
- }
- DEF(AST_Node, noop);
- DEF(AST_LabeledStatement, function(self, tw) {
- self.label = self.label.transform(tw);
- self.body = self.body.transform(tw);
- });
- DEF(AST_SimpleStatement, function(self, tw) {
- self.body = self.body.transform(tw);
- });
- DEF(AST_Block, function(self, tw) {
- self.body = do_list(self.body, tw);
- });
- DEF(AST_Do, function(self, tw) {
- self.body = self.body.transform(tw);
- self.condition = self.condition.transform(tw);
- });
- DEF(AST_While, function(self, tw) {
- self.condition = self.condition.transform(tw);
- self.body = self.body.transform(tw);
- });
- DEF(AST_For, function(self, tw) {
- if (self.init) self.init = self.init.transform(tw);
- if (self.condition) self.condition = self.condition.transform(tw);
- if (self.step) self.step = self.step.transform(tw);
- self.body = self.body.transform(tw);
- });
- DEF(AST_ForEnumeration, function(self, tw) {
- self.init = self.init.transform(tw);
- self.object = self.object.transform(tw);
- self.body = self.body.transform(tw);
- });
- DEF(AST_With, function(self, tw) {
- self.expression = self.expression.transform(tw);
- self.body = self.body.transform(tw);
- });
- DEF(AST_Exit, function(self, tw) {
- if (self.value) self.value = self.value.transform(tw);
- });
- DEF(AST_LoopControl, function(self, tw) {
- if (self.label) self.label = self.label.transform(tw);
- });
- DEF(AST_If, function(self, tw) {
- self.condition = self.condition.transform(tw);
- self.body = self.body.transform(tw);
- if (self.alternative) self.alternative = self.alternative.transform(tw);
- });
- DEF(AST_Switch, function(self, tw) {
- self.expression = self.expression.transform(tw);
- self.body = do_list(self.body, tw);
- });
- DEF(AST_Case, function(self, tw) {
- self.expression = self.expression.transform(tw);
- self.body = do_list(self.body, tw);
- });
- DEF(AST_Try, function(self, tw) {
- self.body = do_list(self.body, tw);
- if (self.bcatch) self.bcatch = self.bcatch.transform(tw);
- if (self.bfinally) self.bfinally = self.bfinally.transform(tw);
- });
- DEF(AST_Catch, function(self, tw) {
- if (self.argname) self.argname = self.argname.transform(tw);
- self.body = do_list(self.body, tw);
- });
- DEF(AST_Definitions, function(self, tw) {
- self.definitions = do_list(self.definitions, tw);
- });
- DEF(AST_VarDef, function(self, tw) {
- self.name = self.name.transform(tw);
- if (self.value) self.value = self.value.transform(tw);
- });
- DEF(AST_DefaultValue, function(self, tw) {
- self.name = self.name.transform(tw);
- self.value = self.value.transform(tw);
- });
- DEF(AST_Lambda, function(self, tw) {
- if (self.name) self.name = self.name.transform(tw);
- self.argnames = do_list(self.argnames, tw);
- if (self.rest) self.rest = self.rest.transform(tw);
- self.body = do_list(self.body, tw);
- });
- function transform_arrow(self, tw) {
- self.argnames = do_list(self.argnames, tw);
- if (self.rest) self.rest = self.rest.transform(tw);
- if (self.value) {
- self.value = self.value.transform(tw);
- } else {
- self.body = do_list(self.body, tw);
- }
- }
- DEF(AST_Arrow, transform_arrow);
- DEF(AST_AsyncArrow, transform_arrow);
- DEF(AST_Class, function(self, tw) {
- if (self.name) self.name = self.name.transform(tw);
- if (self.extends) self.extends = self.extends.transform(tw);
- self.properties = do_list(self.properties, tw);
- });
- DEF(AST_ClassProperty, function(self, tw) {
- if (self.key instanceof AST_Node) self.key = self.key.transform(tw);
- if (self.value) self.value = self.value.transform(tw);
- });
- DEF(AST_Call, function(self, tw) {
- self.expression = self.expression.transform(tw);
- self.args = do_list(self.args, tw);
- });
- DEF(AST_Sequence, function(self, tw) {
- self.expressions = do_list(self.expressions, tw);
- });
- DEF(AST_Await, function(self, tw) {
- self.expression = self.expression.transform(tw);
- });
- DEF(AST_Yield, function(self, tw) {
- if (self.expression) self.expression = self.expression.transform(tw);
- });
- DEF(AST_Dot, function(self, tw) {
- self.expression = self.expression.transform(tw);
- });
- DEF(AST_Sub, function(self, tw) {
- self.expression = self.expression.transform(tw);
- self.property = self.property.transform(tw);
- });
- DEF(AST_Spread, function(self, tw) {
- self.expression = self.expression.transform(tw);
- });
- DEF(AST_Unary, function(self, tw) {
- self.expression = self.expression.transform(tw);
- });
- DEF(AST_Binary, function(self, tw) {
- self.left = self.left.transform(tw);
- self.right = self.right.transform(tw);
- });
- DEF(AST_Conditional, function(self, tw) {
- self.condition = self.condition.transform(tw);
- self.consequent = self.consequent.transform(tw);
- self.alternative = self.alternative.transform(tw);
- });
- DEF(AST_Array, function(self, tw) {
- self.elements = do_list(self.elements, tw);
- });
- DEF(AST_DestructuredArray, function(self, tw) {
- self.elements = do_list(self.elements, tw);
- if (self.rest) self.rest = self.rest.transform(tw);
- });
- DEF(AST_DestructuredKeyVal, function(self, tw) {
- if (self.key instanceof AST_Node) self.key = self.key.transform(tw);
- self.value = self.value.transform(tw);
- });
- DEF(AST_DestructuredObject, function(self, tw) {
- self.properties = do_list(self.properties, tw);
- if (self.rest) self.rest = self.rest.transform(tw);
- });
- DEF(AST_Object, function(self, tw) {
- self.properties = do_list(self.properties, tw);
- });
- DEF(AST_ObjectProperty, function(self, tw) {
- if (self.key instanceof AST_Node) self.key = self.key.transform(tw);
- self.value = self.value.transform(tw);
- });
- DEF(AST_ExportDeclaration, function(self, tw) {
- self.body = self.body.transform(tw);
- });
- DEF(AST_ExportDefault, function(self, tw) {
- self.body = self.body.transform(tw);
- });
- DEF(AST_ExportReferences, function(self, tw) {
- self.properties = do_list(self.properties, tw);
- });
- DEF(AST_Import, function(self, tw) {
- if (self.all) self.all = self.all.transform(tw);
- if (self.default) self.default = self.default.transform(tw);
- if (self.properties) self.properties = do_list(self.properties, tw);
- });
- DEF(AST_Template, function(self, tw) {
- if (self.tag) self.tag = self.tag.transform(tw);
- self.expressions = do_list(self.expressions, tw);
- });
- })(function(node, descend) {
- node.DEFMETHOD("transform", function(tw, in_list) {
- var x, y;
- tw.push(this);
- if (tw.before) x = tw.before(this, descend, in_list);
- if (typeof x === "undefined") {
- x = this;
- descend(x, tw);
- if (tw.after) {
- y = tw.after(x, in_list);
- if (typeof y !== "undefined") x = y;
- }
- }
- tw.pop();
- return x;
- });
- });
|