AutofillElements.cpp [plain text]
#include "config.h"
#include "AutofillElements.h"
#include "FocusController.h"
#include "Page.h"
namespace WebCore {
static inline bool isAutofillableElement(Element& node)
{
if (!is<HTMLInputElement>(node))
return false;
auto inputElement = &downcast<HTMLInputElement>(node);
return inputElement->isTextField() || inputElement->isEmailField();
}
static inline RefPtr<HTMLInputElement> nextAutofillableElement(Node* startNode, FocusController& focusController)
{
if (!is<Element>(startNode))
return nullptr;
RefPtr<Element> nextElement = downcast<Element>(startNode);
do {
nextElement = focusController.nextFocusableElement(*nextElement.get());
} while (nextElement && !isAutofillableElement(*nextElement.get()));
if (!nextElement)
return nullptr;
return &downcast<HTMLInputElement>(*nextElement);
}
static inline RefPtr<HTMLInputElement> previousAutofillableElement(Node* startNode, FocusController& focusController)
{
if (!is<Element>(startNode))
return nullptr;
RefPtr<Element> previousElement = downcast<Element>(startNode);
do {
previousElement = focusController.previousFocusableElement(*previousElement.get());
} while (previousElement && !isAutofillableElement(*previousElement.get()));
if (!previousElement)
return nullptr;
return &downcast<HTMLInputElement>(*previousElement);
}
AutofillElements::AutofillElements(RefPtr<HTMLInputElement>&& username, RefPtr<HTMLInputElement>&& password)
: m_username(username)
, m_password(password)
{
}
Optional<AutofillElements> AutofillElements::computeAutofillElements(Ref<HTMLInputElement> start)
{
if (!start->document().page())
return WTF::nullopt;
FocusController& focusController = start->document().page()->focusController();
if (start->isPasswordField()) {
RefPtr<HTMLInputElement> previousElement = previousAutofillableElement(start.ptr(), focusController);
RefPtr<HTMLInputElement> nextElement = nextAutofillableElement(start.ptr(), focusController);
bool hasDuplicatePasswordElements = (nextElement && nextElement->isPasswordField()) || (previousElement && previousElement->isPasswordField());
if (hasDuplicatePasswordElements)
return WTF::nullopt;
if (previousElement && is<HTMLInputElement>(*previousElement)) {
if (previousElement->isTextField())
return AutofillElements(WTFMove(previousElement), WTFMove(start));
}
} else {
RefPtr<HTMLInputElement> nextElement = nextAutofillableElement(start.ptr(), focusController);
if (nextElement && is<HTMLInputElement>(*nextElement)) {
if (nextElement->isPasswordField()) {
RefPtr<HTMLInputElement> elementAfternextElement = nextAutofillableElement(nextElement.get(), focusController);
bool hasDuplicatePasswordElements = elementAfternextElement && elementAfternextElement->isPasswordField();
if (hasDuplicatePasswordElements)
return WTF::nullopt;
return AutofillElements(WTFMove(start), WTFMove(nextElement));
}
}
}
if (start->isPasswordField()) {
RefPtr<HTMLInputElement> previousElement = previousAutofillableElement(start.ptr(), focusController);
RefPtr<HTMLInputElement> nextElement = nextAutofillableElement(start.ptr(), focusController);
if (!previousElement && !nextElement)
return AutofillElements(nullptr, start.ptr());
}
return WTF::nullopt;
}
void AutofillElements::autofill(String username, String password)
{
if (m_username)
m_username->setValueForUser(username);
if (m_password)
m_password->setValueForUser(password);
}
}