chrome/browser/download/download_shelf_context_menu.cc

   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.
   4
   5 #include "chrome/browser/download/download_shelf_context_menu.h"
   6
   7 #include "chrome/browser/browser_process.h"
   8 #include "chrome/browser/download/download_crx_util.h"
   9 #include "chrome/browser/download/download_item_model.h"
  10 #include "chrome/browser/download/download_prefs.h"
  11 #include "chrome/browser/safe_browsing/download_protection_service.h"
  12 #include "chrome/browser/safe_browsing/safe_browsing_service.h"
  13 #include "chrome/common/extensions/extension.h"
  14 #include "chrome/common/url_constants.h"
  15 #include "content/public/browser/download_item.h"
  16 #include "content/public/browser/download_manager.h"
  17 #include "content/public/browser/page_navigator.h"
  18 #include "grit/generated_resources.h"
  19 #include "ui/base/l10n/l10n_util.h"
  20
  21 using content::DownloadItem;
  22 using extensions::Extension;
  23
  24 DownloadShelfContextMenu::~DownloadShelfContextMenu() {
  25   DetachFromDownloadItem();
  26 }
  27
  28 DownloadShelfContextMenu::DownloadShelfContextMenu(
  29     DownloadItem* download_item,
  30     content::PageNavigator* navigator)
  31     : download_item_(download_item),
  32       navigator_(navigator) {
  33   DCHECK(download_item_);
  34   download_item_->AddObserver(this);
  35 }
  36
  37 ui::SimpleMenuModel* DownloadShelfContextMenu::GetMenuModel() {
  38   ui::SimpleMenuModel* model = NULL;
  39
  40   if (!download_item_)
  41     return NULL;
  42
  43   DownloadItemModel download_model(download_item_);
  44   // We shouldn't be opening a context menu for a dangerous download, unless it
  45   // is a malicious download.
  46   DCHECK(!download_model.IsDangerous() || download_model.IsMalicious());
  47
  48   if (download_model.IsMalicious())
  49     model = GetMaliciousMenuModel();
  50   else if (download_item_->IsComplete())
  51     model = GetFinishedMenuModel();
  52   else if (download_item_->IsInterrupted())
  53     model = GetInterruptedMenuModel();
  54   else
  55     model = GetInProgressMenuModel();
  56   return model;
  57 }
  58
  59 bool DownloadShelfContextMenu::IsCommandIdEnabled(int command_id) const {
  60   if (!download_item_)
  61     return false;
  62
  63   switch (static_cast<ContextMenuCommands>(command_id)) {
  64     case SHOW_IN_FOLDER:
  65       return download_item_->CanShowInFolder();
  66     case OPEN_WHEN_COMPLETE:
  67       return download_item_->CanOpenDownload() &&
  68           !download_crx_util::IsExtensionDownload(*download_item_);
  69     case ALWAYS_OPEN_TYPE:
  70       // For temporary downloads, the target filename might be a temporary
  71       // filename. Don't base an "Always open" decision based on it. Also
  72       // exclude extensions.
  73       return download_item_->CanOpenDownload() &&
  74           !download_crx_util::IsExtensionDownload(*download_item_);
  75     case CANCEL:
  76       return download_item_->IsPartialDownload();
  77     case TOGGLE_PAUSE:
  78       return download_item_->IsInProgress();
  79     case DISCARD:
  80     case KEEP:
  81     case LEARN_MORE_SCANNING:
  82     case LEARN_MORE_INTERRUPTED:
  83       return true;
  84   }
  85   return false;
  86 }
  87
  88 bool DownloadShelfContextMenu::IsCommandIdChecked(int command_id) const {
  89   if (!download_item_)
  90     return false;
  91
  92   switch (command_id) {
  93     case OPEN_WHEN_COMPLETE:
  94       return download_item_->GetOpenWhenComplete() ||
  95           download_crx_util::IsExtensionDownload(*download_item_);
  96     case ALWAYS_OPEN_TYPE:
  97       return download_item_->ShouldOpenFileBasedOnExtension();
  98     case TOGGLE_PAUSE:
  99       return download_item_->IsPaused();
 100   }
 101   return false;
 102 }
 103
 104 void DownloadShelfContextMenu::ExecuteCommand(int command_id, int event_flags) {
 105   if (!download_item_)
 106     return;
 107
 108   switch (static_cast<ContextMenuCommands>(command_id)) {
 109     case SHOW_IN_FOLDER:
 110       download_item_->ShowDownloadInShell();
 111       break;
 112     case OPEN_WHEN_COMPLETE:
 113       download_item_->OpenDownload();
 114       break;
 115     case ALWAYS_OPEN_TYPE: {
 116       DownloadPrefs* prefs = DownloadPrefs::FromBrowserContext(
 117           download_item_->GetBrowserContext());
 118       base::FilePath path = download_item_->GetUserVerifiedFilePath();
 119       if (!IsCommandIdChecked(ALWAYS_OPEN_TYPE))
 120         prefs->EnableAutoOpenBasedOnExtension(path);
 121       else
 122         prefs->DisableAutoOpenBasedOnExtension(path);
 123       break;
 124     }
 125     case CANCEL:
 126       download_item_->Cancel(true /* Cancelled by user */);
 127       break;
 128     case TOGGLE_PAUSE:
 129       // It is possible for the download to complete before the user clicks the
 130       // menu item, recheck if the download is in progress state before toggling
 131       // pause.
 132       if (download_item_->IsPartialDownload()) {
 133         if (download_item_->IsPaused())
 134           download_item_->Resume();
 135         else
 136           download_item_->Pause();
 137       }
 138       break;
 139     case DISCARD:
 140       download_item_->Delete(DownloadItem::DELETE_DUE_TO_USER_DISCARD);
 141       break;
 142     case KEEP:
 143       download_item_->DangerousDownloadValidated();
 144       break;
 145     case LEARN_MORE_SCANNING: {
 146 #if defined(FULL_SAFE_BROWSING)
 147       using safe_browsing::DownloadProtectionService;
 148       SafeBrowsingService* sb_service =
 149           g_browser_process->safe_browsing_service();
 150       DownloadProtectionService* protection_service =
 151           (sb_service ? sb_service->download_protection_service() : NULL);
 152       if (protection_service) {
 153         protection_service->ShowDetailsForDownload(*download_item_, navigator_);
 154       }
 155 #else
 156       // Should only be getting invoked if we are using safe browsing.
 157       NOTREACHED();
 158 #endif
 159       break;
 160     }
 161     case LEARN_MORE_INTERRUPTED:
 162       navigator_->OpenURL(
 163           content::OpenURLParams(GURL(chrome::kDownloadInterruptedLearnMoreURL),
 164                                  content::Referrer(),
 165                                  NEW_FOREGROUND_TAB,
 166                                  content::PAGE_TRANSITION_LINK,
 167                                  false));
 168       break;
 169   }
 170 }
 171
 172 bool DownloadShelfContextMenu::GetAcceleratorForCommandId(
 173     int command_id, ui::Accelerator* accelerator) {
 174   return false;
 175 }
 176
 177 bool DownloadShelfContextMenu::IsItemForCommandIdDynamic(int command_id) const {
 178   return command_id == TOGGLE_PAUSE;
 179 }
 180
 181 string16 DownloadShelfContextMenu::GetLabelForCommandId(int command_id) const {
 182   switch (static_cast<ContextMenuCommands>(command_id)) {
 183     case SHOW_IN_FOLDER:
 184       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_SHOW);
 185     case OPEN_WHEN_COMPLETE:
 186       if (download_item_ && download_item_->IsInProgress())
 187         return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_OPEN_WHEN_COMPLETE);
 188       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_OPEN);
 189     case ALWAYS_OPEN_TYPE:
 190       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE);
 191     case CANCEL:
 192       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_CANCEL);
 193     case TOGGLE_PAUSE:
 194       if (download_item_ && download_item_->IsPaused())
 195         return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_RESUME_ITEM);
 196       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_PAUSE_ITEM);
 197     case DISCARD:
 198       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_DISCARD);
 199     case KEEP:
 200       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_KEEP);
 201     case LEARN_MORE_SCANNING:
 202       return l10n_util::GetStringUTF16(IDS_DOWNLOAD_MENU_LEARN_MORE_SCANNING);
 203     case LEARN_MORE_INTERRUPTED:
 204       return l10n_util::GetStringUTF16(
 205           IDS_DOWNLOAD_MENU_LEARN_MORE_INTERRUPTED);
 206   }
 207   NOTREACHED();
 208   return string16();
 209 }
 210
 211 void DownloadShelfContextMenu::DetachFromDownloadItem() {
 212   if (!download_item_)
 213     return;
 214
 215   download_item_->RemoveObserver(this);
 216   download_item_ = NULL;
 217 }
 218
 219 void DownloadShelfContextMenu::OnDownloadDestroyed(DownloadItem* download) {
 220   DCHECK(download_item_ == download);
 221   DetachFromDownloadItem();
 222 }
 223
 224 ui::SimpleMenuModel* DownloadShelfContextMenu::GetInProgressMenuModel() {
 225   if (in_progress_download_menu_model_)
 226     return in_progress_download_menu_model_.get();
 227
 228   in_progress_download_menu_model_.reset(new ui::SimpleMenuModel(this));
 229
 230   in_progress_download_menu_model_->AddCheckItemWithStringId(
 231       OPEN_WHEN_COMPLETE, IDS_DOWNLOAD_MENU_OPEN_WHEN_COMPLETE);
 232   in_progress_download_menu_model_->AddCheckItemWithStringId(
 233       ALWAYS_OPEN_TYPE, IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE);
 234   in_progress_download_menu_model_->AddSeparator(ui::NORMAL_SEPARATOR);
 235   in_progress_download_menu_model_->AddItemWithStringId(
 236       TOGGLE_PAUSE, IDS_DOWNLOAD_MENU_PAUSE_ITEM);
 237   in_progress_download_menu_model_->AddItemWithStringId(
 238       SHOW_IN_FOLDER, IDS_DOWNLOAD_MENU_SHOW);
 239   in_progress_download_menu_model_->AddSeparator(ui::NORMAL_SEPARATOR);
 240   in_progress_download_menu_model_->AddItemWithStringId(
 241       CANCEL, IDS_DOWNLOAD_MENU_CANCEL);
 242
 243   return in_progress_download_menu_model_.get();
 244 }
 245
 246 ui::SimpleMenuModel* DownloadShelfContextMenu::GetFinishedMenuModel() {
 247   if (finished_download_menu_model_)
 248     return finished_download_menu_model_.get();
 249
 250   finished_download_menu_model_.reset(new ui::SimpleMenuModel(this));
 251
 252   finished_download_menu_model_->AddItemWithStringId(
 253       OPEN_WHEN_COMPLETE, IDS_DOWNLOAD_MENU_OPEN);
 254   finished_download_menu_model_->AddCheckItemWithStringId(
 255       ALWAYS_OPEN_TYPE, IDS_DOWNLOAD_MENU_ALWAYS_OPEN_TYPE);
 256   finished_download_menu_model_->AddSeparator(ui::NORMAL_SEPARATOR);
 257   finished_download_menu_model_->AddItemWithStringId(
 258       SHOW_IN_FOLDER, IDS_DOWNLOAD_MENU_SHOW);
 259   finished_download_menu_model_->AddSeparator(ui::NORMAL_SEPARATOR);
 260   finished_download_menu_model_->AddItemWithStringId(
 261       CANCEL, IDS_DOWNLOAD_MENU_CANCEL);
 262
 263   return finished_download_menu_model_.get();
 264 }
 265
 266 ui::SimpleMenuModel* DownloadShelfContextMenu::GetInterruptedMenuModel() {
 267 #if defined(OS_WIN)
 268   // The Help Center article is currently Windows specific.
 269   // TODO(asanka): Enable this for other platforms when the article is expanded
 270   // for other platforms.
 271   if (interrupted_download_menu_model_)
 272     return interrupted_download_menu_model_.get();
 273
 274   interrupted_download_menu_model_.reset(new ui::SimpleMenuModel(this));
 275
 276   interrupted_download_menu_model_->AddItemWithStringId(
 277       LEARN_MORE_INTERRUPTED, IDS_DOWNLOAD_MENU_LEARN_MORE_INTERRUPTED);
 278
 279   return interrupted_download_menu_model_.get();
 280 #else
 281   return GetInProgressMenuModel();
 282 #endif
 283 }
 284
 285 ui::SimpleMenuModel* DownloadShelfContextMenu::GetMaliciousMenuModel() {
 286   if (malicious_download_menu_model_)
 287     return malicious_download_menu_model_.get();
 288
 289   malicious_download_menu_model_.reset(new ui::SimpleMenuModel(this));
 290
 291   malicious_download_menu_model_->AddItemWithStringId(
 292       DISCARD, IDS_DOWNLOAD_MENU_DISCARD);
 293   malicious_download_menu_model_->AddItemWithStringId(
 294       KEEP, IDS_DOWNLOAD_MENU_KEEP);
 295   malicious_download_menu_model_->AddSeparator(ui::NORMAL_SEPARATOR);
 296   malicious_download_menu_model_->AddItemWithStringId(
 297       LEARN_MORE_SCANNING, IDS_DOWNLOAD_MENU_LEARN_MORE_SCANNING);
 298
 299   return malicious_download_menu_model_.get();
 300 }