1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
|
// Copyright (c) 2011 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/search_engines/search_engine_tab_helper.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/search_engines/template_url.h"
#include "chrome/browser/search_engines/template_url_fetcher.h"
#include "chrome/browser/search_engines/template_url_model.h"
#include "chrome/browser/ui/search_engines/template_url_fetcher_ui_callbacks.h"
#include "chrome/common/render_messages.h"
#include "chrome/common/render_messages_params.h"
#include "content/browser/tab_contents/tab_contents.h"
namespace {
// Returns true if the entry's transition type is FORM_SUBMIT.
bool IsFormSubmit(const NavigationEntry* entry) {
return (PageTransition::StripQualifier(entry->transition_type()) ==
PageTransition::FORM_SUBMIT);
}
} // namespace
SearchEngineTabHelper::SearchEngineTabHelper(TabContents* tab_contents)
: TabContentsObserver(tab_contents) {
DCHECK(tab_contents);
}
SearchEngineTabHelper::~SearchEngineTabHelper() {
}
void SearchEngineTabHelper::DidNavigateMainFramePostCommit(
const NavigationController::LoadCommittedDetails& /*details*/,
const ViewHostMsg_FrameNavigate_Params& params) {
GenerateKeywordIfNecessary(params);
}
bool SearchEngineTabHelper::OnMessageReceived(const IPC::Message& message) {
bool handled = true;
IPC_BEGIN_MESSAGE_MAP(SearchEngineTabHelper, message)
IPC_MESSAGE_HANDLER(ViewHostMsg_PageHasOSDD, OnPageHasOSDD)
IPC_MESSAGE_UNHANDLED(handled = false)
IPC_END_MESSAGE_MAP()
return handled;
}
void SearchEngineTabHelper::OnPageHasOSDD(
int32 page_id,
const GURL& doc_url,
const ViewHostMsg_PageHasOSDD_Type& msg_provider_type) {
// Checks to see if we should generate a keyword based on the OSDD, and if
// necessary uses TemplateURLFetcher to download the OSDD and create a
// keyword.
// Make sure page_id is the current page and other basic checks.
DCHECK(doc_url.is_valid());
if (!tab_contents()->IsActiveEntry(page_id))
return;
if (!tab_contents()->profile()->GetTemplateURLFetcher())
return;
if (tab_contents()->profile()->IsOffTheRecord())
return;
TemplateURLFetcher::ProviderType provider_type;
switch (msg_provider_type.type) {
case ViewHostMsg_PageHasOSDD_Type::AUTODETECTED_PROVIDER:
provider_type = TemplateURLFetcher::AUTODETECTED_PROVIDER;
break;
case ViewHostMsg_PageHasOSDD_Type::EXPLICIT_DEFAULT_PROVIDER:
provider_type = TemplateURLFetcher::EXPLICIT_DEFAULT_PROVIDER;
break;
case ViewHostMsg_PageHasOSDD_Type::EXPLICIT_PROVIDER:
provider_type = TemplateURLFetcher::EXPLICIT_PROVIDER;
break;
default:
NOTREACHED();
return;
}
const NavigationController& controller = tab_contents()->controller();
const NavigationEntry* entry = controller.GetLastCommittedEntry();
DCHECK(entry);
const NavigationEntry* base_entry = entry;
if (IsFormSubmit(base_entry)) {
// If the current page is a form submit, find the last page that was not
// a form submit and use its url to generate the keyword from.
int index = controller.last_committed_entry_index() - 1;
while (index >= 0 && IsFormSubmit(controller.GetEntryAtIndex(index)))
index--;
if (index >= 0)
base_entry = controller.GetEntryAtIndex(index);
else
base_entry = NULL;
}
// We want to use the user typed URL if available since that represents what
// the user typed to get here, and fall back on the regular URL if not.
if (!base_entry)
return;
GURL keyword_url = base_entry->user_typed_url().is_valid() ?
base_entry->user_typed_url() : base_entry->url();
if (!keyword_url.is_valid())
return;
string16 keyword = TemplateURLModel::GenerateKeyword(
keyword_url,
provider_type == TemplateURLFetcher::AUTODETECTED_PROVIDER);
// Download the OpenSearch description document. If this is successful, a
// new keyword will be created when done.
tab_contents()->profile()->GetTemplateURLFetcher()->ScheduleDownload(
keyword,
doc_url,
base_entry->favicon().url(),
new TemplateURLFetcherUICallbacks(this, tab_contents()),
provider_type);
}
void SearchEngineTabHelper::GenerateKeywordIfNecessary(
const ViewHostMsg_FrameNavigate_Params& params) {
if (!params.searchable_form_url.is_valid())
return;
if (tab_contents()->profile()->IsOffTheRecord())
return;
const NavigationController& controller = tab_contents()->controller();
int last_index = controller.last_committed_entry_index();
// When there was no previous page, the last index will be 0. This is
// normally due to a form submit that opened in a new tab.
// TODO(brettw) bug 916126: we should support keywords when form submits
// happen in new tabs.
if (last_index <= 0)
return;
const NavigationEntry* previous_entry =
controller.GetEntryAtIndex(last_index - 1);
if (IsFormSubmit(previous_entry)) {
// Only generate a keyword if the previous page wasn't itself a form
// submit.
return;
}
GURL keyword_url = previous_entry->user_typed_url().is_valid() ?
previous_entry->user_typed_url() : previous_entry->url();
string16 keyword =
TemplateURLModel::GenerateKeyword(keyword_url, true); // autodetected
if (keyword.empty())
return;
TemplateURLModel* url_model =
tab_contents()->profile()->GetTemplateURLModel();
if (!url_model)
return;
if (!url_model->loaded()) {
url_model->Load();
return;
}
const TemplateURL* current_url;
GURL url = params.searchable_form_url;
if (!url_model->CanReplaceKeyword(keyword, url, ¤t_url))
return;
if (current_url) {
if (current_url->originating_url().is_valid()) {
// The existing keyword was generated from an OpenSearch description
// document, don't regenerate.
return;
}
url_model->Remove(current_url);
}
TemplateURL* new_url = new TemplateURL();
new_url->set_keyword(keyword);
new_url->set_short_name(keyword);
new_url->SetURL(url.spec(), 0, 0);
new_url->add_input_encoding(params.searchable_form_encoding);
DCHECK(controller.GetLastCommittedEntry());
const GURL& favicon_url =
controller.GetLastCommittedEntry()->favicon().url();
if (favicon_url.is_valid()) {
new_url->SetFaviconURL(favicon_url);
} else {
// The favicon url isn't valid. This means there really isn't a favicon,
// or the favicon url wasn't obtained before the load started. This assumes
// the later.
// TODO(sky): Need a way to set the favicon that doesn't involve generating
// its url.
new_url->SetFaviconURL(TemplateURL::GenerateFaviconURL(params.referrer));
}
new_url->set_safe_for_autoreplace(true);
url_model->Add(new_url);
}
|
