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

use std::fmt::{self, Display};

use failure::{Backtrace, Context, Fail};
use lambda_runtime::error::HandlerError;

/// The error type for Vicuna operations. Wraps an [`ErrorKind`].
///
/// [`ErrorKind`]: enum.ErrorKind.html
#[derive(Debug)]
pub struct Error {
    inner: Context<ErrorKind>,
}

impl From<ErrorKind> for Error {
    fn from(kind: ErrorKind) -> Error {
        Error {
            inner: Context::new(kind),
        }
    }
}

impl From<Context<ErrorKind>> for Error {
    fn from(inner: Context<ErrorKind>) -> Error {
        Error { inner }
    }
}

impl Fail for Error {
    fn cause(&self) -> Option<&dyn Fail> {
        self.inner.cause()
    }

    fn backtrace(&self) -> Option<&Backtrace> {
        self.inner.backtrace()
    }
}

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

impl From<HandlerError> for Error {
    fn from(err: HandlerError) -> Error {
        ErrorKind::Handler(err).into()
    }
}

impl From<failure::Error> for Error {
    fn from(err: failure::Error) -> Error {
        ErrorKind::Failure(err).into()
    }
}

/// A list specifying kinds of errors Vicuna handlers may encounter.
#[derive(Debug, Fail)]
pub enum ErrorKind {
    /// A `HandlerError`, as provided by the `lambda_rust` crate.
    #[fail(display = "{}", _0)]
    Handler(#[cause] HandlerError),

    /// A `failure::Error`, as provided by the `failure` crate.
    #[fail(display = "{}", _0)]
    Failure(#[cause] failure::Error),
}