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