#pragma once
#include "CSSSelectorList.h"
#include "ExceptionOr.h"
#include "NodeList.h"
#include "SelectorCompiler.h"
#include <wtf/HashMap.h>
#include <wtf/Vector.h>
#include <wtf/text/AtomicStringHash.h>
namespace WebCore {
class CSSSelector;
class ContainerNode;
class Document;
class Element;
class SelectorDataList {
public:
explicit SelectorDataList(const CSSSelectorList&);
bool matches(Element&) const;
Element* closest(Element&) const;
Ref<NodeList> queryAll(ContainerNode& rootNode) const;
Element* queryFirst(ContainerNode& rootNode) const;
private:
struct SelectorData {
SelectorData(const CSSSelector* selector)
: selector(selector)
#if ENABLE(CSS_SELECTOR_JIT) && CSS_SELECTOR_JIT_PROFILING
, m_compiledSelectorUseCount(0)
#endif
{
}
const CSSSelector* selector;
#if ENABLE(CSS_SELECTOR_JIT)
mutable JSC::MacroAssemblerCodeRef compiledSelectorCodeRef;
mutable SelectorCompilationStatus compilationStatus;
#if CSS_SELECTOR_JIT_PROFILING
~SelectorData()
{
if (compiledSelectorCodeRef.code().executableAddress())
dataLogF("SelectorData compiled selector %d \"%s\"\n", m_compiledSelectorUseCount, selector->selectorText().utf8().data());
}
mutable unsigned m_compiledSelectorUseCount;
void compiledSelectorUsed() const { m_compiledSelectorUseCount++; }
#endif
#endif // ENABLE(CSS_SELECTOR_JIT)
};
bool selectorMatches(const SelectorData&, Element&, const ContainerNode& rootNode) const;
Element* selectorClosest(const SelectorData&, Element&, const ContainerNode& rootNode) const;
template <typename SelectorQueryTrait> void execute(ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeFastPathForIdSelector(const ContainerNode& rootNode, const SelectorData&, const CSSSelector* idSelector, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleTagNameSelectorData(const ContainerNode& rootNode, const SelectorData&, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleClassNameSelectorData(const ContainerNode& rootNode, const SelectorData&, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleSelectorData(const ContainerNode& rootNode, const ContainerNode& searchRootNode, const SelectorData&, typename SelectorQueryTrait::OutputType&) const;
template <typename SelectorQueryTrait> void executeSingleMultiSelectorData(const ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
#if ENABLE(CSS_SELECTOR_JIT)
template <typename SelectorQueryTrait> void executeCompiledSimpleSelectorChecker(const ContainerNode& searchRootNode, SelectorCompiler::QuerySelectorSimpleSelectorChecker, typename SelectorQueryTrait::OutputType&, const SelectorData&) const;
template <typename SelectorQueryTrait> void executeCompiledSelectorCheckerWithCheckingContext(const ContainerNode& rootNode, const ContainerNode& searchRootNode, SelectorCompiler::QuerySelectorSelectorCheckerWithCheckingContext, typename SelectorQueryTrait::OutputType&, const SelectorData&) const;
template <typename SelectorQueryTrait> void executeCompiledSingleMultiSelectorData(const ContainerNode& rootNode, typename SelectorQueryTrait::OutputType&) const;
static bool compileSelector(const SelectorData&, const ContainerNode& rootNode);
#endif // ENABLE(CSS_SELECTOR_JIT)
Vector<SelectorData> m_selectors;
mutable enum MatchType {
CompilableSingle,
CompilableSingleWithRootFilter,
CompilableMultipleSelectorMatch,
CompiledSingle,
CompiledSingleWithRootFilter,
CompiledMultipleSelectorMatch,
SingleSelector,
SingleSelectorWithRootFilter,
RightMostWithIdMatch,
TagNameMatch,
ClassNameMatch,
MultipleSelectorMatch,
} m_matchType;
};
class SelectorQuery {
WTF_MAKE_NONCOPYABLE(SelectorQuery);
WTF_MAKE_FAST_ALLOCATED;
public:
explicit SelectorQuery(CSSSelectorList&&);
bool matches(Element&) const;
Element* closest(Element&) const;
Ref<NodeList> queryAll(ContainerNode& rootNode) const;
Element* queryFirst(ContainerNode& rootNode) const;
private:
CSSSelectorList m_selectorList;
SelectorDataList m_selectors;
};
class SelectorQueryCache {
WTF_MAKE_FAST_ALLOCATED;
public:
ExceptionOr<SelectorQuery&> add(const String&, Document&);
private:
HashMap<String, std::unique_ptr<SelectorQuery>> m_entries;
};
inline bool SelectorQuery::matches(Element& element) const
{
return m_selectors.matches(element);
}
inline Element* SelectorQuery::closest(Element& element) const
{
return m_selectors.closest(element);
}
inline Ref<NodeList> SelectorQuery::queryAll(ContainerNode& rootNode) const
{
return m_selectors.queryAll(rootNode);
}
inline Element* SelectorQuery::queryFirst(ContainerNode& rootNode) const
{
return m_selectors.queryFirst(rootNode);
}
}