diff options
Diffstat (limited to 'src/repo/login/extract.rs')
| -rw-r--r-- | src/repo/login/extract.rs | 55 |
1 files changed, 55 insertions, 0 deletions
diff --git a/src/repo/login/extract.rs b/src/repo/login/extract.rs new file mode 100644 index 0000000..a068bc0 --- /dev/null +++ b/src/repo/login/extract.rs @@ -0,0 +1,55 @@ +use axum::{ + extract::{FromRequestParts, State}, + http::{request::Parts, StatusCode}, + response::{IntoResponse, Response}, +}; + +use super::Login; +use crate::{app::App, clock::RequestedAt, error::InternalError, login::extract::IdentityToken}; + +#[async_trait::async_trait] +impl FromRequestParts<App> for Login { + type Rejection = LoginError<InternalError>; + + async fn from_request_parts(parts: &mut Parts, state: &App) -> Result<Self, Self::Rejection> { + // After Rust 1.82 (and #[feature(min_exhaustive_patterns)] lands on + // stable), the following can be replaced: + // + // let Ok(identity_token) = IdentityToken::from_request_parts(parts, state).await; + let identity_token = IdentityToken::from_request_parts(parts, state).await?; + let RequestedAt(used_at) = RequestedAt::from_request_parts(parts, state).await?; + + let secret = identity_token.secret().ok_or(LoginError::Unauthorized)?; + + let app = State::<App>::from_request_parts(parts, state).await?; + let login = app.logins().validate(secret, used_at).await?; + + login.ok_or(LoginError::Unauthorized) + } +} + +pub enum LoginError<E> { + Failure(E), + Unauthorized, +} + +impl<E> IntoResponse for LoginError<E> +where + E: IntoResponse, +{ + fn into_response(self) -> Response { + match self { + Self::Unauthorized => (StatusCode::UNAUTHORIZED, "unauthorized").into_response(), + Self::Failure(e) => e.into_response(), + } + } +} + +impl<E> From<E> for LoginError<InternalError> +where + E: Into<InternalError>, +{ + fn from(err: E) -> Self { + Self::Failure(err.into()) + } +} |
