error.rs 2.27 KB
Newer Older
Guillaume Gomez's avatar
Guillaume Gomez committed
1
// Take a look at the license at the top of the repository in the LICENSE file.
2

3
use thiserror::Error;
4

5 6
use crate::ErrorMessage;
use crate::FlowReturn;
7 8

#[macro_export]
9
macro_rules! panic_to_error(
10 11 12 13 14
    ($element:expr, $panicked:expr, $ret:expr, $code:block) => {{
        use std::panic::{self, AssertUnwindSafe};
        use std::sync::atomic::Ordering;
        use $crate::ElementExtManual;

15
        #[allow(clippy::unused_unit)]
16 17
        {
            if $panicked.load(Ordering::Relaxed) {
18
                $element.post_error_message($crate::error_msg!($crate::LibraryError::Failed, ["Panicked"]));
19 20 21 22 23 24 25 26 27
                $ret
            } else {
                let result = panic::catch_unwind(AssertUnwindSafe(|| $code));

                match result {
                    Ok(result) => result,
                    Err(err) => {
                        $panicked.store(true, Ordering::Relaxed);
                        if let Some(cause) = err.downcast_ref::<&str>() {
28
                            $element.post_error_message($crate::error_msg!($crate::LibraryError::Failed, ["Panicked: {}", cause]));
29
                        } else if let Some(cause) = err.downcast_ref::<String>() {
30
                            $element.post_error_message($crate::error_msg!($crate::LibraryError::Failed, ["Panicked: {}", cause]));
31
                        } else {
32
                            $element.post_error_message($crate::error_msg!($crate::LibraryError::Failed, ["Panicked"]));
33 34
                        }
                        $ret
35 36 37 38 39 40 41
                    }
                }
            }
        }
    }};
);

42
#[derive(Clone, Debug, PartialEq, Eq, Error)]
43
pub enum FlowError {
44
    #[error("Flushing")]
45
    Flushing,
46
    #[error("Eos")]
47
    Eos,
48
    #[error("Not Negotiated")]
49
    NotNegotiated(ErrorMessage),
50
    #[error("Error")]
51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69
    Error(ErrorMessage),
}

impl From<FlowError> for FlowReturn {
    fn from(err: FlowError) -> Self {
        FlowReturn::from(&err)
    }
}

impl<'a> From<&'a FlowError> for FlowReturn {
    fn from(err: &FlowError) -> FlowReturn {
        match *err {
            FlowError::Flushing => FlowReturn::Flushing,
            FlowError::Eos => FlowReturn::Eos,
            FlowError::NotNegotiated(..) => FlowReturn::NotNegotiated,
            FlowError::Error(..) => FlowReturn::Error,
        }
    }
}