nest-rs-guards 0.2.0

Transport-spanning guards for nestrs — one trait, three transports (HTTP, GraphQL, WS), declared once with App::builder().use_guards_global(...).
Documentation 
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

//! Convert a transport-agnostic [`Denial`] to a transport-native error
//! shape — poem [`Response`] for HTTP, [`GraphqlError`] for GraphQL.

use nest_rs_graphql::async_graphql::{Error as GraphqlError, ErrorExtensions};
use nest_rs_http::poem::http::StatusCode;
use nest_rs_http::poem::{Body, Response};

use crate::denial::Denial;

/// Convert a transport-agnostic [`Denial`] to a poem [`Response`].
pub fn denial_to_http_response(denial: Denial) -> Response {
    let status = match denial.http_status() {
        401 => StatusCode::UNAUTHORIZED,
        403 => StatusCode::FORBIDDEN,
        429 => StatusCode::TOO_MANY_REQUESTS,
        _ => StatusCode::INTERNAL_SERVER_ERROR,
    };
    let mut builder = Response::builder().status(status);
    if let Denial::RateLimited {
        retry_after_secs, ..
    } = &denial
    {
        builder = builder.header("Retry-After", retry_after_secs.to_string());
    }
    let message = match &denial {
        Denial::Internal(_) => "internal server error".to_owned(),
        _ => denial.message().to_owned(),
    };
    builder.body(Body::from_string(message))
}

/// Convert a [`Denial`] to an async-graphql error frame.
pub fn denial_to_graphql_error(denial: Denial) -> GraphqlError {
    let code = match denial.http_status() {
        401 => "UNAUTHENTICATED",
        403 => "FORBIDDEN",
        429 => "RATE_LIMITED",
        _ => "INTERNAL",
    };
    let message = match &denial {
        Denial::Internal(_) => "internal server error".to_owned(),
        _ => denial.message().to_owned(),
    };
    GraphqlError::new(message).extend_with(|_, e| e.set("code", code))
}