|
| 1 | +/*********************************************************************** |
| 2 | +
|
| 3 | + A JavaScript tokenizer / parser / beautifier / compressor. |
| 4 | + https://github.com/mishoo/UglifyJS |
| 5 | +
|
| 6 | + -------------------------------- (C) --------------------------------- |
| 7 | +
|
| 8 | + Author: Mihai Bazon |
| 9 | + |
| 10 | + http://mihai.bazon.net/blog |
| 11 | +
|
| 12 | + Distributed under the BSD license: |
| 13 | +
|
| 14 | + Copyright 2012 (c) Mihai Bazon <[email protected]> |
| 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 | +"use strict"; |
| 45 | + |
| 46 | +function TreeTransformer(before, after) { |
| 47 | + TreeWalker.call(this); |
| 48 | + this.before = before; |
| 49 | + this.after = after; |
| 50 | +} |
| 51 | +TreeTransformer.prototype = new TreeWalker; |
| 52 | + |
| 53 | +(function(DEF) { |
| 54 | + function do_list(list, tw) { |
| 55 | + return List(list, function(node) { |
| 56 | + return node.transform(tw, true); |
| 57 | + }); |
| 58 | + } |
| 59 | + |
| 60 | + DEF(AST_Node, noop); |
| 61 | + DEF(AST_LabeledStatement, function(self, tw) { |
| 62 | + self.label = self.label.transform(tw); |
| 63 | + self.body = self.body.transform(tw); |
| 64 | + }); |
| 65 | + DEF(AST_SimpleStatement, function(self, tw) { |
| 66 | + self.body = self.body.transform(tw); |
| 67 | + }); |
| 68 | + DEF(AST_Block, function(self, tw) { |
| 69 | + self.body = do_list(self.body, tw); |
| 70 | + }); |
| 71 | + DEF(AST_Do, function(self, tw) { |
| 72 | + self.body = self.body.transform(tw); |
| 73 | + self.condition = self.condition.transform(tw); |
| 74 | + }); |
| 75 | + DEF(AST_While, function(self, tw) { |
| 76 | + self.condition = self.condition.transform(tw); |
| 77 | + self.body = self.body.transform(tw); |
| 78 | + }); |
| 79 | + DEF(AST_For, function(self, tw) { |
| 80 | + if (self.init) self.init = self.init.transform(tw); |
| 81 | + if (self.condition) self.condition = self.condition.transform(tw); |
| 82 | + if (self.step) self.step = self.step.transform(tw); |
| 83 | + self.body = self.body.transform(tw); |
| 84 | + }); |
| 85 | + DEF(AST_ForEnumeration, function(self, tw) { |
| 86 | + self.init = self.init.transform(tw); |
| 87 | + self.object = self.object.transform(tw); |
| 88 | + self.body = self.body.transform(tw); |
| 89 | + }); |
| 90 | + DEF(AST_With, function(self, tw) { |
| 91 | + self.expression = self.expression.transform(tw); |
| 92 | + self.body = self.body.transform(tw); |
| 93 | + }); |
| 94 | + DEF(AST_Exit, function(self, tw) { |
| 95 | + if (self.value) self.value = self.value.transform(tw); |
| 96 | + }); |
| 97 | + DEF(AST_LoopControl, function(self, tw) { |
| 98 | + if (self.label) self.label = self.label.transform(tw); |
| 99 | + }); |
| 100 | + DEF(AST_If, function(self, tw) { |
| 101 | + self.condition = self.condition.transform(tw); |
| 102 | + self.body = self.body.transform(tw); |
| 103 | + if (self.alternative) self.alternative = self.alternative.transform(tw); |
| 104 | + }); |
| 105 | + DEF(AST_Switch, function(self, tw) { |
| 106 | + self.expression = self.expression.transform(tw); |
| 107 | + self.body = do_list(self.body, tw); |
| 108 | + }); |
| 109 | + DEF(AST_Case, function(self, tw) { |
| 110 | + self.expression = self.expression.transform(tw); |
| 111 | + self.body = do_list(self.body, tw); |
| 112 | + }); |
| 113 | + DEF(AST_Try, function(self, tw) { |
| 114 | + self.body = do_list(self.body, tw); |
| 115 | + if (self.bcatch) self.bcatch = self.bcatch.transform(tw); |
| 116 | + if (self.bfinally) self.bfinally = self.bfinally.transform(tw); |
| 117 | + }); |
| 118 | + DEF(AST_Catch, function(self, tw) { |
| 119 | + if (self.argname) self.argname = self.argname.transform(tw); |
| 120 | + self.body = do_list(self.body, tw); |
| 121 | + }); |
| 122 | + DEF(AST_Definitions, function(self, tw) { |
| 123 | + self.definitions = do_list(self.definitions, tw); |
| 124 | + }); |
| 125 | + DEF(AST_VarDef, function(self, tw) { |
| 126 | + self.name = self.name.transform(tw); |
| 127 | + if (self.value) self.value = self.value.transform(tw); |
| 128 | + }); |
| 129 | + DEF(AST_DefaultValue, function(self, tw) { |
| 130 | + self.name = self.name.transform(tw); |
| 131 | + self.value = self.value.transform(tw); |
| 132 | + }); |
| 133 | + DEF(AST_Lambda, function(self, tw) { |
| 134 | + if (self.name) self.name = self.name.transform(tw); |
| 135 | + self.argnames = do_list(self.argnames, tw); |
| 136 | + if (self.rest) self.rest = self.rest.transform(tw); |
| 137 | + self.body = do_list(self.body, tw); |
| 138 | + }); |
| 139 | + function transform_arrow(self, tw) { |
| 140 | + self.argnames = do_list(self.argnames, tw); |
| 141 | + if (self.rest) self.rest = self.rest.transform(tw); |
| 142 | + if (self.value) { |
| 143 | + self.value = self.value.transform(tw); |
| 144 | + } else { |
| 145 | + self.body = do_list(self.body, tw); |
| 146 | + } |
| 147 | + } |
| 148 | + DEF(AST_Arrow, transform_arrow); |
| 149 | + DEF(AST_AsyncArrow, transform_arrow); |
| 150 | + DEF(AST_Class, function(self, tw) { |
| 151 | + if (self.name) self.name = self.name.transform(tw); |
| 152 | + if (self.extends) self.extends = self.extends.transform(tw); |
| 153 | + self.properties = do_list(self.properties, tw); |
| 154 | + }); |
| 155 | + DEF(AST_ClassProperty, function(self, tw) { |
| 156 | + if (self.key instanceof AST_Node) self.key = self.key.transform(tw); |
| 157 | + if (self.value) self.value = self.value.transform(tw); |
| 158 | + }); |
| 159 | + DEF(AST_Call, function(self, tw) { |
| 160 | + self.expression = self.expression.transform(tw); |
| 161 | + self.args = do_list(self.args, tw); |
| 162 | + }); |
| 163 | + DEF(AST_Sequence, function(self, tw) { |
| 164 | + self.expressions = do_list(self.expressions, tw); |
| 165 | + }); |
| 166 | + DEF(AST_Await, function(self, tw) { |
| 167 | + self.expression = self.expression.transform(tw); |
| 168 | + }); |
| 169 | + DEF(AST_Yield, function(self, tw) { |
| 170 | + if (self.expression) self.expression = self.expression.transform(tw); |
| 171 | + }); |
| 172 | + DEF(AST_Dot, function(self, tw) { |
| 173 | + self.expression = self.expression.transform(tw); |
| 174 | + }); |
| 175 | + DEF(AST_Sub, function(self, tw) { |
| 176 | + self.expression = self.expression.transform(tw); |
| 177 | + self.property = self.property.transform(tw); |
| 178 | + }); |
| 179 | + DEF(AST_Spread, function(self, tw) { |
| 180 | + self.expression = self.expression.transform(tw); |
| 181 | + }); |
| 182 | + DEF(AST_Unary, function(self, tw) { |
| 183 | + self.expression = self.expression.transform(tw); |
| 184 | + }); |
| 185 | + DEF(AST_Binary, function(self, tw) { |
| 186 | + self.left = self.left.transform(tw); |
| 187 | + self.right = self.right.transform(tw); |
| 188 | + }); |
| 189 | + DEF(AST_Conditional, function(self, tw) { |
| 190 | + self.condition = self.condition.transform(tw); |
| 191 | + self.consequent = self.consequent.transform(tw); |
| 192 | + self.alternative = self.alternative.transform(tw); |
| 193 | + }); |
| 194 | + DEF(AST_Array, function(self, tw) { |
| 195 | + self.elements = do_list(self.elements, tw); |
| 196 | + }); |
| 197 | + DEF(AST_DestructuredArray, function(self, tw) { |
| 198 | + self.elements = do_list(self.elements, tw); |
| 199 | + if (self.rest) self.rest = self.rest.transform(tw); |
| 200 | + }); |
| 201 | + DEF(AST_DestructuredKeyVal, function(self, tw) { |
| 202 | + if (self.key instanceof AST_Node) self.key = self.key.transform(tw); |
| 203 | + self.value = self.value.transform(tw); |
| 204 | + }); |
| 205 | + DEF(AST_DestructuredObject, function(self, tw) { |
| 206 | + self.properties = do_list(self.properties, tw); |
| 207 | + if (self.rest) self.rest = self.rest.transform(tw); |
| 208 | + }); |
| 209 | + DEF(AST_Object, function(self, tw) { |
| 210 | + self.properties = do_list(self.properties, tw); |
| 211 | + }); |
| 212 | + DEF(AST_ObjectProperty, function(self, tw) { |
| 213 | + if (self.key instanceof AST_Node) self.key = self.key.transform(tw); |
| 214 | + self.value = self.value.transform(tw); |
| 215 | + }); |
| 216 | + DEF(AST_ExportDeclaration, function(self, tw) { |
| 217 | + self.body = self.body.transform(tw); |
| 218 | + }); |
| 219 | + DEF(AST_ExportDefault, function(self, tw) { |
| 220 | + self.body = self.body.transform(tw); |
| 221 | + }); |
| 222 | + DEF(AST_ExportReferences, function(self, tw) { |
| 223 | + self.properties = do_list(self.properties, tw); |
| 224 | + }); |
| 225 | + DEF(AST_Import, function(self, tw) { |
| 226 | + if (self.all) self.all = self.all.transform(tw); |
| 227 | + if (self.default) self.default = self.default.transform(tw); |
| 228 | + if (self.properties) self.properties = do_list(self.properties, tw); |
| 229 | + }); |
| 230 | + DEF(AST_Template, function(self, tw) { |
| 231 | + if (self.tag) self.tag = self.tag.transform(tw); |
| 232 | + self.expressions = do_list(self.expressions, tw); |
| 233 | + }); |
| 234 | +})(function(node, descend) { |
| 235 | + node.DEFMETHOD("transform", function(tw, in_list) { |
| 236 | + var x, y; |
| 237 | + tw.push(this); |
| 238 | + if (tw.before) x = tw.before(this, descend, in_list); |
| 239 | + if (typeof x === "undefined") { |
| 240 | + x = this; |
| 241 | + descend(x, tw); |
| 242 | + if (tw.after) { |
| 243 | + y = tw.after(x, in_list); |
| 244 | + if (typeof y !== "undefined") x = y; |
| 245 | + } |
| 246 | + } |
| 247 | + tw.pop(); |
| 248 | + return x; |
| 249 | + }); |
| 250 | +}); |
0 commit comments