123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687 |
- 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);
- };
|