88 lines
2.4 KiB
JavaScript
88 lines
2.4 KiB
JavaScript
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);
|
|
};
|