5c7d9e27b1
Proposal: 1. Use anyhow for tests and avoid unwrap 2. Extract a helper for starting a test instance of codex
82 lines
2.6 KiB
Rust
82 lines
2.6 KiB
Rust
use serde::Serialize;
|
||
use tracing::error;
|
||
use tracing::warn;
|
||
|
||
#[derive(Debug, Default)]
|
||
pub(crate) struct UserNotifier {
|
||
notify_command: Option<Vec<String>>,
|
||
}
|
||
|
||
impl UserNotifier {
|
||
pub(crate) fn notify(&self, notification: &UserNotification) {
|
||
if let Some(notify_command) = &self.notify_command
|
||
&& !notify_command.is_empty()
|
||
{
|
||
self.invoke_notify(notify_command, notification)
|
||
}
|
||
}
|
||
|
||
fn invoke_notify(&self, notify_command: &[String], notification: &UserNotification) {
|
||
let Ok(json) = serde_json::to_string(¬ification) else {
|
||
error!("failed to serialise notification payload");
|
||
return;
|
||
};
|
||
|
||
let mut command = std::process::Command::new(¬ify_command[0]);
|
||
if notify_command.len() > 1 {
|
||
command.args(¬ify_command[1..]);
|
||
}
|
||
command.arg(json);
|
||
|
||
// Fire-and-forget – we do not wait for completion.
|
||
if let Err(e) = command.spawn() {
|
||
warn!("failed to spawn notifier '{}': {e}", notify_command[0]);
|
||
}
|
||
}
|
||
|
||
pub(crate) fn new(notify: Option<Vec<String>>) -> Self {
|
||
Self {
|
||
notify_command: notify,
|
||
}
|
||
}
|
||
}
|
||
|
||
/// User can configure a program that will receive notifications. Each
|
||
/// notification is serialized as JSON and passed as an argument to the
|
||
/// program.
|
||
#[derive(Debug, Clone, PartialEq, Serialize)]
|
||
#[serde(tag = "type", rename_all = "kebab-case")]
|
||
pub(crate) enum UserNotification {
|
||
#[serde(rename_all = "kebab-case")]
|
||
AgentTurnComplete {
|
||
turn_id: String,
|
||
|
||
/// Messages that the user sent to the agent to initiate the turn.
|
||
input_messages: Vec<String>,
|
||
|
||
/// The last message sent by the assistant in the turn.
|
||
last_assistant_message: Option<String>,
|
||
},
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
|
||
#[test]
|
||
fn test_user_notification() {
|
||
let notification = UserNotification::AgentTurnComplete {
|
||
turn_id: "12345".to_string(),
|
||
input_messages: vec!["Rename `foo` to `bar` and update the callsites.".to_string()],
|
||
last_assistant_message: Some(
|
||
"Rename complete and verified `cargo build` succeeds.".to_string(),
|
||
),
|
||
};
|
||
let serialized = serde_json::to_string(¬ification).unwrap();
|
||
assert_eq!(
|
||
serialized,
|
||
r#"{"type":"agent-turn-complete","turn-id":"12345","input-messages":["Rename `foo` to `bar` and update the callsites."],"last-assistant-message":"Rename complete and verified `cargo build` succeeds."}"#
|
||
);
|
||
}
|
||
}
|