expected.h 6.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208
  1. #pragma once
  2. #ifdef __cpp_lib_expected
  3. #if __cpp_lib_expected >= 202202L
  4. #include <expected>
  5. namespace jvalidate::detail {
  6. using std::expected;
  7. using std::unexpected;
  8. inline constexpr std::unexpect_t unexpect{};
  9. }
  10. #endif
  11. #else
  12. #include <initializer_list>
  13. #include <optional>
  14. #include <utility>
  15. #include <variant>
  16. #include <jvalidate/detail/expect.h>
  17. namespace jvalidate::detail {
  18. inline constexpr struct unexpect_t {
  19. } unexpect{};
  20. template <typename E> class unexpected {
  21. public:
  22. template <typename Err = E>
  23. constexpr explicit unexpected(Err && e) : error_(std::forward<Err>(e)) {}
  24. template <typename... Args>
  25. constexpr explicit unexpected(std::in_place_t, Args &&... args)
  26. : error_{std::forward<Args>(args)...} {}
  27. template <typename U, typename... Args>
  28. constexpr explicit unexpected(std::in_place_t, std::initializer_list<U> il, Args &&... args)
  29. : error_{il, std::forward<Args>(args)...} {}
  30. constexpr const E & error() const & noexcept { return error_; }
  31. constexpr E & error() & noexcept { return error_; }
  32. constexpr E && error() && noexcept { return std::move(error_); }
  33. constexpr void swap(unexpected & other) noexcept(std::is_nothrow_swappable_v<E>) {
  34. using std::swap;
  35. swap(error(), other.error());
  36. }
  37. template <typename E2> friend constexpr bool operator==(unexpected & x, unexpected<E2> & y) {
  38. return x.error() == y.error();
  39. }
  40. friend constexpr void swap(unexpected & x, unexpected & y) noexcept(noexcept(x.swap(y))) {
  41. return x.swap(y);
  42. }
  43. private:
  44. E error_;
  45. };
  46. template <typename E> unexpected(E) -> unexpected<E>;
  47. template <typename E> class bad_expected_access;
  48. template <> class bad_expected_access<void> : public std::exception {
  49. public:
  50. char const * what() const noexcept override { return "unexpected"; }
  51. };
  52. template <typename E> class bad_expected_access : public bad_expected_access<void> {
  53. public:
  54. bad_expected_access(E error) : error_(std::move(error)) {}
  55. char const * what() const noexcept override { return error_.c_str(); }
  56. constexpr const E & error() const & noexcept { return error_; }
  57. constexpr E & error() & noexcept { return error_; }
  58. constexpr E && error() && noexcept { return std::move(error_); }
  59. private:
  60. E error_;
  61. };
  62. template <typename T, typename E> class expected {
  63. public:
  64. constexpr expected() = default;
  65. template <typename U, typename G>
  66. constexpr explicit(!std::is_convertible_v<const U &, T> && !std::is_convertible_v<const G &, E>)
  67. expected(expected<U, G> const & other) {
  68. if (other.has_value()) {
  69. *this = *other;
  70. } else {
  71. *this = other.error();
  72. }
  73. }
  74. template <typename U, typename G>
  75. constexpr explicit(!std::is_convertible_v<U, T> && !std::is_convertible_v<G, E>)
  76. expected(expected<U, G> && other) {
  77. if (other.has_value()) {
  78. *this = *std::move(other);
  79. } else {
  80. *this = std::move(other).error();
  81. }
  82. }
  83. template <typename U = std::remove_cv_t<T>>
  84. constexpr explicit(!std::is_convertible_v<U, T>) expected(U && v)
  85. : value_(std::in_place_index<0>, std::forward<T>(v)) {}
  86. template <typename G>
  87. constexpr explicit(!std::is_convertible_v<const G &, E>) expected(unexpected<G> const & e)
  88. : value_(std::in_place_index<1>, e.error()) {}
  89. template <typename G>
  90. constexpr explicit(!std::is_convertible_v<G, E>) expected(unexpected<G> && e)
  91. : value_(std::in_place_index<1>, std::move(e).error()) {}
  92. constexpr explicit operator bool() const noexcept { return has_value(); }
  93. constexpr bool has_value() const noexcept { return value_.index() == 0; }
  94. constexpr const T * operator->() const noexcept { return std::get_if<0>(&value_); }
  95. constexpr T * operator->() noexcept { return std::get_if<0>(&value_); }
  96. constexpr const T & operator*() const & noexcept { return *std::get_if<0>(&value_); }
  97. constexpr T & operator*() & noexcept { return *std::get_if<0>(&value_); }
  98. constexpr T && operator*() && noexcept { return std::move(*std::get_if<0>(&value_)); }
  99. constexpr const T & value() const & noexcept {
  100. if (JVALIDATE_LIKELY(has_value())) {
  101. return operator*();
  102. }
  103. throw bad_expected_access(error());
  104. }
  105. constexpr T & value() & noexcept {
  106. if (JVALIDATE_LIKELY(has_value())) {
  107. return operator*();
  108. }
  109. throw bad_expected_access(std::as_const(error()));
  110. }
  111. constexpr T && value() && noexcept {
  112. if (JVALIDATE_LIKELY(has_value())) {
  113. return std::move(*this).operator*();
  114. }
  115. throw bad_expected_access(std::move(error()));
  116. }
  117. constexpr const E & error() const & noexcept { return *std::get_if<1>(&value_); }
  118. constexpr E & error() & noexcept { return *std::get_if<1>(&value_); }
  119. constexpr E && error() && noexcept { return std::move(*std::get_if<1>(&value_)); }
  120. private:
  121. std::variant<T, E> value_;
  122. };
  123. template <typename E> class expected<void, E> {
  124. public:
  125. constexpr expected() = default;
  126. template <typename G>
  127. constexpr explicit(!std::is_convertible_v<const G &, E>)
  128. expected(expected<void, G> const & other) {
  129. if (not other.has_value()) {
  130. *this = other.error();
  131. }
  132. }
  133. template <typename G>
  134. constexpr explicit(!std::is_convertible_v<G, E>) expected(expected<void, G> && other) {
  135. if (not other.has_value()) {
  136. *this = std::move(other).error();
  137. }
  138. }
  139. template <typename G>
  140. constexpr explicit(!std::is_convertible_v<const G &, E>) expected(unexpected<G> const & e)
  141. : value_(std::in_place_index<1>, e.error()) {}
  142. template <typename G>
  143. constexpr explicit(!std::is_convertible_v<G, E>) expected(unexpected<G> && e)
  144. : value_(std::in_place_index<1>, std::move(e).error()) {}
  145. constexpr explicit operator bool() const noexcept { return has_value(); }
  146. constexpr bool has_value() const noexcept { return not value_.has_value(); }
  147. constexpr void operator*() const noexcept {}
  148. void value() const & noexcept {
  149. if (JVALIDATE_LIKELY(has_value())) {
  150. return operator*();
  151. }
  152. throw bad_expected_access(error());
  153. }
  154. void value() && noexcept {
  155. if (JVALIDATE_LIKELY(has_value())) {
  156. return operator*();
  157. }
  158. throw bad_expected_access(std::move(error()));
  159. }
  160. constexpr const E & error() const & noexcept { return *value_; }
  161. constexpr E & error() & noexcept { return *value_; }
  162. constexpr E && error() && noexcept { return std::move(*value_); }
  163. private:
  164. std::optional<E> value_;
  165. };
  166. }
  167. #endif