|
|
- /*
- Copyright (C) 2014 Yusuke Suzuki <utatane.tea@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 HOLDERS AND CONTRIBUTORS "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 <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.
- */
- (function () {
- 'use strict';
-
- var estraverse = require('estraverse');
-
- function isNode(node) {
- if (node == null) {
- return false;
- }
- return typeof node === 'object' && typeof node.type === 'string';
- }
-
- function isProperty(nodeType, key) {
- return (nodeType === estraverse.Syntax.ObjectExpression || nodeType === estraverse.Syntax.ObjectPattern) && key === 'properties';
- }
-
- function Visitor(visitor, options) {
- options = options || {};
-
- this.__visitor = visitor || this;
- this.__childVisitorKeys = options.childVisitorKeys
- ? Object.assign({}, estraverse.VisitorKeys, options.childVisitorKeys)
- : estraverse.VisitorKeys;
- if (options.fallback === 'iteration') {
- this.__fallback = Object.keys;
- } else if (typeof options.fallback === 'function') {
- this.__fallback = options.fallback;
- }
- }
-
- /* Default method for visiting children.
- * When you need to call default visiting operation inside custom visiting
- * operation, you can use it with `this.visitChildren(node)`.
- */
- Visitor.prototype.visitChildren = function (node) {
- var type, children, i, iz, j, jz, child;
-
- if (node == null) {
- return;
- }
-
- type = node.type || estraverse.Syntax.Property;
-
- children = this.__childVisitorKeys[type];
- if (!children) {
- if (this.__fallback) {
- children = this.__fallback(node);
- } else {
- throw new Error('Unknown node type ' + type + '.');
- }
- }
-
- for (i = 0, iz = children.length; i < iz; ++i) {
- child = node[children[i]];
- if (child) {
- if (Array.isArray(child)) {
- for (j = 0, jz = child.length; j < jz; ++j) {
- if (child[j]) {
- if (isNode(child[j]) || isProperty(type, children[i])) {
- this.visit(child[j]);
- }
- }
- }
- } else if (isNode(child)) {
- this.visit(child);
- }
- }
- }
- };
-
- /* Dispatching node. */
- Visitor.prototype.visit = function (node) {
- var type;
-
- if (node == null) {
- return;
- }
-
- type = node.type || estraverse.Syntax.Property;
- if (this.__visitor[type]) {
- this.__visitor[type].call(this, node);
- return;
- }
- this.visitChildren(node);
- };
-
- exports.version = require('./package.json').version;
- exports.Visitor = Visitor;
- exports.visit = function (node, visitor, options) {
- var v = new Visitor(visitor, options);
- v.visit(node);
- };
- }());
- /* vim: set sw=4 ts=4 et tw=80 : */
|