1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

use alloc::boxed::Box;

/// This includes `BoxedError` for general errors, tag errors, and only standard
/// library errors are directly coded.
///
/// The intention with `TimeoutError` is that if it is in the error stack, a
/// timeout occured. When other timeout structs are used, this should be added
/// on.
///
/// `ProbablyNotRootCauseError` is used to signal that the error is probably not
/// the "root cause". This is used by the `super_orchestrator` crate to reduce
/// noise when one error triggers a cascade of errors in a network.
#[derive(Debug, thiserror::Error, Default)]
pub enum ErrorKind {
    // used for special cases where we need something
    #[default]
    #[error("UnitError")]
    UnitError,
    #[error("TimeoutError")]
    TimeoutError,
    #[error("ProbablyNotRootCauseError")]
    ProbablyNotRootCauseError,
    #[error("{0}")]
    StrError(&'static str),
    #[error("{0}")]
    StringError(alloc::string::String),
    #[error("{0}")]
    BoxedError(Box<dyn std::error::Error + Send + Sync>),
    #[error("{0}")]
    TryFromIntError(core::num::TryFromIntError),
    #[error("{0}")]
    StdIoError(std::io::Error),
    #[error("{0}")]
    FromUtf8Error(alloc::string::FromUtf8Error),
    // this is more obscure but I think we should keep it because it deals with bad strings, and
    // we don't want that in our generic string errors.
    #[error("{0}")]
    FromUtf16Error(alloc::string::FromUtf16Error),
    #[error("{0}")]
    ParseIntError(core::num::ParseIntError),
    #[error("{0}")]
    ParseFloatError(core::num::ParseFloatError),
}

use ErrorKind::*;

impl ErrorKind {
    pub fn from_err<E: std::error::Error + Send + Sync + 'static>(e: E) -> Self {
        ErrorKind::BoxedError(Box::new(e))
    }

    pub fn from_box(e: Box<dyn std::error::Error + Send + Sync>) -> Self {
        ErrorKind::BoxedError(e)
    }

    /// Attempts to downcast from a `ErrorKind::BoxedError` to a concrete type.
    /// Returns the input in an `Err` if it was not an `ErrorKind::BoxedError`
    /// or the box would not downcast.
    pub fn downcast<E: std::error::Error + 'static>(self) -> Result<E, Self> {
        if let BoxedError(boxed_err) = self {
            match boxed_err.downcast() {
                Ok(err) => Ok(*err),
                Err(boxed_err) => Err(Self::BoxedError(boxed_err)),
            }
        } else {
            Err(self)
        }
    }
}