HTMLFormControlsCollection.cpp [plain text]
#include "config.h"
#include "HTMLFormControlsCollection.h"
#include "HTMLFieldSetElement.h"
#include "HTMLFormElement.h"
#include "HTMLImageElement.h"
#include "HTMLNames.h"
namespace WebCore {
using namespace HTMLNames;
HTMLFormControlsCollection::HTMLFormControlsCollection(ContainerNode& ownerNode)
: HTMLCollection(ownerNode, FormControls, CustomForwardOnlyTraversal)
, m_cachedElement(nullptr)
, m_cachedElementOffsetInArray(0)
{
ASSERT(is<HTMLFormElement>(ownerNode) || is<HTMLFieldSetElement>(ownerNode));
}
Ref<HTMLFormControlsCollection> HTMLFormControlsCollection::create(ContainerNode& ownerNode, CollectionType)
{
return adoptRef(*new HTMLFormControlsCollection(ownerNode));
}
HTMLFormControlsCollection::~HTMLFormControlsCollection()
{
}
const Vector<FormAssociatedElement*>& HTMLFormControlsCollection::formControlElements() const
{
ASSERT(is<HTMLFormElement>(ownerNode()) || is<HTMLFieldSetElement>(ownerNode()));
if (is<HTMLFormElement>(ownerNode()))
return downcast<HTMLFormElement>(ownerNode()).associatedElements();
return downcast<HTMLFieldSetElement>(ownerNode()).associatedElements();
}
const Vector<HTMLImageElement*>& HTMLFormControlsCollection::formImageElements() const
{
ASSERT(is<HTMLFormElement>(ownerNode()));
return downcast<HTMLFormElement>(ownerNode()).imageElements();
}
static unsigned findFormAssociatedElement(const Vector<FormAssociatedElement*>& elements, const Element& element)
{
for (unsigned i = 0; i < elements.size(); ++i) {
auto& associatedElement = *elements[i];
if (associatedElement.isEnumeratable() && &associatedElement.asHTMLElement() == &element)
return i;
}
return elements.size();
}
Element* HTMLFormControlsCollection::customElementAfter(Element* current) const
{
const Vector<FormAssociatedElement*>& elements = formControlElements();
unsigned start;
if (!current)
start = 0;
else if (m_cachedElement == current)
start = m_cachedElementOffsetInArray + 1;
else
start = findFormAssociatedElement(elements, *current) + 1;
for (unsigned i = start; i < elements.size(); ++i) {
FormAssociatedElement& element = *elements[i];
if (element.isEnumeratable()) {
m_cachedElement = &element.asHTMLElement();
m_cachedElementOffsetInArray = i;
return &element.asHTMLElement();
}
}
return nullptr;
}
static HTMLElement* firstNamedItem(const Vector<FormAssociatedElement*>& elementsArray,
const Vector<HTMLImageElement*>* imageElementsArray, const QualifiedName& attrName, const String& name)
{
ASSERT(attrName == idAttr || attrName == nameAttr);
for (unsigned i = 0; i < elementsArray.size(); ++i) {
HTMLElement& element = elementsArray[i]->asHTMLElement();
if (elementsArray[i]->isEnumeratable() && element.fastGetAttribute(attrName) == name)
return &element;
}
if (!imageElementsArray)
return 0;
for (unsigned i = 0; i < imageElementsArray->size(); ++i) {
HTMLImageElement& element = *(*imageElementsArray)[i];
if (element.fastGetAttribute(attrName) == name)
return &element;
}
return nullptr;
}
HTMLElement* HTMLFormControlsCollection::namedItem(const AtomicString& name) const
{
auto* imageElements = is<HTMLFieldSetElement>(ownerNode()) ? nullptr : &formImageElements();
if (HTMLElement* item = firstNamedItem(formControlElements(), imageElements, idAttr, name))
return item;
return firstNamedItem(formControlElements(), imageElements, nameAttr, name);
}
void HTMLFormControlsCollection::updateNamedElementCache() const
{
if (hasNamedElementCache())
return;
auto cache = std::make_unique<CollectionNamedElementCache>();
bool ownerIsFormElement = is<HTMLFormElement>(ownerNode());
HashSet<AtomicStringImpl*> foundInputElements;
for (auto& elementPtr : formControlElements()) {
FormAssociatedElement& associatedElement = *elementPtr;
if (associatedElement.isEnumeratable()) {
HTMLElement& element = associatedElement.asHTMLElement();
const AtomicString& id = element.getIdAttribute();
if (!id.isEmpty()) {
cache->appendToIdCache(id, element);
if (ownerIsFormElement)
foundInputElements.add(id.impl());
}
const AtomicString& name = element.getNameAttribute();
if (!name.isEmpty() && id != name) {
cache->appendToNameCache(name, element);
if (ownerIsFormElement)
foundInputElements.add(name.impl());
}
}
}
if (ownerIsFormElement) {
for (auto* elementPtr : formImageElements()) {
HTMLImageElement& element = *elementPtr;
const AtomicString& id = element.getIdAttribute();
if (!id.isEmpty() && !foundInputElements.contains(id.impl()))
cache->appendToIdCache(id, element);
const AtomicString& name = element.getNameAttribute();
if (!name.isEmpty() && id != name && !foundInputElements.contains(name.impl()))
cache->appendToNameCache(name, element);
}
}
setNamedItemCache(WTF::move(cache));
}
void HTMLFormControlsCollection::invalidateCache(Document& document) const
{
HTMLCollection::invalidateCache(document);
m_cachedElement = nullptr;
m_cachedElementOffsetInArray = 0;
}
}