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 3// found in the LICENSE file. 4 5#include "chrome/browser/autocomplete/autocomplete_classifier.h" 6 7#include "base/auto_reset.h" 8#include "chrome/browser/autocomplete/autocomplete_controller.h" 9#include "components/metrics/proto/omnibox_event.pb.h" 10#include "components/omnibox/autocomplete_input.h" 11#include "components/omnibox/autocomplete_match.h" 12#include "components/omnibox/autocomplete_provider.h" 13#include "url/gurl.h" 14 15// static 16const int AutocompleteClassifier::kDefaultOmniboxProviders = 17 AutocompleteProvider::TYPE_BOOKMARK | 18 AutocompleteProvider::TYPE_BUILTIN | 19 AutocompleteProvider::TYPE_HISTORY_QUICK | 20 AutocompleteProvider::TYPE_HISTORY_URL | 21 AutocompleteProvider::TYPE_KEYWORD | 22 AutocompleteProvider::TYPE_SEARCH | 23 AutocompleteProvider::TYPE_SHORTCUTS | 24 AutocompleteProvider::TYPE_ZERO_SUGGEST; 25 26AutocompleteClassifier::AutocompleteClassifier( 27 scoped_ptr<AutocompleteController> controller, 28 scoped_ptr<AutocompleteSchemeClassifier> scheme_classifier) 29 : controller_(controller.Pass()), 30 scheme_classifier_(scheme_classifier.Pass()), 31 inside_classify_(false) { 32} 33 34AutocompleteClassifier::~AutocompleteClassifier() { 35 // We should only reach here after Shutdown() has been called. 36 DCHECK(!controller_.get()); 37} 38 39void AutocompleteClassifier::Classify( 40 const base::string16& text, 41 bool prefer_keyword, 42 bool allow_exact_keyword_match, 43 metrics::OmniboxEventProto::PageClassification page_classification, 44 AutocompleteMatch* match, 45 GURL* alternate_nav_url) { 46 DCHECK(!inside_classify_); 47 base::AutoReset<bool> reset(&inside_classify_, true); 48 controller_->Start(AutocompleteInput( 49 text, base::string16::npos, base::string16(), GURL(), 50 page_classification, true, prefer_keyword, 51 allow_exact_keyword_match, false, *scheme_classifier_)); 52 DCHECK(controller_->done()); 53 const AutocompleteResult& result = controller_->result(); 54 if (result.empty()) { 55 if (alternate_nav_url) 56 *alternate_nav_url = GURL(); 57 return; 58 } 59 60 DCHECK(result.default_match() != result.end()); 61 *match = *result.default_match(); 62 if (alternate_nav_url) 63 *alternate_nav_url = result.alternate_nav_url(); 64} 65 66void AutocompleteClassifier::Shutdown() { 67 controller_.reset(); 68} 69