dom/base/nsTraversal.cpp

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
   2  *
   3  * This Source Code Form is subject to the terms of the Mozilla Public
   4  * License, v. 2.0. If a copy of the MPL was not distributed with this
   5  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   6
   7 #include "nsTraversal.h"
   8
   9 #include "nsError.h"
  10 #include "nsINode.h"
  11 #include "mozilla/AutoRestore.h"
  12 #include "mozilla/dom/NodeFilterBinding.h"
  13
  14 #include "nsGkAtoms.h"
  15
  16 using namespace mozilla;
  17 using namespace mozilla::dom;
  18
  19 nsTraversal::nsTraversal(nsINode* aRoot, uint32_t aWhatToShow,
  20                          NodeFilter* aFilter)
  21     : mRoot(aRoot),
  22       mWhatToShow(aWhatToShow),
  23       mFilter(aFilter),
  24       mInAcceptNode(false) {
  25   NS_ASSERTION(aRoot, "invalid root in call to nsTraversal constructor");
  26 }
  27
  28 nsTraversal::~nsTraversal() { /* destructor code */
  29 }
  30
  31 /*
  32  * Tests if and how a node should be filtered. Uses mWhatToShow and
  33  * mFilter to test the node.
  34  * @param aNode     Node to test
  35  * @param aResult   Whether we succeeded
  36  * @returns         Filtervalue. See NodeFilter.webidl
  37  */
  38 int16_t nsTraversal::TestNode(nsINode* aNode, mozilla::ErrorResult& aResult,
  39                               nsCOMPtr<nsINode>* aUnskippedNode) {
  40   if (mInAcceptNode) {
  41     aResult.Throw(NS_ERROR_DOM_INVALID_STATE_ERR);
  42     return 0;
  43   }
  44
  45   uint16_t nodeType = aNode->NodeType();
  46
  47   if (nodeType <= 12 && !((1 << (nodeType - 1)) & mWhatToShow)) {
  48     return NodeFilter_Binding::FILTER_SKIP;
  49   }
  50
  51   if (aUnskippedNode) {
  52     *aUnskippedNode = aNode;
  53   }
  54
  55   if (!mFilter) {
  56     // No filter, just accept
  57     return NodeFilter_Binding::FILTER_ACCEPT;
  58   }
  59
  60   AutoRestore<bool> inAcceptNode(mInAcceptNode);
  61   mInAcceptNode = true;
  62   // No need to pass in an execution reason, since the generated default,
  63   // "NodeFilter.acceptNode", is pretty much exactly what we'd say anyway.
  64   return mFilter->AcceptNode(*aNode, aResult, nullptr,
  65                              CallbackObject::eRethrowExceptions);
  66 }