malvin 0.2.6

Non-interactive research and coding 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
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

use serde_json::Value;

use super::session::NpmPiSession;
use crate::acp::AgentError;

pub(super) async fn auto_reply_extension_ui(
    session: &NpmPiSession,
    request: &Value,
) -> Result<(), AgentError> {
    let Some(id) = request.get("id").and_then(Value::as_str) else {
        return Ok(());
    };
    let method = request.get("method").and_then(Value::as_str).unwrap_or("");
    if is_fire_and_forget(method) {
        return Ok(());
    }
    let response = extension_ui_response(id, method, request);
    super::session_io::write_json(session, &response).await
}

fn is_fire_and_forget(method: &str) -> bool {
    matches!(
        method,
        "notify" | "setStatus" | "setWidget" | "setTitle" | "set_editor_text"
    )
}

fn extension_ui_response(id: &str, method: &str, request: &Value) -> Value {
    match method {
        "confirm" => serde_json::json!({
            "type": "extension_ui_response",
            "id": id,
            "confirmed": true
        }),
        "select" => {
            let first = request
                .get("options")
                .and_then(Value::as_array)
                .and_then(|a| a.first())
                .cloned()
                .unwrap_or(Value::Null);
            serde_json::json!({
                "type": "extension_ui_response",
                "id": id,
                "value": first
            })
        }
        _ => serde_json::json!({
            "type": "extension_ui_response",
            "id": id,
            "cancelled": true
        }),
    }
}

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

    #[test]
    fn confirm_and_cancel_shapes() {
        let confirm = extension_ui_response("1", "confirm", &Value::Null);
        assert_eq!(confirm["confirmed"], true);
        let cancel = extension_ui_response("2", "input", &Value::Null);
        assert_eq!(cancel["cancelled"], true);
        assert!(is_fire_and_forget("notify"));
        assert!(!is_fire_and_forget("confirm"));
    }
}