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