Initial Implementation of Download Notification

chrome/browser/download/download_ui_controller.cc

 // Copyright 2013 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/download_ui_controller.h"
 
 #include "base/callback.h"
+#include "base/command_line.h"
 #include "base/stl_util.h"
 #include "chrome/browser/download/download_item_model.h"
+#include "chrome/browser/download/notification/download_notification_manager.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/browser_tabstrip.h"
+#include "chrome/common/chrome_switches.h"
 #include "content/public/browser/download_item.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/browser/web_contents_delegate.h"
 
 #if defined(OS_ANDROID)
 #include "content/public/browser/android/download_controller_android.h"
 #else
 #include "chrome/browser/profiles/profile.h"
 #endif
 
 namespace {
 
-// DefaultUIControllerDelegate{Android,} is used when a DownloadUIController is
+// DownloadShelfUIControllerDelegate{Android,} is used when a
+// DownloadUIController is
 // constructed without specifying an explicit Delegate.
 #if defined(OS_ANDROID)
 
-class DefaultUIControllerDelegateAndroid
-    : public DownloadUIController::Delegate {
+class AndroidUIControllerDelegate : public DownloadUIController::Delegate {
  public:
-  DefaultUIControllerDelegateAndroid() {}
-  ~DefaultUIControllerDelegateAndroid() override {}
+  AndroidUIControllerDelegate() {}
+  ~AndroidUIControllerDelegate() override {}
 
  private:
   // DownloadUIController::Delegate
   void OnNewDownloadReady(content::DownloadItem* item) override;
 };
 
-void DefaultUIControllerDelegateAndroid::OnNewDownloadReady(
+void AndroidUIControllerDelegate::OnNewDownloadReady(
     content::DownloadItem* item) {
   // The Android DownloadController is only interested in IN_PROGRESS downloads.
   // Ones which are INTERRUPTED etc. can't be handed over to the Android
   // DownloadManager.
   if (item->GetState() != content::DownloadItem::IN_PROGRESS)
     return;
 
   // GET downloads without authentication are delegated to the Android
   // DownloadManager. Chrome is responsible for the rest.  See
   // InterceptDownloadResourceThrottle::ProcessDownloadRequest().
   content::DownloadControllerAndroid::Get()->OnDownloadStarted(item);
 }
 
 #else  // OS_ANDROID
 
-class DefaultUIControllerDelegate : public DownloadUIController::Delegate {
+class DownloadShelfUIControllerDelegate
+    : public DownloadUIController::Delegate {
  public:
-  // |profile| is required to outlive DefaultUIControllerDelegate.
-  explicit DefaultUIControllerDelegate(Profile* profile)
+  // |profile| is required to outlive DownloadShelfUIControllerDelegate.
+  explicit DownloadShelfUIControllerDelegate(Profile* profile)
       : profile_(profile) {}
-  ~DefaultUIControllerDelegate() override {}
+  ~DownloadShelfUIControllerDelegate() override {}
 
  private:
   // DownloadUIController::Delegate
   void OnNewDownloadReady(content::DownloadItem* item) override;
 
   Profile* profile_;
 };
 
-void DefaultUIControllerDelegate::OnNewDownloadReady(
+void DownloadShelfUIControllerDelegate::OnNewDownloadReady(
     content::DownloadItem* item) {
   content::WebContents* web_contents = item->GetWebContents();
   Browser* browser =
       web_contents ? chrome::FindBrowserWithWebContents(web_contents) : NULL;
 
   // As a last resort, use the last active browser for this profile. Not ideal,
   // but better than not showing the download at all.
   if (browser == NULL) {
     browser = chrome::FindLastActiveWithProfile(profile_,
                                                 chrome::GetActiveDesktop());
   }
 
   if (browser)
     browser->ShowDownload(item);
 }
 
 #endif  // !OS_ANDROID
 
 } // namespace
 
 DownloadUIController::Delegate::~Delegate() {
 }
 
 DownloadUIController::DownloadUIController(content::DownloadManager* manager,
                                            scoped_ptr<Delegate> delegate)
     : download_notifier_(manager, this),
       delegate_(delegate.Pass()) {
   if (!delegate_) {
 #if defined(OS_ANDROID)
-    delegate_.reset(new DefaultUIControllerDelegateAndroid());
+    delegate_.reset(new AndroidUIControllerDelegate());
 #else
     // The delegate should not be invoked after the profile has gone away. This
     // should be the case since DownloadUIController is owned by
     // DownloadService, which in turn is a profile keyed service.
-    delegate_.reset(new DefaultUIControllerDelegate(
-        Profile::FromBrowserContext(manager->GetBrowserContext())));
+    if (base::CommandLine::ForCurrentProcess()->HasSwitch(
+            switches::kEnableDownloadNotification)) {
+      delegate_.reset(new DownloadNotificationManager(
+          Profile::FromBrowserContext(manager->GetBrowserContext())));
+    } else {
+      delegate_.reset(new DownloadShelfUIControllerDelegate(
+          Profile::FromBrowserContext(manager->GetBrowserContext())));
+    }
 #endif
   }
 }
 
 DownloadUIController::~DownloadUIController() {
 }
 
 void DownloadUIController::OnDownloadCreated(content::DownloadManager* manager,
                                              content::DownloadItem* item) {
   // SavePackage downloads are created in a state where they can be shown in the
@@ skipping 10 matching lines @@
   if (item_model.WasUINotified() || !item_model.ShouldNotifyUI())
     return;
 
   // Wait until the target path is determined.
   if (item->GetTargetFilePath().empty())
     return;
 
   DownloadItemModel(item).SetWasUINotified(true);
   delegate_->OnNewDownloadReady(item);
 }