summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibFileSystemAccessClient/Client.cpp
blob: 730c1b609a791c77ccdbe6d2f3370b121aff8ace (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*
 * Copyright (c) 2021, timmot <tiwwot@protonmail.com>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

// FIXME: LibIPC Decoder and Encoder are sensitive to include order here
// clang-format off
#include <LibGUI/WindowServerConnection.h>
// clang-format on
#include <AK/LexicalPath.h>
#include <LibCore/File.h>
#include <LibFileSystemAccessClient/Client.h>
#include <LibGUI/Window.h>

namespace FileSystemAccessClient {

static RefPtr<Client> s_the = nullptr;

Client& Client::the()
{
    if (!s_the || !s_the->is_open())
        s_the = Client::construct();
    return *s_the;
}

Result Client::request_file_read_only_approved(i32 parent_window_id, String const& path)
{
    m_promise = Core::Promise<Result>::construct();
    auto parent_window_server_client_id = GUI::WindowServerConnection::the().expose_client_id();
    auto child_window_server_client_id = expose_window_server_client_id();

    GUI::WindowServerConnection::the().async_add_window_stealing_for_client(child_window_server_client_id, parent_window_id);

    ScopeGuard guard([parent_window_id, child_window_server_client_id] {
        GUI::WindowServerConnection::the().async_remove_window_stealing_for_client(child_window_server_client_id, parent_window_id);
    });

    if (path.starts_with('/')) {
        async_request_file_read_only_approved(parent_window_server_client_id, parent_window_id, path);
    } else {
        auto full_path = LexicalPath::join(Core::File::current_working_directory(), path).string();
        async_request_file_read_only_approved(parent_window_server_client_id, parent_window_id, full_path);
    }

    return m_promise->await();
}

Result Client::request_file(i32 parent_window_id, String const& path, Core::OpenMode mode)
{
    m_promise = Core::Promise<Result>::construct();
    auto parent_window_server_client_id = GUI::WindowServerConnection::the().expose_client_id();
    auto child_window_server_client_id = expose_window_server_client_id();

    GUI::WindowServerConnection::the().async_add_window_stealing_for_client(child_window_server_client_id, parent_window_id);

    ScopeGuard guard([parent_window_id, child_window_server_client_id] {
        GUI::WindowServerConnection::the().async_remove_window_stealing_for_client(child_window_server_client_id, parent_window_id);
    });

    if (path.starts_with('/')) {
        async_request_file(parent_window_server_client_id, parent_window_id, path, mode);
    } else {
        auto full_path = LexicalPath::join(Core::File::current_working_directory(), path).string();
        async_request_file(parent_window_server_client_id, parent_window_id, full_path, mode);
    }

    return m_promise->await();
}

Result Client::open_file(i32 parent_window_id, String const& window_title, StringView path, Core::OpenMode requested_access)
{
    m_promise = Core::Promise<Result>::construct();
    auto parent_window_server_client_id = GUI::WindowServerConnection::the().expose_client_id();
    auto child_window_server_client_id = expose_window_server_client_id();

    GUI::WindowServerConnection::the().async_add_window_stealing_for_client(child_window_server_client_id, parent_window_id);

    ScopeGuard guard([parent_window_id, child_window_server_client_id] {
        GUI::WindowServerConnection::the().async_remove_window_stealing_for_client(child_window_server_client_id, parent_window_id);
    });

    async_prompt_open_file(parent_window_server_client_id, parent_window_id, window_title, path, requested_access);

    return m_promise->await();
}

Result Client::save_file(i32 parent_window_id, String const& name, String const ext, Core::OpenMode requested_access)
{
    m_promise = Core::Promise<Result>::construct();
    auto parent_window_server_client_id = GUI::WindowServerConnection::the().expose_client_id();
    auto child_window_server_client_id = expose_window_server_client_id();

    GUI::WindowServerConnection::the().async_add_window_stealing_for_client(child_window_server_client_id, parent_window_id);

    ScopeGuard guard([parent_window_id, child_window_server_client_id] {
        GUI::WindowServerConnection::the().async_remove_window_stealing_for_client(child_window_server_client_id, parent_window_id);
    });

    async_prompt_save_file(parent_window_server_client_id, parent_window_id, name.is_null() ? "Untitled" : name, ext.is_null() ? "txt" : ext, Core::StandardPaths::home_directory(), requested_access);

    return m_promise->await();
}

void Client::handle_prompt_end(i32 error, Optional<IPC::File> const& fd, Optional<String> const& chosen_file)
{
    VERIFY(m_promise);

    if (error == 0) {
        m_promise->resolve({ error, fd->take_fd(), *chosen_file });
    } else {
        m_promise->resolve({ error, {}, chosen_file });
    }
}

void Client::die()
{
    if (m_promise)
        handle_prompt_end(ECONNRESET, {}, "");
}

}