WebSearchableFormData.cpp [plain text]
#include "config.h"
#include "WebSearchableFormData.h"
#include "Document.h"
#include "FormDataBuilder.h"
#include "FormDataList.h"
#include "Frame.h"
#include "HTMLFormControlElement.h"
#include "HTMLFormElement.h"
#include "HTMLInputElement.h"
#include "HTMLNames.h"
#include "HTMLOptionElement.h"
#include "HTMLOptionsCollection.h"
#include "HTMLSelectElement.h"
#include "TextEncoding.h"
#include "WebFormElement.h"
using namespace WebCore;
namespace {
void GetFormEncoding(const HTMLFormElement* form, TextEncoding* encoding)
{
String str(form->getAttribute(HTMLNames::accept_charsetAttr));
str.replace(',', ' ');
Vector<String> charsets;
str.split(' ', charsets);
for (Vector<String>::const_iterator i(charsets.begin()); i != charsets.end(); ++i) {
*encoding = TextEncoding(*i);
if (encoding->isValid())
return;
}
const Frame* frame = form->document()->frame();
*encoding = frame ? TextEncoding(frame->loader()->writer()->encoding()) : Latin1Encoding();
}
bool IsHTTPFormSubmit(const HTMLFormElement* form)
{
String action(form->action());
return form->document()->frame()->loader()->completeURL(action.isNull() ? "" : action).protocol() == "http";
}
HTMLFormControlElement* GetButtonToActivate(HTMLFormElement* form)
{
HTMLFormControlElement* firstSubmitButton = 0;
for (Vector<HTMLFormControlElement*>::const_iterator i(form->formElements.begin()); i != form->formElements.end(); ++i) {
HTMLFormControlElement* formElement = *i;
if (formElement->isActivatedSubmit())
return 0;
if (!firstSubmitButton && formElement->isSuccessfulSubmitButton())
firstSubmitButton = formElement;
}
return firstSubmitButton;
}
bool IsSelectInDefaultState(const HTMLSelectElement* select)
{
const Vector<Element*>& listItems = select->listItems();
if (select->multiple() || select->size() > 1) {
for (Vector<Element*>::const_iterator i(listItems.begin()); i != listItems.end(); ++i) {
if (!(*i)->hasLocalName(HTMLNames::optionTag))
continue;
const HTMLOptionElement* optionElement = static_cast<const HTMLOptionElement*>(*i);
if (optionElement->selected() != optionElement->defaultSelected())
return false;
}
return true;
}
const HTMLOptionElement* initialSelected = 0;
for (Vector<Element*>::const_iterator i(listItems.begin()); i != listItems.end(); ++i) {
if (!(*i)->hasLocalName(HTMLNames::optionTag))
continue;
const HTMLOptionElement* optionElement = static_cast<const HTMLOptionElement*>(*i);
if (optionElement->defaultSelected()) {
initialSelected = optionElement;
break;
}
if (!initialSelected)
initialSelected = optionElement;
}
return initialSelected ? initialSelected->selected() : true;
}
bool IsInDefaultState(const HTMLFormControlElement* formElement)
{
if (formElement->hasTagName(HTMLNames::inputTag)) {
const HTMLInputElement* inputElement = static_cast<const HTMLInputElement*>(formElement);
if (inputElement->inputType() == HTMLInputElement::CHECKBOX || inputElement->inputType() == HTMLInputElement::RADIO)
return inputElement->checked() == inputElement->defaultChecked();
} else if (formElement->hasTagName(HTMLNames::selectTag))
return IsSelectInDefaultState(static_cast<const HTMLSelectElement*>(formElement));
return true;
}
bool HasSuitableTextElement(const HTMLFormElement* form, Vector<char>* encodedString, String* encodingName)
{
TextEncoding encoding;
GetFormEncoding(form, &encoding);
if (!encoding.isValid()) {
return 0;
}
*encodingName = encoding.name();
HTMLInputElement* textElement = 0;
for (Vector<HTMLFormControlElement*>::const_iterator i(form->formElements.begin()); i != form->formElements.end(); ++i) {
HTMLFormControlElement* formElement = *i;
if (formElement->disabled() || formElement->name().isNull())
continue;
if (!IsInDefaultState(formElement) || formElement->hasTagName(HTMLNames::textareaTag))
return 0;
bool isTextElement = false;
if (formElement->hasTagName(HTMLNames::inputTag)) {
switch (static_cast<const HTMLInputElement*>(formElement)->inputType()) {
case HTMLInputElement::TEXT:
case HTMLInputElement::ISINDEX:
isTextElement = true;
break;
case HTMLInputElement::PASSWORD:
case HTMLInputElement::FILE:
return 0;
default:
break;
}
}
FormDataList dataList(encoding);
if (!formElement->appendFormData(dataList, false))
continue;
const Vector<FormDataList::Item>& itemList = dataList.list();
if (isTextElement && !itemList.isEmpty()) {
if (textElement) {
return false;
}
textElement = static_cast<HTMLInputElement*>(formElement);
}
for (Vector<FormDataList::Item>::const_iterator j(itemList.begin()); j != itemList.end(); ++j) {
if (!encodedString->isEmpty() || j->data() != "isindex") {
if (!encodedString->isEmpty())
encodedString->append('&');
FormDataBuilder::encodeStringAsFormData(*encodedString, j->data());
encodedString->append('=');
}
++j;
if (formElement == textElement)
encodedString->append("{searchTerms}", 13);
else
FormDataBuilder::encodeStringAsFormData(*encodedString, j->data());
}
}
return textElement;
}
}
namespace WebKit {
WebSearchableFormData::WebSearchableFormData(const WebFormElement& form)
{
RefPtr<HTMLFormElement> formElement = form.operator PassRefPtr<HTMLFormElement>();
const Frame* frame = formElement->document()->frame();
if (!frame)
return;
if (equalIgnoringCase(formElement->getAttribute(HTMLNames::methodAttr), "post") || !IsHTTPFormSubmit(formElement.get()))
return;
HTMLFormControlElement* firstSubmitButton = GetButtonToActivate(formElement.get());
if (firstSubmitButton) {
firstSubmitButton->setActivatedSubmit(true);
}
Vector<char> encodedString;
String encoding;
bool hasElement = HasSuitableTextElement(formElement.get(), &encodedString, &encoding);
if (firstSubmitButton)
firstSubmitButton->setActivatedSubmit(false);
if (!hasElement) {
return;
}
String action(formElement->action());
KURL url(frame->loader()->completeURL(action.isNull() ? "" : action));
RefPtr<FormData> formData = FormData::create(encodedString);
url.setQuery(formData->flattenToString());
m_url = url;
m_encoding = encoding;
}
}