adam/leptos-use
1
0
Fork 0
mirror of https://github.com/adoyle0/leptos-use.git synced 2025-02-02 02:54:14 -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,
options: UseCookieOptions,
) -> Option<Cookie<'static>> {
let cookies;
let UseCookieOptions {
ssr_cookies_header_getter,
} = options;
#[cfg(feature = "ssr")]
{
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();
}
let cookie = read_cookies_string(ssr_cookies_header_getter);
Cookie::split_parse_encoded(cookies)
.filter_map(|cookie| cookie.ok())
@ -98,14 +88,14 @@ pub fn use_cookie_with_options(
pub struct UseCookieOptions {
/// 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.
ssr_cookies_header_getter: Rc<dyn Fn() -> String>,
ssr_cookies_header_getter: Box<dyn Fn() -> String>,
}
impl Default for UseCookieOptions {
#[allow(dead_code)]
fn default() -> Self {
Self {
ssr_cookies_header_getter: Rc::new(move || {
ssr_cookies_header_getter: Box::new(move || {
#[cfg(feature = "ssr")]
{
#[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()
}
}