adam/leptos-use
1
0
Fork 0
mirror of https://github.com/adoyle0/leptos-use.git synced 2025-02-02 10:54:15 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactoring use_cookie

Browse source
This commit is contained in:
Maccesch 2024-01-29 19:15:55 +00:00
parent e5e6eae9d1
commit 37bf52b5c1
1 changed files with 20 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

36
src/use_cookie.rs
View file

@ -71,21 +71,11 @@ pub fn use_cookie_with_options(
cookie_name: &str, cookie_name: &str,
options: UseCookieOptions, options: UseCookieOptions,
) -> Option<Cookie<'static>> { ) -> Option<Cookie<'static>> {
let cookies; let UseCookieOptions {
ssr_cookies_header_getter,
} = options;
#[cfg(feature = "ssr")] let cookie = read_cookies_string(ssr_cookies_header_getter);
{
cookies = (options.ssr_cookies_header_getter)();
}
#[cfg(not(feature = "ssr"))]
{
use wasm_bindgen::JsCast;
let js_value: wasm_bindgen::JsValue = leptos::document().into();
let document: web_sys::HtmlDocument = js_value.unchecked_into();
cookies = document.cookie().unwrap_or_default();
}
Cookie::split_parse_encoded(cookies) Cookie::split_parse_encoded(cookies)
.filter_map(|cookie| cookie.ok()) .filter_map(|cookie| cookie.ok())
@ -98,14 +88,14 @@ pub fn use_cookie_with_options(
pub struct UseCookieOptions { pub struct UseCookieOptions {
/// Getter function to return the string value of the cookie header. /// Getter function to return the string value of the cookie header.
/// When you use one of the features "axum" or "actix" there's a valid default implementation provided. /// When you use one of the features "axum" or "actix" there's a valid default implementation provided.
ssr_cookies_header_getter: Rc<dyn Fn() -> String>, ssr_cookies_header_getter: Box<dyn Fn() -> String>,
} }
impl Default for UseCookieOptions { impl Default for UseCookieOptions {
#[allow(dead_code)] #[allow(dead_code)]
fn default() -> Self { fn default() -> Self {
Self { Self {
ssr_cookies_header_getter: Rc::new(move || { ssr_cookies_header_getter: Box::new(move || {
#[cfg(feature = "ssr")] #[cfg(feature = "ssr")]
{ {
#[cfg(any(feature = "axum", feature = "actix"))] #[cfg(any(feature = "axum", feature = "actix"))]
@ -154,3 +144,17 @@ impl Default for UseCookieOptions {
} }
} }
} }
fn read_cookies_string(ssr_cookies_header_getter: Box<dyn Fn() -> String>) -> String {
#[cfg(feature = "ssr")]
ssr_cookies_header_getter();
#[cfg(not(feature = "ssr"))]
{
use wasm_bindgen::JsCast;
let js_value: wasm_bindgen::JsValue = leptos::document().into();
let document: web_sys::HtmlDocument = js_value.unchecked_into();
document.cookie().unwrap_or_default()
}
}