blob: 49d5f9fcd04308feed6c62d70b8ac7072187cec9 (
plain) (
blame)
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
|
use mingling_core::{
Program, ProgramCollect,
hook::{ProgramControlUnit, ProgramControls, ProgramHook},
setup::ProgramSetup,
this,
};
use crate::res::ResExitCode;
/// `ExitCodeSetup` — Setup for controlling the program's exit code
///
/// This Setup registers an [`ResExitCode`] resource that tracks the desired exit
/// code for the program. When the program finishes, a hook reads this resource
/// and overrides the program's exit code if it has been modified from its
/// default value of `0`.
///
/// # Usage
///
/// This Setup can be registered using the
/// [`Program`](https://docs.rs/mingling/latest/mingling/struct.Program.html)
/// `with_setup` method, for example:
///
/// ```rust
/// # use mingling::MockProgramCollect as ThisProgram;
/// use mingling::Program;
/// use mingling::setup::ExitCodeSetup;
///
/// let mut program = Program::<ThisProgram>::new();
/// program.with_setup(ExitCodeSetup);
/// ```
///
/// # Behavior
///
/// - Registers an [`ResExitCode`] resource initialised to `0`.
/// - Installs a program-finish hook that:
/// - Reads the current [`ResExitCode`] value.
/// - Overrides the program's exit code with that value if it is non-zero.
/// - Leaves the exit code untouched if the resource still holds its default
/// value of `0`.
pub struct ExitCodeSetup;
impl<C> ProgramSetup<C> for ExitCodeSetup
where
C: ProgramCollect<Enum = C> + 'static,
{
fn setup(self, program: &mut Program<C>) {
// Insert resource
program.with_resource(ResExitCode { exit_code: 0 });
// Insert hook to override exit code before program ends
program.with_hook(ProgramHook::empty().on_finish(|_| {
let this = this::<C>().res_or_default::<ResExitCode>();
let ec = this.exit_code;
// Only override when ResExitCode has been modified
if ec != 0 {
ProgramControlUnit::OverrideExitCode(this.exit_code).into()
} else {
ProgramControls::Empty
}
}));
}
}
|
