| OLD | NEW |
| (Empty) | |
| 1 // Copyright (c) 2011 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 #include "chrome/browser/download/save_package_file_picker.h" |
| 6 |
| 7 #include "chrome/browser/download/download_manager.h" |
| 8 #include "chrome/browser/download/download_prefs.h" |
| 9 #include "chrome/browser/platform_util.h" |
| 10 #include "chrome/browser/profiles/profile.h" |
| 11 #include "content/browser/download/save_package.h" |
| 12 #include "content/browser/tab_contents/tab_contents.h" |
| 13 #include "grit/generated_resources.h" |
| 14 #include "ui/base/l10n/l10n_util.h" |
| 15 |
| 16 namespace { |
| 17 |
| 18 // If false, we don't prompt the user as to where to save the file. This |
| 19 // exists only for testing. |
| 20 bool g_should_prompt_for_filename = true; |
| 21 |
| 22 // Used for mapping between SavePackageType constants and the indexes above. |
| 23 const SavePackage::SavePackageType kIndexToSaveType[] = { |
| 24 SavePackage::SAVE_TYPE_UNKNOWN, |
| 25 SavePackage::SAVE_AS_ONLY_HTML, |
| 26 SavePackage::SAVE_AS_COMPLETE_HTML, |
| 27 }; |
| 28 |
| 29 int SavePackageTypeToIndex(SavePackage::SavePackageType type) { |
| 30 for (size_t i = 0; i < arraysize(kIndexToSaveType); ++i) { |
| 31 if (kIndexToSaveType[i] == type) |
| 32 return i; |
| 33 } |
| 34 NOTREACHED(); |
| 35 return -1; |
| 36 } |
| 37 |
| 38 // Indexes used for specifying which element in the extensions dropdown |
| 39 // the user chooses when picking a save type. |
| 40 const int kSelectFileHtmlOnlyIndex = 1; |
| 41 const int kSelectFileCompleteIndex = 2; |
| 42 |
| 43 // Used for mapping between the IDS_ string identifiers and the indexes above. |
| 44 const int kIndexToIDS[] = { |
| 45 0, IDS_SAVE_PAGE_DESC_HTML_ONLY, IDS_SAVE_PAGE_DESC_COMPLETE, |
| 46 }; |
| 47 |
| 48 } |
| 49 |
| 50 SavePackageFilePicker::SavePackageFilePicker(SavePackage* save_package, |
| 51 const FilePath& suggested_path, |
| 52 bool can_save_as_complete) |
| 53 : save_package_(save_package) { |
| 54 // The TabContents which owns this SavePackage may have disappeared during |
| 55 // the UI->FILE->UI thread hop of |
| 56 // GetSaveInfo->CreateDirectoryOnFileThread->ContinueGetSaveInfo. |
| 57 TabContents* tab_contents = save_package->tab_contents(); |
| 58 if (!tab_contents) { |
| 59 delete this; |
| 60 return; |
| 61 } |
| 62 |
| 63 DownloadPrefs* download_prefs = |
| 64 tab_contents->profile()->GetDownloadManager()->download_prefs(); |
| 65 int file_type_index = SavePackageTypeToIndex( |
| 66 static_cast<SavePackage::SavePackageType>( |
| 67 download_prefs->save_file_type())); |
| 68 DCHECK_NE(-1, file_type_index); |
| 69 |
| 70 SelectFileDialog::FileTypeInfo file_type_info; |
| 71 FilePath::StringType default_extension; |
| 72 |
| 73 // If the contents can not be saved as complete-HTML, do not show the |
| 74 // file filters. |
| 75 if (can_save_as_complete) { |
| 76 bool add_extra_extension = false; |
| 77 FilePath::StringType extra_extension; |
| 78 if (!suggested_path.Extension().empty() && |
| 79 suggested_path.Extension().compare(FILE_PATH_LITERAL("htm")) && |
| 80 suggested_path.Extension().compare(FILE_PATH_LITERAL("html"))) { |
| 81 add_extra_extension = true; |
| 82 extra_extension = suggested_path.Extension().substr(1); |
| 83 } |
| 84 |
| 85 file_type_info.extensions.resize(2); |
| 86 file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back( |
| 87 FILE_PATH_LITERAL("htm")); |
| 88 file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back( |
| 89 FILE_PATH_LITERAL("html")); |
| 90 |
| 91 if (add_extra_extension) { |
| 92 file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back( |
| 93 extra_extension); |
| 94 } |
| 95 |
| 96 file_type_info.extension_description_overrides.push_back( |
| 97 l10n_util::GetStringUTF16(kIndexToIDS[kSelectFileCompleteIndex - 1])); |
| 98 file_type_info.extensions[kSelectFileCompleteIndex - 1].push_back( |
| 99 FILE_PATH_LITERAL("htm")); |
| 100 file_type_info.extensions[kSelectFileCompleteIndex - 1].push_back( |
| 101 FILE_PATH_LITERAL("html")); |
| 102 |
| 103 if (add_extra_extension) { |
| 104 file_type_info.extensions[kSelectFileCompleteIndex - 1].push_back( |
| 105 extra_extension); |
| 106 } |
| 107 |
| 108 file_type_info.extension_description_overrides.push_back( |
| 109 l10n_util::GetStringUTF16(kIndexToIDS[kSelectFileCompleteIndex])); |
| 110 file_type_info.include_all_files = false; |
| 111 default_extension = SavePackage::kDefaultHtmlExtension; |
| 112 } else { |
| 113 file_type_info.extensions.resize(1); |
| 114 file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1].push_back( |
| 115 suggested_path.Extension()); |
| 116 |
| 117 if (!file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1][0].empty()) { |
| 118 // Drop the . |
| 119 file_type_info.extensions[kSelectFileHtmlOnlyIndex - 1][0].erase(0, 1); |
| 120 } |
| 121 |
| 122 file_type_info.include_all_files = true; |
| 123 file_type_index = 1; |
| 124 } |
| 125 |
| 126 if (g_should_prompt_for_filename) { |
| 127 select_file_dialog_ = SelectFileDialog::Create(this); |
| 128 select_file_dialog_->SelectFile(SelectFileDialog::SELECT_SAVEAS_FILE, |
| 129 string16(), |
| 130 suggested_path, |
| 131 &file_type_info, |
| 132 file_type_index, |
| 133 default_extension, |
| 134 tab_contents, |
| 135 platform_util::GetTopLevel( |
| 136 tab_contents->GetNativeView()), |
| 137 NULL); |
| 138 } else { |
| 139 // Just use 'suggested_path' instead of opening the dialog prompt. |
| 140 save_package_->OnPathPicked( |
| 141 suggested_path, kIndexToSaveType[file_type_index]); |
| 142 } |
| 143 } |
| 144 |
| 145 SavePackageFilePicker::~SavePackageFilePicker() { |
| 146 } |
| 147 |
| 148 void SavePackageFilePicker::SetShouldPromptUser(bool should_prompt) { |
| 149 g_should_prompt_for_filename = should_prompt; |
| 150 } |
| 151 |
| 152 void SavePackageFilePicker::FileSelected(const FilePath& path, |
| 153 int index, |
| 154 void* params) { |
| 155 // The option index is not zero-based. |
| 156 DCHECK(index >= kSelectFileHtmlOnlyIndex && |
| 157 index <= kSelectFileCompleteIndex); |
| 158 |
| 159 save_package_->OnPathPicked(path, kIndexToSaveType[index]); |
| 160 delete this; |
| 161 } |
| 162 |
| 163 void SavePackageFilePicker::FileSelectionCanceled(void* params) { |
| 164 delete this; |
| 165 } |
| OLD | NEW |