Test if a selector matches a given element

后端 未结 7 1632
庸人自扰
庸人自扰 2020-11-27 05:58

Is there any way to test if a selector would match a given DOM Element? Preferably, without the use of an external library like Sizzle. This is for a library and I would li

7条回答
  •  南方客
    南方客 (楼主)
    2020-11-27 06:41

    For best performance, use the browser implementations ((moz|webkit|o|ms)matchesSelector) where possible. When you can't do that, here is a manual implementation.

    An important case to consider is testing selectors for elements not attached to the document.

    Here's an approach that handles this situation. If it turns out the the element in question is not attached to the document, crawl up the tree to find the highest ancestor (the last non-null parentNode) and drop that into a DocumentFragment. Then from that DocumentFragment call querySelectorAll and see if the your element is in the resulting NodeList.

    Here is the code.

    The document

    Here's a document structure we'll be working with. We'll grab the .element and test whether it matches the selectors li and .container *.

    
    
      
        

    Header 1

    • one
    • two
    • three

    Header 2

    • one
    • two
    • three
    Footer

    Searching with document.querySelectorAll

    Here is a matchesSelector function that uses document.querySelectorAll.

    // uses document.querySelectorAll
    function matchesSelector(selector, element) {
      var all = document.querySelectorAll(selector);
      for (var i = 0; i < all.length; i++) {
        if (all[i] === element) {
          return true;
        }
      }
      return false;
    }
    

    This works as long as that element is in the document.

    // this works because the element is in the document
    console.log("Part 1");
    var element = document.querySelector(".element");
    console.log(matchesSelector("li", element)); // true
    console.log(matchesSelector(".container *", element)); // true
    

    However, it fails if the element is removed from the document.

    // but they don't work if we remove the article from the document
    console.log("Part 2");
    var article = document.querySelector("article");
    article.parentNode.removeChild(article);
    console.log(matchesSelector("li", element)); // false
    console.log(matchesSelector(".container *", element)); // false
    

    Searching within a DocumentFragment

    The fix requires searching whatever subtree that element happens to be in. Here's an updated function named matchesSelector2.

    // uses a DocumentFragment if element is not attached to the document
    function matchesSelector2(selector, element) {
      if (document.contains(element)) {
        return matchesSelector(selector, element);
      }
      var node = element;
      var root = document.createDocumentFragment();
      while (node.parentNode) {
        node = node.parentNode;
      }
      root.appendChild(node);
      var all = root.querySelectorAll(selector);
      for (var i = 0; i < all.length; i++) {
        if (all[i] === element) {
          root.removeChild(node);
          return true;
        }
      }
      root.removeChild(node);
      return false;
    }
    

    Now we see that matchesSelector2 works even though the element is in a subtree that is detached from the document.

    // but they will work if we use matchesSelector2
    console.log("Part 3");
    console.log(matchesSelector2("li", element)); // true
    console.log(matchesSelector2(".container *", element)); // true
    

    You can see this working at jsfiddle.

    Putting it all together

    Here's the final implementation I came up with:

    function is(element, selector) {
      var node = element;
      var result = false;
      var root, frag;
    
      // crawl up the tree
      while (node.parentNode) {
        node = node.parentNode;
      }
    
      // root must be either a Document or a DocumentFragment
      if (node instanceof Document || node instanceof DocumentFragment) {
        root = node;
      } else {
        root = frag = document.createDocumentFragment();
        frag.appendChild(node);
      }
    
      // see if selector matches
      var matches = root.querySelectorAll(selector);
      for (var i = 0; i < matches.length; i++) {
        if (this === matches.item(i)) {
          result = true;
          break;
        }
      }
    
      // detach from DocumentFragment and return result
      while (frag && frag.firstChild) {
        frag.removeChild(frag.firstChild);
      }
      return result;
    }
    

    An important note is that jQuery's is implementation is much faster. The first optimization I would look into is avoiding crawling up the tree if we don't have to. To do this you could look at the right-most part of the selector and test whether this matches the element. However, beware that if the selector is actually multiple selectors separated by commas, then you'll have to test each one. At this point you're building a CSS selector parser, so you might as well use a library.

提交回复
热议问题