node-source-walk

Synchronously execute a callback on every node of a file's AST and stop walking whenever you see fit.

1npm install node-source-walk

Usage

1const Walker = require('node-source-walk');
2
3const walker = new Walker();
4
5// Assume src is the string contents of myfile.js
6// or the AST of an outside parse of myfile.js
7walker.walk(src, node => {
8  if (node.type === whateverImLookingFor) {
9    // No need to keep traversing since we found what we wanted
10    walker.stopWalking();
11  }
12});

By default, Walker will use @babel/parser (supporting ES6, JSX, Flow, and all other available @babel/parser plugins) and the sourceType: module, but you can change any of the defaults as follows:

1const walker = new Walker({
2  sourceType: 'script',
3  // If you don't like experimental plugins
4  plugins: [
5    'jsx',
6    'flow'
7  ]
8});

• The supplied options are passed through to the parser, so you can configure it according to @babel/parser's documentation.

Swap out the parser

If you want to supply your own parser, you can do:

1const walker = new Walker({
2  parser: mySweetParser
3});

• The custom parser must have a .parse method that takes in a string and returns an object/AST.
• All of the other options supplied to the Walker constructor will be passed along as parser options to your chosen parser.

API

walk(src, callback)

• Recursively walks the given src from top to bottom
• src: the contents of a file or its (already parsed) AST
• callback: a function that is called for every visited node
  • The argument passed to callback will be the currently visited node.

moonwalk(node, callback)

• Recursively walks up an AST starting from the given node. This is a traversal that's in the opposite direction of walk and traverse
• node: a valid AST node
• callback: a function that is called for every node (specifically via visiting the parent(s) of every node recursively)
  • The argument passed to callback will be the currently visited node.

stopWalking()

• Halts further walking of the AST until another manual call of walk or moonwalk
• This is super-beneficial when dealing with large source files (or ASTs)

traverse(node, callback)

• Allows you to traverse an AST node and execute a callback on it
• Callback should expect the first argument to be an AST node, similar to walk's callback

parse(src)

• Uses the options supplied to Walker to parse the given source code string and return its AST using the configured parser (or @babel/parser by default).

License

MIT