silphius/app/astride/traverse.js

export const TRAVERSAL_PATH = {
  ArrayExpression: ['elements'],
  ArrayPattern: ['elements'],
  AssignmentExpression: ['left', 'right'],
  AwaitExpression: ['argument'],
  BinaryExpression: ['left', 'right'],
  BlockStatement: ['body'],
  BreakStatement: [],
  CallExpression: ['arguments', 'callee'],
  ChainExpression: ['expression'],
  ConditionalExpression: ['alternate', 'consequent', 'test'],
  DoWhileStatement: ['body', 'test'],
  ExpressionStatement: ['expression'],
  ForOfStatement: ['body', 'left', 'right'],
  ForStatement: ['body', 'init', 'test', 'update'],
  Identifier: [],
  IfStatement: ['alternate', 'consequent', 'test'],
  MemberExpression: ['object', 'property'],
  Literal: [],
  LogicalExpression: ['left', 'right'],
  ObjectExpression: ['properties'],
  ObjectPattern: ['properties'],
  Program: ['body'],
  Property: ['key', 'value'],
  ReturnStatement: ['argument'],
  SpreadElement: ['argument'],
  UnaryExpression: ['argument'],
  UpdateExpression: ['argument'],
  VariableDeclaration: ['declarations'],
  VariableDeclarator: ['id', 'init'],
  WhileStatement: ['body', 'test'],
};

export default function traverse(node, visitor) {
  /* v8 ignore next 3 */
  if (!(node.type in TRAVERSAL_PATH)) {
    throw new Error(`node type ${node.type} not traversable. (${Object.keys(node).join(', ')})`);
  }
  visitor(node, 'enter');
  const path = TRAVERSAL_PATH[node.type];
  const children = [];
  for (const key of path) {
    children.push(...(Array.isArray(node[key]) ? node[key] : [node[key]]));
  }
  for (const child of children) {
    if (child) {
      traverse(child, visitor);
    }
  }
  visitor(node, 'exit');
}