1// Copyright 2013 The Chromium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5#include "components/autofill/core/browser/autocomplete_history_manager.h" 6 7#include <vector> 8 9#include "base/prefs/pref_service.h" 10#include "base/strings/string16.h" 11#include "base/strings/utf_string_conversions.h" 12#include "components/autofill/core/browser/autofill_client.h" 13#include "components/autofill/core/browser/autofill_driver.h" 14#include "components/autofill/core/browser/autofill_external_delegate.h" 15#include "components/autofill/core/browser/validation.h" 16#include "components/autofill/core/common/autofill_pref_names.h" 17#include "components/autofill/core/common/form_data.h" 18 19namespace autofill { 20namespace { 21 22// Limit on the number of suggestions to appear in the pop-up menu under an 23// text input element in a form. 24const int kMaxAutocompleteMenuItems = 6; 25 26bool IsTextField(const FormFieldData& field) { 27 return 28 field.form_control_type == "text" || 29 field.form_control_type == "search" || 30 field.form_control_type == "tel" || 31 field.form_control_type == "url" || 32 field.form_control_type == "email"; 33} 34 35} // namespace 36 37AutocompleteHistoryManager::AutocompleteHistoryManager( 38 AutofillDriver* driver, 39 AutofillClient* autofill_client) 40 : driver_(driver), 41 database_(autofill_client->GetDatabase()), 42 pending_query_handle_(0), 43 query_id_(0), 44 external_delegate_(NULL), 45 autofill_client_(autofill_client) { 46 DCHECK(autofill_client_); 47} 48 49AutocompleteHistoryManager::~AutocompleteHistoryManager() { 50 CancelPendingQuery(); 51} 52 53void AutocompleteHistoryManager::OnWebDataServiceRequestDone( 54 WebDataServiceBase::Handle h, 55 const WDTypedResult* result) { 56 DCHECK(pending_query_handle_); 57 pending_query_handle_ = 0; 58 59 if (!autofill_client_->IsAutocompleteEnabled()) { 60 SendSuggestions(NULL); 61 return; 62 } 63 64 DCHECK(result); 65 // Returning early here if |result| is NULL. We've seen this happen on 66 // Linux due to NFS dismounting and causing sql failures. 67 // See http://crbug.com/68783. 68 if (!result) { 69 SendSuggestions(NULL); 70 return; 71 } 72 73 DCHECK_EQ(AUTOFILL_VALUE_RESULT, result->GetType()); 74 const WDResult<std::vector<base::string16> >* autofill_result = 75 static_cast<const WDResult<std::vector<base::string16> >*>(result); 76 std::vector<base::string16> suggestions = autofill_result->GetValue(); 77 SendSuggestions(&suggestions); 78} 79 80void AutocompleteHistoryManager::OnGetAutocompleteSuggestions( 81 int query_id, 82 const base::string16& name, 83 const base::string16& prefix, 84 const std::string form_control_type, 85 const std::vector<base::string16>& autofill_values, 86 const std::vector<base::string16>& autofill_labels, 87 const std::vector<base::string16>& autofill_icons, 88 const std::vector<int>& autofill_unique_ids) { 89 CancelPendingQuery(); 90 91 query_id_ = query_id; 92 autofill_values_ = autofill_values; 93 autofill_labels_ = autofill_labels; 94 autofill_icons_ = autofill_icons; 95 autofill_unique_ids_ = autofill_unique_ids; 96 if (!autofill_client_->IsAutocompleteEnabled() || 97 form_control_type == "textarea") { 98 SendSuggestions(NULL); 99 return; 100 } 101 102 if (database_.get()) { 103 pending_query_handle_ = database_->GetFormValuesForElementName( 104 name, prefix, kMaxAutocompleteMenuItems, this); 105 } 106} 107 108void AutocompleteHistoryManager::OnFormSubmitted(const FormData& form) { 109 if (!autofill_client_->IsAutocompleteEnabled()) 110 return; 111 112 if (driver_->IsOffTheRecord()) 113 return; 114 115 // Don't save data that was submitted through JavaScript. 116 if (!form.user_submitted) 117 return; 118 119 // We put the following restriction on stored FormFields: 120 // - non-empty name 121 // - non-empty value 122 // - text field 123 // - value is not a credit card number 124 // - value is not a SSN 125 std::vector<FormFieldData> values; 126 for (std::vector<FormFieldData>::const_iterator iter = 127 form.fields.begin(); 128 iter != form.fields.end(); ++iter) { 129 if (!iter->value.empty() && 130 !iter->name.empty() && 131 IsTextField(*iter) && 132 !autofill::IsValidCreditCardNumber(iter->value) && 133 !autofill::IsSSN(iter->value)) { 134 values.push_back(*iter); 135 } 136 } 137 138 if (!values.empty() && database_.get()) 139 database_->AddFormFields(values); 140} 141 142void AutocompleteHistoryManager::OnRemoveAutocompleteEntry( 143 const base::string16& name, const base::string16& value) { 144 if (database_.get()) 145 database_->RemoveFormValueForElementName(name, value); 146} 147 148void AutocompleteHistoryManager::SetExternalDelegate( 149 AutofillExternalDelegate* delegate) { 150 external_delegate_ = delegate; 151} 152 153void AutocompleteHistoryManager::CancelPendingQuery() { 154 if (pending_query_handle_) { 155 if (database_.get()) 156 database_->CancelRequest(pending_query_handle_); 157 pending_query_handle_ = 0; 158 } 159} 160 161void AutocompleteHistoryManager::SendSuggestions( 162 const std::vector<base::string16>* suggestions) { 163 if (suggestions) { 164 // Combine Autofill and Autocomplete values into values and labels. 165 for (size_t i = 0; i < suggestions->size(); ++i) { 166 bool unique = true; 167 for (size_t j = 0; j < autofill_values_.size(); ++j) { 168 // Don't add duplicate values. 169 if (autofill_values_[j] == (*suggestions)[i]) { 170 unique = false; 171 break; 172 } 173 } 174 175 if (unique) { 176 autofill_values_.push_back((*suggestions)[i]); 177 autofill_labels_.push_back(base::string16()); 178 autofill_icons_.push_back(base::string16()); 179 autofill_unique_ids_.push_back(0); // 0 means no profile. 180 } 181 } 182 } 183 184 external_delegate_->OnSuggestionsReturned(query_id_, 185 autofill_values_, 186 autofill_labels_, 187 autofill_icons_, 188 autofill_unique_ids_); 189 190 query_id_ = 0; 191 autofill_values_.clear(); 192 autofill_labels_.clear(); 193 autofill_icons_.clear(); 194 autofill_unique_ids_.clear(); 195} 196 197} // namespace autofill 198