commit 984a46d0bfa5ed77abeb49e15af10d7db6d2e0dc
Author: gengteng <me@gteng.org>
Date:   Tue Jun 6 09:45:14 2023 +0800

    init

diff --git a/.gitignore b/.gitignore
new file mode 100644
index 0000000..8eb581d
--- /dev/null
+++ b/.gitignore
@@ -0,0 +1,3 @@
+/target
+/Cargo.lock
+/.idea
diff --git a/Cargo.toml b/Cargo.toml
new file mode 100644
index 0000000..1886159
--- /dev/null
+++ b/Cargo.toml
@@ -0,0 +1,26 @@
+[package]
+name = "axum-valid"
+version = "0.1.0"
+description = "Validator for axum"
+authors = ["GengTeng <me@gteng.org>"]
+license = "MIT"
+keywords = [
+    "http",
+    "web",
+    "framework",
+    "validator",
+]
+categories = [
+    "asynchronous",
+    "network-programming",
+    "web-programming",
+]
+edition = "2021"
+
+
+# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
+
+[dependencies]
+axum = "0.6.18"
+serde = "1.0.163"
+validator = "0.16.0"
diff --git a/README.md b/README.md
new file mode 100644
index 0000000..640d081
--- /dev/null
+++ b/README.md
@@ -0,0 +1,2 @@
+# axum-valid
+
diff --git a/src/lib.rs b/src/lib.rs
new file mode 100644
index 0000000..4e15204
--- /dev/null
+++ b/src/lib.rs
@@ -0,0 +1,124 @@
+use axum::body::HttpBody;
+use axum::extract::rejection::{FormRejection, JsonRejection, PathRejection, QueryRejection};
+use axum::extract::{FromRequest, FromRequestParts, Path, Query};
+use axum::http::request::Parts;
+use axum::http::{Request, StatusCode};
+use axum::response::{IntoResponse, Response};
+use axum::{async_trait, BoxError, Form, Json};
+use serde::de::DeserializeOwned;
+use validator::{Validate, ValidationErrors};
+
+#[derive(Debug, Clone, Copy, Default)]
+pub struct Valid<T>(pub T);
+
+pub enum ValidRejection<E> {
+    Valid(ValidationErrors),
+    Inner(E),
+}
+
+impl<E> From<ValidationErrors> for ValidRejection<E> {
+    fn from(value: ValidationErrors) -> Self {
+        Self::Valid(value)
+    }
+}
+
+impl<E: IntoResponse> IntoResponse for ValidRejection<E> {
+    fn into_response(self) -> Response {
+        match self {
+            ValidRejection::Valid(validate_error) => {
+                (StatusCode::BAD_REQUEST, validate_error.to_string()).into_response()
+            }
+            ValidRejection::Inner(json_error) => json_error.into_response(),
+        }
+    }
+}
+
+impl From<JsonRejection> for ValidRejection<JsonRejection> {
+    fn from(value: JsonRejection) -> Self {
+        Self::Inner(value)
+    }
+}
+
+#[async_trait]
+impl<T, S, B> FromRequest<S, B> for Valid<Json<T>>
+where
+    T: DeserializeOwned + Validate,
+    B: HttpBody + Send + 'static,
+    B::Data: Send,
+    B::Error: Into<BoxError>,
+    S: Send + Sync,
+{
+    type Rejection = ValidRejection<JsonRejection>;
+
+    async fn from_request(req: Request<B>, state: &S) -> Result<Self, Self::Rejection> {
+        let json = Json::<T>::from_request(req, state).await?;
+        json.0.validate()?;
+        Ok(Valid(json))
+    }
+}
+
+impl From<QueryRejection> for ValidRejection<QueryRejection> {
+    fn from(value: QueryRejection) -> Self {
+        Self::Inner(value)
+    }
+}
+
+#[async_trait]
+impl<T, S> FromRequestParts<S> for Valid<Query<T>>
+where
+    T: DeserializeOwned + Validate,
+    S: Send + Sync,
+{
+    type Rejection = ValidRejection<QueryRejection>;
+
+    async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
+        let query = Query::<T>::from_request_parts(parts, state).await?;
+        query.validate()?;
+        Ok(Valid(query))
+    }
+}
+
+impl From<PathRejection> for ValidRejection<PathRejection> {
+    fn from(value: PathRejection) -> Self {
+        Self::Inner(value)
+    }
+}
+
+#[async_trait]
+impl<T, S> FromRequestParts<S> for Valid<Path<T>>
+where
+    T: DeserializeOwned + Validate + Send,
+    S: Send + Sync,
+{
+    type Rejection = ValidRejection<PathRejection>;
+
+    async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> {
+        let path = Path::<T>::from_request_parts(parts, state).await?;
+        path.validate()?;
+        Ok(Valid(path))
+    }
+}
+
+impl From<FormRejection> for ValidRejection<FormRejection> {
+    fn from(value: FormRejection) -> Self {
+        Self::Inner(value)
+    }
+}
+
+#[async_trait]
+impl<T, S, B> FromRequest<S, B> for Valid<Form<T>>
+where
+    T: DeserializeOwned + Validate,
+    B: HttpBody + Send + 'static,
+    B::Data: Send,
+    B::Error: Into<BoxError>,
+    S: Send + Sync,
+{
+    type Rejection = ValidRejection<FormRejection>;
+
+    async fn from_request(req: Request<B>, state: &S) -> Result<Self, Self::Rejection> {
+        let form = Form::<T>::from_request(req, state).await?;
+        form.validate()?;
+        Ok(Valid(form))
+    }
+}