sword-layers 0.1.6

Tower layers for the Sword web framework
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

use crate::ResponseFnMapper;
use axum::{
    http::StatusCode,
    response::{IntoResponse, Response},
};
use axum_responses::JsonResponse;
use tower::{ServiceBuilder, util::MapResponseLayer};
use tower_layer::{Identity, Stack};

pub struct NotFoundLayer;

pub type ResponseFnMapperServiceLayer =
    ServiceBuilder<Stack<MapResponseLayer<ResponseFnMapper>, Identity>>;

impl NotFoundLayer {
    pub fn new() -> ResponseFnMapperServiceLayer {
        ServiceBuilder::new().map_response(|r: Response| {
            if r.status() != StatusCode::NOT_FOUND {
                return r;
            }

            JsonResponse::NotFound()
                .message("The requested resource was not found.")
                .into_response()
        })
    }
}