Merge branch 'main' into prop/preserve-state

1 files changed, 16 insertions, 2 deletions

diff --git a/src/token/extract/identity.rs b/src/token/extract/identity.rs
index a69f509..acfd7ae 100644
--- a/src/token/extract/identity.rs
+++ b/src/token/extract/identity.rs
@@ -1,5 +1,5 @@
 use axum::{
-    extract::{FromRequestParts, State},
+    extract::{FromRequestParts, OptionalFromRequestParts, State},
     http::request::Parts,
     response::{IntoResponse, Response},
 };
@@ -20,7 +20,6 @@ pub struct Identity {
     pub login: Login,
 }
 
-#[async_trait::async_trait]
 impl FromRequestParts<App> for Identity {
     type Rejection = LoginError<Internal>;
 
@@ -39,6 +38,21 @@ impl FromRequestParts<App> for Identity {
     }
 }
 
+impl OptionalFromRequestParts<App> for Identity {
+    type Rejection = LoginError<Internal>;
+
+    async fn from_request_parts(
+        parts: &mut Parts,
+        state: &App,
+    ) -> Result<Option<Self>, Self::Rejection> {
+        match <Self as FromRequestParts<App>>::from_request_parts(parts, state).await {
+            Ok(identity) => Ok(Some(identity)),
+            Err(LoginError::Unauthorized) => Ok(None),
+            Err(other) => Err(other),
+        }
+    }
+}
+
 pub enum LoginError<E> {
     Failure(E),
     Unauthorized,