81 lines
2.3 KiB
C++
81 lines
2.3 KiB
C++
/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
|
|
*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#include "nsTraversal.h"
|
|
|
|
#include "nsIDOMNode.h"
|
|
#include "nsError.h"
|
|
#include "nsINode.h"
|
|
#include "mozilla/AutoRestore.h"
|
|
|
|
#include "nsGkAtoms.h"
|
|
|
|
using namespace mozilla;
|
|
using namespace mozilla::dom;
|
|
|
|
nsTraversal::nsTraversal(nsINode *aRoot,
|
|
uint32_t aWhatToShow,
|
|
NodeFilterHolder aFilter) :
|
|
mRoot(aRoot),
|
|
mWhatToShow(aWhatToShow),
|
|
mFilter(Move(aFilter)),
|
|
mInAcceptNode(false)
|
|
{
|
|
NS_ASSERTION(aRoot, "invalid root in call to nsTraversal constructor");
|
|
}
|
|
|
|
nsTraversal::~nsTraversal()
|
|
{
|
|
/* destructor code */
|
|
}
|
|
|
|
/*
|
|
* Tests if and how a node should be filtered. Uses mWhatToShow and
|
|
* mFilter to test the node.
|
|
* @param aNode Node to test
|
|
* @param aResult Whether we succeeded
|
|
* @returns Filtervalue. See nsIDOMNodeFilter.idl
|
|
*/
|
|
int16_t
|
|
nsTraversal::TestNode(nsINode* aNode, mozilla::ErrorResult& aResult)
|
|
{
|
|
if (mInAcceptNode) {
|
|
aResult.Throw(NS_ERROR_DOM_INVALID_STATE_ERR);
|
|
return 0;
|
|
}
|
|
|
|
uint16_t nodeType = aNode->NodeType();
|
|
|
|
if (nodeType <= 12 && !((1 << (nodeType-1)) & mWhatToShow)) {
|
|
return nsIDOMNodeFilter::FILTER_SKIP;
|
|
}
|
|
|
|
if (!mFilter.GetISupports()) {
|
|
// No filter, just accept
|
|
return nsIDOMNodeFilter::FILTER_ACCEPT;
|
|
}
|
|
|
|
if (mFilter.HasWebIDLCallback()) {
|
|
AutoRestore<bool> inAcceptNode(mInAcceptNode);
|
|
mInAcceptNode = true;
|
|
// No need to pass in an execution reason, since the generated default,
|
|
// "NodeFilter.acceptNode", is pretty much exactly what we'd say anyway.
|
|
return mFilter.GetWebIDLCallback()->
|
|
AcceptNode(*aNode, aResult, nullptr,
|
|
CallbackObject::eRethrowExceptions);
|
|
}
|
|
|
|
nsCOMPtr<nsIDOMNode> domNode = do_QueryInterface(aNode);
|
|
AutoRestore<bool> inAcceptNode(mInAcceptNode);
|
|
mInAcceptNode = true;
|
|
int16_t filtered;
|
|
nsresult rv = mFilter.GetXPCOMCallback()->AcceptNode(domNode, &filtered);
|
|
if (NS_FAILED(rv)) {
|
|
aResult.Throw(rv);
|
|
}
|
|
return filtered;
|
|
}
|