lean_ctx/
cloud_client.rs

1use crate::config;
2use crate::models::{ProjectContext, ProjectResolutionRequest, ProjectResolutionResponse, ServerConnection, TelemetryIngestRequest, ToolCallRequest, ToolCallResponse, ToolListResponse};
3use anyhow::{anyhow, Context, Result};
4use serde::de::DeserializeOwned;
5use serde::Serialize;
6use serde_json::{Map, Value};
7
8pub struct ServerClient {
9    connection: ServerConnection,
10}
11
12impl ServerClient {
13    pub fn load() -> Result<Self> {
14        let connection = config::load_connection()?
15            .ok_or_else(|| anyhow!("No server connection saved. Run `nebu-ctx cloud connect`."))?;
16        Ok(Self { connection })
17    }
18
19    pub fn new(connection: ServerConnection) -> Self {
20        Self { connection }
21    }
22
23    pub fn endpoint(&self) -> &str {
24        &self.connection.endpoint
25    }
26
27    pub fn health(&self) -> Result<Value> {
28        self.get_json("/health")
29    }
30
31    pub fn manifest(&self) -> Result<Value> {
32        self.get_json("/v1/manifest")
33    }
34
35    pub fn list_tools(&self) -> Result<ToolListResponse> {
36        self.get_json("/v1/tools")
37    }
38
39    pub fn resolve_project(&self, project_context: &ProjectContext) -> Result<ProjectResolutionResponse> {
40        self.post_json(
41            "/v1/projects/resolve",
42            &ProjectResolutionRequest {
43                fingerprint: project_context.fingerprint.clone(),
44                suggested_slug: Some(project_context.project_slug.clone()),
45                checkout_binding: Some(project_context.checkout_binding.clone()),
46                project_metadata: project_context.project_metadata.clone(),
47            },
48        )
49    }
50
51    pub fn call_tool(&self, tool_name: &str, arguments: Map<String, Value>, project_context: &ProjectContext) -> Result<Value> {
52        let response: ToolCallResponse = self.post_json(
53            "/v1/tools/call",
54            &ToolCallRequest {
55                name: tool_name.to_string(),
56                arguments,
57                project_id: None,
58                project_slug: Some(project_context.project_slug.clone()),
59                repository_fingerprint: Some(project_context.fingerprint.clone()),
60                checkout_binding: Some(project_context.checkout_binding.clone()),
61                project_metadata: project_context.project_metadata.clone(),
62            },
63        )?;
64
65        Ok(response.result)
66    }
67
68    /// Posts a single tool-call telemetry event to the server for dashboard aggregation.
69    /// Only token counts and metadata are sent — no raw content.
70    pub fn ingest_telemetry(&self, request: &TelemetryIngestRequest) -> Result<()> {
71        let _: serde_json::Value = self.post_json("/v1/telemetry/ingest", request)?;
72        Ok(())
73    }
74
75    fn get_json<T>(&self, path: &str) -> Result<T>
76    where
77        T: DeserializeOwned,
78    {
79        let response = ureq::get(&self.url(path))
80            .header("Authorization", &format!("Bearer {}", self.connection.token.trim()))
81            .call()
82            .map_err(|error| anyhow!("Request to {} failed: {}", self.url(path), error))?;
83        Self::read_json(response)
84    }
85
86    fn post_json<TResponse, TRequest>(&self, path: &str, request: &TRequest) -> Result<TResponse>
87    where
88        TResponse: DeserializeOwned,
89        TRequest: Serialize,
90    {
91        let body = serde_json::to_vec(request).context("failed to serialize request")?;
92        let response = ureq::post(&self.url(path))
93            .header("Authorization", &format!("Bearer {}", self.connection.token.trim()))
94            .header("Content-Type", "application/json")
95            .send(body.as_slice())
96            .map_err(|error| anyhow!("Request to {} failed: {}", self.url(path), error))?;
97        Self::read_json(response)
98    }
99
100    /// Deserializes a JSON response body into a target type.
101    fn read_json<T>(response: ureq::http::Response<ureq::Body>) -> Result<T>
102    where
103        T: DeserializeOwned,
104    {
105        let mut body = response.into_body();
106        let payload = body.read_to_string().context("failed to read response body")?;
107        serde_json::from_str(&payload).context("failed to parse server response")
108    }
109
110    /// Combines the normalized endpoint and a relative API path.
111    fn url(&self, path: &str) -> String {
112        format!("{}{}", self.connection.endpoint.trim_end_matches('/'), path)
113    }
114}
115
116/// Posts every current, high-confidence fact from local `knowledge.json` to the
117/// cloud `ctx_knowledge` store. Called after auto-consolidation and from
118/// `handle_stop()` to keep PostgreSQL in sync with the local session outcome.
119/// Silently returns if the cloud is not configured or any call fails.
120pub fn post_knowledge_to_cloud(project_root: &str) {
121    let Ok(client) = ServerClient::load() else { return };
122    let ctx = crate::git_context::discover_project_context(std::path::Path::new(project_root));
123    let knowledge = crate::core::knowledge::ProjectKnowledge::load_or_create(project_root);
124
125    for fact in knowledge.facts.iter().filter(|f| f.is_current() && f.confidence >= 0.7) {
126        let mut args = Map::new();
127        args.insert("action".to_string(), Value::String("remember".to_string()));
128        args.insert("category".to_string(), Value::String(fact.category.clone()));
129        args.insert("key".to_string(), Value::String(fact.key.clone()));
130        args.insert("value".to_string(), Value::String(fact.value.clone()));
131        args.insert("confidence".to_string(), serde_json::json!(fact.confidence));
132        let _ = client.call_tool("ctx_knowledge", args, &ctx);
133    }
134}
135
136/// Posts a session summary to `ctx_brain` when a session is saved.
137/// Silently returns if the cloud is not configured.
138pub fn post_session_to_brain(session: &crate::core::session::SessionState) {
139    let Ok(client) = ServerClient::load() else { return };
140    let current_dir = std::env::current_dir().unwrap_or_default();
141    let ctx = crate::git_context::discover_project_context(&current_dir);
142
143    let task = session.task.as_ref().map(|t| t.description.as_str()).unwrap_or("(no task)");
144    let summary = format!(
145        "session={} task=\"{}\" calls={} tokens_saved={} decisions={} findings={}",
146        session.id,
147        task,
148        session.stats.total_tool_calls,
149        session.stats.total_tokens_saved,
150        session.decisions.len(),
151        session.findings.len(),
152    );
153    let key = format!("session-{}", session.id);
154
155    let mut args = Map::new();
156    args.insert("action".to_string(), Value::String("store".to_string()));
157    args.insert("key".to_string(), Value::String(key));
158    args.insert("value".to_string(), Value::String(summary));
159    let _ = client.call_tool("ctx_brain", args, &ctx);
160}
lean_ctx/cloud_client.rs

lean_ctx/
cloud_client.rs
