src/chrome/browser/ui/views/select_file_dialog_extension.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_UI_VIEWS_SELECT_FILE_DIALOG_EXTENSION_H_
   6 #define CHROME_BROWSER_UI_VIEWS_SELECT_FILE_DIALOG_EXTENSION_H_
   7
   8 #include <vector>
   9
  10 #include "base/compiler_specific.h"
  11 #include "base/memory/ref_counted.h"
  12 #include "chrome/browser/ui/views/extensions/extension_dialog_observer.h"
  13 #include "ui/gfx/native_widget_types.h"  // gfx::NativeWindow
  14 #include "ui/shell_dialogs/select_file_dialog.h"
  15
  16 class ExtensionDialog;
  17 class Profile;
  18
  19 namespace content {
  20 class RenderViewHost;
  21 class WebContents;
  22 }
  23
  24 namespace ui {
  25 struct SelectedFileInfo;
  26 class SelectFilePolicy;
  27 }
  28
  29 // Shows a dialog box for selecting a file or a folder, using the
  30 // file manager extension implementation.
  31 class SelectFileDialogExtension
  32     : public ui::SelectFileDialog,
  33       public ExtensionDialogObserver {
  34  public:
  35   // Opaque ID type for identifying the tab spawned each dialog, unique for
  36   // every WebContents.
  37   typedef const void* RoutingID;
  38   static RoutingID GetRoutingIDFromWebContents(
  39       const content::WebContents* web_contents);
  40
  41   static SelectFileDialogExtension* Create(
  42       ui::SelectFileDialog::Listener* listener,
  43       ui::SelectFilePolicy* policy);
  44
  45   // BaseShellDialog implementation.
  46   virtual bool IsRunning(gfx::NativeWindow owner_window) const override;
  47   virtual void ListenerDestroyed() override;
  48
  49   // ExtensionDialog::Observer implementation.
  50   virtual void ExtensionDialogClosing(ExtensionDialog* dialog) override;
  51   virtual void ExtensionTerminated(ExtensionDialog* dialog) override;
  52
  53   // Routes callback to appropriate SelectFileDialog::Listener based on the
  54   // owning |web_contents|.
  55   static void OnFileSelected(RoutingID routing_id,
  56                              const ui::SelectedFileInfo& file,
  57                              int index);
  58   static void OnMultiFilesSelected(
  59       RoutingID routing_id,
  60       const std::vector<ui::SelectedFileInfo>& files);
  61   static void OnFileSelectionCanceled(RoutingID routing_id);
  62
  63   // For testing, so we can inject JavaScript into the contained view.
  64   content::RenderViewHost* GetRenderViewHost();
  65
  66  protected:
  67   // SelectFileDialog implementation.
  68   virtual void SelectFileImpl(
  69       Type type,
  70       const base::string16& title,
  71       const base::FilePath& default_path,
  72       const FileTypeInfo* file_types,
  73       int file_type_index,
  74       const base::FilePath::StringType& default_extension,
  75       gfx::NativeWindow owning_window,
  76       void* params) override;
  77
  78  private:
  79   friend class SelectFileDialogExtensionBrowserTest;
  80   friend class SelectFileDialogExtensionTest;
  81
  82   // Object is ref-counted, use Create().
  83   explicit SelectFileDialogExtension(SelectFileDialog::Listener* listener,
  84                                      ui::SelectFilePolicy* policy);
  85   virtual ~SelectFileDialogExtension();
  86
  87   // Invokes the appropriate file selection callback on our listener.
  88   void NotifyListener();
  89
  90   // Adds this to the list of pending dialogs, used for testing.
  91   void AddPending(RoutingID routing_id);
  92
  93   // Check if the list of pending dialogs contains dialog for |routing_id|.
  94   static bool PendingExists(RoutingID routing_id);
  95
  96   // Returns true if the dialog has multiple file type choices.
  97   virtual bool HasMultipleFileTypeChoicesImpl() override;
  98
  99   bool has_multiple_file_type_choices_;
 100
 101   // Host for the extension that implements this dialog.
 102   scoped_refptr<ExtensionDialog> extension_dialog_;
 103
 104   // ID of the tab that spawned this dialog, used to route callbacks.
 105   RoutingID routing_id_;
 106
 107   // Pointer to the profile the dialog is running in.
 108   Profile* profile_;
 109
 110   // The window that created the dialog.
 111   gfx::NativeWindow owner_window_;
 112
 113   // We defer the callback into SelectFileDialog::Listener until the window
 114   // closes, to match the semantics of file selection on Windows and Mac.
 115   // These are the data passed to the listener.
 116   enum SelectionType {
 117     CANCEL = 0,
 118     SINGLE_FILE,
 119     MULTIPLE_FILES
 120   };
 121   SelectionType selection_type_;
 122   std::vector<ui::SelectedFileInfo> selection_files_;
 123   int selection_index_;
 124   void* params_;
 125
 126   DISALLOW_COPY_AND_ASSIGN(SelectFileDialogExtension);
 127 };
 128
 129 #endif  // CHROME_BROWSER_UI_VIEWS_SELECT_FILE_DIALOG_EXTENSION_H_