117 lines
4.3 KiB
C++
117 lines
4.3 KiB
C++
// 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 <vector>
|
|
|
|
#include "base/files/file_path.h"
|
|
#include "base/scoped_observation.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 |temporary_files| for later deletion. Passes |files| to
|
|
// the render view host.
|
|
void ProcessSelectedFilesMacOnUIThread(
|
|
const std::vector<ui::SelectedFileInfo>& files,
|
|
const std::vector<base::FilePath>& temporary_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_;
|
|
|
|
base::ScopedObservation<content::RenderWidgetHost,
|
|
content::RenderWidgetHostObserver>
|
|
observation_{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_
|