mirror of
https://github.com/LadybirdBrowser/ladybird.git
synced 2024-09-21 10:19:03 +03:00
44ca55aaf8
Where it was straightforward to do so, I've updated the users to also use ByteStrings for their file paths, but most of them have a temporary String::from_byte_string() call instead.
107 lines
3.4 KiB
C++
107 lines
3.4 KiB
C++
/*
|
|
* Copyright (c) 2021, timmot <tiwwot@protonmail.com>
|
|
* Copyright (c) 2022, Mustafa Quraish <mustafa@serenityos.org>
|
|
*
|
|
* SPDX-License-Identifier: BSD-2-Clause
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <AK/HashMap.h>
|
|
#include <AK/String.h>
|
|
#include <FileSystemAccessServer/FileSystemAccessClientEndpoint.h>
|
|
#include <FileSystemAccessServer/FileSystemAccessServerEndpoint.h>
|
|
#include <LibCore/File.h>
|
|
#include <LibCore/Promise.h>
|
|
#include <LibCore/StandardPaths.h>
|
|
#include <LibGUI/FileTypeFilter.h>
|
|
#include <LibGUI/Window.h>
|
|
#include <LibIPC/ConnectionToServer.h>
|
|
|
|
namespace FileSystemAccessClient {
|
|
|
|
enum ErrorFlag : u32 {
|
|
Devices = 1 << 0,
|
|
Directories = 1 << 1,
|
|
NoEntries = 1 << 2,
|
|
|
|
None = 0,
|
|
};
|
|
|
|
class Client;
|
|
class File {
|
|
public:
|
|
File(Badge<Client>, NonnullOwnPtr<Core::File> stream, ByteString filename)
|
|
: m_stream(move(stream))
|
|
, m_filename(filename)
|
|
{
|
|
}
|
|
|
|
Core::File& stream() const { return *m_stream; }
|
|
NonnullOwnPtr<Core::File> release_stream() { return move(m_stream); }
|
|
ByteString const& filename() const { return m_filename; }
|
|
|
|
private:
|
|
NonnullOwnPtr<Core::File> m_stream;
|
|
ByteString m_filename;
|
|
};
|
|
|
|
using Result = ErrorOr<File>;
|
|
|
|
struct OpenFileOptions {
|
|
StringView window_title = {};
|
|
ByteString path = Core::StandardPaths::home_directory();
|
|
Core::File::OpenMode requested_access = Core::File::OpenMode::Read;
|
|
Optional<Vector<GUI::FileTypeFilter>> allowed_file_types = {};
|
|
};
|
|
|
|
class Client final
|
|
: public IPC::ConnectionToServer<FileSystemAccessClientEndpoint, FileSystemAccessServerEndpoint>
|
|
, public FileSystemAccessClientEndpoint {
|
|
IPC_CLIENT_CONNECTION(Client, "/tmp/session/%sid/portal/filesystemaccess"sv)
|
|
|
|
public:
|
|
Result request_file_read_only_approved(GUI::Window* parent_window, ByteString const& path);
|
|
Result request_file(GUI::Window* parent_window, ByteString const& path, Core::File::OpenMode requested_access);
|
|
Result open_file(GUI::Window* parent_window, OpenFileOptions const& = {});
|
|
Result save_file(GUI::Window* parent_window, ByteString const& name, ByteString const ext, Core::File::OpenMode requested_access = Core::File::OpenMode::Write | Core::File::OpenMode::Truncate);
|
|
|
|
void set_silence_errors(u32 flags) { m_silenced_errors = flags; }
|
|
u32 silenced_errors() const { return m_silenced_errors; }
|
|
|
|
bool is_silencing_devices() { return m_silenced_errors & ErrorFlag::Devices; }
|
|
bool is_silencing_directories() { return m_silenced_errors & ErrorFlag::Directories; }
|
|
bool is_silencing_nonexistent_entries() { return m_silenced_errors & ErrorFlag::NoEntries; }
|
|
|
|
static Client& the();
|
|
|
|
protected:
|
|
void die() override;
|
|
|
|
private:
|
|
explicit Client(NonnullOwnPtr<Core::LocalSocket> socket)
|
|
: IPC::ConnectionToServer<FileSystemAccessClientEndpoint, FileSystemAccessServerEndpoint>(*this, move(socket))
|
|
{
|
|
}
|
|
|
|
virtual void handle_prompt_end(i32 request_id, i32 error, Optional<IPC::File> const& fd, Optional<ByteString> const& chosen_file) override;
|
|
|
|
int get_new_id();
|
|
Result handle_promise(int);
|
|
|
|
template<typename T>
|
|
using PromiseType = RefPtr<Core::Promise<T>>;
|
|
|
|
struct RequestData {
|
|
PromiseType<Result> promise;
|
|
GUI::Window* parent_window { nullptr };
|
|
Core::File::OpenMode mode { Core::File::OpenMode::NotOpen };
|
|
};
|
|
|
|
HashMap<int, RequestData> m_promises {};
|
|
int m_last_id { 0 };
|
|
u32 m_silenced_errors { ErrorFlag::None };
|
|
};
|
|
|
|
}
|