leptos-use/src/use_cookie.rs

use cookie::Cookie;

/// Get a cookie by name, for both SSR and CSR
///
/// ## Demo
///
/// [Link to Demo](https://github.com/Synphonyte/leptos-use/tree/main/examples/use_cookie)
///
/// ## Usage
///
/// This provides you with the cookie that has been set. For more details on how to use the cookie provided, refer: https://docs.rs/cookie/0.18/cookie/struct.Cookie.html
///
/// ```
/// # use leptos::*;
/// # use leptos_use::use_cookie;
/// #
/// # #[component]
/// # fn Demo() -> impl IntoView {
/// if let Some(cookie) = use_cookie("auth") {
///     view! {
///         <div>
///             format!("'auth' cookie set to `{}`", cookie.value())
///         </div>
///     }.into_view()
/// } else {
///     view! {
///         <div>
///             "No 'auth' cookie set"
///         </div>
///     }.into_view()
/// }
/// # }
/// ```
///
/// Server-Side Rendering
///
/// This works equally well on the server or the client.
/// On the server this function gets the cookie from the HTTP request header.
///
/// If you're using `axum` you have to enable the `"axum"` feature in your Cargo.toml.
/// In case it's `actix-web` enable the feature `"actix"`.
pub fn use_cookie(cookie_name: &str) -> Option<Cookie<'static>> {
    let cookies;
    #[cfg(feature = "ssr")]
    {
        use http::HeaderValue;
        use leptos::expect_context;

        let headers;
        #[cfg(feature = "actix")]
        {
            headers = expect_context::<actix_web::HttpRequest>().headers().clone();
        }
        #[cfg(feature = "axum")]
        {
            headers = expect_context::<leptos_axum::RequestParts>().headers;
        }
        cookies = headers
            .get(http::header::COOKIE)
            .cloned()
            .unwrap_or_else(|| HeaderValue::from_static(""))
            .to_str()
            .unwrap_or_default()
            .to_owned();
    }
    #[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)
        .filter_map(|cookie| cookie.ok())
        .find(|cookie| cookie.name() == cookie_name)
        .map(|cookie| cookie.into_owned())
}