findNodesBy.js

findNodesBy.js

/* globals module */

'using strict';

module.exports = function (options) {
  if (arguments.length === 2) {
    options = {
      body: arguments[0],
      match: arguments[1]
    };
  }

  options.by = options.by || 'type';
  options.visitor = options.visitor || function (k, v, p, a) { return v; };

  var nodes = [],
      isMatch = (function () {
        var which = {
          '[object String]':   function (k, v) { return v === options.match; },
          '[object RegExp]':   function (k, v) { return options.match.test(v); },
          '[object Function]': function (k, v, p, a) { return options.match(k, v, p, a); },
        },
        kind = ({}).toString.call(options.match);

        return which[kind];
      }());

  JSON.stringify(
    options.body,
    function (key, value) {
      var node;

      if (value && value[options.by] && isMatch(key, value[options.by], this, nodes)) {
        node = options.visitor(key, value, this, nodes);
        if (node) {
          nodes.push(node);
        }
      }

      return value;
    }
  );

  return nodes;
};