Initial Implementation of Download Notification

chrome/browser/download/notification/download_notification_item.cc

Index: chrome/browser/download/notification/download_notification_item.cc
diff --git a/chrome/browser/download/notification/download_notification_item.cc b/chrome/browser/download/notification/download_notification_item.cc
new file mode 100644
index 0000000000000000000000000000000000000000..df287a520934e89ae19d8c2c7a6f2125c7fd19d1
--- /dev/null
+++ b/chrome/browser/download/notification/download_notification_item.cc
@@ -0,0 +1,384 @@
+// 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.
+
+#include "chrome/browser/download/notification/download_notification_item.h"
+
+#include "base/strings/string_number_conversions.h"
+#include "chrome/browser/download/download_crx_util.h"
+#include "chrome/browser/download/download_item_model.h"
+#include "chrome/grit/chromium_strings.h"
+#include "chrome/grit/generated_resources.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/browser/download_item.h"
+#include "grit/theme_resources.h"
+#include "ui/base/l10n/l10n_util.h"
+#include "ui/base/resource/resource_bundle.h"
+#include "ui/message_center/message_center.h"
+#include "ui/message_center/notification.h"
+#include "ui/message_center/notification_delegate.h"
+
+using message_center::Notification;
+
+namespace {
+
+const char kDownloadNotificationNotifierId[] =
+    "chrome://settings/display/notification/id-notifier";
+const char kDownloadNotificationIdBase[] =
+    "chrome://settings/display/notification/id-";
+
+}  // anonymous namespace
+
+DownloadNotificationItem::DownloadNotificationItem(
+    content::DownloadItem* item,
+    Delegate* delegate) :
+    openable_(false),
+    downloading_(false),
+    image_resource_id_(0),
+    item_(item),
+    delegate_(delegate) {
+  item->AddObserver(this);
+
+  message_center_ = message_center::MessageCenter::Get();
+
+  ui::ResourceBundle& bundle = ui::ResourceBundle::GetSharedInstance();
+
+  const base::string16 timeout_message =
+      l10n_util::GetStringUTF16(IDS_DOWNLOAD_STATUS_CRX_INSTALL_RUNNING);
+  const base::string16 message =
+      l10n_util::GetStringUTF16(IDS_PROMPT_MALICIOUS_DOWNLOAD_URL);
+
+  std::string id(kDownloadNotificationIdBase);
+  id += base::UintToString(item->GetId());
+
+  message_center::RichNotificationData data;
+  notification_.reset(new Notification(
+      message_center::NOTIFICATION_TYPE_PROGRESS,
+      id,
+      message,
+      timeout_message,
+      bundle.GetImageNamed(IDR_DOWNLOAD_NOTIFICATION_DOWNLOADING),
+      base::string16() /* display_source */,
+      message_center::NotifierId(
+          message_center::NotifierId::SYSTEM_COMPONENT,
+          kDownloadNotificationNotifierId),
+      data,
+      this));
+
+  notification_->set_progress(0);
+  notification_->set_never_timeout(false);
+
+  SetNotificationData();
+
+  scoped_ptr<Notification> notification(new Notification(*notification_));
+  message_center_->AddNotification(notification.Pass());
+}
+
+DownloadNotificationItem::~DownloadNotificationItem() {
+}
+
+void DownloadNotificationItem::Close(bool by_user) {
+  bool is_popup = false;
+
+  const std::string id = notification_->id();
+  message_center::NotificationList::PopupNotifications popups =
+      message_center_->GetPopupNotifications();
+  for (auto it = popups.begin(); it != popups.end(); it++) {
+    if ((*it)->id() == id) {
+      is_popup = true;
+    }
+  }
+  LOG(ERROR) << "CLOSE: " << is_popup;
+
+  if (is_popup) {
+    notification_->set_is_read(true);
+    ShowNotificationAgainSoon();
+  } else {
+    item_->RemoveObserver(this);
+    item_->Cancel(by_user);
+    delegate_->OnDownloadNotificationItemDestroying(this);
+  }
+}
+
+void DownloadNotificationItem::ShowNotificationAgainSoon() {
+  content::BrowserThread::PostTask(
+      content::BrowserThread::UI,
+      FROM_HERE,
+      base::Bind(&DownloadNotificationItem::ShowNotificationAgain, this));
+}
+
+void DownloadNotificationItem::ShowNotificationAgain() {
+  scoped_ptr<Notification> notification(new Notification(*notification_));
+  message_center_->AddNotification(notification.Pass());
+  message_center_->MarkSinglePopupAsShown(
+      notification_->id(), notification_->IsRead());
+}
+
+void DownloadNotificationItem::Click() {
+  if (openable_)
+    item_->OpenDownload();
+  if (item_->IsDone()) {
+    message_center_->RemoveNotification(notification_->id(), true);
+  }
+}
+
+bool DownloadNotificationItem::HasClickedListener() {
+  return true;
+}
+
+void DownloadNotificationItem::ButtonClick(int button_index) {
+  if (button_index < 0 ||
+      static_cast<size_t>(button_index) >= button_actions_->size()) {
+    // Out of boundary. Do nothing.
+    return;
+  }
+
+  switch (button_actions_->at(button_index)) {
+    case OPEN_WHEN_COMPLETE:
+      NOTREACHED();
+      break;
+    case PAUSE:
+      item_->Pause();
+      break;
+    case RESUME:
+    case RETRY:
+      item_->Resume();
+      break;
+    case SHOW_IN_FOLDER:
+      item_->ShowDownloadInShell();
+      break;
+    case DISCARD:
+      item_->Cancel(true);
+      break;
+    default:
+      break;
+  }
+}
+
+// DownloadItem::Observer methods
+void DownloadNotificationItem::OnDownloadUpdated(content::DownloadItem* item) {
+  DCHECK_EQ(item, item_);
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+
+  SetNotificationData();
+
+  // Updates notification.
+  scoped_ptr<Notification> notification(new Notification(*notification_));
+  message_center_->UpdateNotification(
+      notification->id(),
+      notification.Pass());
+}
+
+void DownloadNotificationItem::SetNotificationData() {
+  DownloadItemModel* model = new DownloadItemModel(item_);
+
+  if (!downloading_) {
+    if (item_->GetState() != content::DownloadItem::IN_PROGRESS)
+      delegate_->OnDownloadStarted(this);
+  } else {
+    if (item_->GetState() != content::DownloadItem::IN_PROGRESS)
+      delegate_->OnDownloadStopped(this);
+  }
+
+  if (item_->IsDangerous()) {
+    notification_->set_type(message_center::NOTIFICATION_TYPE_SIMPLE);
+    notification_->set_title(GetTitle());
+    notification_->set_message(GetWarningTextLong());
+
+    // Show icon.
+
+    switch (item_->GetDangerType()) {
+      case content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS:
+      case content::DOWNLOAD_DANGER_TYPE_USER_VALIDATED:
+        break;
+      default:
+        SetImageToNotification(IDR_DOWNLOAD_NOTIFICATION_MALICIOUS);
+        break;
+    }
+  } else {
+    notification_->set_title(GetTitle());
+    notification_->set_message(model->GetStatusText());
+
+    switch (item_->GetState()) {
+      case content::DownloadItem::IN_PROGRESS:
+        notification_->set_type(message_center::NOTIFICATION_TYPE_PROGRESS);
+        notification_->set_progress(item_->PercentComplete());
+        SetImageToNotification(IDR_DOWNLOAD_NOTIFICATION_DOWNLOADING);
+        break;
+      case content::DownloadItem::COMPLETE:
+        notification_->set_type(message_center::NOTIFICATION_TYPE_SIMPLE);
+        SetImageToNotification(IDR_DOWNLOAD_NOTIFICATION_DOWNLOADING);
+
+        // TODO(yoshiki): Popup a notification again.
+        break;
+      case content::DownloadItem::CANCELLED:
+        notification_->set_type(message_center::NOTIFICATION_TYPE_SIMPLE);
+        SetImageToNotification(IDR_DOWNLOAD_NOTIFICATION_WARNING);
+        break;
+      case content::DownloadItem::INTERRUPTED:
+        notification_->set_type(message_center::NOTIFICATION_TYPE_SIMPLE);
+        SetImageToNotification(IDR_DOWNLOAD_NOTIFICATION_WARNING);
+
+        // TODO(yoshiki): Popup a notification again.
+        break;
+      case content::DownloadItem::MAX_DOWNLOAD_STATE:  // sentinel
+        NOTREACHED();
+    }
+  }
+
+  std::vector<message_center::ButtonInfo> notification_actions;
+  scoped_ptr<std::vector<DownloadCommands> >
+      actions(GetPossibleActions().Pass());
+
+  openable_ = false;
+  button_actions_.reset(new std::vector<DownloadCommands>);
+  ResourceBundle& bundle = ResourceBundle::GetSharedInstance();
+  for (auto it = actions->begin(); it != actions->end(); it++) {
+    switch (*it) {
+      case OPEN_WHEN_COMPLETE:
+        openable_ = true;
+        break;
+      default:
+        button_actions_->push_back(*it);
+        message_center::ButtonInfo button_info =
+            message_center::ButtonInfo(GetCommandText(*it));
+        button_info.icon = bundle.GetImageNamed(GetCommandIconId(*it));
+        notification_actions.push_back(button_info);
+        break;
+    }
+  }
+  notification_->set_buttons(notification_actions);
+
+  if (item_->IsDone()) {
+    // TODO(yoshiki): If the downloaded file is an image, show the thumbnail.
+  }
+}
+
+void DownloadNotificationItem::OnDownloadOpened(content::DownloadItem* item) {
+  DCHECK_EQ(item, item_);
+  // Do nothing.
+}
+
+void DownloadNotificationItem::OnDownloadDestroyed(
+    content::DownloadItem* item) {
+  DCHECK_EQ(item, item_);
+  LOG(ERROR) << "ONDOWNLOADDESTROYED";
+
+  // TODO(yoshiki): close the notification if necessary.
+
+  delegate_->OnDownloadNotificationItemDestroying(this);
+}
+
+void DownloadNotificationItem::SetImageToNotification(int resource_id) {
+  if (image_resource_id_ == resource_id)
+    return;
+  ui::ResourceBundle& bundle = ui::ResourceBundle::GetSharedInstance();
+  image_resource_id_ = resource_id;
+  notification_->set_small_image(bundle.GetImageNamed(image_resource_id_));
+}
+
+scoped_ptr<std::vector<DownloadCommands> >
+DownloadNotificationItem::GetPossibleActions() const {
+  scoped_ptr<std::vector<DownloadCommands> >
+      actions(new std::vector<DownloadCommands>());
+
+  if (item_->IsDangerous()) {
+    actions->push_back(DISCARD);
+    actions->push_back(SHOW_IN_FOLDER);
+    return actions.Pass();
+  }
+
+  switch (item_->GetState()) {
+    case content::DownloadItem::IN_PROGRESS:
+      actions->push_back(OPEN_WHEN_COMPLETE);
+      if (!item_->IsPaused())
+        actions->push_back(PAUSE);
+      else
+        actions->push_back(RESUME);
+      break;
+    case content::DownloadItem::CANCELLED:
+    case content::DownloadItem::INTERRUPTED:
+      actions->push_back(RETRY);
+      break;
+    case content::DownloadItem::COMPLETE:
+      actions->push_back(OPEN_WHEN_COMPLETE);
+      actions->push_back(SHOW_IN_FOLDER);
+      break;
+    case content::DownloadItem::MAX_DOWNLOAD_STATE:
+      NOTREACHED();
+  }
+  return actions.Pass();
+}
+
+base::string16 DownloadNotificationItem::GetTitle() const {
+  base::string16 title_text;
+  base::string16 file_name =
+      item_->GetFileNameToReportUser().LossyDisplayName();
+  switch (item_->GetState()) {
+    case content::DownloadItem::IN_PROGRESS:
+      title_text = l10n_util::GetStringFUTF16(
+          IDS_DOWNLOAD_STATUS_IN_PROGRESS_TITLE,
+          file_name);
+      break;
+    case content::DownloadItem::COMPLETE:
+      title_text = l10n_util::GetStringFUTF16(
+          IDS_DOWNLOAD_STATUS_DOWNLOADED_TITLE,
+          file_name);
+    case content::DownloadItem::INTERRUPTED:
+      title_text = l10n_util::GetStringFUTF16(
+          IDS_DOWNLOAD_STATUS_DOWNLOADED_TITLE,
+          file_name);
+      break;
+    case content::DownloadItem::CANCELLED:
+      title_text = l10n_util::GetStringFUTF16(
+          IDS_DOWNLOAD_STATUS_DOWNLOAD_FAILED_TITLE,
+          file_name);
+      break;
+    case content::DownloadItem::MAX_DOWNLOAD_STATE:
+      NOTREACHED();
+  }
+  return title_text;
+}
+
+base::string16 DownloadNotificationItem::GetWarningTextLong() const {
+  // Should only be called if IsDangerous().
+  DCHECK(item_->IsDangerous());
+  base::string16 elided_filename =
+      item_->GetFileNameToReportUser().LossyDisplayName();
+  switch (item_->GetDangerType()) {
+    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_URL: {
+      return l10n_util::GetStringUTF16(IDS_PROMPT_MALICIOUS_DOWNLOAD_URL);
+    }
+    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_FILE: {
+      if (download_crx_util::IsExtensionDownload(*item_)) {
+        return l10n_util::GetStringUTF16(
+            IDS_PROMPT_DANGEROUS_DOWNLOAD_EXTENSION);
+      } else {
+        return l10n_util::GetStringFUTF16(IDS_PROMPT_DANGEROUS_DOWNLOAD,
+                                          elided_filename);
+      }
+    }
+    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_CONTENT:
+    case content::DOWNLOAD_DANGER_TYPE_DANGEROUS_HOST: {
+      return l10n_util::GetStringFUTF16(IDS_PROMPT_MALICIOUS_DOWNLOAD_CONTENT,
+                                        elided_filename);
+    }
+    case content::DOWNLOAD_DANGER_TYPE_UNCOMMON_CONTENT: {
+      return l10n_util::GetStringFUTF16(IDS_PROMPT_UNCOMMON_DOWNLOAD_CONTENT,
+                                        elided_filename);
+    }
+    case content::DOWNLOAD_DANGER_TYPE_POTENTIALLY_UNWANTED: {
+      return l10n_util::GetStringFUTF16(
+          IDS_PROMPT_DOWNLOAD_CHANGES_SETTINGS, elided_filename);
+    }
+    case content::DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS:
+    case content::DOWNLOAD_DANGER_TYPE_MAYBE_DANGEROUS_CONTENT:
+    case content::DOWNLOAD_DANGER_TYPE_USER_VALIDATED:
+    case content::DOWNLOAD_DANGER_TYPE_MAX: {
+      break;
+    }
+  }
+  NOTREACHED();
+  return base::string16();
+}
+