1 |
3a515b92
|
cagy
|
/*
|
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 : */
|