1 | /*
|
---|
2 | Copyright (C) 2014 Yusuke Suzuki <utatane.tea@gmail.com>
|
---|
3 |
|
---|
4 | Redistribution and use in source and binary forms, with or without
|
---|
5 | modification, are permitted provided that the following conditions are met:
|
---|
6 |
|
---|
7 | * Redistributions of source code must retain the above copyright
|
---|
8 | notice, this list of conditions and the following disclaimer.
|
---|
9 | * Redistributions in binary form must reproduce the above copyright
|
---|
10 | notice, this list of conditions and the following disclaimer in the
|
---|
11 | documentation and/or other materials provided with the distribution.
|
---|
12 |
|
---|
13 | THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
|
---|
14 | AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
---|
15 | IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
|
---|
16 | ARE DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY
|
---|
17 | DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
---|
18 | (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
---|
19 | LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
---|
20 | ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
---|
21 | (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
---|
22 | THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
---|
23 | */
|
---|
24 | (function () {
|
---|
25 | 'use strict';
|
---|
26 |
|
---|
27 | var estraverse = require('estraverse');
|
---|
28 |
|
---|
29 | function isNode(node) {
|
---|
30 | if (node == null) {
|
---|
31 | return false;
|
---|
32 | }
|
---|
33 | return typeof node === 'object' && typeof node.type === 'string';
|
---|
34 | }
|
---|
35 |
|
---|
36 | function isProperty(nodeType, key) {
|
---|
37 | return (nodeType === estraverse.Syntax.ObjectExpression || nodeType === estraverse.Syntax.ObjectPattern) && key === 'properties';
|
---|
38 | }
|
---|
39 |
|
---|
40 | function Visitor(visitor, options) {
|
---|
41 | options = options || {};
|
---|
42 |
|
---|
43 | this.__visitor = visitor || this;
|
---|
44 | this.__childVisitorKeys = options.childVisitorKeys
|
---|
45 | ? Object.assign({}, estraverse.VisitorKeys, options.childVisitorKeys)
|
---|
46 | : estraverse.VisitorKeys;
|
---|
47 | if (options.fallback === 'iteration') {
|
---|
48 | this.__fallback = Object.keys;
|
---|
49 | } else if (typeof options.fallback === 'function') {
|
---|
50 | this.__fallback = options.fallback;
|
---|
51 | }
|
---|
52 | }
|
---|
53 |
|
---|
54 | /* Default method for visiting children.
|
---|
55 | * When you need to call default visiting operation inside custom visiting
|
---|
56 | * operation, you can use it with `this.visitChildren(node)`.
|
---|
57 | */
|
---|
58 | Visitor.prototype.visitChildren = function (node) {
|
---|
59 | var type, children, i, iz, j, jz, child;
|
---|
60 |
|
---|
61 | if (node == null) {
|
---|
62 | return;
|
---|
63 | }
|
---|
64 |
|
---|
65 | type = node.type || estraverse.Syntax.Property;
|
---|
66 |
|
---|
67 | children = this.__childVisitorKeys[type];
|
---|
68 | if (!children) {
|
---|
69 | if (this.__fallback) {
|
---|
70 | children = this.__fallback(node);
|
---|
71 | } else {
|
---|
72 | throw new Error('Unknown node type ' + type + '.');
|
---|
73 | }
|
---|
74 | }
|
---|
75 |
|
---|
76 | for (i = 0, iz = children.length; i < iz; ++i) {
|
---|
77 | child = node[children[i]];
|
---|
78 | if (child) {
|
---|
79 | if (Array.isArray(child)) {
|
---|
80 | for (j = 0, jz = child.length; j < jz; ++j) {
|
---|
81 | if (child[j]) {
|
---|
82 | if (isNode(child[j]) || isProperty(type, children[i])) {
|
---|
83 | this.visit(child[j]);
|
---|
84 | }
|
---|
85 | }
|
---|
86 | }
|
---|
87 | } else if (isNode(child)) {
|
---|
88 | this.visit(child);
|
---|
89 | }
|
---|
90 | }
|
---|
91 | }
|
---|
92 | };
|
---|
93 |
|
---|
94 | /* Dispatching node. */
|
---|
95 | Visitor.prototype.visit = function (node) {
|
---|
96 | var type;
|
---|
97 |
|
---|
98 | if (node == null) {
|
---|
99 | return;
|
---|
100 | }
|
---|
101 |
|
---|
102 | type = node.type || estraverse.Syntax.Property;
|
---|
103 | if (this.__visitor[type]) {
|
---|
104 | this.__visitor[type].call(this, node);
|
---|
105 | return;
|
---|
106 | }
|
---|
107 | this.visitChildren(node);
|
---|
108 | };
|
---|
109 |
|
---|
110 | exports.version = require('./package.json').version;
|
---|
111 | exports.Visitor = Visitor;
|
---|
112 | exports.visit = function (node, visitor, options) {
|
---|
113 | var v = new Visitor(visitor, options);
|
---|
114 | v.visit(node);
|
---|
115 | };
|
---|
116 | }());
|
---|
117 | /* vim: set sw=4 ts=4 et tw=80 : */
|
---|