AppFlowy/frontend/rust-lib/lib-dispatch/src/module/data.rs

use crate::{
  errors::{DispatchError, InternalError},
  request::{payload::Payload, AFPluginEventRequest, FromAFPluginRequest},
  util::ready::{ready, Ready},
};
use std::{any::type_name, ops::Deref, sync::Arc};

pub struct AFPluginState<T: ?Sized + Send + Sync>(Arc<T>);

impl<T> AFPluginState<T>
where
  T: Send + Sync,
{
  pub fn new(data: T) -> Self {
    AFPluginState(Arc::new(data))
  }

  pub fn get_ref(&self) -> &T {
    self.0.as_ref()
  }
}

impl<T> Deref for AFPluginState<T>
where
  T: ?Sized + Send + Sync,
{
  type Target = Arc<T>;

  fn deref(&self) -> &Arc<T> {
    &self.0
  }
}

impl<T> Clone for AFPluginState<T>
where
  T: ?Sized + Send + Sync,
{
  fn clone(&self) -> AFPluginState<T> {
    AFPluginState(self.0.clone())
  }
}

impl<T> From<Arc<T>> for AFPluginState<T>
where
  T: ?Sized + Send + Sync,
{
  fn from(arc: Arc<T>) -> Self {
    AFPluginState(arc)
  }
}

impl<T> FromAFPluginRequest for AFPluginState<T>
where
  T: ?Sized + Send + Sync + 'static,
{
  type Error = DispatchError;
  type Future = Ready<Result<Self, DispatchError>>;

  #[inline]
  fn from_request(req: &AFPluginEventRequest, _: &mut Payload) -> Self::Future {
    if let Some(state) = req.get_state::<AFPluginState<T>>() {
      ready(Ok(state.clone()))
    } else {
      let msg = format!(
        "Failed to get the plugin state of type: {}",
        type_name::<T>()
      );
      log::error!("{}", msg,);
      ready(Err(InternalError::Other(msg).into()))
    }
  }
}