diff options
Diffstat (limited to 'mingling_core/src/program')
| -rw-r--r-- | mingling_core/src/program/error.rs | 22 | ||||
| -rw-r--r-- | mingling_core/src/program/exec/error.rs | 5 | ||||
| -rw-r--r-- | mingling_core/src/program/hook.rs | 24 |
3 files changed, 49 insertions, 2 deletions
diff --git a/mingling_core/src/program/error.rs b/mingling_core/src/program/error.rs new file mode 100644 index 0000000..5e92a42 --- /dev/null +++ b/mingling_core/src/program/error.rs @@ -0,0 +1,22 @@ +use std::any::Any; +use std::fmt; +use thiserror::Error; + +/// Error type returned when a panic occurs during execution. +#[derive(Error)] +#[error("execution panicked: {payload:?}")] +pub struct ProgramPanic { + pub payload: Box<dyn Any + Send>, +} + +impl ProgramPanic { + pub fn new(payload: Box<dyn Any + Send>) -> Self { + ProgramPanic { payload } + } +} + +impl fmt::Debug for ProgramPanic { + fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { + write!(f, "{:?}", self.payload) + } +} diff --git a/mingling_core/src/program/exec/error.rs b/mingling_core/src/program/exec/error.rs index b4ff378..2d806dc 100644 --- a/mingling_core/src/program/exec/error.rs +++ b/mingling_core/src/program/exec/error.rs @@ -1,4 +1,4 @@ -use crate::error::ChainProcessError; +use crate::error::{ChainProcessError, ProgramPanic}; #[derive(thiserror::Error, Debug)] pub enum ProgramExecuteError { @@ -8,6 +8,9 @@ pub enum ProgramExecuteError { #[error("No Renderer (`{0}`) Found")] RendererNotFound(String), + #[error("Panic: {0:?}")] + Panic(#[from] ProgramPanic), + #[error("Other error: {0}")] Other(String), } diff --git a/mingling_core/src/program/hook.rs b/mingling_core/src/program/hook.rs index afc253c..080c00f 100644 --- a/mingling_core/src/program/hook.rs +++ b/mingling_core/src/program/hook.rs @@ -1,6 +1,6 @@ use std::any::Any; -use crate::{AnyOutput, Program, ProgramCollect, RenderResult}; +use crate::{AnyOutput, Program, ProgramCollect, RenderResult, error::ProgramPanic}; #[derive(Default)] pub struct ProgramHook<C> @@ -30,6 +30,9 @@ where /// Executes before the program ends pub finish: Option<fn() -> i32>, + + /// Executes when the program panics + pub exec_panic: Option<fn(&ProgramPanic)>, } impl<C> Program<C> @@ -126,6 +129,18 @@ where } } + pub(crate) fn run_hook_exec_panic(&self, panic_info: &ProgramPanic) { + if !self.user_context.run_hook { + return; + } + + for hook in &self.hooks { + if let Some(exec_panic) = hook.exec_panic { + exec_panic(panic_info) + } + } + } + pub(crate) fn run_hook_finish(&self) -> i32 { if !self.user_context.run_hook { return 0; @@ -159,6 +174,7 @@ where pre_render: None, post_render: None, finish: None, + exec_panic: None, } } @@ -209,4 +225,10 @@ where let _ = self.finish.insert(handler); self } + + /// Sets the handler for the `exec_panic` event. + pub fn on_exec_panic(mut self, handler: fn(&ProgramPanic)) -> Self { + let _ = self.exec_panic.insert(handler); + self + } } |
