Show More
Commit Description:
Update README.rdoc
Commit Description:
Update README.rdoc
References:
File last commit:
Show/Diff file:
Action:
node_modules/dash-ast/index.js
| 84 lines
| 2.5 KiB
| application/javascript
| JavascriptLexer
|
| r789 | var assert = require('assert') | |||
| module.exports = dashAst | ||||
| /** | ||||
| * Call `cb` on each node in `ast`. If `cb` is an object, `cb.enter` is called before processing a Node's children, | ||||
| * and `cb.leave` is called after processing a Node's children. | ||||
| */ | ||||
| function dashAst (ast, cb) { | ||||
| assert(ast && typeof ast === 'object' && typeof ast.type === 'string', | ||||
| 'dash-ast: ast must be an AST node') | ||||
| if (typeof cb === 'object') { | ||||
| assert(typeof cb.enter === 'function' || typeof cb.leave === 'function', | ||||
| 'dash-ast: visitor must be an object with enter/leave functions') | ||||
| walk(ast, null, cb.enter || undefined, cb.leave || undefined) | ||||
| } else { | ||||
| assert(cb && typeof cb === 'function', | ||||
| 'dash-ast: callback must be a function') | ||||
| walk(ast, null, cb, undefined) | ||||
| } | ||||
| } | ||||
| /** | ||||
| * Call `cb` on each node in `ast`. Each node will have a `.parent` property. | ||||
| */ | ||||
| dashAst.withParent = function dashAstParent (ast, cb) { | ||||
| assert(ast && typeof ast === 'object' && typeof ast.type === 'string', | ||||
| 'dash-ast.withParent: ast must be an AST node') | ||||
| if (typeof cb === 'object') { | ||||
| assert(typeof cb.enter === 'function' || typeof cb.leave === 'function', | ||||
| 'dash-ast.withParent: visitor must be an object with enter/leave functions') | ||||
| var enter = cb.enter | ||||
| var leave = cb.leave | ||||
| walk(ast, null, function (node, parent) { | ||||
| node.parent = parent | ||||
| if (enter !== undefined) return enter(node) | ||||
| }, leave ? function (node) { leave(node) } : undefined) | ||||
| } else { | ||||
| assert(cb && typeof cb === 'function', | ||||
| 'dash-ast.withParent: callback must be a function') | ||||
| walk(ast, null, function (node, parent) { | ||||
| node.parent = parent | ||||
| return cb(node) | ||||
| }, undefined) | ||||
| } | ||||
| } | ||||
| function walk (node, parent, enter, leave) { | ||||
| var cont = enter !== undefined ? enter(node, parent) : undefined | ||||
| if (cont === false) return | ||||
| for (var k in node) { | ||||
| if (has(node, k)) { | ||||
| if (k === 'parent') continue | ||||
| if (isNode(node[k])) { | ||||
| walk(node[k], node, enter, leave) | ||||
| } else if (Array.isArray(node[k])) { | ||||
| walkArray(node[k], node, enter, leave) | ||||
| } | ||||
| } | ||||
| } | ||||
| if (leave !== undefined) leave(node, parent) | ||||
| } | ||||
| function walkArray (nodes, parent, enter, leave) { | ||||
| for (var i = 0; i < nodes.length; i++) { | ||||
| if (isNode(nodes[i])) walk(nodes[i], parent, enter, leave) | ||||
| } | ||||
| } | ||||
| function isNode (node) { | ||||
| return typeof node === 'object' && node && typeof node.type === 'string' | ||||
| } | ||||
| function has (obj, prop) { | ||||
| return Object.prototype.hasOwnProperty.call(obj, prop) | ||||
| } | ||||