malvin 0.2.5

Non-interactive research and coding agent
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

use std::sync::OnceLock;

type DeferTaggedStdoutFn = fn(&str, &str) -> bool;
type DeferHeartbeatFn = fn(&str, &str) -> bool;

static DEFER_TAGGED_STDOUT: OnceLock<DeferTaggedStdoutFn> = OnceLock::new();
static DEFER_HEARTBEAT: OnceLock<DeferHeartbeatFn> = OnceLock::new();

#[allow(dead_code)]
pub(crate) fn register_defer_stdout_hooks(
    tagged: DeferTaggedStdoutFn,
    heartbeat: DeferHeartbeatFn,
) {
    let _ = DEFER_TAGGED_STDOUT.set(tagged);
    let _ = DEFER_HEARTBEAT.set(heartbeat);
}

pub(crate) fn try_defer_tagged_stdout(display: &str, log: &str) -> bool {
    DEFER_TAGGED_STDOUT
        .get()
        .is_some_and(|hook| hook(display, log))
}

pub(crate) fn try_defer_heartbeat(display: &str, log: &str) -> bool {
    DEFER_HEARTBEAT.get().is_some_and(|hook| hook(display, log))
}

#[cfg(test)]
mod tests {
    use super::{register_defer_stdout_hooks, try_defer_heartbeat, try_defer_tagged_stdout};

    fn noop_tagged(_display: &str, _log: &str) -> bool {
        true
    }

    fn noop_heartbeat(_display: &str, _log: &str) -> bool {
        true
    }

    #[test]
    fn register_defer_stdout_hooks_is_safe_when_called() {
        register_defer_stdout_hooks(noop_tagged, noop_heartbeat);
        let _ = try_defer_tagged_stdout("d", "l");
        let _ = try_defer_heartbeat("hb-d", "hb-l");
    }
}