CustomElementRegistry.cpp [plain text]
#include "config.h"
#include "CustomElementRegistry.h"
#include "CustomElementReactionQueue.h"
#include "DOMWindow.h"
#include "Document.h"
#include "JSCustomElementInterface.h"
#include "JSDOMPromiseDeferred.h"
#include "MathMLNames.h"
#include "QualifiedName.h"
#include "ShadowRoot.h"
#include "TypedElementDescendantIterator.h"
#include <JavaScriptCore/JSCJSValueInlines.h>
#include <wtf/text/AtomString.h>
namespace WebCore {
Ref<CustomElementRegistry> CustomElementRegistry::create(DOMWindow& window, ScriptExecutionContext* scriptExecutionContext)
{
return adoptRef(*new CustomElementRegistry(window, scriptExecutionContext));
}
CustomElementRegistry::CustomElementRegistry(DOMWindow& window, ScriptExecutionContext* scriptExecutionContext)
: ContextDestructionObserver(scriptExecutionContext)
, m_window(window)
{
}
CustomElementRegistry::~CustomElementRegistry() = default;
static void enqueueUpgradeInShadowIncludingTreeOrder(ContainerNode& node, JSCustomElementInterface& elementInterface)
{
for (Element* element = ElementTraversal::firstWithin(node); element; element = ElementTraversal::next(*element)) {
if (element->isCustomElementUpgradeCandidate() && element->tagQName() == elementInterface.name())
element->enqueueToUpgrade(elementInterface);
if (auto* shadowRoot = element->shadowRoot()) {
if (shadowRoot->mode() != ShadowRootMode::UserAgent)
enqueueUpgradeInShadowIncludingTreeOrder(*shadowRoot, elementInterface);
}
}
}
void CustomElementRegistry::addElementDefinition(Ref<JSCustomElementInterface>&& elementInterface)
{
AtomString localName = elementInterface->name().localName();
ASSERT(!m_nameMap.contains(localName));
m_constructorMap.add(elementInterface->constructor(), elementInterface.ptr());
m_nameMap.add(localName, elementInterface.copyRef());
if (auto* document = m_window.document())
enqueueUpgradeInShadowIncludingTreeOrder(*document, elementInterface.get());
if (auto promise = m_promiseMap.take(localName))
promise.value()->resolve();
}
JSCustomElementInterface* CustomElementRegistry::findInterface(const Element& element) const
{
return findInterface(element.tagQName());
}
JSCustomElementInterface* CustomElementRegistry::findInterface(const QualifiedName& name) const
{
if (name.namespaceURI() != HTMLNames::xhtmlNamespaceURI)
return nullptr;
return m_nameMap.get(name.localName());
}
JSCustomElementInterface* CustomElementRegistry::findInterface(const AtomString& name) const
{
return m_nameMap.get(name);
}
JSCustomElementInterface* CustomElementRegistry::findInterface(const JSC::JSObject* constructor) const
{
return m_constructorMap.get(constructor);
}
bool CustomElementRegistry::containsConstructor(const JSC::JSObject* constructor) const
{
return m_constructorMap.contains(constructor);
}
JSC::JSValue CustomElementRegistry::get(const AtomString& name)
{
if (auto* elementInterface = m_nameMap.get(name))
return elementInterface->constructor();
return JSC::jsUndefined();
}
static void upgradeElementsInShadowIncludingDescendants(ContainerNode& root)
{
for (auto& element : descendantsOfType<Element>(root)) {
if (element.isCustomElementUpgradeCandidate())
CustomElementReactionQueue::enqueueElementUpgradeIfDefined(element);
if (auto* shadowRoot = element.shadowRoot())
upgradeElementsInShadowIncludingDescendants(*shadowRoot);
}
}
void CustomElementRegistry::upgrade(Node& root)
{
if (!is<ContainerNode>(root))
return;
if (is<Element>(root) && downcast<Element>(root).isCustomElementUpgradeCandidate())
CustomElementReactionQueue::enqueueElementUpgradeIfDefined(downcast<Element>(root));
upgradeElementsInShadowIncludingDescendants(downcast<ContainerNode>(root));
}
}