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_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);
     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