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

use crate::{
    byte_trait::*,
    errors::{DispatchError, InternalError},
    request::{unexpected_none_payload, AFPluginEventRequest, FromAFPluginRequest, Payload},
    response::{AFPluginEventResponse, AFPluginResponder, ResponseBuilder},
    util::ready::{ready, Ready},
};
use bytes::Bytes;
use std::ops;

pub struct AFPluginData<T>(pub T);

impl<T> AFPluginData<T> {
    pub fn into_inner(self) -> T {
        self.0
    }
}

impl<T> ops::Deref for AFPluginData<T> {
    type Target = T;

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

impl<T> ops::DerefMut for AFPluginData<T> {
    fn deref_mut(&mut self) -> &mut T {
        &mut self.0
    }
}

impl<T> FromAFPluginRequest for AFPluginData<T>
where
    T: AFPluginFromBytes + 'static,
{
    type Error = DispatchError;
    type Future = Ready<Result<Self, DispatchError>>;

    #[inline]
    fn from_request(req: &AFPluginEventRequest, payload: &mut Payload) -> Self::Future {
        match payload {
            Payload::None => ready(Err(unexpected_none_payload(req))),
            Payload::Bytes(bytes) => match T::parse_from_bytes(bytes.clone()) {
                Ok(data) => ready(Ok(AFPluginData(data))),
                Err(e) => ready(Err(InternalError::DeserializeFromBytes(format!("{}", e)).into())),
            },
        }
    }
}

impl<T> AFPluginResponder for AFPluginData<T>
where
    T: ToBytes,
{
    fn respond_to(self, _request: &AFPluginEventRequest) -> AFPluginEventResponse {
        match self.into_inner().into_bytes() {
            Ok(bytes) => {
                log::trace!("Serialize Data: {:?} to event response", std::any::type_name::<T>());
                ResponseBuilder::Ok().data(bytes).build()
            }
            Err(e) => e.into(),
        }
    }
}

impl<T> std::convert::TryFrom<&Payload> for AFPluginData<T>
where
    T: AFPluginFromBytes,
{
    type Error = DispatchError;
    fn try_from(payload: &Payload) -> Result<AFPluginData<T>, Self::Error> {
        parse_payload(payload)
    }
}

impl<T> std::convert::TryFrom<Payload> for AFPluginData<T>
where
    T: AFPluginFromBytes,
{
    type Error = DispatchError;
    fn try_from(payload: Payload) -> Result<AFPluginData<T>, Self::Error> {
        parse_payload(&payload)
    }
}

fn parse_payload<T>(payload: &Payload) -> Result<AFPluginData<T>, DispatchError>
where
    T: AFPluginFromBytes,
{
    match payload {
        Payload::None => Err(InternalError::UnexpectedNone(format!(
            "Parse fail, expected payload:{:?}",
            std::any::type_name::<T>()
        ))
        .into()),
        Payload::Bytes(bytes) => {
            let data = T::parse_from_bytes(bytes.clone())?;
            Ok(AFPluginData(data))
        }
    }
}

impl<T> std::convert::TryInto<Payload> for AFPluginData<T>
where
    T: ToBytes,
{
    type Error = DispatchError;

    fn try_into(self) -> Result<Payload, Self::Error> {
        let inner = self.into_inner();
        let bytes = inner.into_bytes()?;
        Ok(Payload::Bytes(bytes))
    }
}

impl ToBytes for AFPluginData<String> {
    fn into_bytes(self) -> Result<Bytes, DispatchError> {
        Ok(Bytes::from(self.0))
    }
}