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

use std::future::Future;

use crate::{
    errors::{DispatchError, InternalError},
    module::{Event, ModuleDataMap},
    request::payload::Payload,
    util::ready::{ready, Ready},
};
use derivative::*;
use futures_core::ready;
use std::{
    fmt::Debug,
    pin::Pin,
    sync::Arc,
    task::{Context, Poll},
};

#[derive(Clone, Debug, Derivative)]
pub struct EventRequest {
    #[allow(dead_code)]
    pub(crate) id: String,
    pub(crate) event: Event,
    #[derivative(Debug = "ignore")]
    pub(crate) module_data: Arc<ModuleDataMap>,
}

impl EventRequest {
    pub fn new<E>(id: String, event: E, module_data: Arc<ModuleDataMap>) -> EventRequest
    where
        E: Into<Event>,
    {
        Self {
            id,
            event: event.into(),
            module_data,
        }
    }

    pub fn module_data<T: 'static>(&self) -> Option<&T>
    where
        T: Send + Sync,
    {
        if let Some(data) = self.module_data.get::<T>() {
            return Some(data);
        }

        None
    }
}

pub trait FromRequest: Sized {
    type Error: Into<DispatchError>;
    type Future: Future<Output = Result<Self, Self::Error>>;

    fn from_request(req: &EventRequest, payload: &mut Payload) -> Self::Future;
}

#[doc(hidden)]
impl FromRequest for () {
    type Error = DispatchError;
    type Future = Ready<Result<(), DispatchError>>;

    fn from_request(_req: &EventRequest, _payload: &mut Payload) -> Self::Future {
        ready(Ok(()))
    }
}

#[doc(hidden)]
impl FromRequest for String {
    type Error = DispatchError;
    type Future = Ready<Result<Self, Self::Error>>;

    fn from_request(req: &EventRequest, payload: &mut Payload) -> Self::Future {
        match &payload {
            Payload::None => ready(Err(unexpected_none_payload(req))),
            Payload::Bytes(buf) => ready(Ok(String::from_utf8_lossy(buf).into_owned())),
        }
    }
}

pub fn unexpected_none_payload(request: &EventRequest) -> DispatchError {
    log::warn!("{:?} expected payload", &request.event);
    InternalError::UnexpectedNone("Expected payload".to_string()).into()
}

#[doc(hidden)]
impl<T> FromRequest for Result<T, T::Error>
where
    T: FromRequest,
{
    type Error = DispatchError;
    type Future = FromRequestFuture<T::Future>;

    fn from_request(req: &EventRequest, payload: &mut Payload) -> Self::Future {
        FromRequestFuture {
            fut: T::from_request(req, payload),
        }
    }
}

#[pin_project::pin_project]
pub struct FromRequestFuture<Fut> {
    #[pin]
    fut: Fut,
}

impl<Fut, T, E> Future for FromRequestFuture<Fut>
where
    Fut: Future<Output = Result<T, E>>,
{
    type Output = Result<Result<T, E>, DispatchError>;

    fn poll(self: Pin<&mut Self>, cx: &mut Context<'_>) -> Poll<Self::Output> {
        let this = self.project();
        let res = ready!(this.fut.poll(cx));
        Poll::Ready(Ok(res))
    }
}