From 7bd2628014eb70f60bdef97d357549eb0a9a66d8 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dani=C3=ABlle=20Huisman?= <danielle@huisman.me>
Date: Fri, 24 Jan 2025 15:16:20 +0100
Subject: [PATCH] feat(shield-axum): add user required extractor

---
 .../integrations/shield-axum/src/extract.rs   | 20 +++++++++++++++++++
 1 file changed, 20 insertions(+)

diff --git a/packages/integrations/shield-axum/src/extract.rs b/packages/integrations/shield-axum/src/extract.rs
index f3bd391..90468b6 100644
--- a/packages/integrations/shield-axum/src/extract.rs
+++ b/packages/integrations/shield-axum/src/extract.rs
@@ -61,3 +61,23 @@ impl<S: Send + Sync, U: User + Clone + 'static> FromRequestParts<S> for ExtractU
             ))
     }
 }
+
+pub struct UserRequired<U: User>(pub U);
+
+#[async_trait]
+impl<S: Send + Sync, U: User + Clone + 'static> FromRequestParts<S> for UserRequired<U> {
+    type Rejection = (StatusCode, &'static str);
+
+    async fn from_request_parts(parts: &mut Parts, _state: &S) -> Result<Self, Self::Rejection> {
+        parts
+            .extensions
+            .get::<Option<U>>()
+            .cloned()
+            .ok_or((
+                StatusCode::INTERNAL_SERVER_ERROR,
+                "Can't extract Shield user. Is `ShieldLayer` enabled?",
+            ))
+            .and_then(|user| user.ok_or((StatusCode::UNAUTHORIZED, "Unauthorized")))
+            .map(UserRequired)
+    }
+}