Use DownloadItemModel for storing chrome/ specific UI data for DownloadItems.

chrome/browser/download/download_item_model.h

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #ifndef CHROME_BROWSER_DOWNLOAD_DOWNLOAD_ITEM_MODEL_H_
 #define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_ITEM_MODEL_H_
 
 #include <string>
 
 #include "base/basictypes.h"
@@ skipping 15 matching lines @@
 //
 // It is intended to be used as a thin wrapper around a |DownloadItem*|. As
 // such, the caller is expected to ensure that the |download| passed into the
 // constructor outlives this |DownloadItemModel|. In addition, multiple
 // DownloadItemModel objects could be wrapping the same DownloadItem.
 class DownloadItemModel {
  public:
   // Constructs a DownloadItemModel. The caller must ensure that |download|
   // outlives this object.
   explicit DownloadItemModel(content::DownloadItem* download);
-  virtual ~DownloadItemModel();
+  ~DownloadItemModel();
 
   // Cancel the task corresponding to the item.
   void CancelTask();
 
-  // Returns a short one-line status string for the download.
-  string16 GetStatusText() const;
-
   // Returns a long descriptive string for a download that's in the INTERRUPTED
   // state. For other downloads, the returned string will be empty.
   string16 GetInterruptReasonText() const;
 
+  // Returns a short one-line status string for the download.
+  string16 GetStatusText() const;
+
   // Returns a string suitable for use as a tooltip. For a regular download, the
   // tooltip is the filename. For an interrupted download, the string states the
   // filename and a short description of the reason for interruption. For
   // example:
   //    Report.pdf
   //    Network disconnected
   // |font| and |max_width| are used to elide the filename and/or interrupt
   // reason as necessary to keep the width of the tooltip text under
   // |max_width|. The tooltip will be at most 2 lines.
   string16 GetTooltipText(const gfx::Font& font, int max_width) const;
 
-  // Rough percent complete. Returns -1 if the progress is unknown.
-  int PercentComplete() const;
-
   // Get the warning text to display for a dangerous download. The |base_width|
   // is the maximum width of an embedded filename (if there is one). The metrics
   // for the filename will be based on |font|. Should only be called if
   // IsDangerous() is true.
   string16 GetWarningText(const gfx::Font& font, int base_width) const;
 
   // Get the caption text for a button for confirming a dangerous download
   // warning.
   string16 GetWarningConfirmButtonText() const;
 
+  // Get the number of bytes that has completed so far. Virtual for testing.
+  int64 GetCompletedBytes() const;
+
+  // Get the total number of bytes for this download. Should return 0 if the
+  // total size of the download is not known. Virual for testing.
+  int64 GetTotalBytes() const;
+
+  // Rough percent complete. Returns -1 if the progress is unknown.
+  int PercentComplete() const;
+
+  // Is this considered a dangerous download?
+  bool IsDangerous() const;
+
   // Is this considered a malicious download? Implies IsDangerous().
   bool IsMalicious() const;
 
-  // Is this considered a dangerous download?
-  bool IsDangerous() const;
+  // Returns |true| if this download should be displayed in the downloads shelf.
+  bool ShouldShowInShelf() const;
 
-  // Get the total number of bytes for this download. Should return 0 if the
-  // total size of the download is not known. Virual for testing.
-  virtual int64 GetTotalBytes() const;
-
-  // Get the number of bytes that has completed so far. Virtual for testing.
-  virtual int64 GetCompletedBytes() const;
+  // Change whether the download should be displayed on the downloads
+  // shelf. Setting this is only effective if the download hasn't already been
+  // displayed in the shelf.
+  void SetShouldShowInShelf(bool should_show);
 
   content::DownloadItem* download() { return download_; }
 
  private:
   // Returns a string representations of the current download progress sizes. If
   // the total size of the download is known, this string looks like: "100/200
   // MB" where the numerator is the transferred size and the denominator is the
   // total size. If the total isn't known, returns the transferred size as a
   // string (e.g.: "100 MB").
   string16 GetProgressSizesString() const;
 
   // Returns a string indicating the status of an in-progress download.
   string16 GetInProgressStatusString() const;
 
   // The DownloadItem that this model represents. Note that DownloadItemModel
   // itself shouldn't maintain any state since there can be more than one
   // DownloadItemModel in use with the same DownloadItem.
   content::DownloadItem* download_;
 
   DISALLOW_COPY_AND_ASSIGN(DownloadItemModel);
 };
 
 #endif  // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_ITEM_MODEL_H_