ComposedTreeAncestorIterator.h [plain text]
#include "HTMLSlotElement.h"
#include "PseudoElement.h"
#include "ShadowRoot.h"
#ifndef ComposedTreeAncestorIterator_h
#define ComposedTreeAncestorIterator_h
namespace WebCore {
class HTMLSlotElement;
class ComposedTreeAncestorIterator {
public:
ComposedTreeAncestorIterator();
ComposedTreeAncestorIterator(Node& current);
Element& operator*() { return get(); }
Element* operator->() { return &get(); }
bool operator==(const ComposedTreeAncestorIterator& other) const { return m_current == other.m_current; }
bool operator!=(const ComposedTreeAncestorIterator& other) const { return m_current != other.m_current; }
ComposedTreeAncestorIterator& operator++() { return traverseParent(); }
Element& get() { return downcast<Element>(*m_current); }
ComposedTreeAncestorIterator& traverseParent();
private:
void traverseParentInShadowTree();
Node* m_current { 0 };
};
inline ComposedTreeAncestorIterator::ComposedTreeAncestorIterator()
{
}
inline ComposedTreeAncestorIterator::ComposedTreeAncestorIterator(Node& current)
: m_current(¤t)
{
ASSERT(!is<ShadowRoot>(m_current));
}
inline ComposedTreeAncestorIterator& ComposedTreeAncestorIterator::traverseParent()
{
auto* parent = m_current->parentNode();
if (!parent) {
m_current = nullptr;
return *this;
}
if (is<ShadowRoot>(*parent)) {
m_current = downcast<ShadowRoot>(*parent).host();
return *this;
}
if (!is<Element>(*parent)) {
m_current = nullptr;
return *this;
};
if (auto* shadowRoot = parent->shadowRoot()) {
m_current = shadowRoot->findAssignedSlot(*m_current);
return *this;
}
m_current = parent;
return *this;
}
class ComposedTreeAncestorAdapter {
public:
using iterator = ComposedTreeAncestorIterator;
ComposedTreeAncestorAdapter(Node& node)
: m_node(node)
{ }
iterator begin()
{
if (is<ShadowRoot>(m_node))
return iterator(*downcast<ShadowRoot>(m_node).host());
if (is<PseudoElement>(m_node))
return iterator(*downcast<PseudoElement>(m_node).hostElement());
return iterator(m_node).traverseParent();
}
iterator end()
{
return iterator();
}
Element* first()
{
auto it = begin();
if (it == end())
return nullptr;
return &it.get();
}
private:
Node& m_node;
};
inline ComposedTreeAncestorAdapter composedTreeAncestors(Node& node)
{
return ComposedTreeAncestorAdapter(node);
}
}
#endif