| OLD | NEW |
|---|
| (Empty) |
| 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 #include "content/shell/shell_download_manager_delegate.h" | |
| 6 | |
| 7 #if defined(TOOLKIT_GTK) | |
| 8 #include <gtk/gtk.h> | |
| 9 #endif | |
| 10 | |
| 11 #if defined(OS_WIN) | |
| 12 #include <windows.h> | |
| 13 #include <commdlg.h> | |
| 14 #endif | |
| 15 | |
| 16 #include "base/bind.h" | |
| 17 #include "base/command_line.h" | |
| 18 #include "base/file_util.h" | |
| 19 #include "base/logging.h" | |
| 20 #include "base/strings/string_util.h" | |
| 21 #include "base/strings/utf_string_conversions.h" | |
| 22 #include "content/public/browser/browser_context.h" | |
| 23 #include "content/public/browser/browser_thread.h" | |
| 24 #include "content/public/browser/download_manager.h" | |
| 25 #include "content/public/browser/web_contents.h" | |
| 26 #include "content/public/browser/web_contents_view.h" | |
| 27 #include "content/shell/common/shell_switches.h" | |
| 28 #include "content/shell/webkit_test_controller.h" | |
| 29 #include "net/base/net_util.h" | |
| 30 | |
| 31 namespace content { | |
| 32 | |
| 33 ShellDownloadManagerDelegate::ShellDownloadManagerDelegate() | |
| 34 : download_manager_(NULL), | |
| 35 suppress_prompting_(false) { | |
| 36 // Balanced in Shutdown(); | |
| 37 AddRef(); | |
| 38 } | |
| 39 | |
| 40 ShellDownloadManagerDelegate::~ShellDownloadManagerDelegate(){ | |
| 41 } | |
| 42 | |
| 43 | |
| 44 void ShellDownloadManagerDelegate::SetDownloadManager( | |
| 45 DownloadManager* download_manager) { | |
| 46 download_manager_ = download_manager; | |
| 47 } | |
| 48 | |
| 49 void ShellDownloadManagerDelegate::Shutdown() { | |
| 50 Release(); | |
| 51 } | |
| 52 | |
| 53 bool ShellDownloadManagerDelegate::DetermineDownloadTarget( | |
| 54 DownloadItem* download, | |
| 55 const DownloadTargetCallback& callback) { | |
| 56 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
| 57 // This assignment needs to be here because even at the call to | |
| 58 // SetDownloadManager, the system is not fully initialized. | |
| 59 if (default_download_path_.empty()) { | |
| 60 default_download_path_ = download_manager_->GetBrowserContext()->GetPath(). | |
| 61 Append(FILE_PATH_LITERAL("Downloads")); | |
| 62 } | |
| 63 | |
| 64 if (!download->GetForcedFilePath().empty()) { | |
| 65 callback.Run(download->GetForcedFilePath(), | |
| 66 DownloadItem::TARGET_DISPOSITION_OVERWRITE, | |
| 67 DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, | |
| 68 download->GetForcedFilePath()); | |
| 69 return true; | |
| 70 } | |
| 71 | |
| 72 base::FilePath generated_name = net::GenerateFileName( | |
| 73 download->GetURL(), | |
| 74 download->GetContentDisposition(), | |
| 75 EmptyString(), | |
| 76 download->GetSuggestedFilename(), | |
| 77 download->GetMimeType(), | |
| 78 "download"); | |
| 79 | |
| 80 BrowserThread::PostTask( | |
| 81 BrowserThread::FILE, | |
| 82 FROM_HERE, | |
| 83 base::Bind( | |
| 84 &ShellDownloadManagerDelegate::GenerateFilename, | |
| 85 this, download->GetId(), callback, generated_name, | |
| 86 default_download_path_)); | |
| 87 return true; | |
| 88 } | |
| 89 | |
| 90 bool ShellDownloadManagerDelegate::ShouldOpenDownload( | |
| 91 DownloadItem* item, | |
| 92 const DownloadOpenDelayedCallback& callback) { | |
| 93 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kDumpRenderTree) && | |
| 94 WebKitTestController::Get()->IsMainWindow(item->GetWebContents()) && | |
| 95 item->GetMimeType() == "text/html") { | |
| 96 WebKitTestController::Get()->OpenURL( | |
| 97 net::FilePathToFileURL(item->GetFullPath())); | |
| 98 } | |
| 99 return true; | |
| 100 } | |
| 101 | |
| 102 void ShellDownloadManagerDelegate::GetNextId( | |
| 103 const DownloadIdCallback& callback) { | |
| 104 static uint32 next_id = DownloadItem::kInvalidId + 1; | |
| 105 callback.Run(next_id++); | |
| 106 } | |
| 107 | |
| 108 void ShellDownloadManagerDelegate::GenerateFilename( | |
| 109 uint32 download_id, | |
| 110 const DownloadTargetCallback& callback, | |
| 111 const base::FilePath& generated_name, | |
| 112 const base::FilePath& suggested_directory) { | |
| 113 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); | |
| 114 if (!base::PathExists(suggested_directory)) | |
| 115 file_util::CreateDirectory(suggested_directory); | |
| 116 | |
| 117 base::FilePath suggested_path(suggested_directory.Append(generated_name)); | |
| 118 BrowserThread::PostTask( | |
| 119 BrowserThread::UI, | |
| 120 FROM_HERE, | |
| 121 base::Bind( | |
| 122 &ShellDownloadManagerDelegate::OnDownloadPathGenerated, | |
| 123 this, download_id, callback, suggested_path)); | |
| 124 } | |
| 125 | |
| 126 void ShellDownloadManagerDelegate::OnDownloadPathGenerated( | |
| 127 uint32 download_id, | |
| 128 const DownloadTargetCallback& callback, | |
| 129 const base::FilePath& suggested_path) { | |
| 130 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
| 131 if (suppress_prompting_) { | |
| 132 // Testing exit. | |
| 133 callback.Run(suggested_path, DownloadItem::TARGET_DISPOSITION_OVERWRITE, | |
| 134 DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, | |
| 135 suggested_path.AddExtension(FILE_PATH_LITERAL(".crdownload"))); | |
| 136 return; | |
| 137 } | |
| 138 | |
| 139 ChooseDownloadPath(download_id, callback, suggested_path); | |
| 140 } | |
| 141 | |
| 142 void ShellDownloadManagerDelegate::ChooseDownloadPath( | |
| 143 uint32 download_id, | |
| 144 const DownloadTargetCallback& callback, | |
| 145 const base::FilePath& suggested_path) { | |
| 146 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | |
| 147 DownloadItem* item = download_manager_->GetDownload(download_id); | |
| 148 if (!item || (item->GetState() != DownloadItem::IN_PROGRESS)) | |
| 149 return; | |
| 150 | |
| 151 base::FilePath result; | |
| 152 #if defined(OS_WIN) && !defined(USE_AURA) | |
| 153 std::wstring file_part = base::FilePath(suggested_path).BaseName().value(); | |
| 154 wchar_t file_name[MAX_PATH]; | |
| 155 base::wcslcpy(file_name, file_part.c_str(), arraysize(file_name)); | |
| 156 OPENFILENAME save_as; | |
| 157 ZeroMemory(&save_as, sizeof(save_as)); | |
| 158 save_as.lStructSize = sizeof(OPENFILENAME); | |
| 159 save_as.hwndOwner = item->GetWebContents()->GetView()->GetNativeView(); | |
| 160 save_as.lpstrFile = file_name; | |
| 161 save_as.nMaxFile = arraysize(file_name); | |
| 162 | |
| 163 std::wstring directory; | |
| 164 if (!suggested_path.empty()) | |
| 165 directory = suggested_path.DirName().value(); | |
| 166 | |
| 167 save_as.lpstrInitialDir = directory.c_str(); | |
| 168 save_as.Flags = OFN_OVERWRITEPROMPT | OFN_EXPLORER | OFN_ENABLESIZING | | |
| 169 OFN_NOCHANGEDIR | OFN_PATHMUSTEXIST; | |
| 170 | |
| 171 if (GetSaveFileName(&save_as)) | |
| 172 result = base::FilePath(std::wstring(save_as.lpstrFile)); | |
| 173 #elif defined(TOOLKIT_GTK) | |
| 174 GtkWidget *dialog; | |
| 175 gfx::NativeWindow parent_window; | |
| 176 std::string base_name = base::FilePath(suggested_path).BaseName().value(); | |
| 177 | |
| 178 parent_window = item->GetWebContents()->GetView()->GetTopLevelNativeWindow(); | |
| 179 dialog = gtk_file_chooser_dialog_new("Save File", | |
| 180 parent_window, | |
| 181 GTK_FILE_CHOOSER_ACTION_SAVE, | |
| 182 GTK_STOCK_CANCEL, GTK_RESPONSE_CANCEL, | |
| 183 GTK_STOCK_SAVE, GTK_RESPONSE_ACCEPT, | |
| 184 NULL); | |
| 185 gtk_file_chooser_set_do_overwrite_confirmation(GTK_FILE_CHOOSER(dialog), | |
| 186 TRUE); | |
| 187 gtk_file_chooser_set_current_name(GTK_FILE_CHOOSER(dialog), | |
| 188 base_name.c_str()); | |
| 189 | |
| 190 if (gtk_dialog_run(GTK_DIALOG(dialog)) == GTK_RESPONSE_ACCEPT) { | |
| 191 char *filename; | |
| 192 filename = gtk_file_chooser_get_filename(GTK_FILE_CHOOSER(dialog)); | |
| 193 result = base::FilePath(filename); | |
| 194 g_free(filename); | |
| 195 } | |
| 196 gtk_widget_destroy(dialog); | |
| 197 #else | |
| 198 NOTIMPLEMENTED(); | |
| 199 #endif | |
| 200 | |
| 201 callback.Run(result, DownloadItem::TARGET_DISPOSITION_PROMPT, | |
| 202 DOWNLOAD_DANGER_TYPE_NOT_DANGEROUS, result); | |
| 203 } | |
| 204 | |
| 205 void ShellDownloadManagerDelegate::SetDownloadBehaviorForTesting( | |
| 206 const base::FilePath& default_download_path) { | |
| 207 default_download_path_ = default_download_path; | |
| 208 suppress_prompting_ = true; | |
| 209 } | |
| 210 | |
| 211 } // namespace content | |
| OLD | NEW |
|---|
Chromium Code Reviews
Issue 23316003:
[content shell] move browser process stuff into browser/ subdir (Closed)
Base URL: svn://svn.chromium.org/chrome/trunk/src