| // 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. |
| |
| #ifndef CONTENT_SHELL_BROWSER_SHELL_DOWNLOAD_MANAGER_DELEGATE_H_ |
| #define CONTENT_SHELL_BROWSER_SHELL_DOWNLOAD_MANAGER_DELEGATE_H_ |
| |
| #include "base/compiler_specific.h" |
| #include "base/memory/ref_counted.h" |
| #include "content/public/browser/download_manager_delegate.h" |
| |
| namespace content { |
| |
| class DownloadManager; |
| |
| class ShellDownloadManagerDelegate |
| : public DownloadManagerDelegate, |
| public base::RefCountedThreadSafe<ShellDownloadManagerDelegate> { |
| public: |
| ShellDownloadManagerDelegate(); |
| |
| void SetDownloadManager(DownloadManager* manager); |
| |
| virtual void Shutdown() OVERRIDE; |
| virtual bool DetermineDownloadTarget( |
| DownloadItem* download, |
| const DownloadTargetCallback& callback) OVERRIDE; |
| virtual bool ShouldOpenDownload( |
| DownloadItem* item, |
| const DownloadOpenDelayedCallback& callback) OVERRIDE; |
| virtual void GetNextId(const DownloadIdCallback& callback) OVERRIDE; |
| |
| // Inhibits prompting and sets the default download path. |
| void SetDownloadBehaviorForTesting( |
| const base::FilePath& default_download_path); |
| |
| protected: |
| // To allow subclasses for testing. |
| virtual ~ShellDownloadManagerDelegate(); |
| |
| private: |
| friend class base::RefCountedThreadSafe<ShellDownloadManagerDelegate>; |
| |
| |
| void GenerateFilename(uint32 download_id, |
| const DownloadTargetCallback& callback, |
| const base::FilePath& generated_name, |
| const base::FilePath& suggested_directory); |
| void OnDownloadPathGenerated(uint32 download_id, |
| const DownloadTargetCallback& callback, |
| const base::FilePath& suggested_path); |
| void ChooseDownloadPath(uint32 download_id, |
| const DownloadTargetCallback& callback, |
| const base::FilePath& suggested_path); |
| |
| DownloadManager* download_manager_; |
| base::FilePath default_download_path_; |
| bool suppress_prompting_; |
| |
| DISALLOW_COPY_AND_ASSIGN(ShellDownloadManagerDelegate); |
| }; |
| |
| } // namespace content |
| |
| #endif // CONTENT_SHELL_BROWSER_SHELL_DOWNLOAD_MANAGER_DELEGATE_H_ |