Home | History | Annotate | Download | only in download 
      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 
     15 namespace content {
     16 class 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.
     24 class 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