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

use crate::{
  byte_trait::AFPluginFromBytes,
  request::AFPluginEventRequest,
  response::{AFPluginEventResponse, ResponseBuilder},
};
use bytes::Bytes;
use dyn_clone::DynClone;

use std::fmt;
use tokio::{sync::mpsc::error::SendError, task::JoinError};

pub trait Error: fmt::Debug + DynClone + Send + Sync {
  fn as_response(&self) -> AFPluginEventResponse;
}

dyn_clone::clone_trait_object!(Error);

impl<T: Error + 'static> From<T> for DispatchError {
  fn from(err: T) -> DispatchError {
    DispatchError {
      inner: Box::new(err),
    }
  }
}

#[derive(Clone)]
pub struct DispatchError {
  inner: Box<dyn Error>,
}

impl DispatchError {
  pub fn inner_error(&self) -> &dyn Error {
    self.inner.as_ref()
  }
}

impl fmt::Display for DispatchError {
  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
    write!(f, "{:?}", &self.inner)
  }
}

impl fmt::Debug for DispatchError {
  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
    write!(f, "{:?}", &self.inner)
  }
}

impl std::error::Error for DispatchError {
  fn source(&self) -> Option<&(dyn std::error::Error + 'static)> {
    None
  }

  fn cause(&self) -> Option<&dyn std::error::Error> {
    None
  }
}

impl From<SendError<AFPluginEventRequest>> for DispatchError {
  fn from(err: SendError<AFPluginEventRequest>) -> Self {
    InternalError::Other(format!("{}", err)).into()
  }
}

impl From<String> for DispatchError {
  fn from(s: String) -> Self {
    InternalError::Other(s).into()
  }
}

#[cfg(feature = "use_protobuf")]
impl From<protobuf::ProtobufError> for DispatchError {
  fn from(e: protobuf::ProtobufError) -> Self {
    InternalError::ProtobufError(format!("{:?}", e)).into()
  }
}

impl AFPluginFromBytes for DispatchError {
  fn parse_from_bytes(bytes: Bytes) -> Result<Self, DispatchError> {
    let s = String::from_utf8(bytes.to_vec()).unwrap();
    Ok(InternalError::DeserializeFromBytes(s).into())
  }
}

impl From<DispatchError> for AFPluginEventResponse {
  fn from(err: DispatchError) -> Self {
    err.inner_error().as_response()
  }
}
#[cfg(feature = "use_serde")]
impl serde::Serialize for DispatchError {
  fn serialize<S>(
    &self,
    serializer: S,
  ) -> Result<<S as serde::Serializer>::Ok, <S as serde::Serializer>::Error>
  where
    S: serde::Serializer,
  {
    serializer.serialize_str(&format!("{}", self))
  }
}

#[derive(Clone, Debug)]
pub(crate) enum InternalError {
  ProtobufError(String),
  UnexpectedNone(String),
  DeserializeFromBytes(String),
  JoinError(String),
  ServiceNotFound(String),
  HandleNotFound(String),
  Other(String),
}

impl fmt::Display for InternalError {
  fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
    match self {
      InternalError::ProtobufError(s) => fmt::Display::fmt(&s, f),
      InternalError::UnexpectedNone(s) => fmt::Display::fmt(&s, f),
      InternalError::DeserializeFromBytes(s) => fmt::Display::fmt(&s, f),
      InternalError::JoinError(s) => fmt::Display::fmt(&s, f),
      InternalError::ServiceNotFound(s) => fmt::Display::fmt(&s, f),
      InternalError::HandleNotFound(s) => fmt::Display::fmt(&s, f),
      InternalError::Other(s) => fmt::Display::fmt(&s, f),
    }
  }
}

impl Error for InternalError {
  fn as_response(&self) -> AFPluginEventResponse {
    ResponseBuilder::Err().data(self.to_string()).build()
  }
}

impl std::convert::From<JoinError> for InternalError {
  fn from(e: JoinError) -> Self {
    InternalError::JoinError(format!("{}", e))
  }
}