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_PREFS_H_ 6 #define CHROME_BROWSER_DOWNLOAD_DOWNLOAD_PREFS_H_ 7 8 #include <set> 9 10 #include "base/files/file_path.h" 11 #include "base/prefs/pref_member.h" 12 13 class PrefService; 14 class Profile; 15 16 namespace content { 17 class BrowserContext; 18 class DownloadManager; 19 } 20 21 namespace user_prefs { 22 class PrefRegistrySyncable; 23 } 24 25 // Stores all download-related preferences. 26 class DownloadPrefs { 27 public: 28 explicit DownloadPrefs(Profile* profile); 29 ~DownloadPrefs(); 30 31 static void RegisterProfilePrefs(user_prefs::PrefRegistrySyncable* registry); 32 33 // Returns the default download directory. 34 static const base::FilePath& GetDefaultDownloadDirectory(); 35 36 // Returns the DownloadPrefs corresponding to the given DownloadManager 37 // or BrowserContext. 38 static DownloadPrefs* FromDownloadManager( 39 content::DownloadManager* download_manager); 40 static DownloadPrefs* FromBrowserContext( 41 content::BrowserContext* browser_context); 42 43 base::FilePath DownloadPath() const; 44 void SetDownloadPath(const base::FilePath& path); 45 base::FilePath SaveFilePath() const; 46 void SetSaveFilePath(const base::FilePath& path); save_file_type()47 int save_file_type() const { return *save_file_type_; } 48 void SetSaveFileType(int type); 49 50 // Returns true if the prompt_for_download preference has been set and the 51 // download location is not managed (which means the user shouldn't be able 52 // to choose another download location). 53 bool PromptForDownload() const; 54 55 // Returns true if the download path preference is managed. 56 bool IsDownloadPathManaged() const; 57 58 // Returns true if there is at least one file extension registered 59 // for auto-open. 60 bool IsAutoOpenUsed() const; 61 62 // Returns true if |path| should be opened automatically based on 63 // |path.Extension()|. 64 bool IsAutoOpenEnabledBasedOnExtension(const base::FilePath& path) const; 65 66 // Enables auto-open based on file extension. Returns true on success. 67 // TODO(phajdan.jr): Add WARN_UNUSED_RESULT here. 68 bool EnableAutoOpenBasedOnExtension(const base::FilePath& file_name); 69 70 // Disables auto-open based on file extension. 71 void DisableAutoOpenBasedOnExtension(const base::FilePath& file_name); 72 73 void ResetAutoOpen(); 74 75 private: 76 void SaveAutoOpenState(); 77 78 Profile* profile_; 79 80 BooleanPrefMember prompt_for_download_; 81 FilePathPrefMember download_path_; 82 FilePathPrefMember save_file_path_; 83 IntegerPrefMember save_file_type_; 84 85 // Set of file extensions to open at download completion. 86 struct AutoOpenCompareFunctor { 87 bool operator()(const base::FilePath::StringType& a, 88 const base::FilePath::StringType& b) const; 89 }; 90 typedef std::set<base::FilePath::StringType, 91 AutoOpenCompareFunctor> AutoOpenSet; 92 AutoOpenSet auto_open_; 93 94 DISALLOW_COPY_AND_ASSIGN(DownloadPrefs); 95 }; 96 97 #endif // CHROME_BROWSER_DOWNLOAD_DOWNLOAD_PREFS_H_ 98