agent-status 2.0.0

Tmux-integrated indicator showing which AI coding agent sessions are waiting on user input.
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

use crate::agents::Agent;

/// omp / oh-my-pi ([omp.sh](https://omp.sh)), a coding-first fork of pi.
///
/// Reads `session_id` from the JSON piped in by the bundled omp extension at
/// `extensions/oh-my-pi.ts`, which fires on omp's `session_start`,
/// `session_switch`, `session_branch`, `before_agent_start`,
/// `tool_execution_start`, `tool_execution_end`, `agent_end`, and
/// `session_shutdown` events.
pub struct OhMyPiAgent;

impl Agent for OhMyPiAgent {
    fn name(&self) -> &'static str {
        "oh-my-pi"
    }

    fn extract_session_id(&self, stdin_json: &str) -> Option<String> {
        let v: serde_json::Value = serde_json::from_str(stdin_json).ok()?;
        let id = v.get("session_id")?.as_str()?;
        if id.is_empty() {
            None
        } else {
            Some(id.to_string())
        }
    }

    fn extract_message(&self, stdin_json: &str) -> Option<String> {
        let v: serde_json::Value = serde_json::from_str(stdin_json).ok()?;
        let m = v.get("message")?.as_str()?;
        if m.is_empty() {
            None
        } else {
            Some(m.to_string())
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn name_is_oh_my_pi() {
        assert_eq!(OhMyPiAgent.name(), "oh-my-pi");
    }

    #[test]
    fn extract_session_id_returns_id() {
        let json = r#"{"session_id":"0196fdb1-1111-7000-8000-aaaaaaaaaaaa","other":"stuff"}"#;
        assert_eq!(
            OhMyPiAgent.extract_session_id(json).as_deref(),
            Some("0196fdb1-1111-7000-8000-aaaaaaaaaaaa")
        );
    }

    #[test]
    fn extract_session_id_returns_none_for_missing_field() {
        assert_eq!(OhMyPiAgent.extract_session_id(r#"{"other":1}"#), None);
    }

    #[test]
    fn extract_session_id_returns_none_for_empty_string() {
        assert_eq!(OhMyPiAgent.extract_session_id(r#"{"session_id":""}"#), None);
    }

    #[test]
    fn extract_session_id_returns_none_for_invalid_json() {
        assert_eq!(OhMyPiAgent.extract_session_id("not json"), None);
    }

    #[test]
    fn extract_message_returns_string_when_present() {
        let json = r#"{"session_id":"x","message":"Running: cargo test"}"#;
        assert_eq!(
            OhMyPiAgent.extract_message(json).as_deref(),
            Some("Running: cargo test")
        );
    }

    #[test]
    fn extract_message_returns_none_when_field_missing() {
        assert!(OhMyPiAgent.extract_message(r#"{"session_id":"x"}"#).is_none());
    }

    #[test]
    fn extract_message_returns_none_when_empty() {
        assert!(OhMyPiAgent.extract_message(r#"{"session_id":"x","message":""}"#).is_none());
    }

    #[test]
    fn extract_message_returns_none_for_non_string_value() {
        assert!(OhMyPiAgent.extract_message(r#"{"session_id":"x","message":null}"#).is_none());
    }

    #[test]
    fn extract_message_returns_none_for_invalid_json() {
        assert!(OhMyPiAgent.extract_message("not json").is_none());
    }
}