aboutsummaryrefslogtreecommitdiff
path: root/toolsrc/include/vcpkg_expected.h
blob: 9637ec0878748938460a1ad75fc59db1de126392 (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
#pragma once

#include "vcpkg_Checks.h"
#include <system_error>

namespace vcpkg
{
    template<class Err>
    struct ErrorHolder
    {
        ErrorHolder() : m_is_error(false) {}
        template<class U>
        ErrorHolder(U&& err) : m_is_error(true), m_err(std::forward<U>(err))
        {
        }

        constexpr bool has_error() const { return m_is_error; }

        const Err& error() const { return m_err; }
        Err& error() { return m_err; }

        CStringView to_string() const { return "value was error"; }

    private:
        bool m_is_error;
        Err m_err;
    };

    template<>
    struct ErrorHolder<std::error_code>
    {
        ErrorHolder() = default;
        ErrorHolder(const std::error_code& err) : m_err(err) {}

        constexpr bool has_error() const { return bool(m_err); }

        const std::error_code& error() const { return m_err; }
        std::error_code& error() { return m_err; }

        CStringView to_string() const { return "value was error"; }

    private:
        std::error_code m_err;
    };

    template<class T, class S>
    class ExpectedT
    {
    public:
        constexpr ExpectedT() = default;

        // Constructors are intentionally implicit

        ExpectedT(const S& s) : m_s(s) {}
        ExpectedT(S&& s) : m_s(std::move(s)) {}

        ExpectedT(const T& t) : m_t(t) {}
        ExpectedT(T&& t) : m_t(std::move(t)) {}

        ExpectedT(const ExpectedT&) = default;
        ExpectedT(ExpectedT&&) = default;
        ExpectedT& operator=(const ExpectedT&) = default;
        ExpectedT& operator=(ExpectedT&&) = default;

        explicit constexpr operator bool() const noexcept { return !m_s.has_error(); }
        constexpr bool has_value() const noexcept { return !m_s.has_error(); }

        T&& value_or_exit(const LineInfo& line_info) &&
        {
            exit_if_error(line_info);
            return std::move(this->m_t);
        }

        const T& value_or_exit(const LineInfo& line_info) const &
        {
            exit_if_error(line_info);
            return this->m_t;
        }

        const S& error() const & { return this->m_s.error(); }

        S&& error() && { return std::move(this->m_s.error()); }

        const T* get() const
        {
            if (!this->has_value())
            {
                return nullptr;
            }
            return &this->m_t;
        }

        T* get()
        {
            if (!this->has_value())
            {
                return nullptr;
            }
            return &this->m_t;
        }

    private:
        void exit_if_error(const LineInfo& line_info) const
        {
            Checks::check_exit(line_info, !m_s.has_error(), m_s.to_string());
        }

        ErrorHolder<S> m_s;
        T m_t;
    };

    template<class T>
    using Expected = ExpectedT<T, std::error_code>;
}