nab 0.7.1

Token-optimized HTTP client for LLMs — fetches any URL as clean markdown
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

//! Shared HTTP client and session-store singletons for `nab-mcp` tools.
//!
//! Both `OnceCell` globals are initialised lazily on first use and live for
//! the process lifetime, so all tool invocations share the same connection
//! pool and cookie jars.

use rust_mcp_sdk::schema::schema_utils::CallToolError;
use tokio::sync::OnceCell;

use nab::{AcceleratedClient, SessionStore};

// ─── Global singletons ────────────────────────────────────────────────────────

/// Process-wide `AcceleratedClient` — connection pool shared by all tools.
static CLIENT: OnceCell<AcceleratedClient> = OnceCell::const_new();

/// Return (or lazily create) the global `AcceleratedClient`.
pub async fn get_client() -> &'static AcceleratedClient {
    CLIENT
        .get_or_init(|| async { AcceleratedClient::new().expect("Failed to create HTTP client") })
        .await
}

/// Process-wide `SessionStore` — cookie jars keyed by session name.
static SESSION_STORE: OnceCell<SessionStore> = OnceCell::const_new();

pub(crate) async fn get_session_store() -> &'static SessionStore {
    SESSION_STORE
        .get_or_init(|| async { SessionStore::new() })
        .await
}

// ─── Session helpers ──────────────────────────────────────────────────────────

/// Look up (or create) the named session, seeding it with browser cookies when
/// `seed_cookies` is non-empty.  Returns the session's dedicated `reqwest::Client`.
pub(crate) async fn resolve_session_client(
    name: &str,
    seed_cookies: Option<&str>,
    seed_url: &str,
) -> Result<reqwest::Client, CallToolError> {
    let store = get_session_store().await;
    let seed = seed_cookies.filter(|s| !s.is_empty());
    let entry = store
        .get_or_create(name, seed, Some(seed_url))
        .await
        .map_err(|e| CallToolError::from_message(e.to_string()))?;
    Ok(entry.client)
}

/// Build a temporary per-call client with its own cookie jar.
///
/// Used by non-session MCP flows that still need cookie seeding and `Set-Cookie`
/// persistence across multiple requests inside a single tool invocation.
pub(crate) async fn build_transient_client(
    seed_cookies: Option<&str>,
    seed_url: &str,
) -> Result<reqwest::Client, CallToolError> {
    let store = SessionStore::new();
    let seed = seed_cookies.filter(|s| !s.is_empty());
    let entry = store
        .get_or_create("transient", seed, Some(seed_url))
        .await
        .map_err(|e| CallToolError::from_message(e.to_string()))?;
    Ok(entry.client)
}