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.
5 #include "chrome/browser/ui/toolbar/back_forward_menu_model.h"
8 #include "base/bind_helpers.h"
9 #include "base/prefs/pref_service.h"
10 #include "base/strings/string_number_conversions.h"
11 #include "build/build_config.h"
12 #include "chrome/browser/favicon/favicon_service_factory.h"
13 #include "chrome/browser/profiles/profile.h"
14 #include "chrome/browser/ui/browser.h"
15 #include "chrome/browser/ui/browser_commands.h"
16 #include "chrome/browser/ui/singleton_tabs.h"
17 #include "chrome/browser/ui/tabs/tab_strip_model.h"
18 #include "chrome/common/favicon/favicon_types.h"
19 #include "chrome/common/pref_names.h"
20 #include "chrome/common/url_constants.h"
21 #include "content/public/browser/favicon_status.h"
22 #include "content/public/browser/navigation_controller.h"
23 #include "content/public/browser/navigation_entry.h"
24 #include "content/public/browser/user_metrics.h"
25 #include "content/public/browser/web_contents.h"
26 #include "grit/generated_resources.h"
27 #include "grit/theme_resources.h"
28 #include "net/base/registry_controlled_domains/registry_controlled_domain.h"
29 #include "ui/base/l10n/l10n_util.h"
30 #include "ui/base/resource/resource_bundle.h"
31 #include "ui/base/window_open_disposition.h"
32 #include "ui/gfx/favicon_size.h"
33 #include "ui/gfx/text_elider.h"
35 using content::NavigationController;
36 using content::NavigationEntry;
37 using content::UserMetricsAction;
38 using content::WebContents;
40 const int BackForwardMenuModel::kMaxHistoryItems = 12;
41 const int BackForwardMenuModel::kMaxChapterStops = 5;
42 static const int kMaxWidth = 700;
44 BackForwardMenuModel::BackForwardMenuModel(Browser* browser,
47 test_web_contents_(NULL),
48 model_type_(model_type),
49 menu_model_delegate_(NULL) {
52 BackForwardMenuModel::~BackForwardMenuModel() {
55 bool BackForwardMenuModel::HasIcons() const {
59 int BackForwardMenuModel::GetItemCount() const {
60 int items = GetHistoryItemCount();
63 int chapter_stops = 0;
65 // Next, we count ChapterStops, if any.
66 if (items == kMaxHistoryItems)
67 chapter_stops = GetChapterStopCount(items);
70 items += chapter_stops + 1; // Chapter stops also need a separator.
72 // If the menu is not empty, add two positions in the end
73 // for a separator and a "Show Full History" item.
80 ui::MenuModel::ItemType BackForwardMenuModel::GetTypeAt(int index) const {
81 return IsSeparator(index) ? TYPE_SEPARATOR : TYPE_COMMAND;
84 ui::MenuSeparatorType BackForwardMenuModel::GetSeparatorTypeAt(
86 return ui::NORMAL_SEPARATOR;
89 int BackForwardMenuModel::GetCommandIdAt(int index) const {
93 string16 BackForwardMenuModel::GetLabelAt(int index) const {
94 // Return label "Show Full History" for the last item of the menu.
95 if (index == GetItemCount() - 1)
96 return l10n_util::GetStringUTF16(IDS_SHOWFULLHISTORY_LINK);
98 // Return an empty string for a separator.
99 if (IsSeparator(index))
102 // Return the entry title, escaping any '&' characters and eliding it if it's
104 NavigationEntry* entry = GetNavigationEntry(index);
106 Profile::FromBrowserContext(GetWebContents()->GetBrowserContext());
107 string16 menu_text(entry->GetTitleForDisplay(
108 profile->GetPrefs()->GetString(prefs::kAcceptLanguages)));
110 gfx::ElideText(menu_text, gfx::Font(), kMaxWidth, gfx::ELIDE_AT_END);
112 #if !defined(OS_MACOSX)
113 for (size_t i = menu_text.find('&'); i != string16::npos;
114 i = menu_text.find('&', i + 2)) {
115 menu_text.insert(i, 1, '&');
122 bool BackForwardMenuModel::IsItemDynamicAt(int index) const {
123 // This object is only used for a single showing of a menu.
127 bool BackForwardMenuModel::GetAcceleratorAt(
129 ui::Accelerator* accelerator) const {
133 bool BackForwardMenuModel::IsItemCheckedAt(int index) const {
137 int BackForwardMenuModel::GetGroupIdAt(int index) const {
141 bool BackForwardMenuModel::GetIconAt(int index, gfx::Image* icon) {
142 if (!ItemHasIcon(index))
145 if (index == GetItemCount() - 1) {
146 *icon = ResourceBundle::GetSharedInstance().GetNativeImageNamed(
147 IDR_HISTORY_FAVICON);
149 NavigationEntry* entry = GetNavigationEntry(index);
150 *icon = entry->GetFavicon().image;
151 if (!entry->GetFavicon().valid && menu_model_delegate()) {
159 ui::ButtonMenuItemModel* BackForwardMenuModel::GetButtonMenuItemAt(
164 bool BackForwardMenuModel::IsEnabledAt(int index) const {
165 return index < GetItemCount() && !IsSeparator(index);
168 ui::MenuModel* BackForwardMenuModel::GetSubmenuModelAt(int index) const {
172 void BackForwardMenuModel::HighlightChangedTo(int index) {
175 void BackForwardMenuModel::ActivatedAt(int index) {
176 ActivatedAt(index, 0);
179 void BackForwardMenuModel::ActivatedAt(int index, int event_flags) {
180 DCHECK(!IsSeparator(index));
182 // Execute the command for the last item: "Show Full History".
183 if (index == GetItemCount() - 1) {
184 content::RecordComputedAction(BuildActionName("ShowFullHistory", -1));
185 chrome::ShowSingletonTabOverwritingNTP(browser_,
186 chrome::GetSingletonTabNavigateParams(
187 browser_, GURL(chrome::kChromeUIHistoryURL)));
191 // Log whether it was a history or chapter click.
192 if (index < GetHistoryItemCount()) {
193 content::RecordComputedAction(
194 BuildActionName("HistoryClick", index));
196 content::RecordComputedAction(
197 BuildActionName("ChapterClick", index - GetHistoryItemCount() - 1));
200 int controller_index = MenuIndexToNavEntryIndex(index);
201 WindowOpenDisposition disposition =
202 ui::DispositionFromEventFlags(event_flags);
203 if (!chrome::NavigateToIndexWithDisposition(browser_,
210 void BackForwardMenuModel::MenuWillShow() {
211 content::RecordComputedAction(BuildActionName("Popup", -1));
212 requested_favicons_.clear();
213 cancelable_task_tracker_.TryCancelAll();
216 bool BackForwardMenuModel::IsSeparator(int index) const {
217 int history_items = GetHistoryItemCount();
218 // If the index is past the number of history items + separator,
219 // we then consider if it is a chapter-stop entry.
220 if (index > history_items) {
221 // We either are in ChapterStop area, or at the end of the list (the "Show
222 // Full History" link).
223 int chapter_stops = GetChapterStopCount(history_items);
224 if (chapter_stops == 0)
225 return false; // We must have reached the "Show Full History" link.
226 // Otherwise, look to see if we have reached the separator for the
227 // chapter-stops. If not, this is a chapter stop.
228 return (index == history_items + 1 + chapter_stops);
231 // Look to see if we have reached the separator for the history items.
232 return index == history_items;
235 void BackForwardMenuModel::SetMenuModelDelegate(
236 ui::MenuModelDelegate* menu_model_delegate) {
237 menu_model_delegate_ = menu_model_delegate;
240 ui::MenuModelDelegate* BackForwardMenuModel::GetMenuModelDelegate() const {
241 return menu_model_delegate_;
244 void BackForwardMenuModel::FetchFavicon(NavigationEntry* entry) {
245 // If the favicon has already been requested for this menu, don't do
247 if (requested_favicons_.find(entry->GetUniqueID()) !=
248 requested_favicons_.end()) {
251 requested_favicons_.insert(entry->GetUniqueID());
252 FaviconService* favicon_service = FaviconServiceFactory::GetForProfile(
253 browser_->profile(), Profile::EXPLICIT_ACCESS);
254 if (!favicon_service)
257 favicon_service->GetFaviconImageForURL(
258 FaviconService::FaviconForURLParams(browser_->profile(),
262 base::Bind(&BackForwardMenuModel::OnFavIconDataAvailable,
263 base::Unretained(this),
264 entry->GetUniqueID()),
265 &cancelable_task_tracker_);
268 void BackForwardMenuModel::OnFavIconDataAvailable(
269 int navigation_entry_unique_id,
270 const chrome::FaviconImageResult& image_result) {
271 if (!image_result.image.IsEmpty()) {
272 // Find the current model_index for the unique id.
273 NavigationEntry* entry = NULL;
274 int model_index = -1;
275 for (int i = 0; i < GetItemCount() - 1; i++) {
278 if (GetNavigationEntry(i)->GetUniqueID() == navigation_entry_unique_id) {
280 entry = GetNavigationEntry(i);
286 // The NavigationEntry wasn't found, this can happen if the user
287 // navigates to another page and a NavigatationEntry falls out of the
288 // range of kMaxHistoryItems.
291 // Now that we have a valid NavigationEntry, decode the favicon and assign
292 // it to the NavigationEntry.
293 entry->GetFavicon().valid = true;
294 entry->GetFavicon().url = image_result.icon_url;
295 entry->GetFavicon().image = image_result.image;
296 if (menu_model_delegate()) {
297 menu_model_delegate()->OnIconChanged(model_index);
302 int BackForwardMenuModel::GetHistoryItemCount() const {
303 WebContents* contents = GetWebContents();
306 if (model_type_ == FORWARD_MENU) {
307 // Only count items from n+1 to end (if n is current entry)
308 items = contents->GetController().GetEntryCount() -
309 contents->GetController().GetCurrentEntryIndex() - 1;
311 items = contents->GetController().GetCurrentEntryIndex();
314 if (items > kMaxHistoryItems)
315 items = kMaxHistoryItems;
322 int BackForwardMenuModel::GetChapterStopCount(int history_items) const {
323 WebContents* contents = GetWebContents();
325 int chapter_stops = 0;
326 int current_entry = contents->GetController().GetCurrentEntryIndex();
328 if (history_items == kMaxHistoryItems) {
329 int chapter_id = current_entry;
330 if (model_type_ == FORWARD_MENU) {
331 chapter_id += history_items;
333 chapter_id -= history_items;
337 chapter_id = GetIndexOfNextChapterStop(chapter_id,
338 model_type_ == FORWARD_MENU);
339 if (chapter_id != -1)
341 } while (chapter_id != -1 && chapter_stops < kMaxChapterStops);
344 return chapter_stops;
347 int BackForwardMenuModel::GetIndexOfNextChapterStop(int start_from,
348 bool forward) const {
349 WebContents* contents = GetWebContents();
350 NavigationController& controller = contents->GetController();
352 int max_count = controller.GetEntryCount();
353 if (start_from < 0 || start_from >= max_count)
354 return -1; // Out of bounds.
357 if (start_from < max_count - 1) {
358 // We want to advance over the current chapter stop, so we add one.
359 // We don't need to do this when direction is backwards.
366 NavigationEntry* start_entry = controller.GetEntryAtIndex(start_from);
367 const GURL& url = start_entry->GetURL();
370 // When going backwards we return the first entry we find that has a
372 for (int i = start_from - 1; i >= 0; --i) {
373 if (!net::registry_controlled_domains::SameDomainOrHost(url,
374 controller.GetEntryAtIndex(i)->GetURL(),
375 net::registry_controlled_domains::EXCLUDE_PRIVATE_REGISTRIES))
378 // We have reached the beginning without finding a chapter stop.
381 // When going forwards we return the entry before the entry that has a
383 for (int i = start_from + 1; i < max_count; ++i) {
384 if (!net::registry_controlled_domains::SameDomainOrHost(url,
385 controller.GetEntryAtIndex(i)->GetURL(),
386 net::registry_controlled_domains::EXCLUDE_PRIVATE_REGISTRIES))
389 // Last entry is always considered a chapter stop.
390 return max_count - 1;
394 int BackForwardMenuModel::FindChapterStop(int offset,
397 if (offset < 0 || skip < 0)
403 WebContents* contents = GetWebContents();
404 int entry = contents->GetController().GetCurrentEntryIndex() + offset;
405 for (int i = 0; i < skip + 1; i++)
406 entry = GetIndexOfNextChapterStop(entry, forward);
411 bool BackForwardMenuModel::ItemHasCommand(int index) const {
412 return index < GetItemCount() && !IsSeparator(index);
415 bool BackForwardMenuModel::ItemHasIcon(int index) const {
416 return index < GetItemCount() && !IsSeparator(index);
419 string16 BackForwardMenuModel::GetShowFullHistoryLabel() const {
420 return l10n_util::GetStringUTF16(IDS_SHOWFULLHISTORY_LINK);
423 WebContents* BackForwardMenuModel::GetWebContents() const {
424 // We use the test web contents if the unit test has specified it.
425 return test_web_contents_ ?
427 browser_->tab_strip_model()->GetActiveWebContents();
430 int BackForwardMenuModel::MenuIndexToNavEntryIndex(int index) const {
431 WebContents* contents = GetWebContents();
432 int history_items = GetHistoryItemCount();
436 // Convert anything above the History items separator.
437 if (index < history_items) {
438 if (model_type_ == FORWARD_MENU) {
439 index += contents->GetController().GetCurrentEntryIndex() + 1;
441 // Back menu is reverse.
442 index = contents->GetController().GetCurrentEntryIndex() - (index + 1);
446 if (index == history_items)
447 return -1; // Don't translate the separator for history items.
449 if (index >= history_items + 1 + GetChapterStopCount(history_items))
450 return -1; // This is beyond the last chapter stop so we abort.
452 // This menu item is a chapter stop located between the two separators.
453 index = FindChapterStop(history_items,
454 model_type_ == FORWARD_MENU,
455 index - history_items - 1);
460 NavigationEntry* BackForwardMenuModel::GetNavigationEntry(int index) const {
461 int controller_index = MenuIndexToNavEntryIndex(index);
462 NavigationController& controller = GetWebContents()->GetController();
463 if (controller_index >= 0 && controller_index < controller.GetEntryCount())
464 return controller.GetEntryAtIndex(controller_index);
470 std::string BackForwardMenuModel::BuildActionName(
471 const std::string& action, int index) const {
472 DCHECK(!action.empty());
474 std::string metric_string;
475 if (model_type_ == FORWARD_MENU)
476 metric_string += "ForwardMenu_";
478 metric_string += "BackMenu_";
479 metric_string += action;
481 // +1 is for historical reasons (indices used to start at 1).
482 metric_string += base::IntToString(index + 1);
484 return metric_string;