1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
// Copyright (c) 2010 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.
#include "chrome/browser/download/download_prefs.h"
#include "base/file_util.h"
#include "base/string_split.h"
#include "base/string_util.h"
#include "base/sys_string_conversions.h"
#include "base/utf_string_conversions.h"
#include "chrome/browser/download/download_util.h"
#include "chrome/browser/prefs/pref_service.h"
#include "chrome/common/pref_names.h"
DownloadPrefs::DownloadPrefs(PrefService* prefs) : prefs_(prefs) {
prompt_for_download_.Init(prefs::kPromptForDownload, prefs, NULL);
download_path_.Init(prefs::kDownloadDefaultDirectory, prefs, NULL);
// We store any file extension that should be opened automatically at
// download completion in this pref.
std::string extensions_to_open =
prefs->GetString(prefs::kDownloadExtensionsToOpen);
std::vector<std::string> extensions;
base::SplitString(extensions_to_open, ':', &extensions);
for (size_t i = 0; i < extensions.size(); ++i) {
#if defined(OS_POSIX)
FilePath path(extensions[i]);
#elif defined(OS_WIN)
FilePath path(UTF8ToWide(extensions[i]));
#endif
if (!extensions[i].empty() && !download_util::IsExecutableFile(path))
auto_open_.insert(path.value());
}
}
DownloadPrefs::~DownloadPrefs() {
SaveAutoOpenState();
}
// static
void DownloadPrefs::RegisterUserPrefs(PrefService* prefs) {
prefs->RegisterBooleanPref(prefs::kPromptForDownload, false);
prefs->RegisterStringPref(prefs::kDownloadExtensionsToOpen, "");
prefs->RegisterBooleanPref(prefs::kDownloadDirUpgraded, false);
// The default download path is userprofile\download.
const FilePath& default_download_path =
download_util::GetDefaultDownloadDirectory();
prefs->RegisterFilePathPref(prefs::kDownloadDefaultDirectory,
default_download_path);
#if defined(OS_CHROMEOS)
// Ensure that the download directory specified in the preferences exists.
BrowserThread::PostTask(
BrowserThread::FILE, FROM_HERE,
NewRunnableFunction(&file_util::CreateDirectory, default_download_path));
#endif // defined(OS_CHROMEOS)
// If the download path is dangerous we forcefully reset it. But if we do
// so we set a flag to make sure we only do it once, to avoid fighting
// the user if he really wants it on an unsafe place such as the desktop.
if (!prefs->GetBoolean(prefs::kDownloadDirUpgraded)) {
FilePath current_download_dir = prefs->GetFilePath(
prefs::kDownloadDefaultDirectory);
if (download_util::DownloadPathIsDangerous(current_download_dir)) {
prefs->SetFilePath(prefs::kDownloadDefaultDirectory,
default_download_path);
}
prefs->SetBoolean(prefs::kDownloadDirUpgraded, true);
}
}
bool DownloadPrefs::IsAutoOpenUsed() const {
return !auto_open_.empty();
}
bool DownloadPrefs::IsAutoOpenEnabledForExtension(
const FilePath::StringType& extension) const {
return auto_open_.find(extension) != auto_open_.end();
}
bool DownloadPrefs::EnableAutoOpenBasedOnExtension(const FilePath& file_path) {
FilePath::StringType extension = file_path.Extension();
if (extension.empty())
return false;
DCHECK(extension[0] == FilePath::kExtensionSeparator);
extension.erase(0, 1);
if (download_util::IsExecutableExtension(extension))
return false;
auto_open_.insert(extension);
SaveAutoOpenState();
return true;
}
void DownloadPrefs::DisableAutoOpenBasedOnExtension(const FilePath& file_path) {
FilePath::StringType extension = file_path.Extension();
if (extension.empty())
return;
DCHECK(extension[0] == FilePath::kExtensionSeparator);
extension.erase(0, 1);
auto_open_.erase(extension);
SaveAutoOpenState();
}
void DownloadPrefs::ResetToDefaults() {
// TODO(phajdan.jr): Should we reset rest of prefs here?
ResetAutoOpen();
}
void DownloadPrefs::ResetAutoOpen() {
auto_open_.clear();
SaveAutoOpenState();
}
void DownloadPrefs::SaveAutoOpenState() {
std::string extensions;
for (AutoOpenSet::iterator it = auto_open_.begin();
it != auto_open_.end(); ++it) {
#if defined(OS_POSIX)
std::string this_extension = *it;
#elif defined(OS_WIN)
// TODO(phajdan.jr): Why we're using Sys conversion here, but not in ctor?
std::string this_extension = base::SysWideToUTF8(*it);
#endif
extensions += this_extension + ":";
}
if (!extensions.empty())
extensions.erase(extensions.size() - 1);
prefs_->SetString(prefs::kDownloadExtensionsToOpen, extensions);
}
bool DownloadPrefs::AutoOpenCompareFunctor::operator()(
const FilePath::StringType& a,
const FilePath::StringType& b) const {
return FilePath::CompareLessIgnoreCase(a, b);
}
|
