summaryrefslogtreecommitdiff
path: root/Userland/Libraries/LibCore/Promise.h
blob: a996eb7d451f904c98e9054b40ffb2e5624117bf (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
/*
 * Copyright (c) 2021, Kyle Pereira <hey@xylepereira.me>
 * Copyright (c) 2022, kleines Filmröllchen <filmroellchen@serenityos.org>
 *
 * SPDX-License-Identifier: BSD-2-Clause
 */

#pragma once

#include <LibCore/EventLoop.h>
#include <LibCore/Object.h>

namespace Core {

template<typename Result>
class Promise : public Object {
    C_OBJECT(Promise);

public:
    Function<ErrorOr<void>(Result&)> on_resolved;

    ErrorOr<void> resolve(Result&& result)
    {
        m_pending_or_error = move(result);

        if (on_resolved)
            return on_resolved(m_pending_or_error->value());
        return {};
    }

    void cancel(Error error)
    {
        m_pending_or_error = move(error);
    }

    bool is_canceled()
    {
        return m_pending_or_error.has_value() && m_pending_or_error->is_error();
    }

    bool is_resolved() const
    {
        return m_pending_or_error.has_value() && !m_pending_or_error->is_error();
    }

    ErrorOr<Result> await()
    {
        while (!m_pending_or_error.has_value())
            Core::EventLoop::current().pump();

        return m_pending_or_error.release_value();
    }

    // Converts a Promise<A> to a Promise<B> using a function func: A -> B
    template<typename T>
    RefPtr<Promise<T>> map(T func(Result&))
    {
        RefPtr<Promise<T>> new_promise = Promise<T>::construct();
        on_resolved = [new_promise, func](Result& result) {
            auto t = func(result);
            return new_promise->resolve(move(t));
        };
        return new_promise;
    }

private:
    Promise() = default;
    Promise(Object* parent)
        : Object(parent)
    {
    }

    Optional<ErrorOr<Result>> m_pending_or_error;
};

}