OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/autocomplete/search_provider.h" | 5 #include "chrome/browser/autocomplete/search_provider.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 #include <cmath> | 8 #include <cmath> |
9 | 9 |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
(...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
55 found_word = true; | 55 found_word = true; |
56 } | 56 } |
57 } | 57 } |
58 } | 58 } |
59 return false; | 59 return false; |
60 } | 60 } |
61 | 61 |
62 }; | 62 }; |
63 | 63 |
64 | 64 |
65 // SearchProvider ------------------------------------------------------------- | 65 // SearchProvider::Providers -------------------------------------------------- |
66 | |
67 // static | |
68 const int SearchProvider::kDefaultProviderURLFetcherID = 1; | |
69 // static | |
70 const int SearchProvider::kKeywordProviderURLFetcherID = 2; | |
71 // static | |
72 bool SearchProvider::query_suggest_immediately_ = false; | |
73 | 66 |
74 void SearchProvider::Providers::Set(const TemplateURL* default_provider, | 67 void SearchProvider::Providers::Set(const TemplateURL* default_provider, |
75 const TemplateURL* keyword_provider) { | 68 const TemplateURL* keyword_provider) { |
76 // TODO(pkasting): http://b/1162970 We shouldn't need to structure-copy | 69 // TODO(pkasting): http://b/1162970 We shouldn't need to structure-copy |
77 // this. Nor should we need |default_provider_| and |keyword_provider_| | 70 // this. Nor should we need |default_provider_| and |keyword_provider_| |
78 // just to know whether the provider changed. | 71 // just to know whether the provider changed. |
79 default_provider_ = default_provider; | 72 default_provider_ = default_provider; |
80 if (default_provider) | 73 if (default_provider) |
81 cached_default_provider_ = *default_provider; | 74 cached_default_provider_ = *default_provider; |
82 keyword_provider_ = keyword_provider; | 75 keyword_provider_ = keyword_provider; |
83 if (keyword_provider) | 76 if (keyword_provider) |
84 cached_keyword_provider_ = *keyword_provider; | 77 cached_keyword_provider_ = *keyword_provider; |
85 } | 78 } |
86 | 79 |
| 80 |
| 81 // SearchProvider ------------------------------------------------------------- |
| 82 |
| 83 // static |
| 84 const int SearchProvider::kDefaultProviderURLFetcherID = 1; |
| 85 // static |
| 86 const int SearchProvider::kKeywordProviderURLFetcherID = 2; |
| 87 // static |
| 88 bool SearchProvider::query_suggest_immediately_ = false; |
| 89 |
87 SearchProvider::SearchProvider(ACProviderListener* listener, Profile* profile) | 90 SearchProvider::SearchProvider(ACProviderListener* listener, Profile* profile) |
88 : AutocompleteProvider(listener, profile, "Search"), | 91 : AutocompleteProvider(listener, profile, "Search"), |
89 suggest_results_pending_(0), | 92 suggest_results_pending_(0), |
90 have_suggest_results_(false), | 93 have_suggest_results_(false), |
91 instant_finalized_(false) { | 94 instant_finalized_(false) { |
92 // We use GetSuggestNumberOfGroups() as the group ID to mean "not in field | 95 // We use GetSuggestNumberOfGroups() as the group ID to mean "not in field |
93 // trial." Field trial groups run from 0 to GetSuggestNumberOfGroups() - 1 | 96 // trial." Field trial groups run from 0 to GetSuggestNumberOfGroups() - 1 |
94 // (inclusive). | 97 // (inclusive). |
95 int suggest_field_trial_group_number = | 98 int suggest_field_trial_group_number = |
96 AutocompleteFieldTrial::GetSuggestNumberOfGroups(); | 99 AutocompleteFieldTrial::GetSuggestNumberOfGroups(); |
(...skipping 138 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
235 bool operator()(const ScoredTerm& a, const ScoredTerm& b) { | 238 bool operator()(const ScoredTerm& a, const ScoredTerm& b) { |
236 // Sort in descending relevance order. | 239 // Sort in descending relevance order. |
237 return a.second > b.second; | 240 return a.second > b.second; |
238 } | 241 } |
239 }; | 242 }; |
240 | 243 |
241 void SearchProvider::Run() { | 244 void SearchProvider::Run() { |
242 // Start a new request with the current input. | 245 // Start a new request with the current input. |
243 DCHECK(!done_); | 246 DCHECK(!done_); |
244 suggest_results_pending_ = 0; | 247 suggest_results_pending_ = 0; |
| 248 if (providers_.valid_suggest_for_default_provider()) { |
| 249 suggest_results_pending_++; |
| 250 default_fetcher_.reset(CreateSuggestFetcher(kDefaultProviderURLFetcherID, |
| 251 providers_.default_provider().suggestions_url_ref(), input_.text())); |
| 252 } |
245 if (providers_.valid_suggest_for_keyword_provider()) { | 253 if (providers_.valid_suggest_for_keyword_provider()) { |
246 suggest_results_pending_++; | 254 suggest_results_pending_++; |
247 keyword_fetcher_.reset(CreateSuggestFetcher(kKeywordProviderURLFetcherID, | 255 keyword_fetcher_.reset(CreateSuggestFetcher(kKeywordProviderURLFetcherID, |
248 providers_.keyword_provider(), keyword_input_text_)); | 256 providers_.keyword_provider().suggestions_url_ref(), |
249 } | 257 keyword_input_text_)); |
250 if (providers_.valid_suggest_for_default_provider()) { | |
251 suggest_results_pending_++; | |
252 default_fetcher_.reset(CreateSuggestFetcher(kDefaultProviderURLFetcherID, | |
253 providers_.default_provider(), input_.text())); | |
254 } | 258 } |
255 // We should only get here if we have a suggest url for the keyword or default | 259 // We should only get here if we have a suggest url for the keyword or default |
256 // providers. | 260 // providers. |
257 DCHECK_GT(suggest_results_pending_, 0); | 261 DCHECK_GT(suggest_results_pending_, 0); |
258 } | 262 } |
259 | 263 |
260 void SearchProvider::Stop() { | 264 void SearchProvider::Stop() { |
261 StopSuggest(); | 265 StopSuggest(); |
262 done_ = true; | 266 done_ = true; |
263 default_provider_suggest_text_.clear(); | 267 default_provider_suggest_text_.clear(); |
(...skipping 64 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
328 // Request history for both the keyword and default provider. We grab many | 332 // Request history for both the keyword and default provider. We grab many |
329 // more matches than we'll ultimately clamp to so that if there are several | 333 // more matches than we'll ultimately clamp to so that if there are several |
330 // recent multi-word matches who scores are lowered (see | 334 // recent multi-word matches who scores are lowered (see |
331 // AddHistoryResultsToMap()), they won't crowd out older, higher-scoring | 335 // AddHistoryResultsToMap()), they won't crowd out older, higher-scoring |
332 // matches. Note that this doesn't fix the problem entirely, but merely | 336 // matches. Note that this doesn't fix the problem entirely, but merely |
333 // limits it to cases with a very large number of such multi-word matches; for | 337 // limits it to cases with a very large number of such multi-word matches; for |
334 // now, this seems OK compared with the complexity of a real fix, which would | 338 // now, this seems OK compared with the complexity of a real fix, which would |
335 // require multiple searches and tracking of "single- vs. multi-word" in the | 339 // require multiple searches and tracking of "single- vs. multi-word" in the |
336 // database. | 340 // database. |
337 int num_matches = kMaxMatches * 5; | 341 int num_matches = kMaxMatches * 5; |
| 342 if (providers_.valid_default_provider()) { |
| 343 url_db->GetMostRecentKeywordSearchTerms(providers_.default_provider().id(), |
| 344 input_.text(), num_matches, &default_history_results_); |
| 345 } |
338 if (providers_.valid_keyword_provider()) { | 346 if (providers_.valid_keyword_provider()) { |
339 url_db->GetMostRecentKeywordSearchTerms(providers_.keyword_provider().id(), | 347 url_db->GetMostRecentKeywordSearchTerms(providers_.keyword_provider().id(), |
340 keyword_input_text_, num_matches, &keyword_history_results_); | 348 keyword_input_text_, num_matches, &keyword_history_results_); |
341 } | 349 } |
342 if (providers_.valid_default_provider()) { | |
343 url_db->GetMostRecentKeywordSearchTerms(providers_.default_provider().id(), | |
344 input_.text(), num_matches, &default_history_results_); | |
345 } | |
346 } | 350 } |
347 | 351 |
348 void SearchProvider::StartOrStopSuggestQuery(bool minimal_changes) { | 352 void SearchProvider::StartOrStopSuggestQuery(bool minimal_changes) { |
349 // Don't send any queries to the server until some time has elapsed after | 353 // Don't send any queries to the server until some time has elapsed after |
350 // the last keypress, to avoid flooding the server with requests we are | 354 // the last keypress, to avoid flooding the server with requests we are |
351 // likely to end up throwing away anyway. | 355 // likely to end up throwing away anyway. |
352 const int kQueryDelayMs = 200; | 356 const int kQueryDelayMs = 200; |
353 | 357 |
354 if (!IsQuerySuitableForSuggest()) { | 358 if (!IsQuerySuitableForSuggest()) { |
355 StopSuggest(); | 359 StopSuggest(); |
(...skipping 23 matching lines...) Expand all Loading... |
379 suggest_results_pending_ = 1; | 383 suggest_results_pending_ = 1; |
380 | 384 |
381 // Kick off a timer that will start the URL fetch if it completes before | 385 // Kick off a timer that will start the URL fetch if it completes before |
382 // the user types another character. | 386 // the user types another character. |
383 int delay = query_suggest_immediately_ ? 0 : kQueryDelayMs; | 387 int delay = query_suggest_immediately_ ? 0 : kQueryDelayMs; |
384 timer_.Start(FROM_HERE, TimeDelta::FromMilliseconds(delay), this, | 388 timer_.Start(FROM_HERE, TimeDelta::FromMilliseconds(delay), this, |
385 &SearchProvider::Run); | 389 &SearchProvider::Run); |
386 } | 390 } |
387 | 391 |
388 bool SearchProvider::IsQuerySuitableForSuggest() const { | 392 bool SearchProvider::IsQuerySuitableForSuggest() const { |
389 // Don't run Suggest in incognito mode, the engine doesn't support it, or | 393 // Don't run Suggest in incognito mode, if the engine doesn't support it, or |
390 // the user has disabled it. | 394 // if the user has disabled it. |
391 if (profile_->IsOffTheRecord() || | 395 if (profile_->IsOffTheRecord() || |
392 (!providers_.valid_suggest_for_keyword_provider() && | 396 (!providers_.valid_suggest_for_default_provider() && |
393 !providers_.valid_suggest_for_default_provider()) || | 397 !providers_.valid_suggest_for_keyword_provider()) || |
394 !profile_->GetPrefs()->GetBoolean(prefs::kSearchSuggestEnabled)) | 398 !profile_->GetPrefs()->GetBoolean(prefs::kSearchSuggestEnabled)) |
395 return false; | 399 return false; |
396 | 400 |
397 // If the input type might be a URL, we take extra care so that private data | 401 // If the input type might be a URL, we take extra care so that private data |
398 // isn't sent to the server. | 402 // isn't sent to the server. |
399 | 403 |
400 // FORCED_QUERY means the user is explicitly asking us to search for this, so | 404 // FORCED_QUERY means the user is explicitly asking us to search for this, so |
401 // we assume it isn't a URL and/or there isn't private data. | 405 // we assume it isn't a URL and/or there isn't private data. |
402 if (input_.type() == AutocompleteInput::FORCED_QUERY) | 406 if (input_.type() == AutocompleteInput::FORCED_QUERY) |
403 return true; | 407 return true; |
(...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
446 default_fetcher_.reset(); | 450 default_fetcher_.reset(); |
447 keyword_suggest_results_.clear(); | 451 keyword_suggest_results_.clear(); |
448 default_suggest_results_.clear(); | 452 default_suggest_results_.clear(); |
449 keyword_navigation_results_.clear(); | 453 keyword_navigation_results_.clear(); |
450 default_navigation_results_.clear(); | 454 default_navigation_results_.clear(); |
451 have_suggest_results_ = false; | 455 have_suggest_results_ = false; |
452 } | 456 } |
453 | 457 |
454 content::URLFetcher* SearchProvider::CreateSuggestFetcher( | 458 content::URLFetcher* SearchProvider::CreateSuggestFetcher( |
455 int id, | 459 int id, |
456 const TemplateURL& provider, | 460 const TemplateURLRef& suggestions_url, |
457 const string16& text) { | 461 const string16& text) { |
458 const TemplateURLRef& suggestions_url = provider.suggestions_url_ref(); | |
459 DCHECK(suggestions_url.SupportsReplacement()); | 462 DCHECK(suggestions_url.SupportsReplacement()); |
460 content::URLFetcher* fetcher = content::URLFetcher::Create(id, | 463 content::URLFetcher* fetcher = content::URLFetcher::Create(id, |
461 GURL(suggestions_url.ReplaceSearchTermsUsingProfile( | 464 GURL(suggestions_url.ReplaceSearchTermsUsingProfile( |
462 profile_, text, TemplateURLRef::NO_SUGGESTIONS_AVAILABLE, | 465 profile_, text, TemplateURLRef::NO_SUGGESTIONS_AVAILABLE, |
463 string16())), | 466 string16())), |
464 content::URLFetcher::GET, this); | 467 content::URLFetcher::GET, this); |
465 fetcher->SetRequestContext(profile_->GetRequestContext()); | 468 fetcher->SetRequestContext(profile_->GetRequestContext()); |
466 fetcher->SetLoadFlags(net::LOAD_DO_NOT_SAVE_COOKIES); | 469 fetcher->SetLoadFlags(net::LOAD_DO_NOT_SAVE_COOKIES); |
467 fetcher->Start(); | 470 fetcher->Start(); |
468 return fetcher; | 471 return fetcher; |
(...skipping 362 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
831 void SearchProvider::AddMatchToMap(const string16& query_string, | 834 void SearchProvider::AddMatchToMap(const string16& query_string, |
832 const string16& input_text, | 835 const string16& input_text, |
833 int relevance, | 836 int relevance, |
834 AutocompleteMatch::Type type, | 837 AutocompleteMatch::Type type, |
835 int accepted_suggestion, | 838 int accepted_suggestion, |
836 bool is_keyword, | 839 bool is_keyword, |
837 bool prevent_inline_autocomplete, | 840 bool prevent_inline_autocomplete, |
838 MatchMap* map) { | 841 MatchMap* map) { |
839 AutocompleteMatch match(this, relevance, false, type); | 842 AutocompleteMatch match(this, relevance, false, type); |
840 std::vector<size_t> content_param_offsets; | 843 std::vector<size_t> content_param_offsets; |
841 const TemplateURL& provider = is_keyword ? providers_.keyword_provider() : | 844 const TemplateURL& provider = is_keyword ? |
842 providers_.default_provider(); | 845 providers_.keyword_provider() : providers_.default_provider(); |
843 match.template_url = &provider; | 846 match.template_url = &provider; |
844 match.contents.assign(query_string); | 847 match.contents.assign(query_string); |
845 // We do intra-string highlighting for suggestions - the suggested segment | 848 // We do intra-string highlighting for suggestions - the suggested segment |
846 // will be highlighted, e.g. for input_text = "you" the suggestion may be | 849 // will be highlighted, e.g. for input_text = "you" the suggestion may be |
847 // "youtube", so we'll bold the "tube" section: you*tube*. | 850 // "youtube", so we'll bold the "tube" section: you*tube*. |
848 if (input_text != query_string) { | 851 if (input_text != query_string) { |
849 size_t input_position = match.contents.find(input_text); | 852 size_t input_position = match.contents.find(input_text); |
850 if (input_position == string16::npos) { | 853 if (input_position == string16::npos) { |
851 // The input text is not a substring of the query string, e.g. input | 854 // The input text is not a substring of the query string, e.g. input |
852 // text is "slasdot" and the query string is "slashdot", so we bold the | 855 // text is "slasdot" and the query string is "slashdot", so we bold the |
(...skipping 108 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
961 | 964 |
962 return match; | 965 return match; |
963 } | 966 } |
964 | 967 |
965 void SearchProvider::UpdateDone() { | 968 void SearchProvider::UpdateDone() { |
966 // We're done when there are no more suggest queries pending (this is set to 1 | 969 // We're done when there are no more suggest queries pending (this is set to 1 |
967 // when the timer is started) and we're not waiting on instant. | 970 // when the timer is started) and we're not waiting on instant. |
968 done_ = ((suggest_results_pending_ == 0) && | 971 done_ = ((suggest_results_pending_ == 0) && |
969 (instant_finalized_ || !InstantController::IsEnabled(profile_))); | 972 (instant_finalized_ || !InstantController::IsEnabled(profile_))); |
970 } | 973 } |
OLD | NEW |