systemprompt-mcp 0.1.14

Core MCP (Model Context Protocol) functionality for systemprompt.io OS
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

pub mod rbac;
pub mod session_manager;

use rmcp::service::RequestContext;
use rmcp::{ErrorData as McpError, RoleServer};
use systemprompt_models::RequestContext as SysRequestContext;
use systemprompt_traits::ContextPropagation;

pub use rbac::{enforce_rbac_from_registry, AuthResult, AuthenticatedRequestContext};
pub use session_manager::DatabaseSessionManager;

pub fn extract_bearer_token(
    mcp_context: &RequestContext<RoleServer>,
) -> Result<Option<String>, McpError> {
    let parts = mcp_context
        .extensions
        .get::<http::request::Parts>()
        .ok_or_else(|| {
            McpError::invalid_request("No HTTP parts in MCP context".to_string(), None)
        })?;

    let auth_header = parts
        .headers
        .get("authorization")
        .and_then(|v| v.to_str().ok());

    if let Some(auth) = auth_header {
        if let Some(token) = auth.strip_prefix("Bearer ") {
            return Ok(Some(token.to_string()));
        }
    }

    Ok(None)
}

pub fn extract_request_context(
    mcp_context: &RequestContext<RoleServer>,
) -> Result<SysRequestContext, McpError> {
    let parts = mcp_context.extensions.get::<http::request::Parts>();

    if let Some(parts) = parts {
        if let Some(request_context) = parts.extensions.get::<SysRequestContext>() {
            return Ok(request_context.clone());
        }

        return SysRequestContext::from_headers(&parts.headers)
            .map_err(|e| McpError::invalid_request(e.to_string(), None));
    }

    Err(McpError::invalid_request(
        "RequestContext missing - no axum parts in MCP context",
        None,
    ))
}