rootle-stdio 0.12.0

External rootle providers: NDJSON-RPC 2.0 over stdio child processes (the LSP model)
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

//! The initialize handshake (protocol v1, doc/provider-protocol.md):
//! one round trip per process generation — protocol check, provider
//! name, capabilities, and the advisory cache budget (v1.2) travel
//! here. `protocol` must be 1; anything else aborts stdio setup and
//! rootle falls back to github.

use super::StdioProvider;
use rootle_provider::{Capabilities, ErrorKind, ProviderError, ProviderResult};
use serde_json::{Value, json};

impl StdioProvider {
    /// The v1 handshake. Also the liveness proof after a rebuild: it
    /// runs ungated so the rebuilder can validate the fresh child
    /// while the transport sits in `Respawning`.
    pub(super) fn handshake(&self) -> ProviderResult<Value> {
        let mut params = json!({ "protocol": 1 });
        if self.cache_bytes > 0 {
            params["cache_bytes"] = json!(self.cache_bytes);
        }
        if let Some(dir) = &self.cache_dir {
            params["cache_dir"] = json!(dir.to_string_lossy());
        }
        let reply = self.exchange("initialize", params, false)?;
        let protocol = reply.get("protocol").and_then(Value::as_u64).unwrap_or(1);
        let session = self.shared.routing.lock().session;
        if protocol != 1 {
            crate::trace::lifecycle("handshake", |fields| {
                fields.insert("session".into(), serde_json::json!(session.value()));
                fields.insert("protocol".into(), serde_json::json!(protocol));
                fields.insert("outcome".into(), serde_json::json!("unsupported_protocol"));
            });
            return Err(ProviderError::new(
                ErrorKind::Provider,
                format!("unsupported provider protocol {protocol}"),
            ));
        }
        crate::trace::lifecycle("handshake", |fields| {
            fields.insert("session".into(), serde_json::json!(session.value()));
            fields.insert("protocol".into(), serde_json::json!(protocol));
            fields.insert("outcome".into(), serde_json::json!("ok"));
        });
        Ok(reply)
    }

    pub(super) fn initialize_from(mut self, reply: Value) -> ProviderResult<Self> {
        if let Some(name) = reply.get("name").and_then(Value::as_str) {
            self.name = format!("stdio:{name}");
        }
        // v1.3: the provider's modeline icon — a builtin name or a
        // literal glyph; absent means text-only.
        if let Some(icon) = reply.get("icon").and_then(Value::as_str) {
            self.icon = Some(icon.to_string());
        }
        if let Some(caps) = reply.get("capabilities") {
            self.capabilities = Capabilities {
                orgs: caps.get("orgs").and_then(Value::as_bool).unwrap_or(true),
                code_search: caps
                    .get("code_search")
                    .and_then(Value::as_bool)
                    .unwrap_or(true),
                // v1.3: absent inherits code_search (back-compat).
                file_search: caps
                    .get("file_search")
                    .and_then(Value::as_bool)
                    .unwrap_or_else(|| {
                        caps.get("code_search")
                            .and_then(Value::as_bool)
                            .unwrap_or(true)
                    }),
                // v1.5: revision awareness — absent means false
                // (default-branch-only), the inverse of the search
                // defaults.
                refs: caps.get("refs").and_then(Value::as_bool).unwrap_or(false),
                log: caps.get("log").and_then(Value::as_bool).unwrap_or(false),
                blame: caps.get("blame").and_then(Value::as_bool).unwrap_or(false),
                commit: caps.get("commit").and_then(Value::as_bool).unwrap_or(false),
            };
        }
        if let Some(bytes) = reply.pointer("/cache/bytes").and_then(Value::as_u64) {
            *self.cache_used.lock() = Some(bytes);
        }
        Ok(self)
    }
}