omk 0.5.0

A Rust runtime for Kimi CLI. Turns prompts into proof-backed engineering runs with gates, worktrees, and replay.
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
100
101
102
103
104
105
106
107
108
109
110

use tracing::{debug, warn};

use super::config::WebhookConfig;
use super::payload::{format_discord, format_slack, format_telegram, NotificationEvent};
use super::transport::{ReqwestWebhookTransport, WebhookTransport};
use crate::wire::protocol::scrub_secret_patterns;

/// Send a notification to all configured webhooks using the default
/// `ReqwestWebhookTransport`.
pub async fn send_notification(config: &WebhookConfig, event: &NotificationEvent) {
    send_notification_with_transport(config, event, &ReqwestWebhookTransport).await;
}

/// Send a notification to all configured webhooks through the given transport.
///
/// This is the trait-backed entrypoint. Production code uses
/// `send_notification`; tests inject `MockWebhookTransport` here.
pub async fn send_notification_with_transport(
    config: &WebhookConfig,
    event: &NotificationEvent,
    transport: &dyn WebhookTransport,
) {
    let payload = serde_json::to_string(event).unwrap_or_default();

    if let Some(url) = &config.discord {
        let (content, embeds) = format_discord(event);
        let body = serde_json::json!({
            "content": content,
            "embeds": embeds,
        });
        if let Err(e) = transport.post_json(url, body).await {
            warn!(url = %url, error = %e, "Failed to send Discord notification");
        }
    }

    if let Some(url) = &config.slack {
        let text = format_slack(event);
        let body = serde_json::json!({ "text": text });
        if let Err(e) = transport.post_json(url, body).await {
            warn!(url = %url, error = %e, "Failed to send Slack notification");
        }
    }

    if let Some(url) = &config.telegram {
        let text = format_telegram(event);
        let body = serde_json::json!({ "text": text, "parse_mode": "Markdown" });
        if let Err(e) = transport.post_json(url, body).await {
            warn!(url = %url, error = %e, "Failed to send Telegram notification");
        }
    }

    debug!(event = %scrub_secret_patterns(&payload), "Notification sent");
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::notifications::webhook::MockWebhookTransport;

    #[tokio::test]
    async fn send_notification_with_empty_config_does_nothing() {
        let config = WebhookConfig {
            discord: None,
            slack: None,
            telegram: None,
        };
        let event = NotificationEvent::Error {
            source: "test".to_string(),
            message: "hello".to_string(),
        };
        let transport = MockWebhookTransport::default();
        send_notification_with_transport(&config, &event, &transport).await;
        assert!(transport.calls.lock().unwrap().is_empty());
    }

    #[tokio::test]
    async fn send_notification_with_transport_records_discord() {
        let event = NotificationEvent::Error {
            source: "test".to_string(),
            message: "hello".to_string(),
        };
        let transport = MockWebhookTransport::default();
        send_notification_with_transport(
            &WebhookConfig {
                discord: Some("https://discord.webhook/test".to_string()),
                slack: None,
                telegram: None,
            },
            &event,
            &transport,
        )
        .await;
        let calls = transport.calls.lock().unwrap();
        assert_eq!(calls.len(), 1);
        assert_eq!(calls[0].0, "https://discord.webhook/test");
    }

    #[tokio::test]
    async fn send_notification_serializes_event_to_debug_log() {
        let event = NotificationEvent::TeamSpawned {
            name: "test-team".to_string(),
            task: "test-task".to_string(),
            workers: 2,
            role: "qa".to_string(),
        };
        let payload = serde_json::to_string(&event).unwrap_or_default();
        assert!(!payload.is_empty());
        assert!(payload.contains("test-team"));
    }
}