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.
5 #include "chrome/browser/download/save_package_file_picker.h"
8 #include "base/command_line.h"
9 #include "base/i18n/file_util_icu.h"
10 #include "base/metrics/histogram.h"
11 #include "base/prefs/pref_member.h"
12 #include "base/prefs/pref_service.h"
13 #include "base/strings/utf_string_conversions.h"
14 #include "chrome/browser/download/chrome_download_manager_delegate.h"
15 #include "chrome/browser/download/download_prefs.h"
16 #include "chrome/browser/platform_util.h"
17 #include "chrome/browser/profiles/profile.h"
18 #include "chrome/browser/ui/chrome_select_file_policy.h"
19 #include "chrome/common/chrome_switches.h"
20 #include "chrome/grit/generated_resources.h"
21 #include "content/public/browser/download_manager.h"
22 #include "content/public/browser/render_process_host.h"
23 #include "content/public/browser/save_page_type.h"
24 #include "content/public/browser/web_contents.h"
25 #include "ui/base/l10n/l10n_util.h"
27 #if defined(OS_CHROMEOS)
28 #include "chrome/browser/chromeos/drive/download_handler.h"
29 #include "chrome/browser/chromeos/drive/file_system_core_util.h"
32 using content::RenderProcessHost
;
33 using content::SavePageType
;
34 using content::WebContents
;
38 // If false, we don't prompt the user as to where to save the file. This
39 // exists only for testing.
40 bool g_should_prompt_for_filename
= true;
42 void OnSavePackageDownloadCreated(content::DownloadItem
* download
) {
43 ChromeDownloadManagerDelegate::DisableSafeBrowsing(download
);
46 #if defined(OS_CHROMEOS)
47 void OnSavePackageDownloadCreatedChromeOS(
49 const base::FilePath
& drive_path
,
50 content::DownloadItem
* download
) {
51 drive::DownloadHandler::GetForProfile(profile
)->SetDownloadParams(
52 drive_path
, download
);
53 OnSavePackageDownloadCreated(download
);
56 // Trampoline callback between SubstituteDriveDownloadPath() and |callback|.
57 void ContinueSettingUpDriveDownload(
58 const content::SavePackagePathPickedCallback
& callback
,
59 content::SavePageType save_type
,
61 const base::FilePath
& drive_path
,
62 const base::FilePath
& drive_tmp_download_path
) {
63 if (drive_tmp_download_path
.empty()) // Substitution failed.
65 callback
.Run(drive_tmp_download_path
, save_type
, base::Bind(
66 &OnSavePackageDownloadCreatedChromeOS
, profile
, drive_path
));
70 // Adds "Webpage, HTML Only" type to FileTypeInfo.
71 void AddHtmlOnlyFileTypeInfo(
72 ui::SelectFileDialog::FileTypeInfo
* file_type_info
,
73 const base::FilePath::StringType
& extra_extension
) {
74 file_type_info
->extension_description_overrides
.push_back(
75 l10n_util::GetStringUTF16(IDS_SAVE_PAGE_DESC_HTML_ONLY
));
77 std::vector
<base::FilePath::StringType
> extensions
;
78 extensions
.push_back(FILE_PATH_LITERAL("html"));
79 extensions
.push_back(FILE_PATH_LITERAL("htm"));
80 if (!extra_extension
.empty())
81 extensions
.push_back(extra_extension
);
82 file_type_info
->extensions
.push_back(extensions
);
85 // Adds "Web Archive, Single File" type to FileTypeInfo.
86 void AddSingleFileFileTypeInfo(
87 ui::SelectFileDialog::FileTypeInfo
* file_type_info
) {
88 file_type_info
->extension_description_overrides
.push_back(
89 l10n_util::GetStringUTF16(IDS_SAVE_PAGE_DESC_SINGLE_FILE
));
91 std::vector
<base::FilePath::StringType
> extensions
;
92 extensions
.push_back(FILE_PATH_LITERAL("mhtml"));
93 file_type_info
->extensions
.push_back(extensions
);
96 // Chrome OS doesn't support HTML-Complete. crbug.com/154823
97 #if !defined(OS_CHROMEOS)
98 // Adds "Webpage, Complete" type to FileTypeInfo.
99 void AddCompleteFileTypeInfo(
100 ui::SelectFileDialog::FileTypeInfo
* file_type_info
,
101 const base::FilePath::StringType
& extra_extension
) {
102 file_type_info
->extension_description_overrides
.push_back(
103 l10n_util::GetStringUTF16(IDS_SAVE_PAGE_DESC_COMPLETE
));
105 std::vector
<base::FilePath::StringType
> extensions
;
106 extensions
.push_back(FILE_PATH_LITERAL("htm"));
107 extensions
.push_back(FILE_PATH_LITERAL("html"));
108 if (!extra_extension
.empty())
109 extensions
.push_back(extra_extension
);
110 file_type_info
->extensions
.push_back(extensions
);
114 } // anonymous namespace
116 bool SavePackageFilePicker::ShouldSaveAsMHTML() const {
117 #if !defined(OS_CHROMEOS)
118 if (!base::CommandLine::ForCurrentProcess()->HasSwitch(
119 switches::kSavePageAsMHTML
))
122 return can_save_as_complete_
;
125 SavePackageFilePicker::SavePackageFilePicker(
126 content::WebContents
* web_contents
,
127 const base::FilePath
& suggested_path
,
128 const base::FilePath::StringType
& default_extension
,
129 bool can_save_as_complete
,
130 DownloadPrefs
* download_prefs
,
131 const content::SavePackagePathPickedCallback
& callback
)
132 : render_process_id_(web_contents
->GetRenderProcessHost()->GetID()),
133 can_save_as_complete_(can_save_as_complete
),
134 download_prefs_(download_prefs
),
135 callback_(callback
) {
136 base::FilePath suggested_path_copy
= suggested_path
;
137 base::FilePath::StringType default_extension_copy
= default_extension
;
138 int file_type_index
= 0;
139 ui::SelectFileDialog::FileTypeInfo file_type_info
;
141 file_type_info
.support_drive
= true;
143 if (can_save_as_complete_
) {
144 // The option index is not zero-based. Put a dummy entry.
145 save_types_
.push_back(content::SAVE_PAGE_TYPE_UNKNOWN
);
147 base::FilePath::StringType extra_extension
;
148 if (ShouldSaveAsMHTML()) {
149 default_extension_copy
= FILE_PATH_LITERAL("mhtml");
150 suggested_path_copy
= suggested_path_copy
.ReplaceExtension(
151 default_extension_copy
);
153 if (!suggested_path_copy
.FinalExtension().empty() &&
154 !suggested_path_copy
.MatchesExtension(FILE_PATH_LITERAL(".htm")) &&
155 !suggested_path_copy
.MatchesExtension(FILE_PATH_LITERAL(".html"))) {
156 extra_extension
= suggested_path_copy
.FinalExtension().substr(1);
160 AddHtmlOnlyFileTypeInfo(&file_type_info
, extra_extension
);
161 save_types_
.push_back(content::SAVE_PAGE_TYPE_AS_ONLY_HTML
);
163 if (ShouldSaveAsMHTML()) {
164 AddSingleFileFileTypeInfo(&file_type_info
);
165 save_types_
.push_back(content::SAVE_PAGE_TYPE_AS_MHTML
);
168 #if !defined(OS_CHROMEOS)
169 AddCompleteFileTypeInfo(&file_type_info
, extra_extension
);
170 save_types_
.push_back(content::SAVE_PAGE_TYPE_AS_COMPLETE_HTML
);
173 file_type_info
.include_all_files
= false;
175 content::SavePageType preferred_save_type
=
176 static_cast<content::SavePageType
>(download_prefs_
->save_file_type());
177 if (ShouldSaveAsMHTML())
178 preferred_save_type
= content::SAVE_PAGE_TYPE_AS_MHTML
;
180 // Select the item saved in the pref.
181 for (size_t i
= 0; i
< save_types_
.size(); ++i
) {
182 if (save_types_
[i
] == preferred_save_type
) {
188 // If the item saved in the pref was not found, use the last item.
189 if (!file_type_index
)
190 file_type_index
= save_types_
.size() - 1;
192 // The contents can not be saved as complete-HTML, so do not show the file
194 file_type_info
.extensions
.resize(1);
195 file_type_info
.extensions
[0].push_back(
196 suggested_path_copy
.FinalExtension());
198 if (!file_type_info
.extensions
[0][0].empty()) {
200 file_type_info
.extensions
[0][0].erase(0, 1);
203 file_type_info
.include_all_files
= true;
207 if (g_should_prompt_for_filename
) {
208 select_file_dialog_
= ui::SelectFileDialog::Create(
209 this, new ChromeSelectFilePolicy(web_contents
));
210 select_file_dialog_
->SelectFile(
211 ui::SelectFileDialog::SELECT_SAVEAS_FILE
,
216 default_extension_copy
,
217 platform_util::GetTopLevel(web_contents
->GetNativeView()),
220 // Just use 'suggested_path_copy' instead of opening the dialog prompt.
221 // Go through FileSelected() for consistency.
222 FileSelected(suggested_path_copy
, file_type_index
, NULL
);
226 SavePackageFilePicker::~SavePackageFilePicker() {
229 void SavePackageFilePicker::SetShouldPromptUser(bool should_prompt
) {
230 g_should_prompt_for_filename
= should_prompt
;
233 void SavePackageFilePicker::FileSelected(
234 const base::FilePath
& path
, int index
, void* unused_params
) {
235 scoped_ptr
<SavePackageFilePicker
> delete_this(this);
236 RenderProcessHost
* process
= RenderProcessHost::FromID(render_process_id_
);
239 SavePageType save_type
= content::SAVE_PAGE_TYPE_UNKNOWN
;
241 if (can_save_as_complete_
) {
242 DCHECK_LT(index
, static_cast<int>(save_types_
.size()));
243 save_type
= save_types_
[index
];
244 if (select_file_dialog_
.get() &&
245 select_file_dialog_
->HasMultipleFileTypeChoices())
246 download_prefs_
->SetSaveFileType(save_type
);
248 UMA_HISTOGRAM_ENUMERATION("Download.SavePageType",
250 content::SAVE_PAGE_TYPE_MAX
);
252 // Use "HTML Only" type as a dummy.
253 save_type
= content::SAVE_PAGE_TYPE_AS_ONLY_HTML
;
256 base::FilePath
path_copy(path
);
257 base::i18n::NormalizeFileNameEncoding(&path_copy
);
259 download_prefs_
->SetSaveFilePath(path_copy
.DirName());
261 #if defined(OS_CHROMEOS)
262 if (drive::util::IsUnderDriveMountPoint(path_copy
)) {
263 // Here's a map to the callback chain:
264 // SubstituteDriveDownloadPath ->
265 // ContinueSettingUpDriveDownload ->
266 // callback_ = SavePackage::OnPathPicked ->
267 // download_created_callback = OnSavePackageDownloadCreatedChromeOS
268 Profile
* profile
= Profile::FromBrowserContext(
269 process
->GetBrowserContext());
270 drive::DownloadHandler
* drive_download_handler
=
271 drive::DownloadHandler::GetForProfile(profile
);
272 drive_download_handler
->SubstituteDriveDownloadPath(
273 path_copy
, NULL
, base::Bind(&ContinueSettingUpDriveDownload
,
282 callback_
.Run(path_copy
, save_type
,
283 base::Bind(&OnSavePackageDownloadCreated
));
286 void SavePackageFilePicker::FileSelectionCanceled(void* unused_params
) {