chrome/browser/download/save_package_file_picker.h

   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 #ifndef CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_FILE_PICKER_H_
   6 #define CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_FILE_PICKER_H_
   7
   8 #include "base/memory/ref_counted.h"
   9 #include "content/public/browser/download_manager_delegate.h"
  10 #include "ui/shell_dialogs/select_file_dialog.h"
  11
  12 class DownloadPrefs;
  13
  14 // Handles showing a dialog to the user to ask for the filename to save a page.
  15 class SavePackageFilePicker : public ui::SelectFileDialog::Listener {
  16  public:
  17   SavePackageFilePicker(
  18       content::WebContents* web_contents,
  19       const base::FilePath& suggested_path,
  20       const base::FilePath::StringType& default_extension,
  21       bool can_save_as_complete,
  22       DownloadPrefs* download_prefs,
  23       const content::SavePackagePathPickedCallback& callback);
  24   virtual ~SavePackageFilePicker();
  25
  26   // Used to disable prompting the user for a directory/filename of the saved
  27   // web page.  This is available for testing.
  28   static void SetShouldPromptUser(bool should_prompt);
  29
  30  private:
  31   // SelectFileDialog::Listener implementation.
  32   virtual void FileSelected(const base::FilePath& path,
  33                             int index,
  34                             void* unused_params) OVERRIDE;
  35   virtual void FileSelectionCanceled(void* unused_params) OVERRIDE;
  36
  37   bool ShouldSaveAsMHTML() const;
  38
  39   // Used to look up the renderer process for this request to get the context.
  40   int render_process_id_;
  41
  42   // Whether the web page can be saved as a complete HTML file.
  43   bool can_save_as_complete_;
  44
  45   DownloadPrefs* download_prefs_;
  46
  47   content::SavePackagePathPickedCallback callback_;
  48
  49   // For managing select file dialogs.
  50   scoped_refptr<ui::SelectFileDialog> select_file_dialog_;
  51
  52   DISALLOW_COPY_AND_ASSIGN(SavePackageFilePicker);
  53 };
  54
  55 #endif  // CHROME_BROWSER_DOWNLOAD_SAVE_PACKAGE_FILE_PICKER_H_