修改禁引可视域

‘liusuyi’

2023-10-30 3c450768402293d86350e8420f365210c83233fa

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
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);
};