chrome/browser/ui/omnibox/omnibox_current_page_delegate_impl.cc

   1 // Copyright 2013 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/ui/omnibox/omnibox_current_page_delegate_impl.h"
   6
   7 #include "base/strings/utf_string_conversions.h"
   8 #include "chrome/browser/extensions/api/omnibox/omnibox_api.h"
   9 #include "chrome/browser/predictors/autocomplete_action_predictor.h"
  10 #include "chrome/browser/predictors/autocomplete_action_predictor_factory.h"
  11 #include "chrome/browser/profiles/profile.h"
  12 #include "chrome/browser/search/search.h"
  13 #include "chrome/browser/search_engines/template_url_service_factory.h"
  14 #include "chrome/browser/sessions/session_tab_helper.h"
  15 #include "chrome/browser/ui/omnibox/omnibox_edit_controller.h"
  16 #include "chrome/browser/ui/search/instant_search_prerenderer.h"
  17 #include "chrome/browser/ui/search/search_tab_helper.h"
  18 #include "components/omnibox/autocomplete_match.h"
  19 #include "components/search/search.h"
  20 #include "components/search_engines/template_url_service.h"
  21 #include "content/public/browser/navigation_controller.h"
  22 #include "content/public/browser/web_contents.h"
  23 #include "ui/base/window_open_disposition.h"
  24 #include "url/gurl.h"
  25
  26 OmniboxCurrentPageDelegateImpl::OmniboxCurrentPageDelegateImpl(
  27     OmniboxEditController* controller,
  28     Profile* profile)
  29     : controller_(controller),
  30       profile_(profile) {}
  31
  32 OmniboxCurrentPageDelegateImpl::~OmniboxCurrentPageDelegateImpl() {}
  33
  34 bool OmniboxCurrentPageDelegateImpl::CurrentPageExists() const {
  35   return (controller_->GetWebContents() != NULL);
  36 }
  37
  38 const GURL& OmniboxCurrentPageDelegateImpl::GetURL() const {
  39   return controller_->GetWebContents()->GetURL();
  40 }
  41
  42 bool OmniboxCurrentPageDelegateImpl::IsInstantNTP() const {
  43   return chrome::IsInstantNTP(controller_->GetWebContents());
  44 }
  45
  46 bool OmniboxCurrentPageDelegateImpl::IsSearchResultsPage() const {
  47   Profile* profile = Profile::FromBrowserContext(
  48       controller_->GetWebContents()->GetBrowserContext());
  49   return TemplateURLServiceFactory::GetForProfile(profile)->
  50       IsSearchResultsPageFromDefaultSearchProvider(GetURL());
  51 }
  52
  53 bool OmniboxCurrentPageDelegateImpl::IsLoading() const {
  54   return controller_->GetWebContents()->IsLoading();
  55 }
  56
  57 content::NavigationController&
  58     OmniboxCurrentPageDelegateImpl::GetNavigationController() const {
  59   return controller_->GetWebContents()->GetController();
  60 }
  61
  62 const SessionID& OmniboxCurrentPageDelegateImpl::GetSessionID() const {
  63   return SessionTabHelper::FromWebContents(
  64       controller_->GetWebContents())->session_id();
  65 }
  66
  67 bool OmniboxCurrentPageDelegateImpl::ProcessExtensionKeyword(
  68     TemplateURL* template_url,
  69     const AutocompleteMatch& match,
  70     WindowOpenDisposition disposition) {
  71   if (template_url->GetType() != TemplateURL::OMNIBOX_API_EXTENSION)
  72     return false;
  73
  74   // Strip the keyword + leading space off the input, but don't exceed
  75   // fill_into_edit.  An obvious case is that the user may not have entered
  76   // a leading space and is asking to launch this extension without any
  77   // additional input.
  78   size_t prefix_length =
  79       std::min(match.keyword.length() + 1, match.fill_into_edit.length());
  80   extensions::ExtensionOmniboxEventRouter::OnInputEntered(
  81       controller_->GetWebContents(),
  82       template_url->GetExtensionId(),
  83       base::UTF16ToUTF8(match.fill_into_edit.substr(prefix_length)),
  84       disposition);
  85
  86   return true;
  87 }
  88
  89 void OmniboxCurrentPageDelegateImpl::OnInputStateChanged() {
  90   if (!controller_->GetWebContents())
  91     return;
  92   SearchTabHelper::FromWebContents(
  93       controller_->GetWebContents())->OmniboxInputStateChanged();
  94 }
  95
  96 void OmniboxCurrentPageDelegateImpl::OnFocusChanged(
  97     OmniboxFocusState state,
  98     OmniboxFocusChangeReason reason) {
  99   if (!controller_->GetWebContents())
 100     return;
 101   SearchTabHelper::FromWebContents(
 102       controller_->GetWebContents())->OmniboxFocusChanged(state, reason);
 103 }
 104
 105 void OmniboxCurrentPageDelegateImpl::DoPrerender(
 106     const AutocompleteMatch& match) {
 107   content::WebContents* web_contents = controller_->GetWebContents();
 108   gfx::Rect container_bounds = web_contents->GetContainerBounds();
 109
 110   InstantSearchPrerenderer* prerenderer =
 111       InstantSearchPrerenderer::GetForProfile(profile_);
 112   if (prerenderer && prerenderer->IsAllowed(match, web_contents)) {
 113     prerenderer->Init(
 114         web_contents->GetController().GetDefaultSessionStorageNamespace(),
 115         container_bounds.size());
 116     return;
 117   }
 118
 119   predictors::AutocompleteActionPredictorFactory::GetForProfile(profile_)->
 120       StartPrerendering(
 121           match.destination_url,
 122           web_contents->GetController().GetDefaultSessionStorageNamespace(),
 123           container_bounds.size());
 124 }
 125
 126 void OmniboxCurrentPageDelegateImpl::SetSuggestionToPrefetch(
 127       const InstantSuggestion& suggestion) {
 128   DCHECK(chrome::IsInstantExtendedAPIEnabled());
 129   content::WebContents* web_contents = controller_->GetWebContents();
 130   if (web_contents &&
 131       SearchTabHelper::FromWebContents(web_contents)->IsSearchResultsPage()) {
 132     if (chrome::ShouldPrefetchSearchResultsOnSRP()) {
 133       SearchTabHelper::FromWebContents(web_contents)->
 134           SetSuggestionToPrefetch(suggestion);
 135     }
 136   } else {
 137     InstantSearchPrerenderer* prerenderer =
 138         InstantSearchPrerenderer::GetForProfile(profile_);
 139     if (prerenderer)
 140       prerenderer->Prerender(suggestion);
 141   }
 142 }