leptos_reactive 0.0.2

#[cfg(any(feature = "hydrate"))]
use std::collections::{HashMap, HashSet};
#[cfg(any(feature = "ssr"))]
use std::{collections::HashMap, future::Future, pin::Pin};

#[cfg(any(feature = "hydrate"))]
use crate::{Scope, StreamingResourceId};

#[derive(Default)]
pub struct SharedContext {
    #[cfg(feature = "hydrate")]
    pub completed: Vec<web_sys::Element>,
    #[cfg(feature = "hydrate")]
    pub events: Vec<()>,
    pub context: Option<HydrationContext>,
    #[cfg(feature = "hydrate")]
    pub registry: HashMap<String, web_sys::Element>,
    #[cfg(feature = "hydrate")]
    pub pending_resources: HashSet<StreamingResourceId>,
    #[cfg(feature = "hydrate")]
    pub resolved_resources: HashMap<StreamingResourceId, String>,
    #[cfg(feature = "ssr")]
    pub pending_fragments: HashMap<String, Pin<Box<dyn Future<Output = String>>>>,
}

impl std::fmt::Debug for SharedContext {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("SharedContext").finish()
    }
}

#[cfg(feature = "hydrate")]
impl PartialEq for SharedContext {
    fn eq(&self, other: &Self) -> bool {
        self.completed == other.completed
            && self.events == other.events
            && self.context == other.context
            && self.registry == other.registry
            && self.pending_resources == other.pending_resources
            && self.resolved_resources == other.resolved_resources
    }
}

#[cfg(feature = "ssr")]
impl PartialEq for SharedContext {
    fn eq(&self, other: &Self) -> bool {
        self.context == other.context
    }
}

#[cfg(not(any(feature = "ssr", feature = "hydrate")))]
impl PartialEq for SharedContext {
    fn eq(&self, other: &Self) -> bool {
        self.context == other.context
    }
}

impl Eq for SharedContext {}

impl SharedContext {
    #[cfg(feature = "hydrate")]
    pub fn new_with_registry(registry: HashMap<String, web_sys::Element>) -> Self {
        let pending_resources = js_sys::Reflect::get(
            &web_sys::window().unwrap(),
            &wasm_bindgen::JsValue::from_str("__LEPTOS_PENDING_RESOURCES"),
        );
        let pending_resources: HashSet<StreamingResourceId> = pending_resources
            .map_err(|_| ())
            .and_then(|pr| serde_wasm_bindgen::from_value(pr).map_err(|_| ()))
            .unwrap_or_default();

        let resolved_resources = js_sys::Reflect::get(
            &web_sys::window().unwrap(),
            &wasm_bindgen::JsValue::from_str("__LEPTOS_RESOLVED_RESOURCES"),
        )
        .unwrap_or(wasm_bindgen::JsValue::NULL);

        let resolved_resources = match serde_wasm_bindgen::from_value(resolved_resources) {
            Ok(v) => v,
            Err(e) => {
                log::debug!(
                    "(create_resource) error deserializing __LEPTOS_RESOLVED_RESOURCES\n\n{e}"
                );
                HashMap::default()
            }
        };

        Self {
            completed: Default::default(),
            events: Default::default(),
            context: Some(HydrationContext {
                id: "".into(),
                count: -1,
            }),
            registry,
            pending_resources,
            resolved_resources,
        }
    }

    pub fn next_hydration_key(&mut self) -> String {
        if let Some(context) = &mut self.context {
            let k = format!("{}{}", context.id, context.count);
            context.count += 1;
            k
        } else {
            self.context = Some(HydrationContext {
                id: "0-".into(),
                count: 1,
            });
            "0-0".into()
        }
    }

    pub fn current_fragment_key(&self) -> String {
        if let Some(context) = &self.context {
            format!("{}{}f", context.id, context.count)
        } else {
            "0f".to_string()
        }
    }
}

#[derive(Clone, Debug, PartialEq, Eq, Default)]
pub struct HydrationContext {
    id: String,
    count: i32,
}

impl HydrationContext {
    pub fn next_hydration_context(&mut self) -> HydrationContext {
        self.count += 1;
        HydrationContext {
            id: format!("{}{}-", self.id, self.count),
            count: 0,
        }
    }
}