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