Migrate exit code control to resource-based system

1 files changed, 40 insertions, 0 deletions

diff --git a/mingling/src/setups/exit_code.rs b/mingling/src/setups/exit_code.rs
new file mode 100644
index 0000000..3267f85
--- /dev/null
+++ b/mingling/src/setups/exit_code.rs
@@ -0,0 +1,40 @@
+use std::marker::PhantomData;
+
+use mingling_core::{ProgramCollect, hook::ProgramHook, setup::ProgramSetup, this};
+
+use crate::res::ExitCode;
+
+/// Provides the ability to control the program's exit code, which is returned when the program ends.
+///
+/// - Use `mingling::update_exit_code` to update the exit code.
+/// - Use `mingling::current_exit_code` to query the current exit code.
+pub struct ExitCodeSetup<C> {
+    _collect: PhantomData<C>,
+}
+
+impl<C> Default for ExitCodeSetup<C>
+where
+    C: ProgramCollect<Enum = C> + 'static,
+{
+    fn default() -> Self {
+        Self {
+            _collect: Default::default(),
+        }
+    }
+}
+
+impl<C> ProgramSetup<C> for ExitCodeSetup<C>
+where
+    C: ProgramCollect<Enum = C> + 'static,
+{
+    fn setup(&mut self, program: &mut crate::Program<C>) {
+        // Insert resource
+        program.with_resource(ExitCode { 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();
+            *this
+        }));
+    }
+}