| var ElementType = require("domelementtype"); |
| var isTag = exports.isTag = ElementType.isTag; |
| |
| exports.testElement = function(options, element){ |
| for(var key in options){ |
| if(!options.hasOwnProperty(key)); |
| else if(key === "tag_name"){ |
| if(!isTag(element) || !options.tag_name(element.name)){ |
| return false; |
| } |
| } else if(key === "tag_type"){ |
| if(!options.tag_type(element.type)) return false; |
| } else if(key === "tag_contains"){ |
| if(isTag(element) || !options.tag_contains(element.data)){ |
| return false; |
| } |
| } else if(!element.attribs || !options[key](element.attribs[key])){ |
| return false; |
| } |
| } |
| return true; |
| }; |
| |
| var Checks = { |
| tag_name: function(name){ |
| if(typeof name === "function"){ |
| return function(elem){ return isTag(elem) && name(elem.name); }; |
| } else if(name === "*"){ |
| return isTag; |
| } else { |
| return function(elem){ return isTag(elem) && elem.name === name; }; |
| } |
| }, |
| tag_type: function(type){ |
| if(typeof type === "function"){ |
| return function(elem){ return type(elem.type); }; |
| } else { |
| return function(elem){ return elem.type === type; }; |
| } |
| }, |
| tag_contains: function(data){ |
| if(typeof data === "function"){ |
| return function(elem){ return !isTag(elem) && data(elem.data); }; |
| } else { |
| return function(elem){ return !isTag(elem) && elem.data === data; }; |
| } |
| } |
| }; |
| |
| function getAttribCheck(attrib, value){ |
| if(typeof value === "function"){ |
| return function(elem){ return elem.attribs && value(elem.attribs[attrib]); }; |
| } else { |
| return function(elem){ return elem.attribs && elem.attribs[attrib] === value; }; |
| } |
| } |
| |
| function combineFuncs(a, b){ |
| return function(elem){ |
| return a(elem) || b(elem); |
| }; |
| } |
| |
| exports.getElements = function(options, element, recurse, limit){ |
| var funcs = Object.keys(options).map(function(key){ |
| var value = options[key]; |
| return key in Checks ? Checks[key](value) : getAttribCheck(key, value); |
| }); |
| |
| return funcs.length === 0 ? [] : this.filter( |
| funcs.reduce(combineFuncs), |
| element, recurse, limit |
| ); |
| }; |
| |
| exports.getElementById = function(id, element, recurse){ |
| if(!Array.isArray(element)) element = [element]; |
| return this.findOne(getAttribCheck("id", id), element, recurse !== false); |
| }; |
| |
| exports.getElementsByTagName = function(name, element, recurse, limit){ |
| return this.filter(Checks.tag_name(name), element, recurse, limit); |
| }; |
| |
| exports.getElementsByTagType = function(type, element, recurse, limit){ |
| return this.filter(Checks.tag_type(type), element, recurse, limit); |
| }; |