WebSearchableFormData.cpp [plain text]
#include "config.h"
#include "WebSearchableFormData.h"
#include "Document.h"
#include "DocumentLoader.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"
#include "WebInputElement.h"
using namespace WebCore;
using namespace HTMLNames;
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;
}
if (!form->document()->loader())
return;
*encoding = TextEncoding(form->document()->encoding());
}
bool IsHTTPFormSubmit(const HTMLFormElement* form)
{
String action(form->action());
return form->document()->completeURL(action.isNull() ? "" : action).protocolIs("http");
}
HTMLFormControlElement* GetButtonToActivate(HTMLFormElement* form)
{
HTMLFormControlElement* firstSubmitButton = 0;
for (Vector<FormAssociatedElement*>::const_iterator i(form->associatedElements().begin()); i != form->associatedElements().end(); ++i) {
if (!(*i)->isFormControlElement())
continue;
HTMLFormControlElement* formElement = static_cast<HTMLFormControlElement*>(*i);
if (formElement->isActivatedSubmit())
return 0;
if (!firstSubmitButton && formElement->isSuccessfulSubmitButton())
firstSubmitButton = formElement;
}
return firstSubmitButton;
}
bool IsSelectInDefaultState(HTMLSelectElement* select)
{
const Vector<HTMLElement*>& listItems = select->listItems();
if (select->multiple() || select->size() > 1) {
for (Vector<HTMLElement*>::const_iterator i(listItems.begin()); i != listItems.end(); ++i) {
if (!(*i)->hasLocalName(HTMLNames::optionTag))
continue;
HTMLOptionElement* optionElement = static_cast<HTMLOptionElement*>(*i);
if (optionElement->selected() != optionElement->hasAttribute(selectedAttr))
return false;
}
return true;
}
HTMLOptionElement* initialSelected = 0;
for (Vector<HTMLElement*>::const_iterator i(listItems.begin()); i != listItems.end(); ++i) {
if (!(*i)->hasLocalName(HTMLNames::optionTag))
continue;
HTMLOptionElement* optionElement = static_cast<HTMLOptionElement*>(*i);
if (optionElement->hasAttribute(selectedAttr)) {
initialSelected = optionElement;
break;
}
if (!initialSelected)
initialSelected = optionElement;
}
return !initialSelected || initialSelected->selected();
}
bool IsInDefaultState(HTMLFormControlElement* formElement)
{
if (formElement->hasTagName(HTMLNames::inputTag)) {
const HTMLInputElement* inputElement = static_cast<HTMLInputElement*>(formElement);
if (inputElement->isCheckbox() || inputElement->isRadioButton())
return inputElement->checked() == inputElement->hasAttribute(checkedAttr);
} else if (formElement->hasTagName(HTMLNames::selectTag))
return IsSelectInDefaultState(static_cast<HTMLSelectElement*>(formElement));
return true;
}
HTMLInputElement* findSuitableSearchInputElement(const HTMLFormElement* form)
{
HTMLInputElement* textElement = 0;
for (Vector<FormAssociatedElement*>::const_iterator i(form->associatedElements().begin()); i != form->associatedElements().end(); ++i) {
if (!(*i)->isFormControlElement())
continue;
HTMLFormControlElement* formElement = static_cast<HTMLFormControlElement*>(*i);
if (formElement->disabled() || formElement->name().isNull())
continue;
if (!IsInDefaultState(formElement) || formElement->hasTagName(HTMLNames::textareaTag))
return 0;
if (formElement->hasTagName(HTMLNames::inputTag) && formElement->willValidate()) {
const HTMLInputElement* input = static_cast<const HTMLInputElement*>(formElement);
if (input->isFileUpload() || input->isPasswordField())
return 0;
if (input->isTextField()) {
if (textElement) {
return 0;
}
textElement = static_cast<HTMLInputElement*>(formElement);
}
}
}
return textElement;
}
bool buildSearchString(const HTMLFormElement* form, Vector<char>* encodedString, TextEncoding* encoding, const HTMLInputElement* textElement)
{
bool isElementFound = false;
for (Vector<FormAssociatedElement*>::const_iterator i(form->associatedElements().begin()); i != form->associatedElements().end(); ++i) {
if (!(*i)->isFormControlElement())
continue;
HTMLFormControlElement* formElement = static_cast<HTMLFormControlElement*>(*i);
if (formElement->disabled() || formElement->name().isNull())
continue;
FormDataList dataList(*encoding);
if (!formElement->appendFormData(dataList, false))
continue;
const Vector<FormDataList::Item>& items = dataList.items();
for (Vector<FormDataList::Item>::const_iterator j(items.begin()); j != items.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);
isElementFound = true;
} else
FormDataBuilder::encodeStringAsFormData(*encodedString, j->data());
}
}
return isElementFound;
}
}
namespace WebKit {
WebSearchableFormData::WebSearchableFormData(const WebFormElement& form, const WebInputElement& selectedInputElement)
{
RefPtr<HTMLFormElement> formElement = form.operator PassRefPtr<HTMLFormElement>();
HTMLInputElement* inputElement = selectedInputElement.operator PassRefPtr<HTMLInputElement>().get();
if (equalIgnoringCase(formElement->getAttribute(methodAttr), "post")
|| (!IsHTTPFormSubmit(formElement.get()) && !inputElement))
return;
Vector<char> encodedString;
TextEncoding encoding;
GetFormEncoding(formElement.get(), &encoding);
if (!encoding.isValid()) {
return;
}
if (!inputElement) {
inputElement = findSuitableSearchInputElement(formElement.get());
if (!inputElement)
return;
}
HTMLFormControlElement* firstSubmitButton = GetButtonToActivate(formElement.get());
if (firstSubmitButton) {
firstSubmitButton->setActivatedSubmit(true);
}
bool isValidSearchString = buildSearchString(formElement.get(), &encodedString, &encoding, inputElement);
if (firstSubmitButton)
firstSubmitButton->setActivatedSubmit(false);
if (!isValidSearchString)
return;
String action(formElement->action());
KURL url(formElement->document()->completeURL(action.isNull() ? "" : action));
RefPtr<FormData> formData = FormData::create(encodedString);
url.setQuery(formData->flattenToString());
m_url = url;
m_encoding = String(encoding.name());
}
}