use crate::state::{Freeze, MakeRx, MakeUnrx};
use serde::{de::DeserializeOwned, Deserialize, Serialize};
use std::collections::HashMap;
use std::hash::Hash;
use std::ops::Deref;
#[cfg(any(client, doc))]
use sycamore::prelude::Scope;
use sycamore::reactive::{create_rc_signal, RcSignal};
#[derive(Serialize, Deserialize, Clone, Debug, PartialEq, Eq)]
pub struct RxHashMapNested<K, V>(HashMap<K, V>)
where
K: Clone + Eq + Hash,
V: MakeRx + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone;
#[derive(Clone, Debug, PartialEq, Eq)]
pub struct RxHashMapNestedRx<K, V>(RcSignal<HashMap<K, V::Rx>>)
where
K: Clone + Serialize + DeserializeOwned + Eq + Hash,
V: MakeRx + Serialize + DeserializeOwned + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone;
impl<K, V> MakeRx for RxHashMapNested<K, V>
where
K: Clone + Serialize + DeserializeOwned + Eq + Hash,
V: MakeRx + Serialize + DeserializeOwned + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone,
{
type Rx = RxHashMapNestedRx<K, V>;
fn make_rx(self) -> Self::Rx {
RxHashMapNestedRx(create_rc_signal(
self.0.into_iter().map(|(k, v)| (k, v.make_rx())).collect(),
))
}
}
impl<K, V> MakeUnrx for RxHashMapNestedRx<K, V>
where
K: Clone + Serialize + DeserializeOwned + Eq + Hash,
V: MakeRx + Serialize + DeserializeOwned + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone,
{
type Unrx = RxHashMapNested<K, V>;
fn make_unrx(self) -> Self::Unrx {
let map = (*self.0.get_untracked()).clone();
RxHashMapNested(map.into_iter().map(|(k, v)| (k, v.make_unrx())).collect())
}
#[cfg(any(client, doc))]
fn compute_suspense(&self, cx: Scope) {
for elem in self.0.get_untracked().values() {
elem.compute_suspense(cx);
}
}
}
impl<K, V> Deref for RxHashMapNested<K, V>
where
K: Clone + Serialize + DeserializeOwned + Eq + Hash,
V: MakeRx + Serialize + DeserializeOwned + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone,
{
type Target = HashMap<K, V>;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl<K, V> Deref for RxHashMapNestedRx<K, V>
where
K: Clone + Serialize + DeserializeOwned + Eq + Hash,
V: MakeRx + Serialize + DeserializeOwned + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone,
{
type Target = RcSignal<HashMap<K, V::Rx>>;
fn deref(&self) -> &Self::Target {
&self.0
}
}
impl<K, V> From<HashMap<K, V>> for RxHashMapNested<K, V>
where
K: Clone + Serialize + DeserializeOwned + Eq + Hash,
V: MakeRx + Serialize + DeserializeOwned + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone,
{
fn from(value: HashMap<K, V>) -> Self {
Self(value)
}
}
impl<K, V> Freeze for RxHashMapNestedRx<K, V>
where
K: Clone + Serialize + DeserializeOwned + Eq + Hash,
V: MakeRx + Serialize + DeserializeOwned + 'static,
V::Rx: MakeUnrx<Unrx = V> + Freeze + Clone,
{
fn freeze(&self) -> String {
let unrx = Self(self.0.clone()).make_unrx();
serde_json::to_string(&unrx).unwrap()
}
}