#pragma once
#include "RenderTreeUpdater.h"
#include "ShadowRoot.h"
#include <wtf/HashMap.h>
#include <wtf/HashSet.h>
#include <wtf/Vector.h>
#include <wtf/text/AtomicString.h>
#include <wtf/text/AtomicStringHash.h>
namespace WebCore {
class Element;
class HTMLSlotElement;
class Node;
class SlotAssignment {
WTF_MAKE_NONCOPYABLE(SlotAssignment); WTF_MAKE_FAST_ALLOCATED;
public:
SlotAssignment();
virtual ~SlotAssignment();
static const AtomicString& defaultSlotName() { return emptyAtom(); }
HTMLSlotElement* findAssignedSlot(const Node&, ShadowRoot&);
void addSlotElementByName(const AtomicString&, HTMLSlotElement&, ShadowRoot&);
void removeSlotElementByName(const AtomicString&, HTMLSlotElement&, ShadowRoot&);
void didChangeSlot(const AtomicString&, ShadowRoot&);
void enqueueSlotChangeEvent(const AtomicString&, ShadowRoot&);
const Vector<Node*>* assignedNodesForSlot(const HTMLSlotElement&, ShadowRoot&);
virtual void hostChildElementDidChange(const Element&, ShadowRoot&);
private:
struct SlotInfo {
WTF_MAKE_FAST_ALLOCATED;
public:
SlotInfo() { }
SlotInfo(HTMLSlotElement& slotElement)
: element(&slotElement)
, elementCount(1)
{ }
bool hasSlotElements() { return !!elementCount; }
bool hasDuplicatedSlotElements() { return elementCount > 1; }
bool shouldResolveSlotElement() { return !element && elementCount; }
HTMLSlotElement* element { nullptr };
unsigned elementCount { 0 };
Vector<Node*> assignedNodes;
};
virtual const AtomicString& slotNameForHostChild(const Node&) const;
HTMLSlotElement* findFirstSlotElement(SlotInfo&, ShadowRoot&);
void resolveAllSlotElements(ShadowRoot&);
void assignSlots(ShadowRoot&);
void assignToSlot(Node& child, const AtomicString& slotName);
HashMap<AtomicString, std::unique_ptr<SlotInfo>> m_slots;
#ifndef NDEBUG
HashSet<HTMLSlotElement*> m_slotElementsForConsistencyCheck;
bool m_needsToResolveSlotElements { false };
#endif
bool m_slotAssignmentsIsValid { false };
};
inline void ShadowRoot::didRemoveAllChildrenOfShadowHost()
{
if (m_slotAssignment) m_slotAssignment->didChangeSlot(nullAtom(), *this);
}
inline void ShadowRoot::didChangeDefaultSlot()
{
if (m_slotAssignment)
m_slotAssignment->didChangeSlot(nullAtom(), *this);
}
inline void ShadowRoot::hostChildElementDidChange(const Element& childElement)
{
if (m_slotAssignment)
m_slotAssignment->hostChildElementDidChange(childElement, *this);
}
inline void ShadowRoot::hostChildElementDidChangeSlotAttribute(Element& element, const AtomicString& oldValue, const AtomicString& newValue)
{
if (!m_slotAssignment)
return;
m_slotAssignment->didChangeSlot(oldValue, *this);
m_slotAssignment->didChangeSlot(newValue, *this);
RenderTreeUpdater::tearDownRenderers(element);
}
}