src/chrome/browser/chromeos/drive/download_handler.h

   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_CHROMEOS_DRIVE_DOWNLOAD_HANDLER_H_
   6 #define CHROME_BROWSER_CHROMEOS_DRIVE_DOWNLOAD_HANDLER_H_
   7
   8 #include "base/callback_forward.h"
   9 #include "base/memory/weak_ptr.h"
  10 #include "chrome/browser/chromeos/drive/file_errors.h"
  11 #include "chrome/browser/download/all_download_item_notifier.h"
  12 #include "content/public/browser/download_manager_delegate.h"
  13
  14 class Profile;
  15
  16 namespace content {
  17 class DownloadItem;
  18 class DownloadManager;
  19 }
  20
  21 namespace drive {
  22
  23 class FileSystemInterface;
  24 class ResourceEntry;
  25
  26 // Observes downloads to temporary local drive folder. Schedules these
  27 // downloads for upload to drive service.
  28 class DownloadHandler : public AllDownloadItemNotifier::Observer {
  29  public:
  30   explicit DownloadHandler(FileSystemInterface* file_system);
  31   virtual ~DownloadHandler();
  32
  33   // Utility method to get DownloadHandler with profile.
  34   static DownloadHandler* GetForProfile(Profile* profile);
  35
  36   // Become an observer of DownloadManager.
  37   void Initialize(content::DownloadManager* download_manager,
  38                   const base::FilePath& drive_tmp_download_path);
  39
  40   // In addition to the DownloadManager passed to Initialize(), observe another
  41   // download manager. This should be called only for the DownloadManager of the
  42   // incognito version of the profile where |file_system_| resides.
  43   void ObserveIncognitoDownloadManager(
  44       content::DownloadManager* download_manager);
  45
  46   // Callback used to return results from SubstituteDriveDownloadPath.
  47   // TODO(hashimoto): Report error with a FileError. crbug.com/171345
  48   typedef base::Callback<void(const base::FilePath&)>
  49       SubstituteDriveDownloadPathCallback;
  50
  51   void SubstituteDriveDownloadPath(
  52       const base::FilePath& drive_path,
  53       content::DownloadItem* download,
  54       const SubstituteDriveDownloadPathCallback& callback);
  55
  56   // Sets drive path, for example, '/special/drive/MyFolder/MyFile',
  57   // to external data in |download|. Also sets display name and
  58   // makes |download| a temporary.
  59   void SetDownloadParams(const base::FilePath& drive_path,
  60                          content::DownloadItem* download);
  61
  62   // Gets the target drive path from external data in |download|.
  63   base::FilePath GetTargetPath(const content::DownloadItem* download);
  64
  65   // Gets the downloaded drive cache file path from external data in |download|.
  66   base::FilePath GetCacheFilePath(const content::DownloadItem* download);
  67
  68   // Checks if there is a Drive upload associated with |download|
  69   bool IsDriveDownload(const content::DownloadItem* download);
  70
  71   // Checks a file corresponding to the download item exists in Drive.
  72   void CheckForFileExistence(
  73       const content::DownloadItem* download,
  74       const content::CheckForFileExistenceCallback& callback);
  75
  76  private:
  77   // AllDownloadItemNotifier::Observer overrides:
  78   virtual void OnDownloadCreated(content::DownloadManager* manager,
  79                                  content::DownloadItem* download) override;
  80   virtual void OnDownloadUpdated(content::DownloadManager* manager,
  81                                  content::DownloadItem* download) override;
  82
  83   // Removes the download.
  84   void RemoveDownload(void* manager_id, int id);
  85
  86   // Callback for FileSystem::CreateDirectory().
  87   // Used to implement SubstituteDriveDownloadPath().
  88   void OnCreateDirectory(const SubstituteDriveDownloadPathCallback& callback,
  89                          FileError error);
  90
  91   // Starts the upload of a downloaded/downloading file.
  92   void UploadDownloadItem(content::DownloadManager* manager,
  93                           content::DownloadItem* download);
  94
  95   // Sets |cache_file_path| as user data of the download item specified by |id|.
  96   void SetCacheFilePath(void* manager_id,
  97                         int id,
  98                         const base::FilePath* cache_file_path,
  99                         FileError error);
 100
 101   // Gets a download manager, given a |manager_id| casted from the pointer to
 102   // the manager. This is used to validate the manager that may be deleted while
 103   // asynchronous task posting. Returns NULL if the manager is already gone.
 104   content::DownloadManager* GetDownloadManager(void* manager_id);
 105
 106   FileSystemInterface* file_system_;  // Owned by DriveIntegrationService.
 107   // Observe the DownloadManager for new downloads.
 108   scoped_ptr<AllDownloadItemNotifier> notifier_;
 109   scoped_ptr<AllDownloadItemNotifier> notifier_incognito_;
 110
 111   // Temporary download location directory.
 112   base::FilePath drive_tmp_download_path_;
 113
 114   // Note: This should remain the last member so it'll be destroyed and
 115   // invalidate its weak pointers before any other members are destroyed.
 116   base::WeakPtrFactory<DownloadHandler> weak_ptr_factory_;
 117
 118   DISALLOW_COPY_AND_ASSIGN(DownloadHandler);
 119 };
 120
 121 }  // namespace drive
 122
 123 #endif  // CHROME_BROWSER_CHROMEOS_DRIVE_DOWNLOAD_HANDLER_H_