jcs.org
Serenity Operating System
1/*
2 * Copyright (c) 2021, timmot <tiwwot@protonmail.com>
3 * Copyright (c) 2022, Mustafa Quraish <mustafa@serenityos.org>
4 *
5 * SPDX-License-Identifier: BSD-2-Clause
6 */
7
8#pragma once
9
10#include <AK/HashMap.h>
11#include <AK/String.h>
12#include <FileSystemAccessServer/FileSystemAccessClientEndpoint.h>
13#include <FileSystemAccessServer/FileSystemAccessServerEndpoint.h>
14#include <LibCore/File.h>
15#include <LibCore/Promise.h>
16#include <LibCore/StandardPaths.h>
17#include <LibGUI/Window.h>
18#include <LibIPC/ConnectionToServer.h>
19
20namespace FileSystemAccessClient {
21
22class Client;
23class File {
24public:
25 File(Badge<Client>, NonnullOwnPtr<Core::File> stream, String filename)
26 : m_stream(move(stream))
27 , m_filename(filename)
28 {
29 }
30
31 Core::File& stream() const { return *m_stream; }
32 NonnullOwnPtr<Core::File> release_stream() { return move(m_stream); }
33 String filename() const { return m_filename; }
34
35private:
36 NonnullOwnPtr<Core::File> m_stream;
37 String m_filename;
38};
39
40using Result = ErrorOr<File>;
41
42class Client final
43 : public IPC::ConnectionToServer<FileSystemAccessClientEndpoint, FileSystemAccessServerEndpoint>
44 , public FileSystemAccessClientEndpoint {
45 IPC_CLIENT_CONNECTION(Client, "/tmp/session/%sid/portal/filesystemaccess"sv)
46
47public:
48 Result request_file_read_only_approved(GUI::Window* parent_window, DeprecatedString const& path);
49 Result request_file(GUI::Window* parent_window, DeprecatedString const& path, Core::File::OpenMode requested_access);
50 Result open_file(GUI::Window* parent_window, DeprecatedString const& window_title = {}, StringView path = Core::StandardPaths::home_directory(), Core::File::OpenMode requested_access = Core::File::OpenMode::Read);
51 Result save_file(GUI::Window* parent_window, DeprecatedString const& name, DeprecatedString const ext, Core::File::OpenMode requested_access = Core::File::OpenMode::Write | Core::File::OpenMode::Truncate);
52
53 static Client& the();
54
55protected:
56 void die() override;
57
58private:
59 explicit Client(NonnullOwnPtr<Core::LocalSocket> socket)
60 : IPC::ConnectionToServer<FileSystemAccessClientEndpoint, FileSystemAccessServerEndpoint>(*this, move(socket))
61 {
62 }
63
64 virtual void handle_prompt_end(i32 request_id, i32 error, Optional<IPC::File> const& fd, Optional<DeprecatedString> const& chosen_file) override;
65
66 int get_new_id();
67 Result handle_promise(int);
68
69 template<typename T>
70 using PromiseType = RefPtr<Core::Promise<T>>;
71
72 struct PromiseAndWindow {
73 PromiseType<Result> promise;
74 GUI::Window* parent_window { nullptr };
75 };
76
77 HashMap<int, PromiseAndWindow> m_promises {};
78 int m_last_id { 0 };
79};
80
81}