[email protected] | e41982a7 | 2012-11-20 07:16:51 | [diff] [blame] | 1 | // Copyright 2012 The Chromium Authors. All rights reserved. |
license.bot | bf09a50 | 2008-08-24 00:55:55 | [diff] [blame] | 2 | // Use of this source code is governed by a BSD-style license that can be |
| 3 | // found in the LICENSE file. |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 4 | |
| 5 | #include "chrome/browser/autocomplete/search_provider.h" |
| 6 | |
[email protected] | 1cb2dac | 2010-03-08 21:49:15 | [diff] [blame] | 7 | #include <algorithm> |
[email protected] | c3a4bd99 | 2010-08-18 20:25:01 | [diff] [blame] | 8 | #include <cmath> |
[email protected] | 1cb2dac | 2010-03-08 21:49:15 | [diff] [blame] | 9 | |
[email protected] | 2041cf34 | 2010-02-19 03:15:59 | [diff] [blame] | 10 | #include "base/callback.h" |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 11 | #include "base/i18n/break_iterator.h" |
[email protected] | 503d0387 | 2011-05-06 08:36:26 | [diff] [blame] | 12 | #include "base/i18n/case_conversion.h" |
[email protected] | d6e58c6e | 2009-10-10 20:40:50 | [diff] [blame] | 13 | #include "base/i18n/icu_string_conversions.h" |
[email protected] | ffbec69 | 2012-02-26 20:26:42 | [diff] [blame] | 14 | #include "base/json/json_string_value_serializer.h" |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 15 | #include "base/message_loop.h" |
[email protected] | f5b95ba9 | 2012-03-27 14:05:19 | [diff] [blame] | 16 | #include "base/metrics/histogram.h" |
[email protected] | dc9a676 | 2010-08-16 07:13:53 | [diff] [blame] | 17 | #include "base/string16.h" |
[email protected] | 371dab1 | 2012-06-01 03:23:55 | [diff] [blame] | 18 | #include "base/string_util.h" |
[email protected] | 1cb2dac | 2010-03-08 21:49:15 | [diff] [blame] | 19 | #include "base/utf_string_conversions.h" |
[email protected] | ea3b9a50 | 2011-04-04 14:19:37 | [diff] [blame] | 20 | #include "chrome/browser/autocomplete/autocomplete_classifier.h" |
[email protected] | 810ffba | 2012-06-12 01:07:48 | [diff] [blame] | 21 | #include "chrome/browser/autocomplete/autocomplete_classifier_factory.h" |
[email protected] | 9ac4009 | 2010-10-27 23:05:26 | [diff] [blame] | 22 | #include "chrome/browser/autocomplete/autocomplete_match.h" |
[email protected] | 5af9bc8 | 2012-06-29 00:53:48 | [diff] [blame] | 23 | #include "chrome/browser/autocomplete/autocomplete_provider_listener.h" |
[email protected] | 73c2b163 | 2012-07-02 22:51:38 | [diff] [blame] | 24 | #include "chrome/browser/autocomplete/autocomplete_result.h" |
[email protected] | 3723e6e | 2012-06-11 21:06:56 | [diff] [blame] | 25 | #include "chrome/browser/autocomplete/history_url_provider.h" |
[email protected] | 2c812ba0 | 2011-07-14 00:23:15 | [diff] [blame] | 26 | #include "chrome/browser/autocomplete/keyword_provider.h" |
[email protected] | 371dab1 | 2012-06-01 03:23:55 | [diff] [blame] | 27 | #include "chrome/browser/autocomplete/url_prefix.h" |
[email protected] | ce560f8 | 2009-06-03 09:39:44 | [diff] [blame] | 28 | #include "chrome/browser/history/history.h" |
[email protected] | 9d2db76 | 2012-06-19 00:01:10 | [diff] [blame] | 29 | #include "chrome/browser/history/history_service_factory.h" |
[email protected] | 10c2d69 | 2012-05-11 05:32:23 | [diff] [blame] | 30 | #include "chrome/browser/history/in_memory_database.h" |
[email protected] | f870a32 | 2009-01-16 21:47:27 | [diff] [blame] | 31 | #include "chrome/browser/net/url_fixer_upper.h" |
[email protected] | 37858e5 | 2010-08-26 00:22:02 | [diff] [blame] | 32 | #include "chrome/browser/prefs/pref_service.h" |
[email protected] | 8ecad5e | 2010-12-02 21:18:33 | [diff] [blame] | 33 | #include "chrome/browser/profiles/profile.h" |
[email protected] | a0ad93ea | 2012-05-07 22:11:53 | [diff] [blame] | 34 | #include "chrome/browser/search_engines/search_engine_type.h" |
[email protected] | 9899a61 | 2012-08-21 23:50:04 | [diff] [blame] | 35 | #include "chrome/browser/search_engines/template_url_prepopulate_data.h" |
[email protected] | 8e5c89a | 2011-06-07 18:13:33 | [diff] [blame] | 36 | #include "chrome/browser/search_engines/template_url_service.h" |
| 37 | #include "chrome/browser/search_engines/template_url_service_factory.h" |
[email protected] | e41982a7 | 2012-11-20 07:16:51 | [diff] [blame] | 38 | #include "chrome/browser/ui/browser_instant_controller.h" |
[email protected] | 5cd06c3e | 2012-09-20 03:25:20 | [diff] [blame] | 39 | #include "chrome/browser/ui/search/search.h" |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 40 | #include "chrome/common/pref_names.h" |
[email protected] | dcf7d35 | 2009-02-26 01:56:02 | [diff] [blame] | 41 | #include "chrome/common/url_constants.h" |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 42 | #include "googleurl/src/url_util.h" |
[email protected] | 34ac8f3 | 2009-02-22 23:03:27 | [diff] [blame] | 43 | #include "grit/generated_resources.h" |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 44 | #include "net/base/escape.h" |
[email protected] | d3cf8682f0 | 2012-02-29 23:29:34 | [diff] [blame] | 45 | #include "net/base/load_flags.h" |
[email protected] | 371dab1 | 2012-06-01 03:23:55 | [diff] [blame] | 46 | #include "net/base/net_util.h" |
[email protected] | 319d9e6f | 2009-02-18 19:47:21 | [diff] [blame] | 47 | #include "net/http/http_response_headers.h" |
[email protected] | 3dc1bc4 | 2012-06-19 08:20:53 | [diff] [blame] | 48 | #include "net/url_request/url_fetcher.h" |
[email protected] | 319d9e6f | 2009-02-18 19:47:21 | [diff] [blame] | 49 | #include "net/url_request/url_request_status.h" |
[email protected] | c051a1b | 2011-01-21 23:30:17 | [diff] [blame] | 50 | #include "ui/base/l10n/l10n_util.h" |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 51 | |
[email protected] | e1acf6f | 2008-10-27 20:43:33 | [diff] [blame] | 52 | using base::Time; |
| 53 | using base::TimeDelta; |
| 54 | |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 55 | namespace { |
| 56 | |
[email protected] | 7706a52 | 2012-08-16 17:42:25 | [diff] [blame] | 57 | // We keep track in a histogram how many suggest requests we send, how |
| 58 | // many suggest requests we invalidate (e.g., due to a user typing |
| 59 | // another character), and how many replies we receive. |
| 60 | // *** ADD NEW ENUMS AFTER ALL PREVIOUSLY DEFINED ONES! *** |
| 61 | // (excluding the end-of-list enum value) |
| 62 | // We do not want values of existing enums to change or else it screws |
| 63 | // up the statistics. |
| 64 | enum SuggestRequestsHistogramValue { |
| 65 | REQUEST_SENT = 1, |
| 66 | REQUEST_INVALIDATED, |
| 67 | REPLY_RECEIVED, |
| 68 | MAX_SUGGEST_REQUEST_HISTOGRAM_VALUE |
| 69 | }; |
| 70 | |
| 71 | // Increments the appropriate value in the histogram by one. |
| 72 | void LogOmniboxSuggestRequest( |
| 73 | SuggestRequestsHistogramValue request_value) { |
| 74 | UMA_HISTOGRAM_ENUMERATION("Omnibox.SuggestRequests", request_value, |
| 75 | MAX_SUGGEST_REQUEST_HISTOGRAM_VALUE); |
| 76 | } |
| 77 | |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 78 | bool HasMultipleWords(const string16& text) { |
| 79 | base::i18n::BreakIterator i(text, base::i18n::BreakIterator::BREAK_WORD); |
| 80 | bool found_word = false; |
| 81 | if (i.Init()) { |
| 82 | while (i.Advance()) { |
| 83 | if (i.IsWord()) { |
| 84 | if (found_word) |
| 85 | return true; |
| 86 | found_word = true; |
| 87 | } |
| 88 | } |
| 89 | } |
| 90 | return false; |
| 91 | } |
| 92 | |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 93 | } // namespace |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 94 | |
[email protected] | 033f342 | 2012-03-13 21:24:18 | [diff] [blame] | 95 | |
[email protected] | 3954c3a | 2012-04-10 20:17:55 | [diff] [blame] | 96 | // SearchProvider::Providers -------------------------------------------------- |
[email protected] | b547666d | 2009-04-23 16:37:58 | [diff] [blame] | 97 | |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 98 | SearchProvider::Providers::Providers(TemplateURLService* template_url_service) |
| 99 | : template_url_service_(template_url_service) { |
| 100 | } |
| 101 | |
| 102 | const TemplateURL* SearchProvider::Providers::GetDefaultProviderURL() const { |
| 103 | return default_provider_.empty() ? NULL : |
| 104 | template_url_service_->GetTemplateURLForKeyword(default_provider_); |
| 105 | } |
| 106 | |
| 107 | const TemplateURL* SearchProvider::Providers::GetKeywordProviderURL() const { |
| 108 | return keyword_provider_.empty() ? NULL : |
| 109 | template_url_service_->GetTemplateURLForKeyword(keyword_provider_); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 110 | } |
| 111 | |
[email protected] | 3954c3a | 2012-04-10 20:17:55 | [diff] [blame] | 112 | |
| 113 | // SearchProvider ------------------------------------------------------------- |
| 114 | |
| 115 | // static |
| 116 | const int SearchProvider::kDefaultProviderURLFetcherID = 1; |
| 117 | // static |
| 118 | const int SearchProvider::kKeywordProviderURLFetcherID = 2; |
[email protected] | 3954c3a | 2012-04-10 20:17:55 | [diff] [blame] | 119 | |
[email protected] | 30f5bc9 | 2012-06-26 04:14:55 | [diff] [blame] | 120 | SearchProvider::SearchProvider(AutocompleteProviderListener* listener, |
| 121 | Profile* profile) |
[email protected] | 35f1f4f0 | 2012-09-11 13:17:00 | [diff] [blame] | 122 | : AutocompleteProvider(listener, profile, |
| 123 | AutocompleteProvider::TYPE_SEARCH), |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 124 | providers_(TemplateURLServiceFactory::GetForProfile(profile)), |
[email protected] | 601858c0 | 2010-09-01 17:08:20 | [diff] [blame] | 125 | suggest_results_pending_(0), |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 126 | has_suggested_relevance_(false), |
| 127 | verbatim_relevance_(-1), |
[email protected] | 8e5cc28 | 2010-12-05 18:11:39 | [diff] [blame] | 128 | have_suggest_results_(false), |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 129 | instant_finalized_(false) { |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 130 | } |
| 131 | |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 132 | void SearchProvider::FinalizeInstantQuery(const string16& input_text, |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 133 | const InstantSuggestion& suggestion) { |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 134 | if (done_ || instant_finalized_) |
| 135 | return; |
| 136 | |
| 137 | instant_finalized_ = true; |
| 138 | UpdateDone(); |
| 139 | |
[email protected] | e918c11 | 2010-12-08 23:03:49 | [diff] [blame] | 140 | if (input_text.empty()) { |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 141 | // We only need to update the listener if we're actually done. |
| 142 | if (done_) |
| 143 | listener_->OnProviderUpdate(false); |
| 144 | return; |
| 145 | } |
| 146 | |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 147 | default_provider_suggestion_ = suggestion; |
[email protected] | 9e78974 | 2011-01-10 23:27:32 | [diff] [blame] | 148 | |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 149 | string16 adjusted_input_text(input_text); |
[email protected] | e918c11 | 2010-12-08 23:03:49 | [diff] [blame] | 150 | AutocompleteInput::RemoveForcedQueryStringIfNecessary(input_.type(), |
| 151 | &adjusted_input_text); |
| 152 | |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 153 | const string16 text = adjusted_input_text + suggestion.text; |
[email protected] | 9acdcdc0 | 2012-05-21 20:58:36 | [diff] [blame] | 154 | bool results_updated = false; |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 155 | // Remove any matches that are identical to |text|. We don't use the |
| 156 | // destination_url for comparison as it varies depending upon the index passed |
| 157 | // to TemplateURL::ReplaceSearchTerms. |
| 158 | for (ACMatches::iterator i = matches_.begin(); i != matches_.end();) { |
| 159 | if (((i->type == AutocompleteMatch::SEARCH_HISTORY) || |
| 160 | (i->type == AutocompleteMatch::SEARCH_SUGGEST)) && |
| 161 | (i->fill_into_edit == text)) { |
[email protected] | e030de6 | 2010-11-24 05:41:19 | [diff] [blame] | 162 | i = matches_.erase(i); |
[email protected] | 9acdcdc0 | 2012-05-21 20:58:36 | [diff] [blame] | 163 | results_updated = true; |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 164 | } else { |
| 165 | ++i; |
| 166 | } |
| 167 | } |
| 168 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 169 | // Add the new instant suggest result. We give it a rank higher than |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 170 | // SEARCH_WHAT_YOU_TYPED so that it gets autocompleted. |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 171 | const int verbatim_relevance = GetVerbatimRelevance(); |
| 172 | if (suggestion.type == INSTANT_SUGGESTION_SEARCH) { |
| 173 | // Instant has a query suggestion. |
| 174 | int did_not_accept_default_suggestion = default_suggest_results_.empty() ? |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 175 | TemplateURLRef::NO_SUGGESTIONS_AVAILABLE : |
| 176 | TemplateURLRef::NO_SUGGESTION_CHOSEN; |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 177 | MatchMap match_map; |
| 178 | AddMatchToMap(text, adjusted_input_text, verbatim_relevance + 1, |
| 179 | AutocompleteMatch::SEARCH_SUGGEST, |
| 180 | did_not_accept_default_suggestion, false, &match_map); |
| 181 | if (!match_map.empty()) { |
| 182 | matches_.push_back(match_map.begin()->second); |
| 183 | results_updated = true; |
| 184 | } |
| 185 | } else { |
| 186 | // Instant has an URL suggestion. |
| 187 | matches_.push_back(NavigationToMatch( |
| 188 | NavigationResult(GURL(UTF16ToUTF8(suggestion.text)), |
| 189 | string16(), |
| 190 | verbatim_relevance + 1), |
| 191 | false)); |
[email protected] | 9acdcdc0 | 2012-05-21 20:58:36 | [diff] [blame] | 192 | results_updated = true; |
| 193 | } |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 194 | |
[email protected] | 9acdcdc0 | 2012-05-21 20:58:36 | [diff] [blame] | 195 | if (results_updated || done_) |
| 196 | listener_->OnProviderUpdate(results_updated); |
[email protected] | 601858c0 | 2010-09-01 17:08:20 | [diff] [blame] | 197 | } |
| 198 | |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 199 | void SearchProvider::Start(const AutocompleteInput& input, |
[email protected] | 8deeb95 | 2008-10-09 18:21:27 | [diff] [blame] | 200 | bool minimal_changes) { |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 201 | matches_.clear(); |
| 202 | |
[email protected] | ea3b9a50 | 2011-04-04 14:19:37 | [diff] [blame] | 203 | instant_finalized_ = |
| 204 | (input.matches_requested() != AutocompleteInput::ALL_MATCHES); |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 205 | |
[email protected] | 6c85aa0 | 2009-02-27 12:08:09 | [diff] [blame] | 206 | // Can't return search/suggest results for bogus input or without a profile. |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 207 | if (!profile_ || (input.type() == AutocompleteInput::INVALID)) { |
[email protected] | e29249dc5 | 2012-07-19 17:33:50 | [diff] [blame] | 208 | Stop(false); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 209 | return; |
| 210 | } |
| 211 | |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 212 | keyword_input_text_.clear(); |
| 213 | const TemplateURL* keyword_provider = |
| 214 | KeywordProvider::GetSubstitutingTemplateURLForInput(profile_, input, |
| 215 | &keyword_input_text_); |
[email protected] | 8d45713 | 2010-11-04 18:13:40 | [diff] [blame] | 216 | if (keyword_input_text_.empty()) |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 217 | keyword_provider = NULL; |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 218 | |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 219 | TemplateURLService* model = providers_.template_url_service(); |
| 220 | DCHECK(model); |
| 221 | model->Load(); |
| 222 | const TemplateURL* default_provider = model->GetDefaultSearchProvider(); |
[email protected] | 9b74ab5 | 2012-03-30 16:08:07 | [diff] [blame] | 223 | if (default_provider && !default_provider->SupportsReplacement()) |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 224 | default_provider = NULL; |
| 225 | |
| 226 | if (keyword_provider == default_provider) |
[email protected] | e17511f | 2011-07-13 14:09:18 | [diff] [blame] | 227 | default_provider = NULL; // No use in querying the same provider twice. |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 228 | |
| 229 | if (!default_provider && !keyword_provider) { |
| 230 | // No valid providers. |
[email protected] | e29249dc5 | 2012-07-19 17:33:50 | [diff] [blame] | 231 | Stop(false); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 232 | return; |
| 233 | } |
| 234 | |
| 235 | // If we're still running an old query but have since changed the query text |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 236 | // or the providers, abort the query. |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 237 | string16 default_provider_keyword(default_provider ? |
| 238 | default_provider->keyword() : string16()); |
| 239 | string16 keyword_provider_keyword(keyword_provider ? |
| 240 | keyword_provider->keyword() : string16()); |
[email protected] | 9e78974 | 2011-01-10 23:27:32 | [diff] [blame] | 241 | if (!minimal_changes || |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 242 | !providers_.equal(default_provider_keyword, keyword_provider_keyword)) { |
[email protected] | 9e78974 | 2011-01-10 23:27:32 | [diff] [blame] | 243 | if (done_) |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 244 | default_provider_suggestion_ = InstantSuggestion(); |
[email protected] | 9e78974 | 2011-01-10 23:27:32 | [diff] [blame] | 245 | else |
[email protected] | e29249dc5 | 2012-07-19 17:33:50 | [diff] [blame] | 246 | Stop(false); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 247 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 248 | |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 249 | providers_.set(default_provider_keyword, keyword_provider_keyword); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 250 | |
| 251 | if (input.text().empty()) { |
| 252 | // User typed "?" alone. Give them a placeholder result indicating what |
| 253 | // this syntax does. |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 254 | if (default_provider) { |
[email protected] | 69c579e | 2010-04-23 20:01:00 | [diff] [blame] | 255 | AutocompleteMatch match; |
| 256 | match.provider = this; |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 257 | match.contents.assign(l10n_util::GetStringUTF16(IDS_EMPTY_KEYWORD_VALUE)); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 258 | match.contents_class.push_back( |
[email protected] | 2c33dd2 | 2010-02-11 21:46:35 | [diff] [blame] | 259 | ACMatchClassification(0, ACMatchClassification::NONE)); |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 260 | match.keyword = providers_.default_provider(); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 261 | matches_.push_back(match); |
| 262 | } |
[email protected] | e29249dc5 | 2012-07-19 17:33:50 | [diff] [blame] | 263 | Stop(false); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 264 | return; |
| 265 | } |
| 266 | |
| 267 | input_ = input; |
| 268 | |
[email protected] | 5cd06c3e | 2012-09-20 03:25:20 | [diff] [blame] | 269 | // Don't run the normal provider flow when the Instant Extended API is |
| 270 | // enabled. (When the Extended API is enabled, the embedded page will handle |
| 271 | // all search suggestions itself.) |
| 272 | // TODO(dcblack): once we are done refactoring the omnibox so we don't need to |
| 273 | // use FinalizeInstantQuery anymore, we can take out this check and remove |
| 274 | // this provider from kInstantExtendedOmniboxProviders. |
| 275 | if (!chrome::search::IsInstantExtendedAPIEnabled(profile_)) { |
| 276 | DoHistoryQuery(minimal_changes); |
| 277 | StartOrStopSuggestQuery(minimal_changes); |
| 278 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 279 | ConvertResultsToAutocompleteMatches(); |
| 280 | } |
| 281 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 282 | SearchProvider::Result::Result(int relevance) : relevance_(relevance) {} |
| 283 | SearchProvider::Result::~Result() {} |
| 284 | |
| 285 | SearchProvider::SuggestResult::SuggestResult(const string16& suggestion, |
| 286 | int relevance) |
| 287 | : Result(relevance), |
| 288 | suggestion_(suggestion) { |
| 289 | } |
| 290 | |
| 291 | SearchProvider::SuggestResult::~SuggestResult() {} |
| 292 | |
| 293 | SearchProvider::NavigationResult::NavigationResult(const GURL& url, |
| 294 | const string16& description, |
| 295 | int relevance) |
| 296 | : Result(relevance), |
| 297 | url_(url), |
| 298 | description_(description) { |
| 299 | DCHECK(url_.is_valid()); |
| 300 | } |
| 301 | |
| 302 | SearchProvider::NavigationResult::~NavigationResult() {} |
| 303 | |
| 304 | class SearchProvider::CompareScoredResults { |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 305 | public: |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 306 | bool operator()(const Result& a, const Result& b) { |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 307 | // Sort in descending relevance order. |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 308 | return a.relevance() > b.relevance(); |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 309 | } |
| 310 | }; |
| 311 | |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 312 | void SearchProvider::Run() { |
| 313 | // Start a new request with the current input. |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 314 | suggest_results_pending_ = 0; |
[email protected] | a0ad93ea | 2012-05-07 22:11:53 | [diff] [blame] | 315 | time_suggest_request_sent_ = base::TimeTicks::Now(); |
[email protected] | 9ff9172 | 2012-09-07 05:29:12 | [diff] [blame] | 316 | |
| 317 | default_fetcher_.reset(CreateSuggestFetcher(kDefaultProviderURLFetcherID, |
| 318 | providers_.GetDefaultProviderURL(), input_.text())); |
| 319 | keyword_fetcher_.reset(CreateSuggestFetcher(kKeywordProviderURLFetcherID, |
| 320 | providers_.GetKeywordProviderURL(), keyword_input_text_)); |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 321 | |
| 322 | // Both the above can fail if the providers have been modified or deleted |
| 323 | // since the query began. |
| 324 | if (suggest_results_pending_ == 0) { |
| 325 | UpdateDone(); |
| 326 | // We only need to update the listener if we're actually done. |
| 327 | if (done_) |
| 328 | listener_->OnProviderUpdate(false); |
| 329 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 330 | } |
| 331 | |
[email protected] | e29249dc5 | 2012-07-19 17:33:50 | [diff] [blame] | 332 | void SearchProvider::Stop(bool clear_cached_results) { |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 333 | StopSuggest(); |
| 334 | done_ = true; |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 335 | default_provider_suggestion_ = InstantSuggestion(); |
[email protected] | e29249dc5 | 2012-07-19 17:33:50 | [diff] [blame] | 336 | |
| 337 | if (clear_cached_results) |
| 338 | ClearResults(); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 339 | } |
| 340 | |
[email protected] | 0e9e878 | 2012-05-15 23:01:51 | [diff] [blame] | 341 | void SearchProvider::AddProviderInfo(ProvidersInfo* provider_info) const { |
| 342 | provider_info->push_back(metrics::OmniboxEventProto_ProviderInfo()); |
| 343 | metrics::OmniboxEventProto_ProviderInfo& new_entry = provider_info->back(); |
| 344 | new_entry.set_provider(AsOmniboxEventProviderType()); |
| 345 | new_entry.set_provider_done(done_); |
| 346 | } |
| 347 | |
[email protected] | 10c2d69 | 2012-05-11 05:32:23 | [diff] [blame] | 348 | void SearchProvider::OnURLFetchComplete(const net::URLFetcher* source) { |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 349 | DCHECK(!done_); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 350 | suggest_results_pending_--; |
[email protected] | 7706a52 | 2012-08-16 17:42:25 | [diff] [blame] | 351 | LogOmniboxSuggestRequest(REPLY_RECEIVED); |
[email protected] | 1cb2dac | 2010-03-08 21:49:15 | [diff] [blame] | 352 | DCHECK_GE(suggest_results_pending_, 0); // Should never go negative. |
[email protected] | ec9207d3 | 2008-09-26 00:51:06 | [diff] [blame] | 353 | const net::HttpResponseHeaders* const response_headers = |
[email protected] | 7cc6e563 | 2011-10-25 17:56:12 | [diff] [blame] | 354 | source->GetResponseHeaders(); |
[email protected] | c530c85 | 2011-10-24 18:18:34 | [diff] [blame] | 355 | std::string json_data; |
| 356 | source->GetResponseAsString(&json_data); |
[email protected] | 6c85aa0 | 2009-02-27 12:08:09 | [diff] [blame] | 357 | // JSON is supposed to be UTF-8, but some suggest service providers send JSON |
| 358 | // files in non-UTF-8 encodings. The actual encoding is usually specified in |
| 359 | // the Content-Type header field. |
[email protected] | ec9207d3 | 2008-09-26 00:51:06 | [diff] [blame] | 360 | if (response_headers) { |
| 361 | std::string charset; |
| 362 | if (response_headers->GetCharset(&charset)) { |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 363 | string16 data_16; |
[email protected] | ec9207d3 | 2008-09-26 00:51:06 | [diff] [blame] | 364 | // TODO(jungshik): Switch to CodePageToUTF8 after it's added. |
[email protected] | c530c85 | 2011-10-24 18:18:34 | [diff] [blame] | 365 | if (base::CodepageToUTF16(json_data, charset.c_str(), |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 366 | base::OnStringConversionError::FAIL, |
| 367 | &data_16)) |
| 368 | json_data = UTF16ToUTF8(data_16); |
[email protected] | ec9207d3 | 2008-09-26 00:51:06 | [diff] [blame] | 369 | } |
| 370 | } |
| 371 | |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 372 | const bool is_keyword = (source == keyword_fetcher_.get()); |
[email protected] | 013e9a0 | 2012-05-18 20:27:10 | [diff] [blame] | 373 | const bool request_succeeded = |
| 374 | source->GetStatus().is_success() && source->GetResponseCode() == 200; |
[email protected] | a0ad93ea | 2012-05-07 22:11:53 | [diff] [blame] | 375 | |
| 376 | // Record response time for suggest requests sent to Google. We care |
| 377 | // only about the common case: the Google default provider used in |
| 378 | // non-keyword mode. |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 379 | const TemplateURL* default_url = providers_.GetDefaultProviderURL(); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 380 | if (!is_keyword && default_url && |
[email protected] | 9899a61 | 2012-08-21 23:50:04 | [diff] [blame] | 381 | (TemplateURLPrepopulateData::GetEngineType(default_url->url()) == |
| 382 | SEARCH_ENGINE_GOOGLE)) { |
[email protected] | 6dc950f | 2012-07-16 19:49:08 | [diff] [blame] | 383 | const TimeDelta elapsed_time = |
[email protected] | 013e9a0 | 2012-05-18 20:27:10 | [diff] [blame] | 384 | base::TimeTicks::Now() - time_suggest_request_sent_; |
| 385 | if (request_succeeded) { |
| 386 | UMA_HISTOGRAM_TIMES("Omnibox.SuggestRequest.Success.GoogleResponseTime", |
| 387 | elapsed_time); |
| 388 | } else { |
| 389 | UMA_HISTOGRAM_TIMES("Omnibox.SuggestRequest.Failure.GoogleResponseTime", |
| 390 | elapsed_time); |
| 391 | } |
[email protected] | b4cebf8 | 2008-12-29 19:59:08 | [diff] [blame] | 392 | } |
| 393 | |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 394 | bool results_updated = false; |
| 395 | if (request_succeeded) { |
| 396 | JSONStringValueSerializer deserializer(json_data); |
| 397 | deserializer.set_allow_trailing_comma(true); |
| 398 | scoped_ptr<Value> data(deserializer.Deserialize(NULL, NULL)); |
| 399 | results_updated = data.get() && ParseSuggestResults(data.get(), is_keyword); |
| 400 | } |
| 401 | |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 402 | ConvertResultsToAutocompleteMatches(); |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 403 | if (done_ || results_updated) |
| 404 | listener_->OnProviderUpdate(results_updated); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 405 | } |
| 406 | |
[email protected] | 601858c0 | 2010-09-01 17:08:20 | [diff] [blame] | 407 | SearchProvider::~SearchProvider() { |
| 408 | } |
| 409 | |
[email protected] | 8d45713 | 2010-11-04 18:13:40 | [diff] [blame] | 410 | void SearchProvider::DoHistoryQuery(bool minimal_changes) { |
| 411 | // The history query results are synchronous, so if minimal_changes is true, |
| 412 | // we still have the last results and don't need to do anything. |
| 413 | if (minimal_changes) |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 414 | return; |
| 415 | |
[email protected] | 8d45713 | 2010-11-04 18:13:40 | [diff] [blame] | 416 | keyword_history_results_.clear(); |
| 417 | default_history_results_.clear(); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 418 | |
[email protected] | 8d45713 | 2010-11-04 18:13:40 | [diff] [blame] | 419 | HistoryService* const history_service = |
[email protected] | 9d2db76 | 2012-06-19 00:01:10 | [diff] [blame] | 420 | HistoryServiceFactory::GetForProfile(profile_, Profile::EXPLICIT_ACCESS); |
[email protected] | 8d45713 | 2010-11-04 18:13:40 | [diff] [blame] | 421 | history::URLDatabase* url_db = history_service ? |
| 422 | history_service->InMemoryDatabase() : NULL; |
| 423 | if (!url_db) |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 424 | return; |
| 425 | |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 426 | // Request history for both the keyword and default provider. We grab many |
| 427 | // more matches than we'll ultimately clamp to so that if there are several |
| 428 | // recent multi-word matches who scores are lowered (see |
| 429 | // AddHistoryResultsToMap()), they won't crowd out older, higher-scoring |
| 430 | // matches. Note that this doesn't fix the problem entirely, but merely |
| 431 | // limits it to cases with a very large number of such multi-word matches; for |
| 432 | // now, this seems OK compared with the complexity of a real fix, which would |
| 433 | // require multiple searches and tracking of "single- vs. multi-word" in the |
| 434 | // database. |
| 435 | int num_matches = kMaxMatches * 5; |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 436 | const TemplateURL* default_url = providers_.GetDefaultProviderURL(); |
| 437 | if (default_url) { |
| 438 | url_db->GetMostRecentKeywordSearchTerms(default_url->id(), input_.text(), |
| 439 | num_matches, &default_history_results_); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 440 | } |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 441 | const TemplateURL* keyword_url = providers_.GetKeywordProviderURL(); |
| 442 | if (keyword_url) { |
| 443 | url_db->GetMostRecentKeywordSearchTerms(keyword_url->id(), |
[email protected] | 3954c3a | 2012-04-10 20:17:55 | [diff] [blame] | 444 | keyword_input_text_, num_matches, &keyword_history_results_); |
| 445 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 446 | } |
| 447 | |
[email protected] | 6dc950f | 2012-07-16 19:49:08 | [diff] [blame] | 448 | void SearchProvider::StartOrStopSuggestQuery(bool minimal_changes) { |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 449 | if (!IsQuerySuitableForSuggest()) { |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 450 | StopSuggest(); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 451 | ClearResults(); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 452 | return; |
| 453 | } |
| 454 | |
| 455 | // For the minimal_changes case, if we finished the previous query and still |
| 456 | // have its results, or are allowed to keep running it, just do that, rather |
| 457 | // than starting a new query. |
| 458 | if (minimal_changes && |
[email protected] | ea3b9a50 | 2011-04-04 14:19:37 | [diff] [blame] | 459 | (have_suggest_results_ || |
| 460 | (!done_ && |
| 461 | input_.matches_requested() == AutocompleteInput::ALL_MATCHES))) |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 462 | return; |
| 463 | |
| 464 | // We can't keep running any previous query, so halt it. |
| 465 | StopSuggest(); |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 466 | |
| 467 | // Remove existing results that cannot inline autocomplete the new input. |
| 468 | RemoveStaleResults(); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 469 | |
| 470 | // We can't start a new query if we're only allowed synchronous results. |
[email protected] | ea3b9a50 | 2011-04-04 14:19:37 | [diff] [blame] | 471 | if (input_.matches_requested() != AutocompleteInput::ALL_MATCHES) |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 472 | return; |
| 473 | |
[email protected] | 515ffa94 | 2012-11-27 20:18:24 | [diff] [blame^] | 474 | // To avoid flooding the suggest server, don't send a query until at least 100 |
| 475 | // ms since the last query. |
| 476 | const int kMinimumTimeBetweenSuggestQueriesMs = 100; |
| 477 | base::TimeTicks next_suggest_time(time_suggest_request_sent_ + |
| 478 | TimeDelta::FromMilliseconds(kMinimumTimeBetweenSuggestQueriesMs)); |
| 479 | base::TimeTicks now(base::TimeTicks::Now()); |
| 480 | if (now >= next_suggest_time) { |
| 481 | Run(); |
| 482 | return; |
| 483 | } |
| 484 | timer_.Start(FROM_HERE, next_suggest_time - now, this, &SearchProvider::Run); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 485 | } |
| 486 | |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 487 | bool SearchProvider::IsQuerySuitableForSuggest() const { |
[email protected] | 3954c3a | 2012-04-10 20:17:55 | [diff] [blame] | 488 | // Don't run Suggest in incognito mode, if the engine doesn't support it, or |
| 489 | // if the user has disabled it. |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 490 | const TemplateURL* default_url = providers_.GetDefaultProviderURL(); |
| 491 | const TemplateURL* keyword_url = providers_.GetKeywordProviderURL(); |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 492 | if (profile_->IsOffTheRecord() || |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 493 | ((!default_url || default_url->suggestions_url().empty()) && |
| 494 | (!keyword_url || keyword_url->suggestions_url().empty())) || |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 495 | !profile_->GetPrefs()->GetBoolean(prefs::kSearchSuggestEnabled)) |
| 496 | return false; |
| 497 | |
[email protected] | cac59d3 | 2010-08-09 23:23:14 | [diff] [blame] | 498 | // If the input type might be a URL, we take extra care so that private data |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 499 | // isn't sent to the server. |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 500 | |
[email protected] | cac59d3 | 2010-08-09 23:23:14 | [diff] [blame] | 501 | // FORCED_QUERY means the user is explicitly asking us to search for this, so |
| 502 | // we assume it isn't a URL and/or there isn't private data. |
| 503 | if (input_.type() == AutocompleteInput::FORCED_QUERY) |
| 504 | return true; |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 505 | |
[email protected] | cac59d3 | 2010-08-09 23:23:14 | [diff] [blame] | 506 | // Next we check the scheme. If this is UNKNOWN/REQUESTED_URL/URL with a |
| 507 | // scheme that isn't http/https/ftp, we shouldn't send it. Sending things |
| 508 | // like file: and data: is both a waste of time and a disclosure of |
| 509 | // potentially private, local data. Other "schemes" may actually be |
| 510 | // usernames, and we don't want to send passwords. If the scheme is OK, we |
| 511 | // still need to check other cases below. If this is QUERY, then the presence |
| 512 | // of these schemes means the user explicitly typed one, and thus this is |
| 513 | // probably a URL that's being entered and happens to currently be invalid -- |
| 514 | // in which case we again want to run our checks below. Other QUERY cases are |
| 515 | // less likely to be URLs and thus we assume we're OK. |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 516 | if (!LowerCaseEqualsASCII(input_.scheme(), chrome::kHttpScheme) && |
| 517 | !LowerCaseEqualsASCII(input_.scheme(), chrome::kHttpsScheme) && |
| 518 | !LowerCaseEqualsASCII(input_.scheme(), chrome::kFtpScheme)) |
[email protected] | cac59d3 | 2010-08-09 23:23:14 | [diff] [blame] | 519 | return (input_.type() == AutocompleteInput::QUERY); |
| 520 | |
| 521 | // Don't send URLs with usernames, queries or refs. Some of these are |
| 522 | // private, and the Suggest server is unlikely to have any useful results |
| 523 | // for any of them. Also don't send URLs with ports, as we may initially |
| 524 | // think that a username + password is a host + port (and we don't want to |
| 525 | // send usernames/passwords), and even if the port really is a port, the |
| 526 | // server is once again unlikely to have and useful results. |
| 527 | const url_parse::Parsed& parts = input_.parts(); |
| 528 | if (parts.username.is_nonempty() || parts.port.is_nonempty() || |
| 529 | parts.query.is_nonempty() || parts.ref.is_nonempty()) |
| 530 | return false; |
| 531 | |
| 532 | // Don't send anything for https except the hostname. Hostnames are OK |
| 533 | // because they are visible when the TCP connection is established, but the |
| 534 | // specific path may reveal private information. |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 535 | if (LowerCaseEqualsASCII(input_.scheme(), chrome::kHttpsScheme) && |
| 536 | parts.path.is_nonempty()) |
[email protected] | cac59d3 | 2010-08-09 23:23:14 | [diff] [blame] | 537 | return false; |
[email protected] | 83c72648 | 2008-09-10 06:36:34 | [diff] [blame] | 538 | |
| 539 | return true; |
| 540 | } |
| 541 | |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 542 | void SearchProvider::StopSuggest() { |
[email protected] | 7706a52 | 2012-08-16 17:42:25 | [diff] [blame] | 543 | // Increment the appropriate field in the histogram by the number of |
| 544 | // pending requests that were invalidated. |
| 545 | for (int i = 0; i < suggest_results_pending_; i++) |
| 546 | LogOmniboxSuggestRequest(REQUEST_INVALIDATED); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 547 | suggest_results_pending_ = 0; |
[email protected] | 2d31666 | 2008-09-03 18:18:14 | [diff] [blame] | 548 | timer_.Stop(); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 549 | // Stop any in-progress URL fetches. |
| 550 | keyword_fetcher_.reset(); |
| 551 | default_fetcher_.reset(); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 552 | } |
| 553 | |
| 554 | void SearchProvider::ClearResults() { |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 555 | keyword_suggest_results_.clear(); |
| 556 | default_suggest_results_.clear(); |
| 557 | keyword_navigation_results_.clear(); |
| 558 | default_navigation_results_.clear(); |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 559 | has_suggested_relevance_ = false; |
| 560 | verbatim_relevance_ = -1; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 561 | have_suggest_results_ = false; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 562 | } |
| 563 | |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 564 | void SearchProvider::RemoveStaleResults() { |
| 565 | RemoveStaleSuggestResults(&keyword_suggest_results_, true); |
| 566 | RemoveStaleSuggestResults(&default_suggest_results_, false); |
| 567 | RemoveStaleNavigationResults(&keyword_navigation_results_, true); |
| 568 | RemoveStaleNavigationResults(&default_navigation_results_, false); |
| 569 | } |
| 570 | |
| 571 | void SearchProvider::RemoveStaleSuggestResults(SuggestResults* list, |
| 572 | bool is_keyword) { |
| 573 | const string16& input = is_keyword ? keyword_input_text_ : input_.text(); |
| 574 | for (SuggestResults::iterator i = list->begin(); i < list->end();) |
| 575 | i = StartsWith(i->suggestion(), input, false) ? (i + 1) : list->erase(i); |
| 576 | } |
| 577 | |
| 578 | void SearchProvider::RemoveStaleNavigationResults(NavigationResults* list, |
| 579 | bool is_keyword) { |
| 580 | const string16& input = is_keyword ? keyword_input_text_ : input_.text(); |
| 581 | for (NavigationResults::iterator i = list->begin(); i < list->end();) { |
| 582 | const string16 fill(AutocompleteInput::FormattedStringWithEquivalentMeaning( |
| 583 | i->url(), StringForURLDisplay(i->url(), true, false))); |
| 584 | i = URLPrefix::BestURLPrefix(fill, input) ? (i + 1) : list->erase(i); |
| 585 | } |
| 586 | } |
| 587 | |
| 588 | void SearchProvider::ApplyCalculatedRelevance() { |
| 589 | ApplyCalculatedSuggestRelevance(&keyword_suggest_results_, true); |
| 590 | ApplyCalculatedSuggestRelevance(&default_suggest_results_, false); |
| 591 | ApplyCalculatedNavigationRelevance(&keyword_navigation_results_, true); |
| 592 | ApplyCalculatedNavigationRelevance(&default_navigation_results_, false); |
| 593 | has_suggested_relevance_ = false; |
| 594 | verbatim_relevance_ = -1; |
| 595 | } |
| 596 | |
| 597 | void SearchProvider::ApplyCalculatedSuggestRelevance(SuggestResults* list, |
| 598 | bool is_keyword) { |
| 599 | for (size_t i = 0; i < list->size(); ++i) { |
| 600 | (*list)[i].set_relevance(CalculateRelevanceForSuggestion(is_keyword) + |
| 601 | (list->size() - i - 1)); |
| 602 | } |
| 603 | } |
| 604 | |
| 605 | void SearchProvider::ApplyCalculatedNavigationRelevance(NavigationResults* list, |
| 606 | bool is_keyword) { |
| 607 | for (size_t i = 0; i < list->size(); ++i) { |
| 608 | (*list)[i].set_relevance(CalculateRelevanceForNavigation(is_keyword) + |
| 609 | (list->size() - i - 1)); |
| 610 | } |
| 611 | } |
| 612 | |
[email protected] | 15fb2aa | 2012-05-22 22:52:59 | [diff] [blame] | 613 | net::URLFetcher* SearchProvider::CreateSuggestFetcher( |
[email protected] | 7cc6e563 | 2011-10-25 17:56:12 | [diff] [blame] | 614 | int id, |
[email protected] | 9ff9172 | 2012-09-07 05:29:12 | [diff] [blame] | 615 | const TemplateURL* template_url, |
[email protected] | 7cc6e563 | 2011-10-25 17:56:12 | [diff] [blame] | 616 | const string16& text) { |
[email protected] | 9ff9172 | 2012-09-07 05:29:12 | [diff] [blame] | 617 | if (!template_url || template_url->suggestions_url().empty()) |
| 618 | return NULL; |
| 619 | |
| 620 | // Bail if the suggestion URL is invalid with the given replacements. |
| 621 | GURL suggest_url(template_url->suggestions_url_ref().ReplaceSearchTerms( |
| 622 | TemplateURLRef::SearchTermsArgs(text))); |
| 623 | if (!suggest_url.is_valid()) |
| 624 | return NULL; |
| 625 | |
| 626 | suggest_results_pending_++; |
| 627 | LogOmniboxSuggestRequest(REQUEST_SENT); |
| 628 | |
| 629 | net::URLFetcher* fetcher = |
| 630 | net::URLFetcher::Create(id, suggest_url, net::URLFetcher::GET, this); |
[email protected] | 7cc6e563 | 2011-10-25 17:56:12 | [diff] [blame] | 631 | fetcher->SetRequestContext(profile_->GetRequestContext()); |
[email protected] | d3cf8682f0 | 2012-02-29 23:29:34 | [diff] [blame] | 632 | fetcher->SetLoadFlags(net::LOAD_DO_NOT_SAVE_COOKIES); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 633 | fetcher->Start(); |
| 634 | return fetcher; |
| 635 | } |
| 636 | |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 637 | bool SearchProvider::ParseSuggestResults(Value* root_val, bool is_keyword) { |
| 638 | // TODO(pkasting): Fix |have_suggest_results_|; see http://crbug.com/130631 |
| 639 | have_suggest_results_ = false; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 640 | |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 641 | string16 query; |
| 642 | ListValue* root_list = NULL; |
| 643 | ListValue* results = NULL; |
| 644 | const string16& input_text = is_keyword ? keyword_input_text_ : input_.text(); |
| 645 | if (!root_val->GetAsList(&root_list) || !root_list->GetString(0, &query) || |
| 646 | (query != input_text) || !root_list->GetList(1, &results)) |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 647 | return false; |
| 648 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 649 | // 3rd element: Description list. |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 650 | ListValue* descriptions = NULL; |
| 651 | root_list->GetList(2, &descriptions); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 652 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 653 | // 4th element: Disregard the query URL list for now. |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 654 | |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 655 | // Reset suggested relevance information from the default provider. |
| 656 | if (!is_keyword) { |
| 657 | has_suggested_relevance_ = false; |
| 658 | verbatim_relevance_ = -1; |
| 659 | } |
| 660 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 661 | // 5th element: Optional key-value pairs from the Suggest server. |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 662 | ListValue* types = NULL; |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 663 | ListValue* relevances = NULL; |
| 664 | DictionaryValue* extras = NULL; |
| 665 | if (root_list->GetDictionary(4, &extras)) { |
| 666 | extras->GetList("google:suggesttype", &types); |
| 667 | |
| 668 | // Only accept relevance suggestions if Instant is disabled. |
[email protected] | e41982a7 | 2012-11-20 07:16:51 | [diff] [blame] | 669 | if (!is_keyword && |
| 670 | !chrome::BrowserInstantController::IsInstantEnabled(profile_)) { |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 671 | // Discard this list if its size does not match that of the suggestions. |
| 672 | if (extras->GetList("google:suggestrelevance", &relevances) && |
| 673 | relevances->GetSize() != results->GetSize()) |
| 674 | relevances = NULL; |
| 675 | |
| 676 | extras->GetInteger("google:verbatimrelevance", &verbatim_relevance_); |
| 677 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 678 | } |
| 679 | |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 680 | SuggestResults* suggest_results = |
| 681 | is_keyword ? &keyword_suggest_results_ : &default_suggest_results_; |
| 682 | NavigationResults* navigation_results = |
| 683 | is_keyword ? &keyword_navigation_results_ : &default_navigation_results_; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 684 | |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 685 | // Clear the previous results now that new results are available. |
| 686 | suggest_results->clear(); |
| 687 | navigation_results->clear(); |
| 688 | |
| 689 | string16 result, title; |
| 690 | std::string type; |
| 691 | int relevance = -1; |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 692 | for (size_t index = 0; results->GetString(index, &result); ++index) { |
[email protected] | 8e81f509 | 2010-09-29 23:19:40 | [diff] [blame] | 693 | // Google search may return empty suggestions for weird input characters, |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 694 | // they make no sense at all and can cause problems in our code. |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 695 | if (result.empty()) |
[email protected] | 8e81f509 | 2010-09-29 23:19:40 | [diff] [blame] | 696 | continue; |
| 697 | |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 698 | // Apply valid suggested relevance scores; discard invalid lists. |
| 699 | if (relevances != NULL && !relevances->GetInteger(index, &relevance)) |
| 700 | relevances = NULL; |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 701 | if (types && types->GetString(index, &type) && (type == "NAVIGATION")) { |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 702 | // Do not blindly trust the URL coming from the server to be valid. |
| 703 | GURL url(URLFixerUpper::FixupURL(UTF16ToUTF8(result), std::string())); |
| 704 | if (url.is_valid()) { |
| 705 | if (descriptions != NULL) |
| 706 | descriptions->GetString(index, &title); |
| 707 | navigation_results->push_back(NavigationResult(url, title, relevance)); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 708 | } |
| 709 | } else { |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 710 | // TODO(kochi): Improve calculator result presentation. |
| 711 | suggest_results->push_back(SuggestResult(result, relevance)); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 712 | } |
| 713 | } |
| 714 | |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 715 | // Apply calculated relevance scores if a valid list was not provided. |
| 716 | if (relevances == NULL) { |
| 717 | ApplyCalculatedSuggestRelevance(suggest_results, is_keyword); |
| 718 | ApplyCalculatedNavigationRelevance(navigation_results, is_keyword); |
| 719 | } else if (!is_keyword) { |
| 720 | has_suggested_relevance_ = true; |
| 721 | } |
| 722 | |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 723 | have_suggest_results_ = true; |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 724 | return true; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 725 | } |
| 726 | |
| 727 | void SearchProvider::ConvertResultsToAutocompleteMatches() { |
| 728 | // Convert all the results to matches and add them to a map, so we can keep |
| 729 | // the most relevant match for each result. |
| 730 | MatchMap map; |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 731 | const Time no_time; |
| 732 | int did_not_accept_keyword_suggestion = keyword_suggest_results_.empty() ? |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 733 | TemplateURLRef::NO_SUGGESTIONS_AVAILABLE : |
| 734 | TemplateURLRef::NO_SUGGESTION_CHOSEN; |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 735 | // Keyword what you typed results are handled by the KeywordProvider. |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 736 | |
[email protected] | 382a064 | 2012-06-06 06:13:52 | [diff] [blame] | 737 | int verbatim_relevance = GetVerbatimRelevance(); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 738 | int did_not_accept_default_suggestion = default_suggest_results_.empty() ? |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 739 | TemplateURLRef::NO_SUGGESTIONS_AVAILABLE : |
| 740 | TemplateURLRef::NO_SUGGESTION_CHOSEN; |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 741 | if (verbatim_relevance > 0) { |
| 742 | AddMatchToMap(input_.text(), input_.text(), verbatim_relevance, |
| 743 | AutocompleteMatch::SEARCH_WHAT_YOU_TYPED, |
| 744 | did_not_accept_default_suggestion, false, &map); |
| 745 | } |
[email protected] | b9ce853 | 2012-05-24 20:51:15 | [diff] [blame] | 746 | const size_t what_you_typed_size = map.size(); |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 747 | if (!default_provider_suggestion_.text.empty() && |
| 748 | default_provider_suggestion_.type == INSTANT_SUGGESTION_SEARCH) |
| 749 | AddMatchToMap(input_.text() + default_provider_suggestion_.text, |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 750 | input_.text(), verbatim_relevance + 1, |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 751 | AutocompleteMatch::SEARCH_SUGGEST, |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 752 | did_not_accept_default_suggestion, false, &map); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 753 | |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 754 | AddHistoryResultsToMap(keyword_history_results_, true, |
| 755 | did_not_accept_keyword_suggestion, &map); |
| 756 | AddHistoryResultsToMap(default_history_results_, false, |
| 757 | did_not_accept_default_suggestion, &map); |
| 758 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 759 | AddSuggestResultsToMap(keyword_suggest_results_, true, &map); |
| 760 | AddSuggestResultsToMap(default_suggest_results_, false, &map); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 761 | |
| 762 | // Now add the most relevant matches from the map to |matches_|. |
| 763 | matches_.clear(); |
| 764 | for (MatchMap::const_iterator i(map.begin()); i != map.end(); ++i) |
| 765 | matches_.push_back(i->second); |
| 766 | |
[email protected] | 93b7383 | 2012-10-18 20:18:38 | [diff] [blame] | 767 | if (!default_provider_suggestion_.text.empty() && |
| 768 | default_provider_suggestion_.type == INSTANT_SUGGESTION_URL) |
| 769 | matches_.push_back(NavigationToMatch( |
| 770 | NavigationResult(GURL(UTF16ToUTF8(default_provider_suggestion_.text)), |
| 771 | string16(), |
| 772 | verbatim_relevance + 1), |
| 773 | false)); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 774 | AddNavigationResultsToMatches(keyword_navigation_results_, true); |
| 775 | AddNavigationResultsToMatches(default_navigation_results_, false); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 776 | |
[email protected] | b9ce853 | 2012-05-24 20:51:15 | [diff] [blame] | 777 | // Allow an additional match for "what you typed" if it's present. |
| 778 | const size_t max_total_matches = kMaxMatches + what_you_typed_size; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 779 | std::partial_sort(matches_.begin(), |
| 780 | matches_.begin() + std::min(max_total_matches, matches_.size()), |
| 781 | matches_.end(), &AutocompleteMatch::MoreRelevant); |
[email protected] | 3723e6e | 2012-06-11 21:06:56 | [diff] [blame] | 782 | |
| 783 | // If the top match is effectively 'verbatim' but exceeds the calculated |
| 784 | // verbatim relevance, and REQUESTED_URL |input_| has a |desired_tld| |
| 785 | // (for example ".com" when the CTRL key is pressed for REQUESTED_URL input), |
| 786 | // promote a URL_WHAT_YOU_TYPED match to the top. Otherwise, these matches can |
| 787 | // stomp the HistoryURLProvider's similar transient URL_WHAT_YOU_TYPED match, |
| 788 | // and CTRL+ENTER will invoke the search instead of the expected navigation. |
| 789 | if ((has_suggested_relevance_ || verbatim_relevance_ >= 0) && |
| 790 | input_.type() == AutocompleteInput::REQUESTED_URL && |
| 791 | !input_.desired_tld().empty() && !matches_.empty() && |
| 792 | matches_.front().relevance > CalculateRelevanceForVerbatim() && |
| 793 | matches_.front().fill_into_edit == input_.text()) { |
| 794 | AutocompleteMatch match = HistoryURLProvider::SuggestExactInput( |
| 795 | this, input_, !HasHTTPScheme(input_.text())); |
| 796 | match.relevance = matches_.front().relevance + 1; |
| 797 | matches_.insert(matches_.begin(), match); |
| 798 | } |
| 799 | |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 800 | if (matches_.size() > max_total_matches) |
[email protected] | a28e9566 | 2008-11-12 19:19:02 | [diff] [blame] | 801 | matches_.erase(matches_.begin() + max_total_matches, matches_.end()); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 802 | |
[email protected] | 382a064 | 2012-06-06 06:13:52 | [diff] [blame] | 803 | // Check constraints that may be violated by suggested relevances. |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 804 | if (!matches_.empty() && |
[email protected] | e6acd00 | 2012-06-16 22:27:47 | [diff] [blame] | 805 | (has_suggested_relevance_ || verbatim_relevance_ >= 0)) { |
[email protected] | 382a064 | 2012-06-06 06:13:52 | [diff] [blame] | 806 | bool reconstruct_matches = false; |
[email protected] | 1beee34 | 2012-06-19 22:22:28 | [diff] [blame] | 807 | if (matches_.front().type != AutocompleteMatch::SEARCH_WHAT_YOU_TYPED && |
| 808 | matches_.front().type != AutocompleteMatch::URL_WHAT_YOU_TYPED && |
| 809 | matches_.front().inline_autocomplete_offset == string16::npos && |
| 810 | matches_.front().fill_into_edit != input_.text()) { |
| 811 | // Disregard suggested relevances if the top match is not SWYT, inlinable, |
| 812 | // or URL_WHAT_YOU_TYPED (which may be top match regardless of inlining). |
| 813 | // For example, input "foo" should not invoke a search for "bar", which |
| 814 | // would happen if the "bar" search match outranked all other matches. |
| 815 | ApplyCalculatedRelevance(); |
| 816 | reconstruct_matches = true; |
| 817 | } else if (matches_.front().relevance < CalculateRelevanceForVerbatim()) { |
| 818 | // Disregard the suggested verbatim relevance if the top score is below |
| 819 | // the usual verbatim value. For example, a BarProvider may rely on |
| 820 | // SearchProvider's verbatim or inlineable matches for input "foo" to |
| 821 | // always outrank its own lowly-ranked non-inlineable "bar" match. |
| 822 | verbatim_relevance_ = -1; |
| 823 | reconstruct_matches = true; |
[email protected] | e6acd00 | 2012-06-16 22:27:47 | [diff] [blame] | 824 | } |
| 825 | if (input_.type() == AutocompleteInput::URL && |
| 826 | matches_.front().relevance > CalculateRelevanceForVerbatim() && |
| 827 | (matches_.front().type == AutocompleteMatch::SEARCH_SUGGEST || |
| 828 | matches_.front().type == AutocompleteMatch::SEARCH_WHAT_YOU_TYPED)) { |
| 829 | // Disregard the suggested search and verbatim relevances if the input |
| 830 | // type is URL and the top match is a highly-ranked search suggestion. |
| 831 | // For example, prevent a search for "foo.com" from outranking another |
| 832 | // provider's navigation for "foo.com" or "foo.com/url_from_history". |
| 833 | // Reconstruction will also ensure that the new top match is inlineable. |
| 834 | ApplyCalculatedSuggestRelevance(&keyword_suggest_results_, true); |
| 835 | ApplyCalculatedSuggestRelevance(&default_suggest_results_, false); |
[email protected] | 382a064 | 2012-06-06 06:13:52 | [diff] [blame] | 836 | verbatim_relevance_ = -1; |
| 837 | reconstruct_matches = true; |
| 838 | } |
| 839 | if (reconstruct_matches) { |
| 840 | ConvertResultsToAutocompleteMatches(); |
| 841 | return; |
| 842 | } |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 843 | } |
| 844 | |
[email protected] | cc63dea | 2008-08-21 20:56:31 | [diff] [blame] | 845 | UpdateStarredStateOfMatches(); |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 846 | UpdateDone(); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 847 | } |
| 848 | |
| 849 | void SearchProvider::AddNavigationResultsToMatches( |
| 850 | const NavigationResults& navigation_results, |
| 851 | bool is_keyword) { |
| 852 | if (!navigation_results.empty()) { |
[email protected] | 6c53584 | 2012-05-15 05:20:55 | [diff] [blame] | 853 | // TODO(kochi|msw): Add more navigational results if they get more |
| 854 | // meaningful relevance values; see http://b/1170574. |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 855 | // CompareScoredResults sorts by descending relevance; so use min_element. |
[email protected] | 6c53584 | 2012-05-15 05:20:55 | [diff] [blame] | 856 | NavigationResults::const_iterator result( |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 857 | std::min_element(navigation_results.begin(), |
[email protected] | 6c53584 | 2012-05-15 05:20:55 | [diff] [blame] | 858 | navigation_results.end(), |
| 859 | CompareScoredResults())); |
| 860 | matches_.push_back(NavigationToMatch(*result, is_keyword)); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 861 | } |
| 862 | } |
| 863 | |
| 864 | void SearchProvider::AddHistoryResultsToMap(const HistoryResults& results, |
| 865 | bool is_keyword, |
| 866 | int did_not_accept_suggestion, |
| 867 | MatchMap* map) { |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 868 | if (results.empty()) |
| 869 | return; |
| 870 | |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 871 | bool prevent_inline_autocomplete = input_.prevent_inline_autocomplete() || |
| 872 | (input_.type() == AutocompleteInput::URL); |
| 873 | const string16& input_text = is_keyword ? keyword_input_text_ : input_.text(); |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 874 | bool input_multiple_words = HasMultipleWords(input_text); |
| 875 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 876 | SuggestResults scored_results; |
| 877 | if (!prevent_inline_autocomplete && input_multiple_words) { |
| 878 | // ScoreHistoryResults() allows autocompletion of multi-word, 1-visit |
| 879 | // queries if the input also has multiple words. But if we were already |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 880 | // autocompleting a multi-word, multi-visit query, and the current input is |
| 881 | // still a prefix of it, then changing the autocompletion suddenly feels |
| 882 | // wrong. To detect this case, first score as if only one word has been |
| 883 | // typed, then check for a best result that is an autocompleted, multi-word |
| 884 | // query. If we find one, then just keep that score set. |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 885 | scored_results = ScoreHistoryResults(results, prevent_inline_autocomplete, |
| 886 | false, input_text, is_keyword); |
| 887 | if ((scored_results[0].relevance() < |
| 888 | AutocompleteResult::kLowestDefaultScore) || |
| 889 | !HasMultipleWords(scored_results[0].suggestion())) |
| 890 | scored_results.clear(); // Didn't detect the case above, score normally. |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 891 | } |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 892 | if (scored_results.empty()) |
| 893 | scored_results = ScoreHistoryResults(results, prevent_inline_autocomplete, |
| 894 | input_multiple_words, input_text, |
| 895 | is_keyword); |
| 896 | for (SuggestResults::const_iterator i(scored_results.begin()); |
| 897 | i != scored_results.end(); ++i) { |
| 898 | AddMatchToMap(i->suggestion(), input_text, i->relevance(), |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 899 | AutocompleteMatch::SEARCH_HISTORY, did_not_accept_suggestion, |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 900 | is_keyword, map); |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 901 | } |
| 902 | } |
| 903 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 904 | SearchProvider::SuggestResults SearchProvider::ScoreHistoryResults( |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 905 | const HistoryResults& results, |
| 906 | bool base_prevent_inline_autocomplete, |
| 907 | bool input_multiple_words, |
| 908 | const string16& input_text, |
| 909 | bool is_keyword) { |
[email protected] | 810ffba | 2012-06-12 01:07:48 | [diff] [blame] | 910 | AutocompleteClassifier* classifier = |
| 911 | AutocompleteClassifierFactory::GetForProfile(profile_); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 912 | SuggestResults scored_results; |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 913 | for (HistoryResults::const_iterator i(results.begin()); i != results.end(); |
| 914 | ++i) { |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 915 | // Don't autocomplete multi-word queries that have only been seen once |
| 916 | // unless the user has typed more than one word. |
| 917 | bool prevent_inline_autocomplete = base_prevent_inline_autocomplete || |
| 918 | (!input_multiple_words && (i->visits < 2) && HasMultipleWords(i->term)); |
| 919 | |
[email protected] | ea3b9a50 | 2011-04-04 14:19:37 | [diff] [blame] | 920 | // Don't autocomplete search terms that would normally be treated as URLs |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 921 | // when typed. For example, if the user searched for "google.com" and types |
| 922 | // "goog", don't autocomplete to the search term "google.com". Otherwise, |
| 923 | // the input will look like a URL but act like a search, which is confusing. |
[email protected] | cc44736 | 2011-04-06 03:57:48 | [diff] [blame] | 924 | // NOTE: We don't check this in the following cases: |
| 925 | // * When inline autocomplete is disabled, we won't be inline |
| 926 | // autocompleting this term, so we don't need to worry about confusion as |
| 927 | // much. This also prevents calling Classify() again from inside the |
| 928 | // classifier (which will corrupt state and likely crash), since the |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 929 | // classifier always disables inline autocomplete. |
[email protected] | cc44736 | 2011-04-06 03:57:48 | [diff] [blame] | 930 | // * When the user has typed the whole term, the "what you typed" history |
| 931 | // match will outrank us for URL-like inputs anyway, so we need not do |
| 932 | // anything special. |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 933 | if (!prevent_inline_autocomplete && classifier && (i->term != input_text)) { |
[email protected] | ea3b9a50 | 2011-04-04 14:19:37 | [diff] [blame] | 934 | AutocompleteMatch match; |
[email protected] | 72874a8d | 2011-05-11 03:48:54 | [diff] [blame] | 935 | classifier->Classify(i->term, string16(), false, false, &match, NULL); |
[email protected] | 2905f74 | 2011-10-13 03:51:58 | [diff] [blame] | 936 | prevent_inline_autocomplete = |
[email protected] | 749e7ae0 | 2012-09-05 18:47:46 | [diff] [blame] | 937 | !AutocompleteMatch::IsSearchType(match.type); |
[email protected] | ea3b9a50 | 2011-04-04 14:19:37 | [diff] [blame] | 938 | } |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 939 | |
| 940 | int relevance = CalculateRelevanceForHistory(i->time, is_keyword, |
| 941 | prevent_inline_autocomplete); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 942 | scored_results.push_back(SuggestResult(i->term, relevance)); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 943 | } |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 944 | |
| 945 | // History returns results sorted for us. However, we may have docked some |
| 946 | // results' scores, so things are no longer in order. Do a stable sort to get |
| 947 | // things back in order without otherwise disturbing results with equal |
| 948 | // scores, then force the scores to be unique, so that the order in which |
| 949 | // they're shown is deterministic. |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 950 | std::stable_sort(scored_results.begin(), scored_results.end(), |
| 951 | CompareScoredResults()); |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 952 | int last_relevance = 0; |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 953 | for (SuggestResults::iterator i(scored_results.begin()); |
| 954 | i != scored_results.end(); ++i) { |
| 955 | if ((i != scored_results.begin()) && (i->relevance() >= last_relevance)) |
| 956 | i->set_relevance(last_relevance - 1); |
| 957 | last_relevance = i->relevance(); |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 958 | } |
| 959 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 960 | return scored_results; |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 961 | } |
| 962 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 963 | void SearchProvider::AddSuggestResultsToMap(const SuggestResults& results, |
| 964 | bool is_keyword, |
| 965 | MatchMap* map) { |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 966 | const string16& input_text = is_keyword ? keyword_input_text_ : input_.text(); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 967 | for (size_t i = 0; i < results.size(); ++i) { |
[email protected] | d7ad477 | 2012-06-01 03:12:54 | [diff] [blame] | 968 | AddMatchToMap(results[i].suggestion(), input_text, results[i].relevance(), |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 969 | AutocompleteMatch::SEARCH_SUGGEST, i, is_keyword, map); |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 970 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 971 | } |
| 972 | |
[email protected] | 382a064 | 2012-06-06 06:13:52 | [diff] [blame] | 973 | int SearchProvider::GetVerbatimRelevance() const { |
[email protected] | dc6943b | 2012-06-19 06:39:56 | [diff] [blame] | 974 | // Use the suggested verbatim relevance score if it is non-negative (valid), |
| 975 | // if inline autocomplete isn't prevented (always show verbatim on backspace), |
[email protected] | 1beee34 | 2012-06-19 22:22:28 | [diff] [blame] | 976 | // and if it won't suppress verbatim, leaving no default provider matches. |
| 977 | // Otherwise, if the default provider returned no matches and was still able |
[email protected] | dc6943b | 2012-06-19 06:39:56 | [diff] [blame] | 978 | // to suppress verbatim, the user would have no search/nav matches and may be |
[email protected] | 1beee34 | 2012-06-19 22:22:28 | [diff] [blame] | 979 | // left unable to search using their default provider from the omnibox. |
[email protected] | dc6943b | 2012-06-19 06:39:56 | [diff] [blame] | 980 | // Check for results on each verbatim calculation, as results from older |
| 981 | // queries (on previous input) may be trimmed for failing to inline new input. |
| 982 | if (verbatim_relevance_ >= 0 && !input_.prevent_inline_autocomplete() && |
[email protected] | 1beee34 | 2012-06-19 22:22:28 | [diff] [blame] | 983 | (verbatim_relevance_ > 0 || |
| 984 | !default_suggest_results_.empty() || |
[email protected] | dc6943b | 2012-06-19 06:39:56 | [diff] [blame] | 985 | !default_navigation_results_.empty())) { |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 986 | return verbatim_relevance_; |
[email protected] | dc6943b | 2012-06-19 06:39:56 | [diff] [blame] | 987 | } |
[email protected] | 382a064 | 2012-06-06 06:13:52 | [diff] [blame] | 988 | return CalculateRelevanceForVerbatim(); |
| 989 | } |
[email protected] | d1f0a7f | 2012-06-05 10:26:42 | [diff] [blame] | 990 | |
[email protected] | 382a064 | 2012-06-06 06:13:52 | [diff] [blame] | 991 | int SearchProvider::CalculateRelevanceForVerbatim() const { |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 992 | if (!providers_.keyword_provider().empty()) |
[email protected] | 52d08b1 | 2009-10-19 18:42:36 | [diff] [blame] | 993 | return 250; |
| 994 | |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 995 | switch (input_.type()) { |
| 996 | case AutocompleteInput::UNKNOWN: |
[email protected] | 52d08b1 | 2009-10-19 18:42:36 | [diff] [blame] | 997 | case AutocompleteInput::QUERY: |
| 998 | case AutocompleteInput::FORCED_QUERY: |
| 999 | return 1300; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1000 | |
| 1001 | case AutocompleteInput::REQUESTED_URL: |
[email protected] | 52d08b1 | 2009-10-19 18:42:36 | [diff] [blame] | 1002 | return 1150; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1003 | |
| 1004 | case AutocompleteInput::URL: |
[email protected] | 52d08b1 | 2009-10-19 18:42:36 | [diff] [blame] | 1005 | return 850; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1006 | |
| 1007 | default: |
| 1008 | NOTREACHED(); |
| 1009 | return 0; |
| 1010 | } |
| 1011 | } |
| 1012 | |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 1013 | int SearchProvider::CalculateRelevanceForHistory( |
| 1014 | const Time& time, |
| 1015 | bool is_keyword, |
| 1016 | bool prevent_inline_autocomplete) const { |
[email protected] | aa613d6 | 2010-11-09 20:40:18 | [diff] [blame] | 1017 | // The relevance of past searches falls off over time. There are two distinct |
| 1018 | // equations used. If the first equation is used (searches to the primary |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 1019 | // provider that we want to inline autocomplete), the score starts at 1399 and |
| 1020 | // falls to 1300. If the second equation is used the relevance of a search 15 |
| 1021 | // minutes ago is discounted 50 points, while the relevance of a search two |
| 1022 | // weeks ago is discounted 450 points. |
[email protected] | aa613d6 | 2010-11-09 20:40:18 | [diff] [blame] | 1023 | double elapsed_time = std::max((Time::Now() - time).InSecondsF(), 0.); |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 1024 | bool is_primary_provider = providers_.is_primary_provider(is_keyword); |
| 1025 | if (is_primary_provider && !prevent_inline_autocomplete) { |
[email protected] | aa613d6 | 2010-11-09 20:40:18 | [diff] [blame] | 1026 | // Searches with the past two days get a different curve. |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 1027 | const double autocomplete_time = 2 * 24 * 60 * 60; |
[email protected] | aa613d6 | 2010-11-09 20:40:18 | [diff] [blame] | 1028 | if (elapsed_time < autocomplete_time) { |
[email protected] | e17511f | 2011-07-13 14:09:18 | [diff] [blame] | 1029 | return (is_keyword ? 1599 : 1399) - static_cast<int>(99 * |
[email protected] | aa613d6 | 2010-11-09 20:40:18 | [diff] [blame] | 1030 | std::pow(elapsed_time / autocomplete_time, 2.5)); |
| 1031 | } |
| 1032 | elapsed_time -= autocomplete_time; |
| 1033 | } |
| 1034 | |
[email protected] | c3a4bd99 | 2010-08-18 20:25:01 | [diff] [blame] | 1035 | const int score_discount = |
| 1036 | static_cast<int>(6.5 * std::pow(elapsed_time, 0.3)); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1037 | |
[email protected] | 6c85aa0 | 2009-02-27 12:08:09 | [diff] [blame] | 1038 | // Don't let scores go below 0. Negative relevance scores are meaningful in |
| 1039 | // a different way. |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1040 | int base_score; |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 1041 | if (is_primary_provider) |
[email protected] | 52d08b1 | 2009-10-19 18:42:36 | [diff] [blame] | 1042 | base_score = (input_.type() == AutocompleteInput::URL) ? 750 : 1050; |
[email protected] | 5112455 | 2011-07-16 01:37:10 | [diff] [blame] | 1043 | else |
| 1044 | base_score = 200; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1045 | return std::max(0, base_score - score_discount); |
| 1046 | } |
| 1047 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 1048 | int SearchProvider::CalculateRelevanceForSuggestion(bool for_keyword) const { |
| 1049 | return !providers_.is_primary_provider(for_keyword) ? 100 : |
| 1050 | ((input_.type() == AutocompleteInput::URL) ? 300 : 600); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1051 | } |
| 1052 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 1053 | int SearchProvider::CalculateRelevanceForNavigation(bool for_keyword) const { |
| 1054 | return providers_.is_primary_provider(for_keyword) ? 800 : 150; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1055 | } |
| 1056 | |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 1057 | void SearchProvider::AddMatchToMap(const string16& query_string, |
| 1058 | const string16& input_text, |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1059 | int relevance, |
[email protected] | 4c1fb7ec | 2008-11-13 00:19:00 | [diff] [blame] | 1060 | AutocompleteMatch::Type type, |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1061 | int accepted_suggestion, |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 1062 | bool is_keyword, |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1063 | MatchMap* map) { |
[email protected] | 9251368 | 2011-09-01 06:16:52 | [diff] [blame] | 1064 | AutocompleteMatch match(this, relevance, false, type); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1065 | std::vector<size_t> content_param_offsets; |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 1066 | // Bail out now if we don't actually have a valid provider. |
| 1067 | match.keyword = is_keyword ? |
[email protected] | 3954c3a | 2012-04-10 20:17:55 | [diff] [blame] | 1068 | providers_.keyword_provider() : providers_.default_provider(); |
[email protected] | dbff44658 | 2012-10-30 00:20:26 | [diff] [blame] | 1069 | const TemplateURL* provider_url = match.GetTemplateURL(profile_, false); |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 1070 | if (provider_url == NULL) |
| 1071 | return; |
| 1072 | |
[email protected] | 7083326 | 2011-01-05 23:40:44 | [diff] [blame] | 1073 | match.contents.assign(query_string); |
[email protected] | fb5153c5 | 2009-07-31 19:40:33 | [diff] [blame] | 1074 | // We do intra-string highlighting for suggestions - the suggested segment |
| 1075 | // will be highlighted, e.g. for input_text = "you" the suggestion may be |
| 1076 | // "youtube", so we'll bold the "tube" section: you*tube*. |
| 1077 | if (input_text != query_string) { |
[email protected] | fb5153c5 | 2009-07-31 19:40:33 | [diff] [blame] | 1078 | size_t input_position = match.contents.find(input_text); |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 1079 | if (input_position == string16::npos) { |
[email protected] | fb5153c5 | 2009-07-31 19:40:33 | [diff] [blame] | 1080 | // The input text is not a substring of the query string, e.g. input |
| 1081 | // text is "slasdot" and the query string is "slashdot", so we bold the |
| 1082 | // whole thing. |
| 1083 | match.contents_class.push_back( |
| 1084 | ACMatchClassification(0, ACMatchClassification::MATCH)); |
[email protected] | ec237916 | 2009-06-09 23:58:17 | [diff] [blame] | 1085 | } else { |
[email protected] | fb5153c5 | 2009-07-31 19:40:33 | [diff] [blame] | 1086 | // TODO(beng): ACMatchClassification::MATCH now seems to just mean |
| 1087 | // "bold" this. Consider modifying the terminology. |
| 1088 | // We don't iterate over the string here annotating all matches because |
| 1089 | // it looks odd to have every occurrence of a substring that may be as |
| 1090 | // short as a single character highlighted in a query suggestion result, |
| 1091 | // e.g. for input text "s" and query string "southwest airlines", it |
| 1092 | // looks odd if both the first and last s are highlighted. |
| 1093 | if (input_position != 0) { |
| 1094 | match.contents_class.push_back( |
| 1095 | ACMatchClassification(0, ACMatchClassification::NONE)); |
| 1096 | } |
| 1097 | match.contents_class.push_back( |
| 1098 | ACMatchClassification(input_position, ACMatchClassification::DIM)); |
| 1099 | size_t next_fragment_position = input_position + input_text.length(); |
| 1100 | if (next_fragment_position < query_string.length()) { |
| 1101 | match.contents_class.push_back( |
| 1102 | ACMatchClassification(next_fragment_position, |
| 1103 | ACMatchClassification::NONE)); |
| 1104 | } |
[email protected] | ec237916 | 2009-06-09 23:58:17 | [diff] [blame] | 1105 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1106 | } else { |
[email protected] | fb5153c5 | 2009-07-31 19:40:33 | [diff] [blame] | 1107 | // Otherwise, we're dealing with the "default search" result which has no |
[email protected] | 7083326 | 2011-01-05 23:40:44 | [diff] [blame] | 1108 | // completion. |
[email protected] | fb5153c5 | 2009-07-31 19:40:33 | [diff] [blame] | 1109 | match.contents_class.push_back( |
| 1110 | ACMatchClassification(0, ACMatchClassification::NONE)); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1111 | } |
| 1112 | |
| 1113 | // When the user forced a query, we need to make sure all the fill_into_edit |
| 1114 | // values preserve that property. Otherwise, if the user starts editing a |
| 1115 | // suggestion, non-Search results will suddenly appear. |
[email protected] | 8f2249b | 2012-08-29 02:27:24 | [diff] [blame] | 1116 | if (input_.type() == AutocompleteInput::FORCED_QUERY) |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 1117 | match.fill_into_edit.assign(ASCIIToUTF16("?")); |
[email protected] | 8f2249b | 2012-08-29 02:27:24 | [diff] [blame] | 1118 | if (is_keyword) |
[email protected] | 033f342 | 2012-03-13 21:24:18 | [diff] [blame] | 1119 | match.fill_into_edit.append(match.keyword + char16(' ')); |
[email protected] | 8f2249b | 2012-08-29 02:27:24 | [diff] [blame] | 1120 | if (!input_.prevent_inline_autocomplete() && |
| 1121 | StartsWith(query_string, input_text, false)) { |
| 1122 | match.inline_autocomplete_offset = |
| 1123 | match.fill_into_edit.length() + input_text.length(); |
[email protected] | c0048b4 | 2009-05-04 21:47:17 | [diff] [blame] | 1124 | } |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1125 | match.fill_into_edit.append(query_string); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1126 | |
[email protected] | 85b8d6f | 2012-05-08 20:53:47 | [diff] [blame] | 1127 | const TemplateURLRef& search_url = provider_url->url_ref(); |
[email protected] | 360ba05 | 2012-04-04 17:26:13 | [diff] [blame] | 1128 | DCHECK(search_url.SupportsReplacement()); |
[email protected] | bca359b | 2012-06-24 07:53:04 | [diff] [blame] | 1129 | match.search_terms_args.reset( |
| 1130 | new TemplateURLRef::SearchTermsArgs(query_string)); |
| 1131 | match.search_terms_args->original_query = input_text; |
| 1132 | match.search_terms_args->accepted_suggestion = accepted_suggestion; |
| 1133 | // This is the destination URL sans assisted query stats. This must be set |
| 1134 | // so the AutocompleteController can properly de-dupe; the controller will |
| 1135 | // eventually overwrite it before it reaches the user. |
| 1136 | match.destination_url = |
| 1137 | GURL(search_url.ReplaceSearchTerms(*match.search_terms_args.get())); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1138 | |
| 1139 | // Search results don't look like URLs. |
[email protected] | 2905f74 | 2011-10-13 03:51:58 | [diff] [blame] | 1140 | match.transition = is_keyword ? |
| 1141 | content::PAGE_TRANSITION_KEYWORD : content::PAGE_TRANSITION_GENERATED; |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1142 | |
| 1143 | // Try to add |match| to |map|. If a match for |query_string| is already in |
| 1144 | // |map|, replace it if |match| is more relevant. |
| 1145 | // NOTE: Keep this ToLower() call in sync with url_database.cc. |
| 1146 | const std::pair<MatchMap::iterator, bool> i = map->insert( |
[email protected] | a2fedb1e | 2011-01-25 15:23:36 | [diff] [blame] | 1147 | std::pair<string16, AutocompleteMatch>( |
[email protected] | 503d0387 | 2011-05-06 08:36:26 | [diff] [blame] | 1148 | base::i18n::ToLower(query_string), match)); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1149 | // NOTE: We purposefully do a direct relevance comparison here instead of |
| 1150 | // using AutocompleteMatch::MoreRelevant(), so that we'll prefer "items added |
| 1151 | // first" rather than "items alphabetically first" when the scores are equal. |
| 1152 | // The only case this matters is when a user has results with the same score |
| 1153 | // that differ only by capitalization; because the history system returns |
| 1154 | // results sorted by recency, this means we'll pick the most recent such |
| 1155 | // result even if the precision of our relevance score is too low to |
| 1156 | // distinguish the two. |
| 1157 | if (!i.second && (match.relevance > i.first->second.relevance)) |
| 1158 | i.first->second = match; |
| 1159 | } |
| 1160 | |
| 1161 | AutocompleteMatch SearchProvider::NavigationToMatch( |
| 1162 | const NavigationResult& navigation, |
[email protected] | 257ab71 | 2009-04-14 17:16:24 | [diff] [blame] | 1163 | bool is_keyword) { |
[email protected] | 371dab1 | 2012-06-01 03:23:55 | [diff] [blame] | 1164 | const string16& input = is_keyword ? keyword_input_text_ : input_.text(); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 1165 | AutocompleteMatch match(this, navigation.relevance(), false, |
[email protected] | 4c1fb7ec | 2008-11-13 00:19:00 | [diff] [blame] | 1166 | AutocompleteMatch::NAVSUGGEST); |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 1167 | match.destination_url = navigation.url(); |
[email protected] | 371dab1 | 2012-06-01 03:23:55 | [diff] [blame] | 1168 | |
| 1169 | // First look for the user's input inside the fill_into_edit as it would be |
| 1170 | // without trimming the scheme, so we can find matches at the beginning of the |
| 1171 | // scheme. |
| 1172 | const string16 untrimmed_fill_into_edit( |
| 1173 | AutocompleteInput::FormattedStringWithEquivalentMeaning(navigation.url(), |
| 1174 | StringForURLDisplay(navigation.url(), true, false))); |
| 1175 | const URLPrefix* prefix = |
| 1176 | URLPrefix::BestURLPrefix(untrimmed_fill_into_edit, input); |
| 1177 | size_t match_start = (prefix == NULL) ? |
| 1178 | untrimmed_fill_into_edit.find(input) : prefix->prefix.length(); |
| 1179 | size_t inline_autocomplete_offset = (prefix == NULL) ? |
| 1180 | string16::npos : (match_start + input.length()); |
| 1181 | bool trim_http = !HasHTTPScheme(input) && (!prefix || (match_start != 0)); |
| 1182 | |
| 1183 | // Preserve the forced query '?' prefix in |match.fill_into_edit|. |
| 1184 | // Otherwise, user edits to a suggestion would show non-Search results. |
| 1185 | if (input_.type() == AutocompleteInput::FORCED_QUERY) { |
| 1186 | match.fill_into_edit = ASCIIToUTF16("?"); |
| 1187 | if (inline_autocomplete_offset != string16::npos) |
| 1188 | ++inline_autocomplete_offset; |
| 1189 | } |
| 1190 | |
| 1191 | const std::string languages( |
| 1192 | profile_->GetPrefs()->GetString(prefs::kAcceptLanguages)); |
| 1193 | const net::FormatUrlTypes format_types = |
| 1194 | net::kFormatUrlOmitAll & ~(trim_http ? 0 : net::kFormatUrlOmitHTTP); |
| 1195 | match.fill_into_edit += |
| 1196 | AutocompleteInput::FormattedStringWithEquivalentMeaning(navigation.url(), |
| 1197 | net::FormatUrl(navigation.url(), languages, format_types, |
| 1198 | net::UnescapeRule::SPACES, NULL, NULL, |
| 1199 | &inline_autocomplete_offset)); |
| 1200 | if (!input_.prevent_inline_autocomplete()) |
| 1201 | match.inline_autocomplete_offset = inline_autocomplete_offset; |
| 1202 | DCHECK((match.inline_autocomplete_offset == string16::npos) || |
| 1203 | (match.inline_autocomplete_offset <= match.fill_into_edit.length())); |
| 1204 | |
| 1205 | match.contents = net::FormatUrl(navigation.url(), languages, |
| 1206 | format_types, net::UnescapeRule::SPACES, NULL, NULL, &match_start); |
| 1207 | // If the first match in the untrimmed string was inside a scheme that we |
| 1208 | // trimmed, look for a subsequent match. |
| 1209 | if (match_start == string16::npos) |
| 1210 | match_start = match.contents.find(input); |
| 1211 | // Safe if |match_start| is npos; also safe if the input is longer than the |
| 1212 | // remaining contents after |match_start|. |
| 1213 | AutocompleteMatch::ClassifyLocationInString(match_start, input.length(), |
| 1214 | match.contents.length(), ACMatchClassification::URL, |
| 1215 | &match.contents_class); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1216 | |
[email protected] | 55ce8f1 | 2012-05-09 04:44:08 | [diff] [blame] | 1217 | match.description = navigation.description(); |
[email protected] | 371dab1 | 2012-06-01 03:23:55 | [diff] [blame] | 1218 | AutocompleteMatch::ClassifyMatchInString(input, match.description, |
| 1219 | ACMatchClassification::NONE, &match.description_class); |
initial.commit | 09911bf | 2008-07-26 23:55:29 | [diff] [blame] | 1220 | return match; |
| 1221 | } |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 1222 | |
| 1223 | void SearchProvider::UpdateDone() { |
[email protected] | 2cdf117 | 2012-08-26 12:21:33 | [diff] [blame] | 1224 | // We're done when the timer isn't running, there are no suggest queries |
| 1225 | // pending, and we're not waiting on instant. |
| 1226 | done_ = (!timer_.IsRunning() && (suggest_results_pending_ == 0) && |
[email protected] | b67d0a4 | 2012-09-04 20:57:35 | [diff] [blame] | 1227 | (instant_finalized_ || |
[email protected] | 249a035 | 2012-11-26 21:06:19 | [diff] [blame] | 1228 | (!chrome::BrowserInstantController::IsInstantEnabled(profile_) && |
| 1229 | !chrome::search::IsInstantExtendedAPIEnabled(profile_)))); |
[email protected] | 4ab4c7c | 2010-11-24 04:49:34 | [diff] [blame] | 1230 | } |