From 2eba732e17cf71dcf8a54b428a1c5e344582c2c2 Mon Sep 17 00:00:00 2001
From: 魏曹先生 <1992414357@qq.com>
Date: Sat, 9 May 2026 17:37:37 +0800
Subject: Add exit code control and hook lifecycle features

---
 .../src/program/setup/exit_code_control.rs         | 36 ++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 mingling_core/src/program/setup/exit_code_control.rs

(limited to 'mingling_core/src/program/setup')

diff --git a/mingling_core/src/program/setup/exit_code_control.rs b/mingling_core/src/program/setup/exit_code_control.rs
new file mode 100644
index 0000000..0844a24
--- /dev/null
+++ b/mingling_core/src/program/setup/exit_code_control.rs
@@ -0,0 +1,36 @@
+use std::sync::atomic::{AtomicI32, Ordering};
+
+use crate::{ProgramCollect, hook::ProgramAnonymousHook, setup::ProgramSetup};
+
+static EXIT_CODE: AtomicI32 = AtomicI32::new(0);
+
+/// 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;
+
+impl<C> ProgramSetup<C> for ExitCodeSetup
+where
+    C: ProgramCollect<Enum = C>,
+{
+    fn setup(&mut self, program: &mut crate::Program<C>) {
+        program
+            .with_hook_anonymous(ProgramAnonymousHook::empty().on_finish(|| current_exit_code()));
+    }
+}
+
+/// Updates the program's exit code.
+///
+/// This function sets the value that will be returned when the program exits.
+/// The new code will take effect immediately and be used when the program finishes.
+pub fn update_exit_code(code: i32) {
+    EXIT_CODE.store(code, Ordering::SeqCst);
+}
+
+/// Returns the current exit code.
+///
+/// This function queries the value that will be returned when the program exits.
+pub fn current_exit_code() -> i32 {
+    EXIT_CODE.load(Ordering::SeqCst)
+}
-- 
cgit