use bytes::Bytes; use derive_more::Display; use flowy_derive::{ProtoBuf, ProtoBuf_Enum}; use flowy_dispatch::prelude::{EventResponse, ResponseBuilder}; use flowy_document::errors::DocError; use flowy_net::errors::ErrorCode as ServerErrorCode; use std::{convert::TryInto, fmt}; #[derive(Debug, Default, Clone, ProtoBuf)] pub struct WorkspaceError { #[pb(index = 1)] pub code: ErrorCode, #[pb(index = 2)] pub msg: String, } impl WorkspaceError { pub fn new(code: ErrorCode, msg: &str) -> Self { Self { code, msg: msg.to_owned() } } } #[derive(Debug, Clone, ProtoBuf_Enum, Display, PartialEq, Eq)] pub enum ErrorCode { #[display(fmt = "Unknown")] Unknown = 0, #[display(fmt = "Workspace name is invalid")] WorkspaceNameInvalid = 1, #[display(fmt = "Workspace id is invalid")] WorkspaceIdInvalid = 2, #[display(fmt = "Color style of the App is invalid")] AppColorStyleInvalid = 3, #[display(fmt = "Workspace desc is invalid")] WorkspaceDescInvalid = 4, #[display(fmt = "Current workspace not found")] CurrentWorkspaceNotFound = 5, #[display(fmt = "Id of the App is invalid")] AppIdInvalid = 10, #[display(fmt = "Name of the App is invalid")] AppNameInvalid = 11, #[display(fmt = "Name of the View is invalid")] ViewNameInvalid = 20, #[display(fmt = "Thumbnail of the view is invalid")] ViewThumbnailInvalid = 21, #[display(fmt = "Id of the View is invalid")] ViewIdInvalid = 22, #[display(fmt = "Description of the View is invalid")] ViewDescInvalid = 23, #[display(fmt = "View data is invalid")] ViewDataInvalid = 24, #[display(fmt = "UserIn is empty")] UserIdIsEmpty = 100, #[display(fmt = "User unauthorized")] UserUnauthorized = 101, #[display(fmt = "Server error")] InternalError = 1000, #[display(fmt = "Record not found")] RecordNotFound = 1001, } impl std::default::Default for ErrorCode { fn default() -> Self { ErrorCode::Unknown } } impl std::convert::From for WorkspaceError { fn from(error: DocError) -> Self { ErrorBuilder::new(ErrorCode::InternalError).error(error).build() } } impl std::convert::From for WorkspaceError { fn from(error: flowy_net::errors::ServerError) -> Self { let code = server_error_to_workspace_error(error.code); ErrorBuilder::new(code).error(error.msg).build() } } impl std::convert::From for WorkspaceError { fn from(error: flowy_database::Error) -> Self { ErrorBuilder::new(ErrorCode::InternalError).error(error).build() } } impl flowy_dispatch::Error for WorkspaceError { fn as_response(&self) -> EventResponse { let bytes: Bytes = self.clone().try_into().unwrap(); ResponseBuilder::Err().data(bytes).build() } } impl fmt::Display for WorkspaceError { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { write!(f, "{:?}: {}", &self.code, &self.msg) } } pub type ErrorBuilder = flowy_infra::errors::Builder; impl flowy_infra::errors::Build for WorkspaceError { fn build(code: ErrorCode, msg: String) -> Self { let msg = if msg.is_empty() { format!("{}", code) } else { msg }; WorkspaceError::new(code, &msg) } } fn server_error_to_workspace_error(code: ServerErrorCode) -> ErrorCode { match code { ServerErrorCode::UserUnauthorized => ErrorCode::UserUnauthorized, ServerErrorCode::RecordNotFound => ErrorCode::RecordNotFound, _ => ErrorCode::InternalError, } }