DatasetDOMStringMap.cpp [plain text]
#include "config.h"
#include "DatasetDOMStringMap.h"
#include "Element.h"
#include "ExceptionCode.h"
#include <wtf/ASCIICType.h>
#include <wtf/text/AtomicString.h>
#include <wtf/text/StringBuilder.h>
#include <wtf/text/WTFString.h>
namespace WebCore {
static bool isValidAttributeName(const String& name)
{
if (!name.startsWith("data-"))
return false;
unsigned length = name.length();
for (unsigned i = 5; i < length; ++i) {
if (isASCIIUpper(name[i]))
return false;
}
return true;
}
static String convertAttributeNameToPropertyName(const String& name)
{
StringBuilder stringBuilder;
unsigned length = name.length();
for (unsigned i = 5; i < length; ++i) {
UChar character = name[i];
if (character != '-')
stringBuilder.append(character);
else {
if ((i + 1 < length) && isASCIILower(name[i + 1])) {
stringBuilder.append(toASCIIUpper(name[i + 1]));
++i;
} else
stringBuilder.append(character);
}
}
return stringBuilder.toString();
}
static bool propertyNameMatchesAttributeName(const String& propertyName, const String& attributeName)
{
if (!attributeName.startsWith("data-"))
return false;
unsigned propertyLength = propertyName.length();
unsigned attributeLength = attributeName.length();
unsigned a = 5;
unsigned p = 0;
bool wordBoundary = false;
while (a < attributeLength && p < propertyLength) {
const UChar currentAttributeNameChar = attributeName[a];
if (currentAttributeNameChar == '-' && a + 1 < attributeLength && attributeName[a + 1] != '-')
wordBoundary = true;
else {
if ((wordBoundary ? toASCIIUpper(currentAttributeNameChar) : currentAttributeNameChar) != propertyName[p])
return false;
p++;
wordBoundary = false;
}
a++;
}
return (a == attributeLength && p == propertyLength);
}
static bool isValidPropertyName(const String& name)
{
unsigned length = name.length();
for (unsigned i = 0; i < length; ++i) {
if (name[i] == '-' && (i + 1 < length) && isASCIILower(name[i + 1]))
return false;
}
return true;
}
template<typename CharacterType>
static inline AtomicString convertPropertyNameToAttributeName(const StringImpl& name)
{
const CharacterType dataPrefix[] = { 'd', 'a', 't', 'a', '-' };
Vector<CharacterType, 32> buffer;
unsigned length = name.length();
buffer.reserveInitialCapacity(WTF_ARRAY_LENGTH(dataPrefix) + length);
buffer.append(dataPrefix, WTF_ARRAY_LENGTH(dataPrefix));
const CharacterType* characters = name.characters<CharacterType>();
for (unsigned i = 0; i < length; ++i) {
CharacterType character = characters[i];
if (isASCIIUpper(character)) {
buffer.append('-');
buffer.append(toASCIILower(character));
} else
buffer.append(character);
}
return AtomicString(buffer.data(), buffer.size());
}
static AtomicString convertPropertyNameToAttributeName(const String& name)
{
if (name.isNull())
return nullAtom;
StringImpl* nameImpl = name.impl();
if (nameImpl->is8Bit())
return convertPropertyNameToAttributeName<LChar>(*nameImpl);
return convertPropertyNameToAttributeName<UChar>(*nameImpl);
}
void DatasetDOMStringMap::ref()
{
m_element.ref();
}
void DatasetDOMStringMap::deref()
{
m_element.deref();
}
void DatasetDOMStringMap::getNames(Vector<String>& names)
{
if (!m_element.hasAttributes())
return;
for (const Attribute& attribute : m_element.attributesIterator()) {
if (isValidAttributeName(attribute.localName()))
names.append(convertAttributeNameToPropertyName(attribute.localName()));
}
}
const AtomicString& DatasetDOMStringMap::item(const String& propertyName, bool& isValid)
{
isValid = false;
if (m_element.hasAttributes()) {
AttributeIteratorAccessor attributeIteratorAccessor = m_element.attributesIterator();
if (attributeIteratorAccessor.attributeCount() == 1) {
const Attribute& attribute = *attributeIteratorAccessor.begin();
if (propertyNameMatchesAttributeName(propertyName, attribute.localName())) {
isValid = true;
return attribute.value();
}
} else {
AtomicString attributeName = convertPropertyNameToAttributeName(propertyName);
for (const Attribute& attribute : attributeIteratorAccessor) {
if (attribute.localName() == attributeName) {
isValid = true;
return attribute.value();
}
}
}
}
return nullAtom;
}
void DatasetDOMStringMap::setItem(const String& name, const String& value, ExceptionCode& ec)
{
if (!isValidPropertyName(name)) {
ec = SYNTAX_ERR;
return;
}
m_element.setAttribute(convertPropertyNameToAttributeName(name), value, ec);
}
bool DatasetDOMStringMap::deleteItem(const String& name)
{
return m_element.removeAttribute(convertPropertyNameToAttributeName(name));
}
}