diff options
| author | Owen Jacobson <owen@grimoire.ca> | 2024-09-04 00:28:35 -0400 |
|---|---|---|
| committer | Owen Jacobson <owen@grimoire.ca> | 2024-09-04 00:28:35 -0400 |
| commit | 2965a788cfcf4a0386cb8832e0d96491bf54c1d3 (patch) | |
| tree | 096b00f64b092396e99d04ebe124fa15d734f6c1 /src/login/extract.rs | |
| parent | 289e99ba977ebe6c4599141bc368c17f9905ffcc (diff) | |
Display a different / page depending on whether the current identity is valid or not.
This is mostly a proof of concept for the implementation of form login implemented in previous commits, but it _is_ useful as it controls whether the / page shows login, or shows logout.
From here, chat is next!
Diffstat (limited to 'src/login/extract.rs')
| -rw-r--r-- | src/login/extract.rs | 69 |
1 files changed, 0 insertions, 69 deletions
diff --git a/src/login/extract.rs b/src/login/extract.rs deleted file mode 100644 index d39e3df..0000000 --- a/src/login/extract.rs +++ /dev/null @@ -1,69 +0,0 @@ -use std::convert::Infallible; - -use axum::{ - extract::FromRequestParts, - http::request::Parts, - response::{IntoResponseParts, ResponseParts}, -}; -use axum_extra::extract::cookie::{Cookie, CookieJar}; - -// The usage pattern here - receive the extractor as an argument, return it in -// the response - is heavily modelled after CookieJar's own intended usage. -pub struct IdentityToken { - cookies: CookieJar, -} - -impl IdentityToken { - /// Get the identity token sent in the request, if any. If the identity was - /// not sent, or if it has previously been [clear]ed, then this will return - /// [None]. If the identity has previously been [set], then this will return - /// that token. - pub fn token(&self) -> Option<&str> { - self.cookies.get(IDENTITY_COOKIE).map(Cookie::value) - } - - /// Positively set the identity token, and ensure that it will be sent back - /// to the client when this extractor is included in a response. - pub fn set(self, token: &str) -> Self { - let identity_cookie = Cookie::build((IDENTITY_COOKIE, String::from(token))) - .http_only(true) - .permanent() - .build(); - - IdentityToken { - cookies: self.cookies.add(identity_cookie), - } - } - - /// Remove the identity token and ensure that it will be cleared when this - /// extractor is included in a response. - pub fn clear(self) -> Self { - IdentityToken { - cookies: self.cookies.remove(IDENTITY_COOKIE), - } - } -} - -const IDENTITY_COOKIE: &str = "identity"; - -#[async_trait::async_trait] -impl<S> FromRequestParts<S> for IdentityToken -where - S: Send + Sync, -{ - type Rejection = Infallible; - - async fn from_request_parts(parts: &mut Parts, state: &S) -> Result<Self, Self::Rejection> { - let cookies = CookieJar::from_request_parts(parts, state).await?; - Ok(IdentityToken { cookies }) - } -} - -impl IntoResponseParts for IdentityToken { - type Error = Infallible; - - fn into_response_parts(self, res: ResponseParts) -> Result<ResponseParts, Self::Error> { - let IdentityToken { cookies } = self; - cookies.into_response_parts(res) - } -} |
