robit 0.1.20

Terminal UI for the robit AI automation 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

//! TuiFrontend — implements the Frontend trait, forwarding events through channels.

use async_trait::async_trait;
use robit_agent::{AgentEvent, Frontend, ToolCallInfo};
use tokio::sync::{mpsc, oneshot};

/// A confirmation request sent from the Frontend to the TUI event loop.
pub struct ConfirmRequest {
    pub tool_info: ToolCallInfo,
    pub responder: oneshot::Sender<bool>,
}

/// TUI implementation of the Frontend trait.
/// All methods send data through channels — the TUI event loop handles rendering.
pub struct TuiFrontend {
    pub event_tx: mpsc::Sender<AgentEvent>,
    pub confirm_tx: mpsc::Sender<ConfirmRequest>,
}

#[async_trait]
impl Frontend for TuiFrontend {
    async fn on_event(&self, event: AgentEvent) -> robit_agent::error::Result<()> {
        let _ = self.event_tx.send(event).await;
        Ok(())
    }

    async fn request_tool_confirmation(
        &self,
        info: &ToolCallInfo,
    ) -> robit_agent::error::Result<bool> {
        let (tx, rx) = oneshot::channel();
        let _ = self
            .confirm_tx
            .send(ConfirmRequest {
                tool_info: info.clone(),
                responder: tx,
            })
            .await;
        match rx.await {
            Ok(approved) => Ok(approved),
            Err(_) => Ok(false),
        }
    }
}