handle invald payload Json reqeust with rejection schema response

Cargo.toml

@@ -51,7 +51,7 @@ rusty-sidekiq = { version = "0.8.2", default-features = false }
 async-trait = "0.1.74"
 bb8 = "0.8.1"
 
-axum = "0.7.1"
+axum = { version = "0.7.1", features = ["macros"] }
 axum-extra = { version = "0.9", features = ["cookie"] }
 regex = "1"
 lazy_static = "1.4.0"

src/controller/format.rs

@@ -8,8 +8,7 @@
 //! Rust struct.
 //!
 //! ```rust
-//! use loco_rs::{controller::format, Result};
-//! use axum::Json;
+//! use loco_rs::{controller::{Json, format}, Result};
 //!
 //! pub struct Health {
 //!     pub ok: bool,
@@ -20,19 +19,17 @@
 //! }
 //! ```
 
+use crate::{controller::Json, Result};
 use axum::{
     body::Body,
     http::{response::Builder, HeaderName, HeaderValue},
     response::{Html, Response},
-    Json,
 };
 use axum_extra::extract::cookie::Cookie;
 use bytes::{BufMut, BytesMut};
 use hyper::{header, StatusCode};
 use serde::Serialize;
 
-use crate::Result;
-
 /// Returns an empty response.
 ///
 /// # Example:
@@ -84,10 +81,9 @@ pub fn text(t: &str) -> Result<String> {
 ///
 /// ```rust
 /// use loco_rs::{
-///     controller::format,
+///     controller::{Json, format},
 ///     Result,
 /// };
-/// use axum::Json;
 ///
 /// pub struct Health {
 ///     pub ok: bool,

src/controller/health.rs

@@ -2,11 +2,11 @@
 //! reporting. These routes are commonly used to monitor the health of the
 //! application and its dependencies.
 
-use axum::{extract::State, routing::get, Json};
+use axum::{extract::State, routing::get};
 use serde::Serialize;
 
 use super::{format, routes::Routes};
-use crate::{app::AppContext, redis, Result};
+use crate::{app::AppContext, controller::Json, redis, Result};
 
 /// Represents the health status of the application.
 #[derive(Serialize)]

src/controller/mod.rs

@@ -78,7 +78,6 @@ pub use app_routes::{AppRoutes, ListRoutes};
 use axum::{
     http::StatusCode,
     response::{IntoResponse, Response},
-    Json,
 };
 use colored::Colorize;
 pub use routes::Routes;
@@ -98,10 +97,9 @@ use crate::{errors::Error, Result};
 /// # Example
 ///
 /// ```rust
-/// use axum::Json;
 /// use loco_rs::{
 ///     Result,
-///     controller::{format, unauthorized}
+///     controller::{format, Json, unauthorized}
 /// };
 ///
 /// async fn login() -> Result<Json<()>> {
@@ -153,6 +151,17 @@ impl ErrorDetail {
     }
 }
 
+use axum::extract::FromRequest;
+#[derive(Debug, FromRequest)]
+#[from_request(via(axum::Json), rejection(Error))]
+pub struct Json<T>(pub T);
+
+impl<T: Serialize> IntoResponse for Json<T> {
+    fn into_response(self) -> axum::response::Response {
+        axum::Json(self.0).into_response()
+    }
+}
+
 impl IntoResponse for Error {
     /// Convert an `Error` into an HTTP response.
     fn into_response(self) -> Response {

src/controller/ping.rs

@@ -2,11 +2,11 @@
 //! reporting. These routes are commonly used to monitor the health of the
 //! application and its dependencies.
 
-use axum::{routing::get, Json};
+use axum::routing::get;
 use serde::Serialize;
 
 use super::{format, routes::Routes};
-use crate::Result;
+use crate::{controller::Json, Result};
 
 /// Represents the health status of the application.
 #[derive(Serialize)]

src/errors.rs

@@ -1,9 +1,12 @@
 //! # Application Error Handling
 
-use axum::http::{
-    header::{InvalidHeaderName, InvalidHeaderValue},
-    method::InvalidMethod,
-    StatusCode,
+use axum::{
+    extract::rejection::JsonRejection,
+    http::{
+        header::{InvalidHeaderName, InvalidHeaderValue},
+        method::InvalidMethod,
+        StatusCode,
+    },
 };
 use lettre::{address::AddressError, transport::smtp};
 
@@ -47,6 +50,9 @@ pub enum Error {
     #[error(transparent)]
     JSON(serde_json::Error),
 
+    #[error(transparent)]
+    JsonRejection(#[from] JsonRejection),
+
     #[error("cannot parse `{1}`: {0}")]
     YAMLFile(#[source] serde_yaml::Error, String),
 

src/prelude.rs

@@ -2,7 +2,6 @@ pub use async_trait::async_trait;
 pub use axum::{
     extract::{Form, Path, State},
     routing::{delete, get, post, put},
-    Json,
 };
 pub use axum_extra::extract::cookie;
 pub use chrono::NaiveDateTime as DateTime;
@@ -14,7 +13,7 @@ pub use sea_orm::{ActiveModelTrait, EntityTrait, IntoActiveModel, ModelTrait, Se
 pub use crate::controller::middleware::auth;
 pub use crate::{
     app::AppContext,
-    controller::{format, not_found, unauthorized, Routes},
+    controller::{format, not_found, unauthorized, Json, Routes},
     errors::Error,
     mailer,
     mailer::Mailer,