mahler-core 0.24.1

//! Error handling types

use std::fmt;
use std::ops::{Deref, DerefMut};

#[doc(hidden)]
type BoxError = Box<dyn std::error::Error + Send + Sync>;

/// A list specifying categories of Mahler errors
///
/// used with the [`Error`] type
#[derive(Debug, Copy, Clone, PartialEq, Eq)]
pub enum ErrorKind {
    /// An error happened trying to serialize or deserialize into provided type
    Serialization,
    /// The route defined for the Job does not point to a valid state component
    InvalidRoute,
    /// The job argument cannot be deserialized into the target type
    CannotDeserializeArg,
    /// A condition was not met to run this Job
    ConditionNotMet,
    /// The Job referenced by method could no be found in the worker domain.
    NotFound,
    /// An argument required for the job is missing in method invocation
    MissingArgs,
    /// An I/O error happened with the job at runtime
    Runtime,
    /// An internal error happened, this is probably a bug in mahler or one of its dependencies
    Internal,
    /// An input to the worker is not valid
    InvalidInput,
}

impl ErrorKind {
    pub(crate) fn as_str(&self) -> &'static str {
        use ErrorKind::*;
        match *self {
            Serialization => "serialization failed",
            InvalidRoute => "invalid route",
            CannotDeserializeArg => "argument cannot be deserialized into target type",
            ConditionNotMet => "condition failed",
            NotFound => "job not found",
            MissingArgs => "missing arguments",
            Runtime => "runtime error",
            Internal => "internal error",
            InvalidInput => "invalid input",
        }
    }
}

impl fmt::Display for ErrorKind {
    /// Shows a human-readable description of the `ErrorKind`.
    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
        fmt.write_str(self.as_str())
    }
}

impl From<ErrorKind> for Error {
    fn from(kind: ErrorKind) -> Self {
        Error { kind, error: None }
    }
}

impl From<serde_json::Error> for Error {
    fn from(e: serde_json::Error) -> Self {
        Error::new(ErrorKind::Serialization, e)
    }
}

/// A mahler Error
///
/// All possible error types are described by [`ErrorKind`]
#[derive(Debug)]
pub struct Error {
    kind: ErrorKind,
    error: Option<BoxError>,
}

impl std::error::Error for Error {}

impl fmt::Display for Error {
    /// Shows a human-readable description of the `Error`.
    fn fmt(&self, fmt: &mut fmt::Formatter<'_>) -> fmt::Result {
        if let Some(source) = self.error.as_ref() {
            write!(fmt, "{source}")
        } else {
            write!(fmt, "{}", self.kind.as_str())
        }
    }
}

impl Error {
    pub fn new<E: Into<BoxError>>(kind: ErrorKind, error: E) -> Self {
        Self {
            kind,
            error: Some(error.into()),
        }
    }

    pub fn kind(&self) -> ErrorKind {
        self.kind
    }

    pub fn source(&self) -> Option<&BoxError> {
        self.error.as_ref()
    }

    pub fn runtime<E: Into<BoxError>>(error: E) -> Self {
        Self::new(ErrorKind::Runtime, error)
    }

    pub fn internal<E: Into<BoxError>>(error: E) -> Self {
        Self::new(ErrorKind::Internal, error)
    }
}

/// A workflow error aggregating multiple errors
/// in the execution of concurrent branches of the
/// graph
#[derive(Debug)]
pub struct AggregateError<E>(pub Vec<E>);

impl<E: std::error::Error> std::error::Error for AggregateError<E> {}

impl<E: std::error::Error> From<Vec<E>> for AggregateError<E> {
    fn from(vec: Vec<E>) -> Self {
        AggregateError(vec)
    }
}

impl<E> fmt::Display for AggregateError<E>
where
    E: fmt::Display,
{
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        for e in &self.0 {
            writeln!(f, "- {e}")?;
        }
        Ok(())
    }
}

impl<E> Deref for AggregateError<E> {
    type Target = Vec<E>;

    fn deref(&self) -> &Self::Target {
        &self.0
    }
}

impl<E> DerefMut for AggregateError<E> {
    fn deref_mut(&mut self) -> &mut Self::Target {
        &mut self.0
    }
}