Add Password Autofill Manager to New Autofill

chrome/browser/autofill/password_autofill_manager.cc

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "chrome/browser/autofill/password_autofill_manager.h"
+#include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h"
+#include "chrome/common/autofill_messages.h"
+#include "content/browser/renderer_host/render_view_host.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/renderer/render_view.h"
+#include "ui/base/keycodes/keyboard_codes.h"
+
+////////////////////////////////////////////////////////////////////////////////
+// PasswordAutofillManager, public:
+
+PasswordAutofillManager::PasswordAutofillManager(
+    TabContentsWrapper* tab_contents_wrapper)
+    : tab_contents_wrapper_(tab_contents_wrapper) {
+}
+
+PasswordAutofillManager::~PasswordAutofillManager() {
+}
+
+bool PasswordAutofillManager::WouldHandleKeyDown(
+    const webkit::forms::FormField& field) {
+  LoginToPasswordInfoMap::iterator iter = login_to_password_info_.find(field);
+  if (iter == login_to_password_info_.end())
+    return false;
+
+  return true;
+}
+
+bool PasswordAutofillManager::DidAcceptAutofillSuggestion(
+    const webkit::forms::FormField& field,
+    const string16& value) {
+  webkit::forms::FormField input;
+  PasswordInfo password;
+  if (!FindLoginInfo(field, &input, &password))
+    return false;
+
+  if (WillFillUserNameAndPassword(input, password.fill_data)) {
+    if (tab_contents_wrapper_ && tab_contents_wrapper_->web_contents() &&
+        tab_contents_wrapper_->web_contents()->GetRenderViewHost()) {
+      RenderViewHost* host =
+          tab_contents_wrapper_->web_contents()->GetRenderViewHost();
+      host->Send(new AutofillMsg_PasswordAcceptAutofillSuggestion(
+          host->GetRoutingID(),
+          value));
+    }
+    return true;
+  }
+
+  return false;
+}
+
+bool PasswordAutofillManager::DidSelectAutofillSuggestion(
+    const webkit::forms::FormField& field) {
+  webkit::forms::FormField input;
+  PasswordInfo password;
+  return FindLoginInfo(field, &input, &password);
+}
+
+bool PasswordAutofillManager::DidClearAutofillSelection(
+    const webkit::forms::FormField& field) {
+  webkit::forms::FormField input;
+  PasswordInfo password;
+  return FindLoginInfo(field, &input, &password);
+}
+
+void PasswordAutofillManager::FillPasswordForm(
+    const webkit::forms::FormField& username_element,
+    const webkit::forms::PasswordFormFillData& fill_data,
+    int frame_id) {
+  PasswordInfo password_info;
+  password_info.frame_id = frame_id;
+  password_info.fill_data = fill_data;
+
+  login_to_password_info_[username_element] = password_info;
+}
+
+void PasswordAutofillManager::FrameClosing(long long frame_id) {
+  for (LoginToPasswordInfoMap::iterator iter = login_to_password_info_.begin();
+       iter != login_to_password_info_.end();) {
+    if (iter->second.frame_id == frame_id)
+      login_to_password_info_.erase(iter++);
+    else
+      ++iter;
+  }
+}
+
+////////////////////////////////////////////////////////////////////////////////
+// PasswordAutofillManager, private:
+
+bool PasswordAutofillManager::WillFillUserNameAndPassword(
+    const webkit::forms::FormField& username_element,
+    const webkit::forms::PasswordFormFillData& fill_data) {
+  string16 current_username = username_element.value;
+
+  // Look for any suitable matches to current field text.
+  if (fill_data.basic_data.fields[0].value == current_username) {
+    return true;
+  } else {
+    // Scan additional logins for a match.
+    webkit::forms::PasswordFormFillData::LoginCollection::const_iterator iter;
+    for (iter = fill_data.additional_logins.begin();
+         iter != fill_data.additional_logins.end(); ++iter) {
+      if (iter->first == current_username)
+        return true;
+    }
+  }
+
+  return false;
+}
+
+bool PasswordAutofillManager::FindLoginInfo(
+    const webkit::forms::FormField& field,
+    webkit::forms::FormField* found_input,
+    PasswordInfo* found_password) {
+  LoginToPasswordInfoMap::iterator iter = login_to_password_info_.find(field);
+  if (iter == login_to_password_info_.end())
+    return false;
+
+  *found_input = field;
+  *found_password = iter->second;
+  return true;
+}