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

use std::collections::HashMap;

use tokio::sync::{oneshot, Mutex};

use crate::runtime::conversation::bus::PreflightAction;

pub type TicketId = String;

#[derive(Debug)]
pub struct PreflightInbox {
    pending: Mutex<HashMap<TicketId, oneshot::Sender<PreflightAction>>>,
}

pub struct PreflightTicket {
    pub id: TicketId,
    pub rx: oneshot::Receiver<PreflightAction>,
}

impl std::fmt::Debug for PreflightTicket {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("PreflightTicket")
            .field("id", &self.id)
            .finish_non_exhaustive()
    }
}

impl PreflightInbox {
    pub fn new() -> Self {
        Self {
            pending: Mutex::new(HashMap::new()),
        }
    }

    pub async fn arm(&self) -> PreflightTicket {
        let id = uuid::Uuid::new_v4().to_string();
        let (tx, rx) = oneshot::channel();
        self.pending.lock().await.insert(id.clone(), tx);
        PreflightTicket { id, rx }
    }

    pub async fn submit(&self, ticket_id: TicketId, action: PreflightAction) {
        if let Some(tx) = self.pending.lock().await.remove(&ticket_id) {
            let _ = tx.send(action);
        }
    }

    pub async fn cancel(&self, ticket_id: &str) {
        self.pending.lock().await.remove(ticket_id);
    }
}

impl Default for PreflightInbox {
    fn default() -> Self {
        Self::new()
    }
}