chrome/browser/autocomplete/autocomplete_classifier.cc

   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 "chrome/browser/autocomplete/autocomplete_input.h"
  10 #include "chrome/browser/autocomplete/autocomplete_match.h"
  11 #include "chrome/browser/autocomplete/autocomplete_provider.h"
  12 #include "googleurl/src/gurl.h"
  13
  14 // static
  15 const int AutocompleteClassifier::kDefaultOmniboxProviders =
  16     AutocompleteProvider::TYPE_BOOKMARK |
  17     AutocompleteProvider::TYPE_BUILTIN |
  18     AutocompleteProvider::TYPE_HISTORY_CONTENTS |
  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
  26 // static
  27 const int AutocompleteClassifier::kInstantExtendedOmniboxProviders =
  28     AutocompleteProvider::TYPE_BUILTIN |
  29     AutocompleteProvider::TYPE_HISTORY_QUICK |
  30     AutocompleteProvider::TYPE_HISTORY_URL |
  31     // TODO: remove TYPE_SEARCH once it's no longer needed to pass
  32     // the Instant suggestion through via FinalizeInstantQuery.
  33     AutocompleteProvider::TYPE_SEARCH |
  34     AutocompleteProvider::TYPE_ZERO_SUGGEST;
  35
  36 AutocompleteClassifier::AutocompleteClassifier(Profile* profile)
  37     : controller_(new AutocompleteController(profile, NULL,
  38                                              kDefaultOmniboxProviders)),
  39       inside_classify_(false) {
  40 }
  41
  42 AutocompleteClassifier::~AutocompleteClassifier() {
  43   // We should only reach here after Shutdown() has been called.
  44   DCHECK(!controller_.get());
  45 }
  46
  47 void AutocompleteClassifier::Classify(const string16& text,
  48                                       const string16& desired_tld,
  49                                       bool prefer_keyword,
  50                                       bool allow_exact_keyword_match,
  51                                       AutocompleteMatch* match,
  52                                       GURL* alternate_nav_url) {
  53   DCHECK(!inside_classify_);
  54   AutoReset<bool> reset(&inside_classify_, true);
  55   controller_->Start(text, desired_tld, true, prefer_keyword,
  56                      allow_exact_keyword_match, AutocompleteInput::BEST_MATCH);
  57   DCHECK(controller_->done());
  58   const AutocompleteResult& result = controller_->result();
  59   if (result.empty()) {
  60     if (alternate_nav_url)
  61       *alternate_nav_url = GURL();
  62     return;
  63   }
  64
  65   DCHECK(result.default_match() != result.end());
  66   *match = *result.default_match();
  67   if (alternate_nav_url)
  68     *alternate_nav_url = result.alternate_nav_url();
  69 }
  70
  71 void AutocompleteClassifier::Shutdown() {
  72   controller_.reset();
  73 }