aleo_development_server/helpers/
or_reject.rs

1// Copyright (C) 2019-2023 Aleo Systems Inc.
2// This file is part of the Aleo SDK library.
3
4// The Aleo SDK library is free software: you can redistribute it and/or modify
5// it under the terms of the GNU General Public License as published by
6// the Free Software Foundation, either version 3 of the License, or
7// (at your option) any later version.
8
9// The Aleo SDK library is distributed in the hope that it will be useful,
10// but WITHOUT ANY WARRANTY; without even the implied warranty of
11// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12// GNU General Public License for more details.
13
14// You should have received a copy of the GNU General Public License
15// along with the Aleo SDK library. If not, see <https://www.gnu.org/licenses/>.
16
17use crate::RestError;
18
19use anyhow::Result;
20use warp::{reject, Rejection};
21
22/// A trait to unwrap a `Result` or `Reject`.
23pub trait OrReject<T> {
24    /// Returns the result if it is successful, otherwise returns a rejection.
25    fn or_reject(self) -> Result<T, Rejection>;
26}
27
28impl<T> OrReject<T> for anyhow::Result<T> {
29    /// Returns the result if it is successful, otherwise returns a rejection.
30    fn or_reject(self) -> Result<T, Rejection> {
31        self.map_err(|e| reject::custom(RestError::Request(e.to_string())))
32    }
33}
34
35impl<T> OrReject<T> for Result<T, tokio::task::JoinError> {
36    /// Returns the result if it is successful, otherwise returns a rejection.
37    fn or_reject(self) -> Result<T, Rejection> {
38        self.map_err(|e| reject::custom(RestError::Request(e.to_string())))
39    }
40}