WebCore/dom/SelectorNodeList.cpp

   1 /*
   2  * Copyright (C) 2007, 2008 Apple Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  * 1.  Redistributions of source code must retain the above copyright
   9  *     notice, this list of conditions and the following disclaimer.
  10  * 2.  Redistributions in binary form must reproduce the above copyright
  11  *     notice, this list of conditions and the following disclaimer in the
  12  *     documentation and/or other materials provided with the distribution.
  13  * 3.  Neither the name of Apple Computer, Inc. ("Apple") nor the names of
  14  *     its contributors may be used to endorse or promote products derived
  15  *     from this software without specific prior written permission.
  16  *
  17  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  18  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  19  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  20  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  21  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  22  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  23  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  24  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  26  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include "config.h"
  30 #include "SelectorNodeList.h"
  31
  32 #include "CSSSelector.h"
  33 #include "CSSSelectorList.h"
  34 #include "CSSStyleSelector.h"
  35 #include "Document.h"
  36 #include "Element.h"
  37 #include "HTMLNames.h"
  38
  39 namespace WebCore {
  40
  41 using namespace HTMLNames;
  42
  43 PassRefPtr<StaticNodeList> createSelectorNodeList(Node* rootNode, const CSSSelectorList& querySelectorList)
  44 {
  45     Vector<RefPtr<Node> > nodes;
  46     Document* document = rootNode->document();
  47     CSSSelector* onlySelector = querySelectorList.hasOneSelector() ? querySelectorList.first() : 0;
  48     bool strictParsing = !document->inCompatMode();
  49
  50     CSSStyleSelector::SelectorChecker selectorChecker(document, strictParsing);
  51
  52     if (strictParsing && rootNode->inDocument() && onlySelector && onlySelector->m_match == CSSSelector::Id && !document->containsMultipleElementsWithId(onlySelector->m_value)) {
  53         ASSERT(querySelectorList.first()->attribute() == idAttr);
  54         Element* element = document->getElementById(onlySelector->m_value);
  55         if (element && (rootNode->isDocumentNode() || element->isDescendantOf(rootNode)) && selectorChecker.checkSelector(onlySelector, element))
  56             nodes.append(element);
  57     } else {
  58         for (Node* n = rootNode->firstChild(); n; n = n->traverseNextNode(rootNode)) {
  59             if (n->isElementNode()) {
  60                 Element* element = static_cast<Element*>(n);
  61                 for (CSSSelector* selector = querySelectorList.first(); selector; selector = CSSSelectorList::next(selector)) {
  62                     if (selectorChecker.checkSelector(selector, element)) {
  63                         nodes.append(n);
  64                         break;
  65                     }
  66                 }
  67             }
  68         }
  69     }
  70
  71     return StaticNodeList::adopt(nodes);
  72 }
  73
  74 } // namespace WebCore