cheerio-select/lib/select.js

Selector engine for cheerio

/*
 * Module dependencies
 */
var CSSselect = require('CSSselect'),
    isArray = Array.isArray;

/*
 * Select function
 */
exports = module.exports = function(query, dom) {
  dom = normalize(dom);

  return CSSselect.iterate(query, dom);
};

/*
 * Normalize the dom
 */
var normalize = exports.normalize = function(dom) {
  dom = dom.cheerio ? dom.toArray() : dom;
  dom = isArray(dom) ? dom : [dom];

  var len = dom.length,
      out = [],
      elem;

  for(var i = 0; i < len; i++) {
    elem = dom[i];
    if(elem.type === 'root') {
      out = out.concat(elem.children || []);
    } else {
      out.push(elem);
    }
  }

  return out;
};