1dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 2c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// Use of this source code is governed by a BSD-style license that can be 3c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// found in the LICENSE file. 4c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 5c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/download/download_shelf.h" 6c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 7c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "base/file_util.h" 8c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/download/download_item.h" 9c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/download/download_item_model.h" 10c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/download/download_manager.h" 11c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/download/download_util.h" 12c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/browser/metrics/user_metrics.h" 13dc0f95d653279beabeb9817299e2902918ba123eKristian Monsen#include "chrome/browser/ui/webui/downloads_ui.h" 14c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "chrome/common/url_constants.h" 15c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch#include "grit/generated_resources.h" 1672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "ui/base/l10n/l10n_util.h" 17c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 18c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch// DownloadShelfContextMenu ---------------------------------------------------- 19c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 20c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochDownloadShelfContextMenu::DownloadShelfContextMenu( 21c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch BaseDownloadItemModel* download_model) 22c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch : download_(download_model->download()), 23c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch model_(download_model) { 24c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 25c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 26c407dc5cd9bdc5668497f21b26b09d988ab439deBen MurdochDownloadShelfContextMenu::~DownloadShelfContextMenu() { 27c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 28c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 2921d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian MonsenDownloadItem* DownloadShelfContextMenu::download() const { 3021d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen return download_; 3121d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen} 3221d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsen 33c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochbool DownloadShelfContextMenu::IsCommandIdChecked(int command_id) const { 34c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch switch (command_id) { 353345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick case OPEN_WHEN_COMPLETE: 36c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return download_->open_when_complete(); 373345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick case ALWAYS_OPEN_TYPE: 383345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return download_->ShouldOpenFileBasedOnExtension(); 39ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen case TOGGLE_PAUSE: 40ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen return download_->is_paused(); 41c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 42c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return false; 43c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 44c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 45c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochstring16 DownloadShelfContextMenu::GetLabelForCommandId(int command_id) const { 46c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch switch (command_id) { 47c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case SHOW_IN_FOLDER: 48c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_SHOW); 49c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case OPEN_WHEN_COMPLETE: 50ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen if (download_->IsInProgress()) 51c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_OPEN_WHEN_COMPLETE); 52c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_OPEN); 53c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case ALWAYS_OPEN_TYPE: 54c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE); 55c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case CANCEL: 56c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_CANCEL); 57c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case TOGGLE_PAUSE: { 58c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (download_->is_paused()) 59c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_RESUME_ITEM); 60c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch else 61c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_PAUSE_ITEM); 62c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 63c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch default: 64c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch NOTREACHED(); 65c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 66c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return string16(); 67c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 68c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 69c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochbool DownloadShelfContextMenu::IsCommandIdEnabled(int command_id) const { 70c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch switch (command_id) { 71c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case SHOW_IN_FOLDER: 72c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case OPEN_WHEN_COMPLETE: 73ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen return !download_->IsCancelled(); 74c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case ALWAYS_OPEN_TYPE: 753345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick return download_->CanOpenDownload(); 76c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case CANCEL: 77ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen return download_->IsPartialDownload(); 78c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case TOGGLE_PAUSE: 79ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen return download_->IsInProgress(); 80c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch default: 81c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return command_id > 0 && command_id < MENU_LAST; 82c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 83c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 84c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 85c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochvoid DownloadShelfContextMenu::ExecuteCommand(int command_id) { 86c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch switch (command_id) { 87c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case SHOW_IN_FOLDER: 883345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick download_->ShowDownloadInShell(); 89c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch break; 90c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case OPEN_WHEN_COMPLETE: 913345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick download_->OpenDownload(); 92c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch break; 93c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case ALWAYS_OPEN_TYPE: { 943345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick download_->OpenFilesBasedOnExtension( 953345a6884c488ff3a535c2c9acdd33d74b37e311Iain Merrick !IsCommandIdChecked(ALWAYS_OPEN_TYPE)); 96c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch break; 97c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 98c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case CANCEL: 99c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch model_->CancelTask(); 100c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch break; 101c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch case TOGGLE_PAUSE: 102c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // It is possible for the download to complete before the user clicks the 103c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // menu item, recheck if the download is in progress state before toggling 104c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch // pause. 105ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen if (download_->IsPartialDownload()) 106c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch download_->TogglePause(); 107c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch break; 108c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch default: 109c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch NOTREACHED(); 110c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch } 111c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 112c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 113c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdochbool DownloadShelfContextMenu::GetAcceleratorForCommandId( 11472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen int command_id, ui::Accelerator* accelerator) { 115c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return false; 116c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 117c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 11821d179b334e59e9a3bfcaed4c4430bef1bc5759dKristian Monsenbool DownloadShelfContextMenu::IsItemForCommandIdDynamic( 119c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch int command_id) const { 120c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return command_id == TOGGLE_PAUSE; 121c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 122c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 12372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsenui::SimpleMenuModel* DownloadShelfContextMenu::GetInProgressMenuModel() { 124c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (in_progress_download_menu_model_.get()) 125c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return in_progress_download_menu_model_.get(); 126c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 12772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen in_progress_download_menu_model_.reset(new ui::SimpleMenuModel(this)); 128c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 129c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch in_progress_download_menu_model_->AddCheckItemWithStringId( 130c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch OPEN_WHEN_COMPLETE, IDS_DOWNLOAD_MENU_OPEN_WHEN_COMPLETE); 131c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch in_progress_download_menu_model_->AddCheckItemWithStringId( 132c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ALWAYS_OPEN_TYPE, IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE); 133c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch in_progress_download_menu_model_->AddSeparator(); 134c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch in_progress_download_menu_model_->AddItemWithStringId( 135c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch TOGGLE_PAUSE, IDS_DOWNLOAD_MENU_PAUSE_ITEM); 136c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch in_progress_download_menu_model_->AddItemWithStringId( 137c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SHOW_IN_FOLDER, IDS_DOWNLOAD_MENU_SHOW); 138c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch in_progress_download_menu_model_->AddSeparator(); 139c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch in_progress_download_menu_model_->AddItemWithStringId( 140c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch CANCEL, IDS_DOWNLOAD_MENU_CANCEL); 141c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 142c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return in_progress_download_menu_model_.get(); 143c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 144c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 14572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsenui::SimpleMenuModel* DownloadShelfContextMenu::GetFinishedMenuModel() { 146c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch if (finished_download_menu_model_.get()) 147c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return finished_download_menu_model_.get(); 148c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 14972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen finished_download_menu_model_.reset(new ui::SimpleMenuModel(this)); 150c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 151c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch finished_download_menu_model_->AddItemWithStringId( 152c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch OPEN_WHEN_COMPLETE, IDS_DOWNLOAD_MENU_OPEN); 153c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch finished_download_menu_model_->AddCheckItemWithStringId( 154c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch ALWAYS_OPEN_TYPE, IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE); 155c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch finished_download_menu_model_->AddSeparator(); 156c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch finished_download_menu_model_->AddItemWithStringId( 157c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch SHOW_IN_FOLDER, IDS_DOWNLOAD_MENU_SHOW); 158c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch finished_download_menu_model_->AddSeparator(); 159c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch finished_download_menu_model_->AddItemWithStringId( 160c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch CANCEL, IDS_DOWNLOAD_MENU_CANCEL); 161c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch 162c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch return finished_download_menu_model_.get(); 163c407dc5cd9bdc5668497f21b26b09d988ab439deBen Murdoch} 164