#pragma once
#include "HTMLConstructionSite.h"
#include "HTMLParserOptions.h"
#include <wtf/text/StringBuilder.h>
#include <wtf/text/TextPosition.h>
namespace WebCore {
class JSCustomElementInterface;
class HTMLDocumentParser;
class ScriptElement;
struct CustomElementConstructionData {
CustomElementConstructionData(Ref<JSCustomElementInterface>&&, const AtomicString& name, Vector<Attribute>&&);
~CustomElementConstructionData();
Ref<JSCustomElementInterface> elementInterface;
AtomicString name;
Vector<Attribute> attributes;
};
class HTMLTreeBuilder {
WTF_MAKE_FAST_ALLOCATED;
public:
HTMLTreeBuilder(HTMLDocumentParser&, HTMLDocument&, ParserContentPolicy, const HTMLParserOptions&);
HTMLTreeBuilder(HTMLDocumentParser&, DocumentFragment&, Element& contextElement, ParserContentPolicy, const HTMLParserOptions&);
void setShouldSkipLeadingNewline(bool);
~HTMLTreeBuilder();
bool isParsingFragment() const;
void constructTree(AtomicHTMLToken&&);
bool isParsingTemplateContents() const;
bool hasParserBlockingScriptWork() const;
RefPtr<ScriptElement> takeScriptToProcess(TextPosition& scriptStartPosition);
std::unique_ptr<CustomElementConstructionData> takeCustomElementConstructionData() { return WTFMove(m_customElementToConstruct); }
void didCreateCustomOrCallbackElement(Ref<Element>&&, CustomElementConstructionData&);
void finished();
private:
class ExternalCharacterTokenBuffer;
enum class InsertionMode {
Initial,
BeforeHTML,
BeforeHead,
InHead,
InHeadNoscript,
AfterHead,
TemplateContents,
InBody,
Text,
InTable,
InTableText,
InCaption,
InColumnGroup,
InTableBody,
InRow,
InCell,
InSelect,
InSelectInTable,
AfterBody,
InFrameset,
AfterFrameset,
AfterAfterBody,
AfterAfterFrameset,
};
bool isParsingFragmentOrTemplateContents() const;
#if ENABLE(TELEPHONE_NUMBER_DETECTION) && PLATFORM(IOS)
void insertPhoneNumberLink(const String&);
void linkifyPhoneNumbers(const String&);
#endif
void processToken(AtomicHTMLToken&&);
void processDoctypeToken(AtomicHTMLToken&&);
void processStartTag(AtomicHTMLToken&&);
void processEndTag(AtomicHTMLToken&&);
void processComment(AtomicHTMLToken&&);
void processCharacter(AtomicHTMLToken&&);
void processEndOfFile(AtomicHTMLToken&&);
bool processStartTagForInHead(AtomicHTMLToken&&);
void processStartTagForInBody(AtomicHTMLToken&&);
void processStartTagForInTable(AtomicHTMLToken&&);
void processEndTagForInBody(AtomicHTMLToken&&);
void processEndTagForInTable(AtomicHTMLToken&&);
void processEndTagForInTableBody(AtomicHTMLToken&&);
void processEndTagForInRow(AtomicHTMLToken&&);
void processEndTagForInCell(AtomicHTMLToken&&);
void processHtmlStartTagForInBody(AtomicHTMLToken&&);
bool processBodyEndTagForInBody(AtomicHTMLToken&&);
bool processTableEndTagForInTable();
bool processCaptionEndTagForInCaption();
bool processColgroupEndTagForInColumnGroup();
bool processTrEndTagForInRow();
void processAnyOtherEndTagForInBody(AtomicHTMLToken&&);
void processCharacterBuffer(ExternalCharacterTokenBuffer&);
inline void processCharacterBufferForInBody(ExternalCharacterTokenBuffer&);
void processFakeStartTag(const QualifiedName&, Vector<Attribute>&& attributes = Vector<Attribute>());
void processFakeEndTag(const QualifiedName&);
void processFakeEndTag(const AtomicString&);
void processFakeCharacters(const String&);
void processFakePEndTagIfPInButtonScope();
void processGenericRCDATAStartTag(AtomicHTMLToken&&);
void processGenericRawTextStartTag(AtomicHTMLToken&&);
void processScriptStartTag(AtomicHTMLToken&&);
void defaultForInitial();
void defaultForBeforeHTML();
void defaultForBeforeHead();
void defaultForInHead();
void defaultForInHeadNoscript();
void defaultForAfterHead();
void defaultForInTableText();
bool shouldProcessTokenInForeignContent(const AtomicHTMLToken&);
void processTokenInForeignContent(AtomicHTMLToken&&);
HTMLStackItem& adjustedCurrentStackItem() const;
void callTheAdoptionAgency(AtomicHTMLToken&);
void closeTheCell();
template <bool shouldClose(const HTMLStackItem&)> void processCloseWhenNestedTag(AtomicHTMLToken&&);
void parseError(const AtomicHTMLToken&);
void resetInsertionModeAppropriately();
void insertGenericHTMLElement(AtomicHTMLToken&&);
void processTemplateStartTag(AtomicHTMLToken&&);
bool processTemplateEndTag(AtomicHTMLToken&&);
bool processEndOfFileForInTemplateContents(AtomicHTMLToken&&);
class FragmentParsingContext {
public:
FragmentParsingContext();
FragmentParsingContext(DocumentFragment&, Element& contextElement);
DocumentFragment* fragment() const;
Element& contextElement() const;
HTMLStackItem& contextElementStackItem() const;
private:
DocumentFragment* m_fragment { nullptr };
RefPtr<HTMLStackItem> m_contextElementStackItem;
};
HTMLDocumentParser& m_parser;
const HTMLParserOptions m_options;
const FragmentParsingContext m_fragmentContext;
HTMLConstructionSite m_tree;
InsertionMode m_insertionMode { InsertionMode::Initial };
InsertionMode m_originalInsertionMode { InsertionMode::Initial };
Vector<InsertionMode, 1> m_templateInsertionModes;
StringBuilder m_pendingTableCharacters;
RefPtr<ScriptElement> m_scriptToProcess; TextPosition m_scriptToProcessStartPosition;
std::unique_ptr<CustomElementConstructionData> m_customElementToConstruct;
bool m_shouldSkipLeadingNewline { false };
bool m_framesetOk { true };
#if !ASSERT_DISABLED
bool m_destroyed { false };
bool m_destructionProhibited { true };
#endif
};
inline HTMLTreeBuilder::~HTMLTreeBuilder()
{
#if !ASSERT_DISABLED
ASSERT(!m_destroyed);
ASSERT(!m_destructionProhibited);
m_destroyed = true;
#endif
}
inline void HTMLTreeBuilder::setShouldSkipLeadingNewline(bool shouldSkip)
{
ASSERT(!m_destroyed);
m_shouldSkipLeadingNewline = shouldSkip;
}
inline bool HTMLTreeBuilder::isParsingFragment() const
{
ASSERT(!m_destroyed);
return !!m_fragmentContext.fragment();
}
inline bool HTMLTreeBuilder::hasParserBlockingScriptWork() const
{
ASSERT(!m_destroyed);
ASSERT(!(m_scriptToProcess && m_customElementToConstruct));
return m_scriptToProcess || m_customElementToConstruct;
}
inline DocumentFragment* HTMLTreeBuilder::FragmentParsingContext::fragment() const
{
return m_fragment;
}
}