Home | History | Annotate | Download | only in download 
      1 // Copyright (c) 2012 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_SERVICE_H_
      6 #define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SERVICE_H_
      7 
      8 #include "base/basictypes.h"
      9 #include "base/callback_forward.h"
     10 #include "base/memory/scoped_ptr.h"
     11 #include "components/keyed_service/core/keyed_service.h"
     12 
     13 class ChromeDownloadManagerDelegate;
     14 class DownloadHistory;
     15 class DownloadUIController;
     16 class ExtensionDownloadsEventRouter;
     17 class Profile;
     18 
     19 namespace content {
     20 class DownloadManager;
     21 }
     22 
     23 namespace extensions {
     24 class ExtensionDownloadsEventRouter;
     25 }
     26 
     27 // Owning class for ChromeDownloadManagerDelegate.
     28 class DownloadService : public KeyedService {
     29  public:
     30   explicit DownloadService(Profile* profile);
     31   virtual ~DownloadService();
     32 
     33   // Get the download manager delegate, creating it if it doesn't already exist.
     34   ChromeDownloadManagerDelegate* GetDownloadManagerDelegate();
     35 
     36   // Get the interface to the history system. Returns NULL if profile is
     37   // incognito or if the DownloadManager hasn't been created yet or if there is
     38   // no HistoryService for profile. Virtual for testing.
     39   virtual DownloadHistory* GetDownloadHistory();
     40 
     41 #if defined(ENABLE_EXTENSIONS)
     42   extensions::ExtensionDownloadsEventRouter* GetExtensionEventRouter() {
     43     return extension_event_router_.get();
     44   }
     45 #endif
     46 
     47   // Has a download manager been created?
     48   bool HasCreatedDownloadManager();
     49 
     50   // Number of non-malicious downloads associated with this instance of the
     51   // service.
     52   int NonMaliciousDownloadCount() const;
     53 
     54   // Cancels all in-progress downloads for this profile.
     55   void CancelDownloads();
     56 
     57   // Number of non-malicious downloads associated with all profiles.
     58   static int NonMaliciousDownloadCountAllProfiles();
     59 
     60   // Cancels all in-progress downloads for all profiles.
     61   static void CancelAllDownloads();
     62 
     63   // Sets the DownloadManagerDelegate associated with this object and
     64   // its DownloadManager.  Takes ownership of |delegate|, and destroys
     65   // the previous delegate.  For testing.
     66   void SetDownloadManagerDelegateForTesting(
     67       scoped_ptr<ChromeDownloadManagerDelegate> delegate);
     68 
     69   // Will be called to release references on other services as part
     70   // of Profile shutdown.
     71   virtual void Shutdown() OVERRIDE;
     72 
     73   // Returns false if at least one extension has disabled the shelf, true
     74   // otherwise.
     75   bool IsShelfEnabled();
     76 
     77  private:
     78   bool download_manager_created_;
     79   Profile* profile_;
     80 
     81   // ChromeDownloadManagerDelegate may be the target of callbacks from
     82   // the history service/DB thread and must be kept alive for those
     83   // callbacks.
     84   scoped_ptr<ChromeDownloadManagerDelegate> manager_delegate_;
     85 
     86   scoped_ptr<DownloadHistory> download_history_;
     87 
     88   // The UI controller is responsible for observing the download manager and
     89   // notifying the UI of any new downloads. Its lifetime matches that of the
     90   // associated download manager.
     91   // Note on destruction order: download_ui_ depends on download_history_ and
     92   // should be destroyed before the latter.
     93   scoped_ptr<DownloadUIController> download_ui_;
     94 
     95   // On Android, GET downloads are not handled by the DownloadManager.
     96   // Once we have extensions on android, we probably need the EventRouter
     97   // in ContentViewDownloadDelegate which knows about both GET and POST
     98   // downloads.
     99 #if defined(ENABLE_EXTENSIONS)
    100   // The ExtensionDownloadsEventRouter dispatches download creation, change, and
    101   // erase events to extensions. Like ChromeDownloadManagerDelegate, it's a
    102   // chrome-level concept and its lifetime should match DownloadManager. There
    103   // should be a separate EDER for on-record and off-record managers.
    104   // There does not appear to be a separate ExtensionSystem for on-record and
    105   // off-record profiles, so ExtensionSystem cannot own the EDER.
    106   scoped_ptr<extensions::ExtensionDownloadsEventRouter> extension_event_router_;
    107 #endif
    108 
    109   DISALLOW_COPY_AND_ASSIGN(DownloadService);
    110 };
    111 
    112 #endif  // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_SERVICE_H_
    113