[Download Notification] Show preview if downloaded file is image

chrome/browser/download/notification/download_notification_item.h

 // Copyright 2015 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_NOTIFICATION_DOWNLOAD_NOTIFICATION_ITEM_H_
 #define CHROME_BROWSER_DOWNLOAD_NOTIFICATION_DOWNLOAD_NOTIFICATION_ITEM_H_
 
 #include "chrome/browser/download/notification/download_notification_item.h"
 
 #include "base/strings/string_number_conversions.h"
 #include "chrome/browser/download/download_commands.h"
+#include "chrome/browser/image_decoder.h"
 #include "chrome/browser/notifications/notification.h"
 #include "chrome/browser/notifications/notification_delegate.h"
 #include "chrome/browser/notifications/notification_test_util.h"
 #include "content/public/browser/download_item.h"
 #include "grit/theme_resources.h"
+#include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/message_center/message_center.h"
 #include "ui/message_center/message_center_observer.h"
 
 namespace test {
 class DownloadNotificationItemTest;
 }
 
-class DownloadNotificationItem : public content::DownloadItem::Observer {
+class DownloadNotificationItem : public content::DownloadItem::Observer,
+                                 public ImageDecoder::ImageRequest {
  public:
   class Delegate {
    public:
     virtual void OnCreated(DownloadNotificationItem* item) = 0;
     virtual void OnDownloadStarted(DownloadNotificationItem* item) = 0;
     virtual void OnDownloadStopped(DownloadNotificationItem* item) = 0;
     virtual void OnDownloadRemoved(DownloadNotificationItem* item) = 0;
   };
 
   static const char kDownloadNotificationOrigin[];
 
   DownloadNotificationItem(content::DownloadItem* item,
                            Profile* profile,
                            Delegate* delegate);
 
   ~DownloadNotificationItem() override;
 
  private:
   friend class test::DownloadNotificationItemTest;
 
   enum NotificationUpdateType {
     ADD_NEW,
     UPDATE_EXISTING
   };
 
+  enum ImageDecodeStatus {
+    NOT_STARTED,
+    IN_PROGRESS,
+    DONE,
+    FAILED,
+    NOT_IMAGE
+  };
+
   class NotificationWatcher : public NotificationDelegate {
    public:
     explicit NotificationWatcher(DownloadNotificationItem* item);
 
     // NotificationDelegate overrides:
     void Close(bool by_user) override;
     void Click() override;
     bool HasClickedListener() override;
     void ButtonClick(int button_index) override;
     std::string id() const override;
 
    private:
     ~NotificationWatcher() override;
 
     DownloadNotificationItem* item_;
   };
 
   // For testing. This is set by SetStubNotificationUIManagerForTesting method.
   static StubNotificationUIManager* stub_notification_ui_manager_for_testing_;
 
   // Replaces the NotificationUIManager for tests.
   static void SetStubNotificationUIManagerForTesting(
       StubNotificationUIManager* stub_notification_ui_manager) {
     stub_notification_ui_manager_for_testing_ = stub_notification_ui_manager;
   }
 
   // Methods called from NotificationWatcher.
   void OnNotificationClick();
   void OnNotificationButtonClick(int button_index);
+  void OnNotificationClose(bool by_user);
 
   // DownloadItem::Observer overrides:
   void OnDownloadUpdated(content::DownloadItem* item) override;
   void OnDownloadOpened(content::DownloadItem* item) override;
   void OnDownloadRemoved(content::DownloadItem* item) override;
   void OnDownloadDestroyed(content::DownloadItem* item) override;
 
   void CloseNotificationByUser();
   void CloseNotificationByNonUser();
   void UpdateNotificationData(NotificationUpdateType type);
-  void SetNotificationImage(int resource_id);
+
+  // Set icon of the notification.
+  void SetNotificationIcon(int resource_id);
+
+  // Set preview image of the notification. Must be called on IO thread.
+  void OnImageLoaded(std::string image_data);

[Lei Zhang, 2015/06/12 07:26:37]

pass by const ref.

[yoshiki, 2015/06/12 08:00:20]

Done.

+
+  // ImageDecoder::ImageRequest overrides:
+  void OnImageDecoded(const SkBitmap& decoded_image) override;
+  void OnDecodeImageFailed() override;
 
   NotificationUIManager* notification_ui_manager() const;
 
   // Returns a short one-line status string for the download.
   base::string16 GetTitle() const;
 
   // Returns a short one-line status string for a download command.
   base::string16 GetCommandLabel(DownloadCommands::Command command) const;
 
   // Get the warning test to notify a dangerous download. Should only be called
   // if IsDangerous() is true.
   base::string16 GetWarningText() const;
 
   Browser* GetBrowser();
 
   // Returns the list of possible extra (all except the default) actions.
   scoped_ptr<std::vector<DownloadCommands::Command>> GetExtraActions() const;
 
   int image_resource_id_ = 0;
   content::DownloadItem::DownloadState previous_download_state_ =
       content::DownloadItem::MAX_DOWNLOAD_STATE;  // As uninitialized state
   Profile* profile_;
   scoped_refptr<NotificationWatcher> watcher_;
   scoped_ptr<Notification> notification_;
   content::DownloadItem* item_;
   scoped_ptr<std::vector<DownloadCommands::Command>> button_actions_;
   Delegate* const delegate_;
 
+  // Status of the preview image decode.
+  ImageDecodeStatus image_decode_status_ = NOT_STARTED;
+
+  base::WeakPtrFactory<DownloadNotificationItem> weak_factory_;
+
   DISALLOW_COPY_AND_ASSIGN(DownloadNotificationItem);
 };
 
 #endif  // CHROME_BROWSER_DOWNLOAD_NOTIFICATION_DOWNLOAD_NOTIFICATION_ITEM_H_