atlasserver 0.1.2

Atlas is a rust library for the purpose of composing REST APIs out of re-usable and extensible modules, specifically with supporting the networking needs of online gaming services in mind.
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

use crate::schema::{self, RejectionResponse};
use std::convert::Infallible;
use warp::{hyper::StatusCode, reject::Reject, Rejection, Reply};

#[derive(Debug)]
pub enum SessionFailure {
	Invalid,
	SessionNotFound,
}

impl Reject for SessionFailure {}

#[allow(clippy::missing_errors_doc)]
//TODO: this a nightly only warning currently
#[allow(clippy::unused_async)]
pub async fn handle_rejection(
	err: Rejection,
) -> Result<impl Reply, Infallible> {
	err.find::<SessionFailure>().map_or_else(
        || {
			tracing::error!("unhandled rejection {:?}", err);

            Ok(warp::reply::with_status(
                crate::pbwarp::protobuf_reply(&RejectionResponse::default(), None),
                StatusCode::INTERNAL_SERVER_ERROR,
            ))
        },
        |session_failure| {
            let mut rejection = schema::RejectionResponse::default();

            match session_failure {
                SessionFailure::Invalid => rejection.set_sessionFilterRejection(
                    schema::RejectionResponse_SessionFilterRejection::INVALID,
                ),
                SessionFailure::SessionNotFound => rejection.set_sessionFilterRejection(
                    schema::RejectionResponse_SessionFilterRejection::SESSION_NOT_FOUND,
                ),
            };

            Ok(warp::reply::with_status(
                crate::pbwarp::protobuf_reply(&rejection, None),
                StatusCode::ACCEPTED,
            ))
        },
    )
}