#include "config.h"
#include "DOMTokenList.h"
#include "ExceptionCode.h"
#include "HTMLParserIdioms.h"
#include <wtf/text/StringBuilder.h>
namespace WebCore {
bool DOMTokenList::validateToken(const AtomicString& token, ExceptionCode& ec)
{
if (token.isEmpty()) {
ec = SYNTAX_ERR;
return false;
}
unsigned length = token.length();
for (unsigned i = 0; i < length; ++i) {
if (isHTMLSpace(token[i])) {
ec = INVALID_CHARACTER_ERR;
return false;
}
}
return true;
}
bool DOMTokenList::validateTokens(const Vector<String>& tokens, ExceptionCode& ec)
{
for (size_t i = 0; i < tokens.size(); ++i) {
if (!validateToken(tokens[i], ec))
return false;
}
return true;
}
bool DOMTokenList::contains(const AtomicString& token, ExceptionCode& ec) const
{
if (!validateToken(token, ec))
return false;
return containsInternal(token);
}
void DOMTokenList::add(const AtomicString& token, ExceptionCode& ec)
{
Vector<String> tokens;
tokens.append(token.string());
add(tokens, ec);
}
void DOMTokenList::add(const Vector<String>& tokens, ExceptionCode& ec)
{
Vector<String> filteredTokens;
for (size_t i = 0; i < tokens.size(); ++i) {
if (!validateToken(tokens[i], ec))
return;
if (!containsInternal(tokens[i]) && !filteredTokens.contains(tokens[i]))
filteredTokens.append(tokens[i]);
}
if (filteredTokens.isEmpty())
return;
setValue(addTokens(value(), filteredTokens));
}
void DOMTokenList::remove(const AtomicString& token, ExceptionCode& ec)
{
Vector<String> tokens;
tokens.append(token.string());
remove(tokens, ec);
}
void DOMTokenList::remove(const Vector<String>& tokens, ExceptionCode& ec)
{
if (!validateTokens(tokens, ec))
return;
bool found = false;
for (size_t i = 0; i < tokens.size(); ++i) {
if (containsInternal(tokens[i])) {
found = true;
break;
}
}
if (found)
setValue(removeTokens(value(), tokens));
}
bool DOMTokenList::toggle(const AtomicString& token, ExceptionCode& ec)
{
if (!validateToken(token, ec))
return false;
if (containsInternal(token)) {
removeInternal(token);
return false;
}
addInternal(token);
return true;
}
bool DOMTokenList::toggle(const AtomicString& token, bool force, ExceptionCode& ec)
{
if (!validateToken(token, ec))
return false;
if (force)
addInternal(token);
else
removeInternal(token);
return force;
}
void DOMTokenList::addInternal(const AtomicString& token)
{
if (!containsInternal(token))
setValue(addToken(value(), token));
}
void DOMTokenList::removeInternal(const AtomicString& token)
{
if (!containsInternal(token))
return;
setValue(removeToken(value(), token));
}
String DOMTokenList::addToken(const AtomicString& input, const AtomicString& token)
{
Vector<String> tokens;
tokens.append(token.string());
return addTokens(input, tokens);
}
String DOMTokenList::addTokens(const AtomicString& input, const Vector<String>& tokens)
{
bool needsSpace = false;
StringBuilder builder;
if (!input.isEmpty()) {
builder.append(input);
needsSpace = !isHTMLSpace(input[input.length() - 1]);
}
for (size_t i = 0; i < tokens.size(); ++i) {
if (needsSpace)
builder.append(' ');
builder.append(tokens[i]);
needsSpace = true;
}
return builder.toString();
}
String DOMTokenList::removeToken(const AtomicString& input, const AtomicString& token)
{
Vector<String> tokens;
tokens.append(token.string());
return removeTokens(input, tokens);
}
String DOMTokenList::removeTokens(const AtomicString& input, const Vector<String>& tokens)
{
unsigned inputLength = input.length();
StringBuilder output; output.reserveCapacity(inputLength);
unsigned position = 0;
while (position < inputLength) {
if (isHTMLSpace(input[position])) { output.append(input[position++]); continue; }
StringBuilder s;
while (position < inputLength && isNotHTMLSpace(input[position]))
s.append(input[position++]);
if (tokens.contains(s.toStringPreserveCapacity())) {
while (position < inputLength && isHTMLSpace(input[position]))
++position;
size_t j = output.length();
while (j > 0 && isHTMLSpace(output[j - 1]))
--j;
output.resize(j);
if (position < inputLength && !output.isEmpty())
output.append(' ');
} else
output.append(s.toStringPreserveCapacity()); }
return output.toString();
}
}