fix: transparently package bundles as zip archives (#25030)

shell/browser/file_select_helper.h

@@ -0,0 +1,118 @@
+// Copyright (c) 2020 Microsoft, Inc. All rights reserved.
+// Use of this source code is governed by the MIT license that can be
+// found in the LICENSE file.
+
+#ifndef SHELL_BROWSER_FILE_SELECT_HELPER_H_
+#define SHELL_BROWSER_FILE_SELECT_HELPER_H_
+
+#include <memory>
+
+#include <string>
+#include <utility>
+#include <vector>
+
+#include "base/files/file_path.h"
+#include "chrome/common/pref_names.h"
+#include "content/public/browser/file_select_listener.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/render_process_host.h"
+#include "content/public/browser/render_view_host.h"
+#include "content/public/browser/render_widget_host.h"
+#include "content/public/browser/render_widget_host_observer.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/browser/web_contents_observer.h"
+#include "gin/dictionary.h"
+#include "net/base/directory_lister.h"
+#include "shell/browser/electron_browser_context.h"
+#include "shell/browser/ui/file_dialog.h"
+#include "shell/common/gin_helper/dictionary.h"
+#include "ui/shell_dialogs/selected_file_info.h"
+
+using blink::mojom::FileChooserParams;
+
+class FileSelectHelper : public content::WebContentsObserver,
+                         public content::RenderWidgetHostObserver,
+                         public net::DirectoryLister::DirectoryListerDelegate {
+ public:
+  FileSelectHelper(content::RenderFrameHost* render_frame_host,
+                   scoped_refptr<content::FileSelectListener> listener,
+                   FileChooserParams::Mode mode);
+  ~FileSelectHelper() override;
+
+  // WebDialogHelper::RunFileChooser
+
+  void ShowOpenDialog(const file_dialog::DialogSettings& settings);
+
+  void ShowSaveDialog(const file_dialog::DialogSettings& settings);
+
+ private:
+  // net::DirectoryLister::DirectoryListerDelegate overrides.
+  void OnListFile(
+      const net::DirectoryLister::DirectoryListerData& data) override;
+  void OnListDone(int error) override;
+
+  void DeleteTemporaryFiles();
+
+  void EnumerateDirectory();
+
+  void OnOpenDialogDone(gin_helper::Dictionary result);
+  void OnSaveDialogDone(gin_helper::Dictionary result);
+
+  void OnFilesSelected(
+      std::vector<blink::mojom::FileChooserFileInfoPtr> file_info,
+      base::FilePath base_dir);
+
+  void RunFileChooserEnd();
+
+  void ConvertToFileChooserFileInfoList(
+      const std::vector<ui::SelectedFileInfo>& files);
+
+#if defined(OS_MAC)
+  // Must be called from a MayBlock() task. Each selected file that is a package
+  // will be zipped, and the zip will be passed to the render view host in place
+  // of the package.
+  void ProcessSelectedFilesMac(const std::vector<ui::SelectedFileInfo>& files);
+
+  // Saves the paths of |zipped_files| for later deletion. Passes |files| to the
+  // render view host.
+  void ProcessSelectedFilesMacOnUIThread(
+      const std::vector<ui::SelectedFileInfo>& files,
+      const std::vector<base::FilePath>& zipped_files);
+
+  // Zips the package at |path| into a temporary destination. Returns the
+  // temporary destination, if the zip was successful. Otherwise returns an
+  // empty path.
+  static base::FilePath ZipPackage(const base::FilePath& path);
+#endif  // defined(OS_MAC)
+
+  // content::RenderWidgetHostObserver:
+  void RenderWidgetHostDestroyed(
+      content::RenderWidgetHost* widget_host) override;
+
+  // content::WebContentsObserver:
+  void RenderFrameHostChanged(content::RenderFrameHost* old_host,
+                              content::RenderFrameHost* new_host) override;
+  void RenderFrameDeleted(content::RenderFrameHost* deleted_host) override;
+  void WebContentsDestroyed() override;
+
+  content::RenderFrameHost* render_frame_host_;
+  content::WebContents* web_contents_;
+  scoped_refptr<content::FileSelectListener> listener_;
+  FileChooserParams::Mode mode_;
+
+  ScopedObserver<content::RenderWidgetHost, content::RenderWidgetHostObserver>
+      observer_{this};
+
+  // Temporary files only used on OSX. This class is responsible for deleting
+  // these files when they are no longer needed.
+  std::vector<base::FilePath> temporary_files_;
+
+  // DirectoryLister-specific members
+  std::unique_ptr<net::DirectoryLister> lister_;
+  base::FilePath lister_base_dir_;
+  std::vector<base::FilePath> lister_paths_;
+
+  base::WeakPtrFactory<FileSelectHelper> weak_ptr_factory_{this};
+};
+
+#endif  // SHELL_BROWSER_FILE_SELECT_HELPER_H_