#include "config.h"
#include "DOMFormData.h"
#include "HTMLFormControlElement.h"
#include "HTMLFormElement.h"
#include <wtf/Optional.h>
namespace WebCore {
DOMFormData::DOMFormData(const TextEncoding& encoding)
: m_encoding(encoding)
{
}
DOMFormData::DOMFormData(HTMLFormElement* form)
: m_encoding(UTF8Encoding())
{
if (!form)
return;
ASSERT(isMainThread());
for (auto& element : form->copyAssociatedElementsVector()) {
if (!element->asHTMLElement().isDisabledFormControl())
element->appendFormData(*this, true);
}
}
auto DOMFormData::createFileEntry(const String& name, Blob& blob, const String& filename) -> Item
{
if (!blob.isFile())
return { name, File::create(blob.scriptExecutionContext(), blob, filename.isNull() ? "blob"_s : filename) };
if (!filename.isNull())
return { name, File::create(blob.scriptExecutionContext(), downcast<File>(blob), filename) };
return { name, RefPtr<File> { &downcast<File>(blob) } };
}
void DOMFormData::append(const String& name, const String& value)
{
m_items.append({ name, value });
}
void DOMFormData::append(const String& name, Blob& blob, const String& filename)
{
m_items.append(createFileEntry(name, blob, filename));
}
void DOMFormData::remove(const String& name)
{
m_items.removeAllMatching([&name] (const auto& item) {
return item.name == name;
});
}
auto DOMFormData::get(const String& name) -> Optional<FormDataEntryValue>
{
for (auto& item : m_items) {
if (item.name == name)
return item.data;
}
return WTF::nullopt;
}
auto DOMFormData::getAll(const String& name) -> Vector<FormDataEntryValue>
{
Vector<FormDataEntryValue> result;
for (auto& item : m_items) {
if (item.name == name)
result.append(item.data);
}
return result;
}
bool DOMFormData::has(const String& name)
{
for (auto& item : m_items) {
if (item.name == name)
return true;
}
return false;
}
void DOMFormData::set(const String& name, const String& value)
{
set(name, { name, value });
}
void DOMFormData::set(const String& name, Blob& blob, const String& filename)
{
set(name, createFileEntry(name, blob, filename));
}
void DOMFormData::set(const String& name, Item&& item)
{
Optional<size_t> initialMatchLocation;
for (size_t i = 0; i < m_items.size(); ++i) {
if (name == m_items[i].name) {
initialMatchLocation = i;
break;
}
}
if (initialMatchLocation) {
m_items[*initialMatchLocation] = WTFMove(item);
m_items.removeAllMatching([&name] (const auto& item) {
return item.name == name;
}, *initialMatchLocation + 1);
return;
}
m_items.append(WTFMove(item));
}
DOMFormData::Iterator::Iterator(DOMFormData& target)
: m_target(target)
{
}
Optional<KeyValuePair<String, DOMFormData::FormDataEntryValue>> DOMFormData::Iterator::next()
{
auto& items = m_target->items();
if (m_index >= items.size())
return WTF::nullopt;
auto& item = items[m_index++];
return makeKeyValuePair(item.name, item.data);
}
}