horus 0.5.0

A small, modular Rust framework for building coding agents
Documentation 
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

use std::process::Stdio;

use crate::{Error, Result};

/// Reaps every process in one Seatbelt instance when the host-owned stdin lease closes.
#[cfg(target_os = "macos")]
pub const MACOS_COMMAND_WRAPPER: &str = r#"
exec 3<&0
(
  trap '' HUP INT TERM
  while IFS= read -r _ <&3; do :; done
  kill -KILL -- -1 2>/dev/null
) &
test -n "$!" || exit 125
exec 3<&-
exec "$@" </dev/null
"#;

/// Kills one command's process group when execution completes or is cancelled.
///
/// The command must be configured with `process_group(0)` before it is spawned.
pub struct ProcessGroupGuard {
    id: u32,
    armed: bool,
}

impl ProcessGroupGuard {
    /// Arms cleanup for the child's process group.
    pub fn new(child: &tokio::process::Child) -> Result<Self> {
        let id = child
            .id()
            .ok_or_else(|| Error::Sandbox("command process ID unavailable".into()))?;
        Ok(Self { id, armed: true })
    }

    /// Kills the process group once.
    pub fn kill(&mut self) {
        if self.armed {
            let _ = std::process::Command::new("/bin/kill")
                .args(["-KILL", "--", &format!("-{}", self.id)])
                .stdin(Stdio::null())
                .stdout(Stdio::null())
                .stderr(Stdio::null())
                .status();
            self.armed = false;
        }
    }
}

impl Drop for ProcessGroupGuard {
    fn drop(&mut self) {
        self.kill();
    }
}