use std::any::type_name;
use std::rc::Rc;
use serde::{Deserialize, Serialize};
use yew::{
format::Json,
services::{storage::Area, StorageService},
};
use super::{Store, StoreLink};
pub trait Persistent: Serialize + for<'a> Deserialize<'a> {
fn key() -> &'static str {
type_name::<Self>()
}
fn area() -> Area {
Area::Local
}
}
#[derive(Default)]
pub struct PersistentStore<T> {
state: Rc<T>,
storage: Option<StorageService>,
}
impl<T> PersistentStore<T>
where
T: Persistent + Default,
{
pub fn new() -> Self {
let mut this: Self = Default::default();
this.storage = StorageService::new(T::area()).ok();
this.load_state();
this
}
pub fn load_state(&mut self) {
let result = self.storage.as_mut().map(|s| s.restore(T::key()));
if let Some(Json(Ok(state))) = result {
self.state = state;
}
}
pub fn save_state(&mut self) {
if let Some(storage) = &mut self.storage {
storage.store(T::key(), Json(&self.state));
}
}
}
impl<T> Store for PersistentStore<T>
where
T: Default + Clone + Persistent + 'static,
{
type Model = T;
type Message = ();
type Input = ();
type Output = ();
fn new(_link: StoreLink<Self>) -> Self {
Self::new()
}
fn state(&mut self) -> &mut Rc<Self::Model> {
&mut self.state
}
fn changed(&mut self) {
self.save_state();
}
}
impl<T> Clone for PersistentStore<T>
where
T: Default + Clone + Persistent,
{
fn clone(&self) -> Self {
let mut new = Self::new();
new.state = self.state.clone();
new
}
}
impl<T: Persistent> Persistent for Option<T> {
fn key() -> &'static str {
T::key()
}
fn area() -> Area {
T::area()
}
}
impl<T: Persistent> Persistent for Rc<T> {
fn key() -> &'static str {
T::key()
}
fn area() -> Area {
T::area()
}
}