1// Copyright (c) 2011 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#ifndef CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SHELF_CONTEXT_MENU_H_ 6#define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SHELF_CONTEXT_MENU_H_ 7 8#include "base/basictypes.h" 9#include "base/compiler_specific.h" 10#include "base/memory/scoped_ptr.h" 11#include "base/strings/string16.h" 12#include "content/public/browser/download_item.h" 13#include "ui/base/models/simple_menu_model.h" 14 15namespace content { 16class PageNavigator; 17} 18 19// This class is responsible for the download shelf context menu. Platform 20// specific subclasses are responsible for creating and running the menu. 21// 22// The DownloadItem corresponding to the context menu is observed for removal or 23// destruction. 24class DownloadShelfContextMenu : public ui::SimpleMenuModel::Delegate, 25 public content::DownloadItem::Observer { 26 public: 27 enum ContextMenuCommands { 28 SHOW_IN_FOLDER = 1, // Open a folder view window with the item selected. 29 OPEN_WHEN_COMPLETE, // Open the download when it's finished. 30 ALWAYS_OPEN_TYPE, // Default this file extension to always open. 31 PLATFORM_OPEN, // Open using platform handler. 32 CANCEL, // Cancel the download. 33 TOGGLE_PAUSE, // Pause or resume a download. 34 DISCARD, // Discard the malicious download. 35 KEEP, // Keep the malicious download. 36 LEARN_MORE_SCANNING, // Show information about download scanning. 37 LEARN_MORE_INTERRUPTED,// Show information about interrupted downloads. 38 }; 39 40 virtual ~DownloadShelfContextMenu(); 41 42 content::DownloadItem* download_item() const { return download_item_; } 43 44 protected: 45 DownloadShelfContextMenu(content::DownloadItem* download_item, 46 content::PageNavigator* navigator); 47 48 // Returns the correct menu model depending on the state of the download item. 49 // Returns NULL if the download was destroyed. 50 ui::SimpleMenuModel* GetMenuModel(); 51 52 // ui::SimpleMenuModel::Delegate: 53 virtual bool IsCommandIdEnabled(int command_id) const OVERRIDE; 54 virtual bool IsCommandIdChecked(int command_id) const OVERRIDE; 55 virtual bool IsCommandIdVisible(int command_id) const OVERRIDE; 56 virtual void ExecuteCommand(int command_id, int event_flags) OVERRIDE; 57 virtual bool GetAcceleratorForCommandId( 58 int command_id, 59 ui::Accelerator* accelerator) OVERRIDE; 60 virtual bool IsItemForCommandIdDynamic(int command_id) const OVERRIDE; 61 virtual base::string16 GetLabelForCommandId(int command_id) const OVERRIDE; 62 63 private: 64 // Detaches self from |download_item_|. Called when the DownloadItem is 65 // destroyed or when this object is being destroyed. 66 void DetachFromDownloadItem(); 67 68 // content::DownloadItem::Observer 69 virtual void OnDownloadDestroyed(content::DownloadItem* download) OVERRIDE; 70 71 ui::SimpleMenuModel* GetInProgressMenuModel(); 72 ui::SimpleMenuModel* GetFinishedMenuModel(); 73 ui::SimpleMenuModel* GetInterruptedMenuModel(); 74 ui::SimpleMenuModel* GetMaybeMaliciousMenuModel(); 75 ui::SimpleMenuModel* GetMaliciousMenuModel(); 76 77 int GetAlwaysOpenStringId() const; 78 79#if defined(OS_WIN) 80 bool IsDownloadPdf() const; 81 bool CanOpenPdfInReader() const; 82#endif 83 84 // We show slightly different menus if the download is in progress vs. if the 85 // download has finished. 86 scoped_ptr<ui::SimpleMenuModel> in_progress_download_menu_model_; 87 scoped_ptr<ui::SimpleMenuModel> finished_download_menu_model_; 88 scoped_ptr<ui::SimpleMenuModel> interrupted_download_menu_model_; 89 scoped_ptr<ui::SimpleMenuModel> maybe_malicious_download_menu_model_; 90 scoped_ptr<ui::SimpleMenuModel> malicious_download_menu_model_; 91 92 // Information source. 93 content::DownloadItem* download_item_; 94 95 // Used to open tabs. 96 content::PageNavigator* navigator_; 97 98#if defined(OS_WIN) 99 bool is_pdf_reader_up_to_date_; 100#endif 101 102 DISALLOW_COPY_AND_ASSIGN(DownloadShelfContextMenu); 103}; 104 105#endif // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SHELF_CONTEXT_MENU_H_ 106