mcp_host/
lib.rs

1//! mcp-host: Production-grade MCP server implementation
2//!
3//! This crate provides a comprehensive Model Context Protocol (MCP) server
4//! implementation with sophisticated features like circuit breakers, visibility
5//! predicates, session customization, and more.
6//!
7//! ## Features
8//!
9//! - `stdio` - STDIO transport (default)
10//! - `http` - HTTP transport via rama
11//! - `macros` - Procedural macros (#[mcp_tool], #[mcp_tool_router])
12//! - `telemetry` - OpenTelemetry integration
13//! - `file-watcher` - File watching support
14//! - `full` - All features
15//!
16//! ## Quick Start
17//!
18//! ```rust,ignore
19//! use mcp_host::prelude::*;
20//! use mcp_host::macros::text;
21//! use schemars::JsonSchema;
22//! use serde::Deserialize;
23//!
24//! #[derive(Deserialize, JsonSchema)]
25//! struct GreetParams {
26//!     name: String,
27//! }
28//!
29//! struct MyServer;
30//!
31//! #[mcp_tool_router]
32//! impl MyServer {
33//!     #[mcp_tool(name = "greet")]
34//!     async fn greet(&self, _ctx: Ctx<'_>, params: Parameters<GreetParams>) -> ToolResult {
35//!         Ok(vec![text(&format!("Hello, {}!", params.0.name))])
36//!     }
37//! }
38//!
39//! #[tokio::main]
40//! async fn main() {
41//!     let server = mcp_host::server::builder::server("my-server", "1.0.0")
42//!         .with_tools(true)
43//!         .build();
44//!
45//!     let my_server = std::sync::Arc::new(MyServer);
46//!     MyServer::tool_router().register_all(server.tool_registry(), my_server);
47//!
48//!     server.run(StdioTransport::new()).await.unwrap();
49//! }
50//! ```
51//!
52//! Enable macros with: `mcp-host = { version = "0.1", features = ["macros"] }`
53
54pub mod content;
55pub mod logging;
56#[cfg(feature = "macros")]
57pub mod macros;
58pub mod managers;
59pub mod protocol;
60pub mod registry;
61pub mod server;
62pub mod transport;
63pub mod utils;
64
65// Re-export commonly used types
66pub mod prelude {
67    // Protocol types
68    pub use crate::protocol::capabilities::{ClientCapabilities, ServerCapabilities};
69    pub use crate::protocol::types::{JsonRpcError, JsonRpcRequest, JsonRpcResponse};
70
71    // Content types and builders
72    pub use crate::content::annotations::Annotations;
73    pub use crate::content::builder::{audio, image, resource, text};
74    pub use crate::content::resource::ResourceContent;
75    pub use crate::content::types::{
76        AudioContent, Content, ImageContent, ResourceLink, TextContent,
77    };
78
79    // Server types
80    pub use crate::server::builder::{server, ServerBuilder};
81    pub use crate::server::core::Server;
82    pub use crate::server::multiplexer::{
83        ClientRequester, CreateMessageParams, CreateMessageResult, JsonRpcClientRequest,
84        ListRootsResult, ModelHint, ModelPreferences, MultiplexerError, RequestMultiplexer, Root,
85        SamplingContent, SamplingMessage,
86    };
87    pub use crate::server::profile::SessionProfile;
88    pub use crate::server::session::Session;
89    pub use crate::server::visibility::{
90        Environment, ExecutionContext, SimpleEnvironment, VisibilityContext,
91    };
92
93    // Prompt message types
94    pub use crate::content::prompt::{PromptContent, PromptMessage, PromptRole};
95
96    // Completion types
97    pub use crate::protocol::types::{
98        CompleteRequest, CompleteResult, CompletionArgument, CompletionInfo, CompletionRef,
99        CompletionValue,
100    };
101
102    // Utility functions
103    pub use crate::utils::{
104        base64_decode, base64_encode, byte_to_line_col, collect_files, is_safe_path,
105    };
106
107    // Transport traits
108    pub use crate::transport::traits::{
109        IncomingMessage, JsonRpcNotification, Transport, TransportError,
110    };
111
112    // Registry types
113    pub use crate::registry::prompts::{
114        GetPromptResult, Prompt, PromptArgument, PromptError, PromptInfo, PromptManager,
115    };
116    pub use crate::registry::resources::{
117        Resource, ResourceError, ResourceInfo, ResourceManager, ResourceRetryConfig,
118        ResourceTemplate, ResourceTemplateInfo,
119    };
120    pub use crate::registry::tools::{
121        Tool, ToolBreakerConfig, ToolError, ToolHelpers, ToolInfo, ToolRegistry,
122    };
123
124    // Middleware
125    pub use crate::server::middleware::{RateLimiter, RateLimiterConfig};
126
127    // Logging
128    pub use crate::logging::{send_log_notification, LogLevel, LoggerConfig, McpLogger};
129
130    // Managers
131    pub use crate::managers::cancellation::{CancellationManager, CancellationToken};
132    pub use crate::managers::completion::{CompletionError, CompletionManager, CompletionProvider};
133    pub use crate::managers::progress::{ProgressToken, ProgressTracker};
134    pub use crate::managers::subscription::SubscriptionManager;
135    pub use crate::managers::task::{TaskError, TaskStore};
136
137    // === Feature-gated exports ===
138
139    // STDIO transport
140    #[cfg(feature = "stdio")]
141    pub use crate::transport::stdio::StdioTransport;
142
143    // HTTP transport
144    #[cfg(feature = "http")]
145    pub use crate::transport::http::{HttpTransport, HttpTransportConfig, HttpTransportError};
146
147    // Tool router (requires macros feature)
148    #[cfg(feature = "macros")]
149    pub use crate::registry::router::{tool_info, McpToolRouter};
150
151    // Macro support types (requires macros feature)
152    #[cfg(feature = "macros")]
153    pub use crate::macros::{schema_for, Ctx, IntoToolResult, Parameters, ToolResult};
154
155    // Procedural macros (requires macros feature)
156    #[cfg(feature = "macros")]
157    pub use mcp_host_macros::{mcp_tool, mcp_tool_router};
158}
mcp_host/lib.rs

mcp_host/
lib.rs
