d5939db
use time::Duration;
use axum_extra::extract::cookie::Cookie;
use axum_extra::extract::cookie::CookieJar;
use axum_extra::extract::cookie::SameSite;
use crate::config::AuthConfig;
use crate::features::TempKey;
pub(crate) fn temp_key_to_cookie(config: &AuthConfig, key: &TempKey) -> Cookie<'static> {
let mut builder = Cookie::build((key.name.clone(), key.value.clone()))
.http_only(key.http_only)
.secure(key.secure)
.max_age(Duration::seconds(key.ttl_seconds as i64))
.path(key.path.clone().unwrap_or_else(|| "/".to_string()))
.same_site(parse_same_site(&key.same_site, &config.cookie_same_site));
if let Some(domain) = key.domain.clone().or_else(|| config.cookie_domain.clone()) {
builder = builder.domain(domain);
}
builder.build()
}
fn parse_same_site(cookie_specific: &str, fallback: &str) -> SameSite {
let raw = if cookie_specific.is_empty() {
fallback
} else {
cookie_specific
};
match raw.to_ascii_lowercase().as_str() {
"strict" => SameSite::Strict,
"lax" => SameSite::Lax,
"none" => SameSite::None,
_ => SameSite::Strict,
}
}
pub(crate) fn read_cookie_value(jar: &CookieJar, name: &str) -> Option<String> {
jar.get(name).map(|c| c.value().to_string())
}
pub(crate) fn remove_cookie(jar: CookieJar, name: &str) -> CookieJar {
jar.remove(Cookie::from(name.to_string()))
}