chrome/browser/download/download_file_picker.cc

   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 "chrome/browser/download/download_file_picker.h"
   6
   7 #include "base/metrics/histogram.h"
   8 #include "chrome/browser/download/download_prefs.h"
   9 #include "chrome/browser/platform_util.h"
  10 #include "chrome/browser/ui/chrome_select_file_policy.h"
  11 #include "content/public/browser/browser_context.h"
  12 #include "content/public/browser/download_item.h"
  13 #include "content/public/browser/download_manager.h"
  14 #include "content/public/browser/web_contents.h"
  15 #include "grit/generated_resources.h"
  16 #include "ui/base/l10n/l10n_util.h"
  17
  18 using content::DownloadItem;
  19 using content::DownloadManager;
  20 using content::WebContents;
  21
  22 namespace {
  23
  24 enum FilePickerResult {
  25   FILE_PICKER_SAME,
  26   FILE_PICKER_DIFFERENT_DIR,
  27   FILE_PICKER_DIFFERENT_NAME,
  28   FILE_PICKER_CANCEL,
  29   FILE_PICKER_MAX,
  30 };
  31
  32 // Record how the File Picker was used during a download. This UMA is only
  33 // recorded for profiles that do not always prompt for save locations on
  34 // downloads.
  35 void RecordFilePickerResult(const base::FilePath& suggested_path,
  36                             const base::FilePath& actual_path) {
  37   FilePickerResult result;
  38   if (suggested_path == actual_path)
  39     result = FILE_PICKER_SAME;
  40   else if (actual_path.empty())
  41     result = FILE_PICKER_CANCEL;
  42   else if (suggested_path.DirName() != actual_path.DirName())
  43     result = FILE_PICKER_DIFFERENT_DIR;
  44   else
  45     result = FILE_PICKER_DIFFERENT_NAME;
  46
  47   UMA_HISTOGRAM_ENUMERATION("Download.FilePickerResult",
  48                             result,
  49                             FILE_PICKER_MAX);
  50 }
  51
  52 }  // namespace
  53
  54 DownloadFilePicker::DownloadFilePicker(
  55     DownloadItem* item,
  56     const base::FilePath& suggested_path,
  57     const FileSelectedCallback& callback)
  58     : suggested_path_(suggested_path),
  59       file_selected_callback_(callback),
  60       should_record_file_picker_result_(false) {
  61   const DownloadPrefs* prefs =
  62       DownloadPrefs::FromBrowserContext(item->GetBrowserContext());
  63   DCHECK(prefs);
  64   // Only record UMA if we aren't prompting the user for all downloads.
  65   should_record_file_picker_result_ = !prefs->PromptForDownload();
  66
  67   WebContents* web_contents = item->GetWebContents();
  68   select_file_dialog_ = ui::SelectFileDialog::Create(
  69       this, new ChromeSelectFilePolicy(web_contents));
  70   ui::SelectFileDialog::FileTypeInfo file_type_info;
  71   // Platform file pickers, notably on Mac and Windows, tend to break
  72   // with double extensions like .tar.gz, so only pass in normal ones.
  73   base::FilePath::StringType extension = suggested_path_.FinalExtension();
  74   if (!extension.empty()) {
  75     extension.erase(extension.begin());  // drop the .
  76     file_type_info.extensions.resize(1);
  77     file_type_info.extensions[0].push_back(extension);
  78   }
  79   file_type_info.include_all_files = true;
  80   file_type_info.support_drive = true;
  81   gfx::NativeWindow owning_window = web_contents ?
  82       platform_util::GetTopLevel(web_contents->GetNativeView()) : NULL;
  83
  84   select_file_dialog_->SelectFile(ui::SelectFileDialog::SELECT_SAVEAS_FILE,
  85                                   base::string16(),
  86                                   suggested_path_,
  87                                   &file_type_info,
  88                                   0,
  89                                   base::FilePath::StringType(),
  90                                   owning_window,
  91                                   NULL);
  92 }
  93
  94 DownloadFilePicker::~DownloadFilePicker() {
  95 }
  96
  97 void DownloadFilePicker::OnFileSelected(const base::FilePath& path) {
  98   if (should_record_file_picker_result_)
  99     RecordFilePickerResult(suggested_path_, path);
 100   file_selected_callback_.Run(path);
 101   delete this;
 102 }
 103
 104 void DownloadFilePicker::FileSelected(const base::FilePath& path,
 105                                       int index,
 106                                       void* params) {
 107   OnFileSelected(path);
 108   // Deletes |this|
 109 }
 110
 111 void DownloadFilePicker::FileSelectionCanceled(void* params) {
 112   OnFileSelected(base::FilePath());
 113   // Deletes |this|
 114 }
 115
 116 // static
 117 void DownloadFilePicker::ShowFilePicker(DownloadItem* item,
 118                                         const base::FilePath& suggested_path,
 119                                         const FileSelectedCallback& callback) {
 120   new DownloadFilePicker(item, suggested_path, callback);
 121   // DownloadFilePicker deletes itself.
 122 }