blob: 43bd446db1a95e3aedba5e185328420bc025d67a [file] [log] [blame]
Leo Repp58b9f112021-11-22 11:57:47 +01001var ElementType = require("domelementtype");
2var isTag = exports.isTag = ElementType.isTag;
3
4exports.testElement = function(options, element){
5 for(var key in options){
6 if(!options.hasOwnProperty(key));
7 else if(key === "tag_name"){
8 if(!isTag(element) || !options.tag_name(element.name)){
9 return false;
10 }
11 } else if(key === "tag_type"){
12 if(!options.tag_type(element.type)) return false;
13 } else if(key === "tag_contains"){
14 if(isTag(element) || !options.tag_contains(element.data)){
15 return false;
16 }
17 } else if(!element.attribs || !options[key](element.attribs[key])){
18 return false;
19 }
20 }
21 return true;
22};
23
24var Checks = {
25 tag_name: function(name){
26 if(typeof name === "function"){
27 return function(elem){ return isTag(elem) && name(elem.name); };
28 } else if(name === "*"){
29 return isTag;
30 } else {
31 return function(elem){ return isTag(elem) && elem.name === name; };
32 }
33 },
34 tag_type: function(type){
35 if(typeof type === "function"){
36 return function(elem){ return type(elem.type); };
37 } else {
38 return function(elem){ return elem.type === type; };
39 }
40 },
41 tag_contains: function(data){
42 if(typeof data === "function"){
43 return function(elem){ return !isTag(elem) && data(elem.data); };
44 } else {
45 return function(elem){ return !isTag(elem) && elem.data === data; };
46 }
47 }
48};
49
50function getAttribCheck(attrib, value){
51 if(typeof value === "function"){
52 return function(elem){ return elem.attribs && value(elem.attribs[attrib]); };
53 } else {
54 return function(elem){ return elem.attribs && elem.attribs[attrib] === value; };
55 }
56}
57
58function combineFuncs(a, b){
59 return function(elem){
60 return a(elem) || b(elem);
61 };
62}
63
64exports.getElements = function(options, element, recurse, limit){
65 var funcs = Object.keys(options).map(function(key){
66 var value = options[key];
67 return key in Checks ? Checks[key](value) : getAttribCheck(key, value);
68 });
69
70 return funcs.length === 0 ? [] : this.filter(
71 funcs.reduce(combineFuncs),
72 element, recurse, limit
73 );
74};
75
76exports.getElementById = function(id, element, recurse){
77 if(!Array.isArray(element)) element = [element];
78 return this.findOne(getAttribCheck("id", id), element, recurse !== false);
79};
80
81exports.getElementsByTagName = function(name, element, recurse, limit){
82 return this.filter(Checks.tag_name(name), element, recurse, limit);
83};
84
85exports.getElementsByTagType = function(type, element, recurse, limit){
86 return this.filter(Checks.tag_type(type), element, recurse, limit);
87};