Spaces:
Running
Running
/* | |
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 : */ | |