nebu-ctx 0.5.5

Lean-ctx runtime adapted for the NebuCtx Cloud-backed product.
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
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use crate::models::ServerConnection;
use anyhow::{Context, Result};
use std::fs;
use std::path::PathBuf;

const CONNECTION_FILE: &str = "server_connection.json";

pub fn config_dir() -> PathBuf {
    let home = preferred_home_dir().unwrap_or_else(|| PathBuf::from("."));
    home.join(".nebu-ctx").join("cloud")
}

pub(crate) fn preferred_home_dir() -> Option<PathBuf> {
    for variable_name in ["NEBU_CTX_HOME", "HOME", "USERPROFILE"] {
        if let Ok(value) = std::env::var(variable_name) {
            let trimmed = value.trim();
            if !trimmed.is_empty() {
                return Some(PathBuf::from(trimmed));
            }
        }
    }

    dirs::home_dir()
}

pub(crate) fn preferred_os_home_dir() -> Option<PathBuf> {
    for variable_name in ["HOME", "USERPROFILE"] {
        if let Ok(value) = std::env::var(variable_name) {
            let trimmed = value.trim();
            if !trimmed.is_empty() {
                return Some(PathBuf::from(trimmed));
            }
        }
    }

    dirs::home_dir()
}

pub fn connection_path() -> PathBuf {
    config_dir().join(CONNECTION_FILE)
}

pub fn normalize_server_endpoint(endpoint: &str) -> String {
    let trimmed = endpoint.trim().trim_end_matches('/');
    if let Some(prefix) = trimmed.strip_suffix("/mcp") {
        return prefix.to_string();
    }

    if let Some(prefix) = trimmed.strip_suffix("/v1/tools/call") {
        return prefix.to_string();
    }

    if let Some(prefix) = trimmed.strip_suffix("/v1/tools") {
        return prefix.to_string();
    }

    if let Some(prefix) = trimmed.strip_suffix("/v1/manifest") {
        return prefix.to_string();
    }

    if let Some(prefix) = trimmed.strip_suffix("/health") {
        return prefix.to_string();
    }

    trimmed.to_string()
}

pub fn save_connection(endpoint: &str, token: &str) -> Result<ServerConnection> {
    let connection = ServerConnection {
        endpoint: normalize_server_endpoint(endpoint),
        token: token.trim().to_string(),
    };

    fs::create_dir_all(config_dir()).context("failed to create client config directory")?;
    let json = serde_json::to_string_pretty(&connection).context("failed to serialize connection")?;
    fs::write(connection_path(), json).context("failed to write connection file")?;
    Ok(connection)
}

pub fn load_connection() -> Result<Option<ServerConnection>> {
    let path = connection_path();
    if !path.exists() {
        return Ok(None);
    }

    let data = fs::read_to_string(&path).with_context(|| format!("failed to read {}", path.display()))?;
    let connection = serde_json::from_str(&data).context("failed to parse connection file")?;
    Ok(Some(connection))
}

pub fn clear_connection() -> Result<()> {
    let path = connection_path();
    if path.exists() {
        fs::remove_file(&path).with_context(|| format!("failed to remove {}", path.display()))?;
    }

    Ok(())
}

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

    #[test]
    fn normalize_server_endpoint_trims_known_api_paths() {
        assert_eq!(normalize_server_endpoint("http://localhost:4242/mcp"), "http://localhost:4242");
        assert_eq!(normalize_server_endpoint("http://localhost:4242/v1/tools/call"), "http://localhost:4242");
        assert_eq!(normalize_server_endpoint("http://localhost:4242/v1/tools"), "http://localhost:4242");
        assert_eq!(normalize_server_endpoint("http://localhost:4242/health"), "http://localhost:4242");
    }
}