Home | History | Annotate | Download | only in dom 
      1 /*
      2  * Copyright (C) 2011, 2013 Apple Inc. All rights reserved.
      3  * Copyright (C) 2014 Samsung Electronics. All rights reserved.
      4  *
      5  * Redistribution and use in source and binary forms, with or without
      6  * modification, are permitted provided that the following conditions
      7  * are met:
      8  *
      9  * 1.  Redistributions of source code must retain the above copyright
     10  *     notice, this list of conditions and the following disclaimer.
     11  * 2.  Redistributions in binary form must reproduce the above copyright
     12  *     notice, this list of conditions and the following disclaimer in the
     13  *     documentation and/or other materials provided with the distribution.
     14  *
     15  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
     16  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
     17  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
     18  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
     19  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
     20  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
     21  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
     22  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
     23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
     24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
     25  */
     26 
     27 #ifndef SelectorQuery_h
     28 #define SelectorQuery_h
     29 
     30 #include "core/css/CSSSelectorList.h"
     31 #include "platform/heap/Handle.h"
     32 #include "wtf/HashMap.h"
     33 #include "wtf/Vector.h"
     34 #include "wtf/text/AtomicStringHash.h"
     35 
     36 namespace WebCore {
     37 
     38 class CSSSelector;
     39 class ContainerNode;
     40 class Document;
     41 class Element;
     42 class ExceptionState;
     43 class Node;
     44 class SimpleNodeList;
     45 class StaticNodeList;
     46 class SpaceSplitString;
     47 
     48 class SelectorDataList {
     49 public:
     50     void initialize(const CSSSelectorList&);
     51     bool matches(Element&) const;
     52     PassRefPtrWillBeRawPtr<StaticNodeList> queryAll(ContainerNode& rootNode) const;
     53     PassRefPtrWillBeRawPtr<Element> queryFirst(ContainerNode& rootNode) const;
     54 
     55 private:
     56     bool canUseFastQuery(const ContainerNode& rootNode) const;
     57     bool selectorMatches(const CSSSelector&, Element&, const ContainerNode&) const;
     58 
     59     template <typename SelectorQueryTrait>
     60     void collectElementsByClassName(ContainerNode& rootNode, const AtomicString& className, typename SelectorQueryTrait::OutputType&) const;
     61     template <typename SelectorQueryTrait>
     62     void collectElementsByTagName(ContainerNode& rootNode, const QualifiedName& tagName, typename SelectorQueryTrait::OutputType&) const;
     63 
     64     template <typename SelectorQueryTrait>
     65     void findTraverseRootsAndExecute(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
     66 
     67     enum MatchTraverseRootState { DoesNotMatchTraverseRoots, MatchesTraverseRoots };
     68     template <typename SelectorQueryTrait>
     69     void executeForTraverseRoot(const CSSSelector&, ContainerNode* traverseRoot, MatchTraverseRootState, ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
     70     template <typename SelectorQueryTrait, typename SimpleElementListType>
     71     void executeForTraverseRoots(const CSSSelector&, SimpleElementListType& traverseRoots, MatchTraverseRootState, ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
     72 
     73     template <typename SelectorQueryTrait>
     74     bool selectorListMatches(ContainerNode& rootNode, Element&, typename SelectorQueryTrait::OutputType&) const;
     75     template <typename SelectorQueryTrait>
     76     void executeSlow(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
     77     template <typename SelectorQueryTrait>
     78     void executeSlowTraversingShadowTree(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
     79     template <typename SelectorQueryTrait>
     80     void execute(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
     81     const CSSSelector* selectorForIdLookup(const CSSSelector&) const;
     82 
     83     Vector<const CSSSelector*> m_selectors;
     84     bool m_crossesTreeBoundary;
     85 };
     86 
     87 class SelectorQuery {
     88     WTF_MAKE_NONCOPYABLE(SelectorQuery);
     89     WTF_MAKE_FAST_ALLOCATED;
     90 public:
     91     static PassOwnPtr<SelectorQuery> adopt(CSSSelectorList&);
     92 
     93     bool matches(Element&) const;
     94     PassRefPtrWillBeRawPtr<StaticNodeList> queryAll(ContainerNode& rootNode) const;
     95     PassRefPtrWillBeRawPtr<Element> queryFirst(ContainerNode& rootNode) const;
     96 private:
     97     explicit SelectorQuery(CSSSelectorList&);
     98 
     99     SelectorDataList m_selectors;
    100     CSSSelectorList m_selectorList;
    101 };
    102 
    103 class SelectorQueryCache {
    104     WTF_MAKE_FAST_ALLOCATED;
    105 public:
    106     SelectorQuery* add(const AtomicString&, const Document&, ExceptionState&);
    107     void invalidate();
    108 
    109 private:
    110     HashMap<AtomicString, OwnPtr<SelectorQuery> > m_entries;
    111 };
    112 
    113 }
    114 
    115 #endif
    116