Home | History | Annotate | Download | only in browser 
      1 // Copyright 2013 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 CONTENT_SHELL_BROWSER_SHELL_DOWNLOAD_MANAGER_DELEGATE_H_
      6 #define CONTENT_SHELL_BROWSER_SHELL_DOWNLOAD_MANAGER_DELEGATE_H_
      7 
      8 #include "base/callback_forward.h"
      9 #include "base/compiler_specific.h"
     10 #include "base/memory/weak_ptr.h"
     11 #include "content/public/browser/download_manager_delegate.h"
     12 
     13 namespace content {
     14 
     15 class DownloadManager;
     16 
     17 class ShellDownloadManagerDelegate : public DownloadManagerDelegate {
     18  public:
     19   ShellDownloadManagerDelegate();
     20   virtual ~ShellDownloadManagerDelegate();
     21 
     22   void SetDownloadManager(DownloadManager* manager);
     23 
     24   virtual void Shutdown() OVERRIDE;
     25   virtual bool DetermineDownloadTarget(
     26       DownloadItem* download,
     27       const DownloadTargetCallback& callback) OVERRIDE;
     28   virtual bool ShouldOpenDownload(
     29       DownloadItem* item,
     30       const DownloadOpenDelayedCallback& callback) OVERRIDE;
     31   virtual void GetNextId(const DownloadIdCallback& callback) OVERRIDE;
     32 
     33   // Inhibits prompting and sets the default download path.
     34   void SetDownloadBehaviorForTesting(
     35       const base::FilePath& default_download_path);
     36 
     37  private:
     38   friend class base::RefCountedThreadSafe<ShellDownloadManagerDelegate>;
     39 
     40   typedef base::Callback<void(const base::FilePath&)>
     41       FilenameDeterminedCallback;
     42 
     43   static void GenerateFilename(const GURL& url,
     44                                const std::string& content_disposition,
     45                                const std::string& suggested_filename,
     46                                const std::string& mime_type,
     47                                const base::FilePath& suggested_directory,
     48                                const FilenameDeterminedCallback& callback);
     49   void OnDownloadPathGenerated(uint32 download_id,
     50                                const DownloadTargetCallback& callback,
     51                                const base::FilePath& suggested_path);
     52   void ChooseDownloadPath(uint32 download_id,
     53                           const DownloadTargetCallback& callback,
     54                           const base::FilePath& suggested_path);
     55 
     56   DownloadManager* download_manager_;
     57   base::FilePath default_download_path_;
     58   bool suppress_prompting_;
     59   base::WeakPtrFactory<ShellDownloadManagerDelegate> weak_ptr_factory_;
     60 
     61   DISALLOW_COPY_AND_ASSIGN(ShellDownloadManagerDelegate);
     62 };
     63 
     64 }  // namespace content
     65 
     66 #endif  // CONTENT_SHELL_BROWSER_SHELL_DOWNLOAD_MANAGER_DELEGATE_H_
     67