1 | /*
|
---|
2 | Copyright (C) 2012-2013 Yusuke Suzuki <utatane.tea@gmail.com>
|
---|
3 | Copyright (C) 2012 Ariya Hidayat <ariya.hidayat@gmail.com>
|
---|
4 |
|
---|
5 | Redistribution and use in source and binary forms, with or without
|
---|
6 | modification, are permitted provided that the following conditions are met:
|
---|
7 |
|
---|
8 | * Redistributions of source code must retain the above copyright
|
---|
9 | notice, this list of conditions and the following disclaimer.
|
---|
10 | * Redistributions in binary form must reproduce the above copyright
|
---|
11 | notice, this list of conditions and the following disclaimer in the
|
---|
12 | documentation and/or other materials provided with the distribution.
|
---|
13 |
|
---|
14 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
---|
15 | AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
---|
16 | IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
---|
17 | ARE DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
|
---|
18 | DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
---|
19 | (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
---|
20 | LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
---|
21 | ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
---|
22 | (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
---|
23 | THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
---|
24 | */
|
---|
25 | /*jslint vars:false, bitwise:true*/
|
---|
26 | /*jshint indent:4*/
|
---|
27 | /*global exports:true*/
|
---|
28 | (function clone(exports) {
|
---|
29 | 'use strict';
|
---|
30 |
|
---|
31 | var Syntax,
|
---|
32 | VisitorOption,
|
---|
33 | VisitorKeys,
|
---|
34 | BREAK,
|
---|
35 | SKIP,
|
---|
36 | REMOVE;
|
---|
37 |
|
---|
38 | function deepCopy(obj) {
|
---|
39 | var ret = {}, key, val;
|
---|
40 | for (key in obj) {
|
---|
41 | if (obj.hasOwnProperty(key)) {
|
---|
42 | val = obj[key];
|
---|
43 | if (typeof val === 'object' && val !== null) {
|
---|
44 | ret[key] = deepCopy(val);
|
---|
45 | } else {
|
---|
46 | ret[key] = val;
|
---|
47 | }
|
---|
48 | }
|
---|
49 | }
|
---|
50 | return ret;
|
---|
51 | }
|
---|
52 |
|
---|
53 | // based on LLVM libc++ upper_bound / lower_bound
|
---|
54 | // MIT License
|
---|
55 |
|
---|
56 | function upperBound(array, func) {
|
---|
57 | var diff, len, i, current;
|
---|
58 |
|
---|
59 | len = array.length;
|
---|
60 | i = 0;
|
---|
61 |
|
---|
62 | while (len) {
|
---|
63 | diff = len >>> 1;
|
---|
64 | current = i + diff;
|
---|
65 | if (func(array[current])) {
|
---|
66 | len = diff;
|
---|
67 | } else {
|
---|
68 | i = current + 1;
|
---|
69 | len -= diff + 1;
|
---|
70 | }
|
---|
71 | }
|
---|
72 | return i;
|
---|
73 | }
|
---|
74 |
|
---|
75 | Syntax = {
|
---|
76 | AssignmentExpression: 'AssignmentExpression',
|
---|
77 | AssignmentPattern: 'AssignmentPattern',
|
---|
78 | ArrayExpression: 'ArrayExpression',
|
---|
79 | ArrayPattern: 'ArrayPattern',
|
---|
80 | ArrowFunctionExpression: 'ArrowFunctionExpression',
|
---|
81 | AwaitExpression: 'AwaitExpression', // CAUTION: It's deferred to ES7.
|
---|
82 | BlockStatement: 'BlockStatement',
|
---|
83 | BinaryExpression: 'BinaryExpression',
|
---|
84 | BreakStatement: 'BreakStatement',
|
---|
85 | CallExpression: 'CallExpression',
|
---|
86 | CatchClause: 'CatchClause',
|
---|
87 | ClassBody: 'ClassBody',
|
---|
88 | ClassDeclaration: 'ClassDeclaration',
|
---|
89 | ClassExpression: 'ClassExpression',
|
---|
90 | ComprehensionBlock: 'ComprehensionBlock', // CAUTION: It's deferred to ES7.
|
---|
91 | ComprehensionExpression: 'ComprehensionExpression', // CAUTION: It's deferred to ES7.
|
---|
92 | ConditionalExpression: 'ConditionalExpression',
|
---|
93 | ContinueStatement: 'ContinueStatement',
|
---|
94 | DebuggerStatement: 'DebuggerStatement',
|
---|
95 | DirectiveStatement: 'DirectiveStatement',
|
---|
96 | DoWhileStatement: 'DoWhileStatement',
|
---|
97 | EmptyStatement: 'EmptyStatement',
|
---|
98 | ExportAllDeclaration: 'ExportAllDeclaration',
|
---|
99 | ExportDefaultDeclaration: 'ExportDefaultDeclaration',
|
---|
100 | ExportNamedDeclaration: 'ExportNamedDeclaration',
|
---|
101 | ExportSpecifier: 'ExportSpecifier',
|
---|
102 | ExpressionStatement: 'ExpressionStatement',
|
---|
103 | ForStatement: 'ForStatement',
|
---|
104 | ForInStatement: 'ForInStatement',
|
---|
105 | ForOfStatement: 'ForOfStatement',
|
---|
106 | FunctionDeclaration: 'FunctionDeclaration',
|
---|
107 | FunctionExpression: 'FunctionExpression',
|
---|
108 | GeneratorExpression: 'GeneratorExpression', // CAUTION: It's deferred to ES7.
|
---|
109 | Identifier: 'Identifier',
|
---|
110 | IfStatement: 'IfStatement',
|
---|
111 | ImportExpression: 'ImportExpression',
|
---|
112 | ImportDeclaration: 'ImportDeclaration',
|
---|
113 | ImportDefaultSpecifier: 'ImportDefaultSpecifier',
|
---|
114 | ImportNamespaceSpecifier: 'ImportNamespaceSpecifier',
|
---|
115 | ImportSpecifier: 'ImportSpecifier',
|
---|
116 | Literal: 'Literal',
|
---|
117 | LabeledStatement: 'LabeledStatement',
|
---|
118 | LogicalExpression: 'LogicalExpression',
|
---|
119 | MemberExpression: 'MemberExpression',
|
---|
120 | MetaProperty: 'MetaProperty',
|
---|
121 | MethodDefinition: 'MethodDefinition',
|
---|
122 | ModuleSpecifier: 'ModuleSpecifier',
|
---|
123 | NewExpression: 'NewExpression',
|
---|
124 | ObjectExpression: 'ObjectExpression',
|
---|
125 | ObjectPattern: 'ObjectPattern',
|
---|
126 | Program: 'Program',
|
---|
127 | Property: 'Property',
|
---|
128 | RestElement: 'RestElement',
|
---|
129 | ReturnStatement: 'ReturnStatement',
|
---|
130 | SequenceExpression: 'SequenceExpression',
|
---|
131 | SpreadElement: 'SpreadElement',
|
---|
132 | Super: 'Super',
|
---|
133 | SwitchStatement: 'SwitchStatement',
|
---|
134 | SwitchCase: 'SwitchCase',
|
---|
135 | TaggedTemplateExpression: 'TaggedTemplateExpression',
|
---|
136 | TemplateElement: 'TemplateElement',
|
---|
137 | TemplateLiteral: 'TemplateLiteral',
|
---|
138 | ThisExpression: 'ThisExpression',
|
---|
139 | ThrowStatement: 'ThrowStatement',
|
---|
140 | TryStatement: 'TryStatement',
|
---|
141 | UnaryExpression: 'UnaryExpression',
|
---|
142 | UpdateExpression: 'UpdateExpression',
|
---|
143 | VariableDeclaration: 'VariableDeclaration',
|
---|
144 | VariableDeclarator: 'VariableDeclarator',
|
---|
145 | WhileStatement: 'WhileStatement',
|
---|
146 | WithStatement: 'WithStatement',
|
---|
147 | YieldExpression: 'YieldExpression'
|
---|
148 | };
|
---|
149 |
|
---|
150 | VisitorKeys = {
|
---|
151 | AssignmentExpression: ['left', 'right'],
|
---|
152 | AssignmentPattern: ['left', 'right'],
|
---|
153 | ArrayExpression: ['elements'],
|
---|
154 | ArrayPattern: ['elements'],
|
---|
155 | ArrowFunctionExpression: ['params', 'body'],
|
---|
156 | AwaitExpression: ['argument'], // CAUTION: It's deferred to ES7.
|
---|
157 | BlockStatement: ['body'],
|
---|
158 | BinaryExpression: ['left', 'right'],
|
---|
159 | BreakStatement: ['label'],
|
---|
160 | CallExpression: ['callee', 'arguments'],
|
---|
161 | CatchClause: ['param', 'body'],
|
---|
162 | ClassBody: ['body'],
|
---|
163 | ClassDeclaration: ['id', 'superClass', 'body'],
|
---|
164 | ClassExpression: ['id', 'superClass', 'body'],
|
---|
165 | ComprehensionBlock: ['left', 'right'], // CAUTION: It's deferred to ES7.
|
---|
166 | ComprehensionExpression: ['blocks', 'filter', 'body'], // CAUTION: It's deferred to ES7.
|
---|
167 | ConditionalExpression: ['test', 'consequent', 'alternate'],
|
---|
168 | ContinueStatement: ['label'],
|
---|
169 | DebuggerStatement: [],
|
---|
170 | DirectiveStatement: [],
|
---|
171 | DoWhileStatement: ['body', 'test'],
|
---|
172 | EmptyStatement: [],
|
---|
173 | ExportAllDeclaration: ['source'],
|
---|
174 | ExportDefaultDeclaration: ['declaration'],
|
---|
175 | ExportNamedDeclaration: ['declaration', 'specifiers', 'source'],
|
---|
176 | ExportSpecifier: ['exported', 'local'],
|
---|
177 | ExpressionStatement: ['expression'],
|
---|
178 | ForStatement: ['init', 'test', 'update', 'body'],
|
---|
179 | ForInStatement: ['left', 'right', 'body'],
|
---|
180 | ForOfStatement: ['left', 'right', 'body'],
|
---|
181 | FunctionDeclaration: ['id', 'params', 'body'],
|
---|
182 | FunctionExpression: ['id', 'params', 'body'],
|
---|
183 | GeneratorExpression: ['blocks', 'filter', 'body'], // CAUTION: It's deferred to ES7.
|
---|
184 | Identifier: [],
|
---|
185 | IfStatement: ['test', 'consequent', 'alternate'],
|
---|
186 | ImportExpression: ['source'],
|
---|
187 | ImportDeclaration: ['specifiers', 'source'],
|
---|
188 | ImportDefaultSpecifier: ['local'],
|
---|
189 | ImportNamespaceSpecifier: ['local'],
|
---|
190 | ImportSpecifier: ['imported', 'local'],
|
---|
191 | Literal: [],
|
---|
192 | LabeledStatement: ['label', 'body'],
|
---|
193 | LogicalExpression: ['left', 'right'],
|
---|
194 | MemberExpression: ['object', 'property'],
|
---|
195 | MetaProperty: ['meta', 'property'],
|
---|
196 | MethodDefinition: ['key', 'value'],
|
---|
197 | ModuleSpecifier: [],
|
---|
198 | NewExpression: ['callee', 'arguments'],
|
---|
199 | ObjectExpression: ['properties'],
|
---|
200 | ObjectPattern: ['properties'],
|
---|
201 | Program: ['body'],
|
---|
202 | Property: ['key', 'value'],
|
---|
203 | RestElement: [ 'argument' ],
|
---|
204 | ReturnStatement: ['argument'],
|
---|
205 | SequenceExpression: ['expressions'],
|
---|
206 | SpreadElement: ['argument'],
|
---|
207 | Super: [],
|
---|
208 | SwitchStatement: ['discriminant', 'cases'],
|
---|
209 | SwitchCase: ['test', 'consequent'],
|
---|
210 | TaggedTemplateExpression: ['tag', 'quasi'],
|
---|
211 | TemplateElement: [],
|
---|
212 | TemplateLiteral: ['quasis', 'expressions'],
|
---|
213 | ThisExpression: [],
|
---|
214 | ThrowStatement: ['argument'],
|
---|
215 | TryStatement: ['block', 'handler', 'finalizer'],
|
---|
216 | UnaryExpression: ['argument'],
|
---|
217 | UpdateExpression: ['argument'],
|
---|
218 | VariableDeclaration: ['declarations'],
|
---|
219 | VariableDeclarator: ['id', 'init'],
|
---|
220 | WhileStatement: ['test', 'body'],
|
---|
221 | WithStatement: ['object', 'body'],
|
---|
222 | YieldExpression: ['argument']
|
---|
223 | };
|
---|
224 |
|
---|
225 | // unique id
|
---|
226 | BREAK = {};
|
---|
227 | SKIP = {};
|
---|
228 | REMOVE = {};
|
---|
229 |
|
---|
230 | VisitorOption = {
|
---|
231 | Break: BREAK,
|
---|
232 | Skip: SKIP,
|
---|
233 | Remove: REMOVE
|
---|
234 | };
|
---|
235 |
|
---|
236 | function Reference(parent, key) {
|
---|
237 | this.parent = parent;
|
---|
238 | this.key = key;
|
---|
239 | }
|
---|
240 |
|
---|
241 | Reference.prototype.replace = function replace(node) {
|
---|
242 | this.parent[this.key] = node;
|
---|
243 | };
|
---|
244 |
|
---|
245 | Reference.prototype.remove = function remove() {
|
---|
246 | if (Array.isArray(this.parent)) {
|
---|
247 | this.parent.splice(this.key, 1);
|
---|
248 | return true;
|
---|
249 | } else {
|
---|
250 | this.replace(null);
|
---|
251 | return false;
|
---|
252 | }
|
---|
253 | };
|
---|
254 |
|
---|
255 | function Element(node, path, wrap, ref) {
|
---|
256 | this.node = node;
|
---|
257 | this.path = path;
|
---|
258 | this.wrap = wrap;
|
---|
259 | this.ref = ref;
|
---|
260 | }
|
---|
261 |
|
---|
262 | function Controller() { }
|
---|
263 |
|
---|
264 | // API:
|
---|
265 | // return property path array from root to current node
|
---|
266 | Controller.prototype.path = function path() {
|
---|
267 | var i, iz, j, jz, result, element;
|
---|
268 |
|
---|
269 | function addToPath(result, path) {
|
---|
270 | if (Array.isArray(path)) {
|
---|
271 | for (j = 0, jz = path.length; j < jz; ++j) {
|
---|
272 | result.push(path[j]);
|
---|
273 | }
|
---|
274 | } else {
|
---|
275 | result.push(path);
|
---|
276 | }
|
---|
277 | }
|
---|
278 |
|
---|
279 | // root node
|
---|
280 | if (!this.__current.path) {
|
---|
281 | return null;
|
---|
282 | }
|
---|
283 |
|
---|
284 | // first node is sentinel, second node is root element
|
---|
285 | result = [];
|
---|
286 | for (i = 2, iz = this.__leavelist.length; i < iz; ++i) {
|
---|
287 | element = this.__leavelist[i];
|
---|
288 | addToPath(result, element.path);
|
---|
289 | }
|
---|
290 | addToPath(result, this.__current.path);
|
---|
291 | return result;
|
---|
292 | };
|
---|
293 |
|
---|
294 | // API:
|
---|
295 | // return type of current node
|
---|
296 | Controller.prototype.type = function () {
|
---|
297 | var node = this.current();
|
---|
298 | return node.type || this.__current.wrap;
|
---|
299 | };
|
---|
300 |
|
---|
301 | // API:
|
---|
302 | // return array of parent elements
|
---|
303 | Controller.prototype.parents = function parents() {
|
---|
304 | var i, iz, result;
|
---|
305 |
|
---|
306 | // first node is sentinel
|
---|
307 | result = [];
|
---|
308 | for (i = 1, iz = this.__leavelist.length; i < iz; ++i) {
|
---|
309 | result.push(this.__leavelist[i].node);
|
---|
310 | }
|
---|
311 |
|
---|
312 | return result;
|
---|
313 | };
|
---|
314 |
|
---|
315 | // API:
|
---|
316 | // return current node
|
---|
317 | Controller.prototype.current = function current() {
|
---|
318 | return this.__current.node;
|
---|
319 | };
|
---|
320 |
|
---|
321 | Controller.prototype.__execute = function __execute(callback, element) {
|
---|
322 | var previous, result;
|
---|
323 |
|
---|
324 | result = undefined;
|
---|
325 |
|
---|
326 | previous = this.__current;
|
---|
327 | this.__current = element;
|
---|
328 | this.__state = null;
|
---|
329 | if (callback) {
|
---|
330 | result = callback.call(this, element.node, this.__leavelist[this.__leavelist.length - 1].node);
|
---|
331 | }
|
---|
332 | this.__current = previous;
|
---|
333 |
|
---|
334 | return result;
|
---|
335 | };
|
---|
336 |
|
---|
337 | // API:
|
---|
338 | // notify control skip / break
|
---|
339 | Controller.prototype.notify = function notify(flag) {
|
---|
340 | this.__state = flag;
|
---|
341 | };
|
---|
342 |
|
---|
343 | // API:
|
---|
344 | // skip child nodes of current node
|
---|
345 | Controller.prototype.skip = function () {
|
---|
346 | this.notify(SKIP);
|
---|
347 | };
|
---|
348 |
|
---|
349 | // API:
|
---|
350 | // break traversals
|
---|
351 | Controller.prototype['break'] = function () {
|
---|
352 | this.notify(BREAK);
|
---|
353 | };
|
---|
354 |
|
---|
355 | // API:
|
---|
356 | // remove node
|
---|
357 | Controller.prototype.remove = function () {
|
---|
358 | this.notify(REMOVE);
|
---|
359 | };
|
---|
360 |
|
---|
361 | Controller.prototype.__initialize = function(root, visitor) {
|
---|
362 | this.visitor = visitor;
|
---|
363 | this.root = root;
|
---|
364 | this.__worklist = [];
|
---|
365 | this.__leavelist = [];
|
---|
366 | this.__current = null;
|
---|
367 | this.__state = null;
|
---|
368 | this.__fallback = null;
|
---|
369 | if (visitor.fallback === 'iteration') {
|
---|
370 | this.__fallback = Object.keys;
|
---|
371 | } else if (typeof visitor.fallback === 'function') {
|
---|
372 | this.__fallback = visitor.fallback;
|
---|
373 | }
|
---|
374 |
|
---|
375 | this.__keys = VisitorKeys;
|
---|
376 | if (visitor.keys) {
|
---|
377 | this.__keys = Object.assign(Object.create(this.__keys), visitor.keys);
|
---|
378 | }
|
---|
379 | };
|
---|
380 |
|
---|
381 | function isNode(node) {
|
---|
382 | if (node == null) {
|
---|
383 | return false;
|
---|
384 | }
|
---|
385 | return typeof node === 'object' && typeof node.type === 'string';
|
---|
386 | }
|
---|
387 |
|
---|
388 | function isProperty(nodeType, key) {
|
---|
389 | return (nodeType === Syntax.ObjectExpression || nodeType === Syntax.ObjectPattern) && 'properties' === key;
|
---|
390 | }
|
---|
391 |
|
---|
392 | Controller.prototype.traverse = function traverse(root, visitor) {
|
---|
393 | var worklist,
|
---|
394 | leavelist,
|
---|
395 | element,
|
---|
396 | node,
|
---|
397 | nodeType,
|
---|
398 | ret,
|
---|
399 | key,
|
---|
400 | current,
|
---|
401 | current2,
|
---|
402 | candidates,
|
---|
403 | candidate,
|
---|
404 | sentinel;
|
---|
405 |
|
---|
406 | this.__initialize(root, visitor);
|
---|
407 |
|
---|
408 | sentinel = {};
|
---|
409 |
|
---|
410 | // reference
|
---|
411 | worklist = this.__worklist;
|
---|
412 | leavelist = this.__leavelist;
|
---|
413 |
|
---|
414 | // initialize
|
---|
415 | worklist.push(new Element(root, null, null, null));
|
---|
416 | leavelist.push(new Element(null, null, null, null));
|
---|
417 |
|
---|
418 | while (worklist.length) {
|
---|
419 | element = worklist.pop();
|
---|
420 |
|
---|
421 | if (element === sentinel) {
|
---|
422 | element = leavelist.pop();
|
---|
423 |
|
---|
424 | ret = this.__execute(visitor.leave, element);
|
---|
425 |
|
---|
426 | if (this.__state === BREAK || ret === BREAK) {
|
---|
427 | return;
|
---|
428 | }
|
---|
429 | continue;
|
---|
430 | }
|
---|
431 |
|
---|
432 | if (element.node) {
|
---|
433 |
|
---|
434 | ret = this.__execute(visitor.enter, element);
|
---|
435 |
|
---|
436 | if (this.__state === BREAK || ret === BREAK) {
|
---|
437 | return;
|
---|
438 | }
|
---|
439 |
|
---|
440 | worklist.push(sentinel);
|
---|
441 | leavelist.push(element);
|
---|
442 |
|
---|
443 | if (this.__state === SKIP || ret === SKIP) {
|
---|
444 | continue;
|
---|
445 | }
|
---|
446 |
|
---|
447 | node = element.node;
|
---|
448 | nodeType = node.type || element.wrap;
|
---|
449 | candidates = this.__keys[nodeType];
|
---|
450 | if (!candidates) {
|
---|
451 | if (this.__fallback) {
|
---|
452 | candidates = this.__fallback(node);
|
---|
453 | } else {
|
---|
454 | throw new Error('Unknown node type ' + nodeType + '.');
|
---|
455 | }
|
---|
456 | }
|
---|
457 |
|
---|
458 | current = candidates.length;
|
---|
459 | while ((current -= 1) >= 0) {
|
---|
460 | key = candidates[current];
|
---|
461 | candidate = node[key];
|
---|
462 | if (!candidate) {
|
---|
463 | continue;
|
---|
464 | }
|
---|
465 |
|
---|
466 | if (Array.isArray(candidate)) {
|
---|
467 | current2 = candidate.length;
|
---|
468 | while ((current2 -= 1) >= 0) {
|
---|
469 | if (!candidate[current2]) {
|
---|
470 | continue;
|
---|
471 | }
|
---|
472 | if (isProperty(nodeType, candidates[current])) {
|
---|
473 | element = new Element(candidate[current2], [key, current2], 'Property', null);
|
---|
474 | } else if (isNode(candidate[current2])) {
|
---|
475 | element = new Element(candidate[current2], [key, current2], null, null);
|
---|
476 | } else {
|
---|
477 | continue;
|
---|
478 | }
|
---|
479 | worklist.push(element);
|
---|
480 | }
|
---|
481 | } else if (isNode(candidate)) {
|
---|
482 | worklist.push(new Element(candidate, key, null, null));
|
---|
483 | }
|
---|
484 | }
|
---|
485 | }
|
---|
486 | }
|
---|
487 | };
|
---|
488 |
|
---|
489 | Controller.prototype.replace = function replace(root, visitor) {
|
---|
490 | var worklist,
|
---|
491 | leavelist,
|
---|
492 | node,
|
---|
493 | nodeType,
|
---|
494 | target,
|
---|
495 | element,
|
---|
496 | current,
|
---|
497 | current2,
|
---|
498 | candidates,
|
---|
499 | candidate,
|
---|
500 | sentinel,
|
---|
501 | outer,
|
---|
502 | key;
|
---|
503 |
|
---|
504 | function removeElem(element) {
|
---|
505 | var i,
|
---|
506 | key,
|
---|
507 | nextElem,
|
---|
508 | parent;
|
---|
509 |
|
---|
510 | if (element.ref.remove()) {
|
---|
511 | // When the reference is an element of an array.
|
---|
512 | key = element.ref.key;
|
---|
513 | parent = element.ref.parent;
|
---|
514 |
|
---|
515 | // If removed from array, then decrease following items' keys.
|
---|
516 | i = worklist.length;
|
---|
517 | while (i--) {
|
---|
518 | nextElem = worklist[i];
|
---|
519 | if (nextElem.ref && nextElem.ref.parent === parent) {
|
---|
520 | if (nextElem.ref.key < key) {
|
---|
521 | break;
|
---|
522 | }
|
---|
523 | --nextElem.ref.key;
|
---|
524 | }
|
---|
525 | }
|
---|
526 | }
|
---|
527 | }
|
---|
528 |
|
---|
529 | this.__initialize(root, visitor);
|
---|
530 |
|
---|
531 | sentinel = {};
|
---|
532 |
|
---|
533 | // reference
|
---|
534 | worklist = this.__worklist;
|
---|
535 | leavelist = this.__leavelist;
|
---|
536 |
|
---|
537 | // initialize
|
---|
538 | outer = {
|
---|
539 | root: root
|
---|
540 | };
|
---|
541 | element = new Element(root, null, null, new Reference(outer, 'root'));
|
---|
542 | worklist.push(element);
|
---|
543 | leavelist.push(element);
|
---|
544 |
|
---|
545 | while (worklist.length) {
|
---|
546 | element = worklist.pop();
|
---|
547 |
|
---|
548 | if (element === sentinel) {
|
---|
549 | element = leavelist.pop();
|
---|
550 |
|
---|
551 | target = this.__execute(visitor.leave, element);
|
---|
552 |
|
---|
553 | // node may be replaced with null,
|
---|
554 | // so distinguish between undefined and null in this place
|
---|
555 | if (target !== undefined && target !== BREAK && target !== SKIP && target !== REMOVE) {
|
---|
556 | // replace
|
---|
557 | element.ref.replace(target);
|
---|
558 | }
|
---|
559 |
|
---|
560 | if (this.__state === REMOVE || target === REMOVE) {
|
---|
561 | removeElem(element);
|
---|
562 | }
|
---|
563 |
|
---|
564 | if (this.__state === BREAK || target === BREAK) {
|
---|
565 | return outer.root;
|
---|
566 | }
|
---|
567 | continue;
|
---|
568 | }
|
---|
569 |
|
---|
570 | target = this.__execute(visitor.enter, element);
|
---|
571 |
|
---|
572 | // node may be replaced with null,
|
---|
573 | // so distinguish between undefined and null in this place
|
---|
574 | if (target !== undefined && target !== BREAK && target !== SKIP && target !== REMOVE) {
|
---|
575 | // replace
|
---|
576 | element.ref.replace(target);
|
---|
577 | element.node = target;
|
---|
578 | }
|
---|
579 |
|
---|
580 | if (this.__state === REMOVE || target === REMOVE) {
|
---|
581 | removeElem(element);
|
---|
582 | element.node = null;
|
---|
583 | }
|
---|
584 |
|
---|
585 | if (this.__state === BREAK || target === BREAK) {
|
---|
586 | return outer.root;
|
---|
587 | }
|
---|
588 |
|
---|
589 | // node may be null
|
---|
590 | node = element.node;
|
---|
591 | if (!node) {
|
---|
592 | continue;
|
---|
593 | }
|
---|
594 |
|
---|
595 | worklist.push(sentinel);
|
---|
596 | leavelist.push(element);
|
---|
597 |
|
---|
598 | if (this.__state === SKIP || target === SKIP) {
|
---|
599 | continue;
|
---|
600 | }
|
---|
601 |
|
---|
602 | nodeType = node.type || element.wrap;
|
---|
603 | candidates = this.__keys[nodeType];
|
---|
604 | if (!candidates) {
|
---|
605 | if (this.__fallback) {
|
---|
606 | candidates = this.__fallback(node);
|
---|
607 | } else {
|
---|
608 | throw new Error('Unknown node type ' + nodeType + '.');
|
---|
609 | }
|
---|
610 | }
|
---|
611 |
|
---|
612 | current = candidates.length;
|
---|
613 | while ((current -= 1) >= 0) {
|
---|
614 | key = candidates[current];
|
---|
615 | candidate = node[key];
|
---|
616 | if (!candidate) {
|
---|
617 | continue;
|
---|
618 | }
|
---|
619 |
|
---|
620 | if (Array.isArray(candidate)) {
|
---|
621 | current2 = candidate.length;
|
---|
622 | while ((current2 -= 1) >= 0) {
|
---|
623 | if (!candidate[current2]) {
|
---|
624 | continue;
|
---|
625 | }
|
---|
626 | if (isProperty(nodeType, candidates[current])) {
|
---|
627 | element = new Element(candidate[current2], [key, current2], 'Property', new Reference(candidate, current2));
|
---|
628 | } else if (isNode(candidate[current2])) {
|
---|
629 | element = new Element(candidate[current2], [key, current2], null, new Reference(candidate, current2));
|
---|
630 | } else {
|
---|
631 | continue;
|
---|
632 | }
|
---|
633 | worklist.push(element);
|
---|
634 | }
|
---|
635 | } else if (isNode(candidate)) {
|
---|
636 | worklist.push(new Element(candidate, key, null, new Reference(node, key)));
|
---|
637 | }
|
---|
638 | }
|
---|
639 | }
|
---|
640 |
|
---|
641 | return outer.root;
|
---|
642 | };
|
---|
643 |
|
---|
644 | function traverse(root, visitor) {
|
---|
645 | var controller = new Controller();
|
---|
646 | return controller.traverse(root, visitor);
|
---|
647 | }
|
---|
648 |
|
---|
649 | function replace(root, visitor) {
|
---|
650 | var controller = new Controller();
|
---|
651 | return controller.replace(root, visitor);
|
---|
652 | }
|
---|
653 |
|
---|
654 | function extendCommentRange(comment, tokens) {
|
---|
655 | var target;
|
---|
656 |
|
---|
657 | target = upperBound(tokens, function search(token) {
|
---|
658 | return token.range[0] > comment.range[0];
|
---|
659 | });
|
---|
660 |
|
---|
661 | comment.extendedRange = [comment.range[0], comment.range[1]];
|
---|
662 |
|
---|
663 | if (target !== tokens.length) {
|
---|
664 | comment.extendedRange[1] = tokens[target].range[0];
|
---|
665 | }
|
---|
666 |
|
---|
667 | target -= 1;
|
---|
668 | if (target >= 0) {
|
---|
669 | comment.extendedRange[0] = tokens[target].range[1];
|
---|
670 | }
|
---|
671 |
|
---|
672 | return comment;
|
---|
673 | }
|
---|
674 |
|
---|
675 | function attachComments(tree, providedComments, tokens) {
|
---|
676 | // At first, we should calculate extended comment ranges.
|
---|
677 | var comments = [], comment, len, i, cursor;
|
---|
678 |
|
---|
679 | if (!tree.range) {
|
---|
680 | throw new Error('attachComments needs range information');
|
---|
681 | }
|
---|
682 |
|
---|
683 | // tokens array is empty, we attach comments to tree as 'leadingComments'
|
---|
684 | if (!tokens.length) {
|
---|
685 | if (providedComments.length) {
|
---|
686 | for (i = 0, len = providedComments.length; i < len; i += 1) {
|
---|
687 | comment = deepCopy(providedComments[i]);
|
---|
688 | comment.extendedRange = [0, tree.range[0]];
|
---|
689 | comments.push(comment);
|
---|
690 | }
|
---|
691 | tree.leadingComments = comments;
|
---|
692 | }
|
---|
693 | return tree;
|
---|
694 | }
|
---|
695 |
|
---|
696 | for (i = 0, len = providedComments.length; i < len; i += 1) {
|
---|
697 | comments.push(extendCommentRange(deepCopy(providedComments[i]), tokens));
|
---|
698 | }
|
---|
699 |
|
---|
700 | // This is based on John Freeman's implementation.
|
---|
701 | cursor = 0;
|
---|
702 | traverse(tree, {
|
---|
703 | enter: function (node) {
|
---|
704 | var comment;
|
---|
705 |
|
---|
706 | while (cursor < comments.length) {
|
---|
707 | comment = comments[cursor];
|
---|
708 | if (comment.extendedRange[1] > node.range[0]) {
|
---|
709 | break;
|
---|
710 | }
|
---|
711 |
|
---|
712 | if (comment.extendedRange[1] === node.range[0]) {
|
---|
713 | if (!node.leadingComments) {
|
---|
714 | node.leadingComments = [];
|
---|
715 | }
|
---|
716 | node.leadingComments.push(comment);
|
---|
717 | comments.splice(cursor, 1);
|
---|
718 | } else {
|
---|
719 | cursor += 1;
|
---|
720 | }
|
---|
721 | }
|
---|
722 |
|
---|
723 | // already out of owned node
|
---|
724 | if (cursor === comments.length) {
|
---|
725 | return VisitorOption.Break;
|
---|
726 | }
|
---|
727 |
|
---|
728 | if (comments[cursor].extendedRange[0] > node.range[1]) {
|
---|
729 | return VisitorOption.Skip;
|
---|
730 | }
|
---|
731 | }
|
---|
732 | });
|
---|
733 |
|
---|
734 | cursor = 0;
|
---|
735 | traverse(tree, {
|
---|
736 | leave: function (node) {
|
---|
737 | var comment;
|
---|
738 |
|
---|
739 | while (cursor < comments.length) {
|
---|
740 | comment = comments[cursor];
|
---|
741 | if (node.range[1] < comment.extendedRange[0]) {
|
---|
742 | break;
|
---|
743 | }
|
---|
744 |
|
---|
745 | if (node.range[1] === comment.extendedRange[0]) {
|
---|
746 | if (!node.trailingComments) {
|
---|
747 | node.trailingComments = [];
|
---|
748 | }
|
---|
749 | node.trailingComments.push(comment);
|
---|
750 | comments.splice(cursor, 1);
|
---|
751 | } else {
|
---|
752 | cursor += 1;
|
---|
753 | }
|
---|
754 | }
|
---|
755 |
|
---|
756 | // already out of owned node
|
---|
757 | if (cursor === comments.length) {
|
---|
758 | return VisitorOption.Break;
|
---|
759 | }
|
---|
760 |
|
---|
761 | if (comments[cursor].extendedRange[0] > node.range[1]) {
|
---|
762 | return VisitorOption.Skip;
|
---|
763 | }
|
---|
764 | }
|
---|
765 | });
|
---|
766 |
|
---|
767 | return tree;
|
---|
768 | }
|
---|
769 |
|
---|
770 | exports.version = require('./package.json').version;
|
---|
771 | exports.Syntax = Syntax;
|
---|
772 | exports.traverse = traverse;
|
---|
773 | exports.replace = replace;
|
---|
774 | exports.attachComments = attachComments;
|
---|
775 | exports.VisitorKeys = VisitorKeys;
|
---|
776 | exports.VisitorOption = VisitorOption;
|
---|
777 | exports.Controller = Controller;
|
---|
778 | exports.cloneEnvironment = function () { return clone({}); };
|
---|
779 |
|
---|
780 | return exports;
|
---|
781 | }(exports));
|
---|
782 | /* vim: set sw=4 ts=4 et tw=80 : */
|
---|