mcp_server/
errors.rs

1use thiserror::Error;
2
3pub type BoxError = Box<dyn std::error::Error + Sync + Send>;
4
5#[derive(Error, Debug)]
6pub enum TransportError {
7    #[error("IO error: {0}")]
8    Io(#[from] std::io::Error),
9
10    #[error("JSON serialization error: {0}")]
11    Json(#[from] serde_json::Error),
12
13    #[error("Invalid UTF-8 sequence: {0}")]
14    Utf8(#[from] std::string::FromUtf8Error),
15
16    #[error("Protocol error: {0}")]
17    Protocol(String),
18
19    #[error("Invalid message format: {0}")]
20    InvalidMessage(String),
21}
22
23#[derive(Error, Debug)]
24pub enum ServerError {
25    #[error("Transport error: {0}")]
26    Transport(#[from] TransportError),
27
28    #[error("Service error: {0}")]
29    Service(String),
30
31    #[error("Internal error: {0}")]
32    Internal(String),
33
34    #[error("Request timed out")]
35    Timeout(#[from] tower::timeout::error::Elapsed),
36}
37
38#[derive(Error, Debug)]
39pub enum RouterError {
40    #[error("Method not found: {0}")]
41    MethodNotFound(String),
42
43    #[error("Invalid parameters: {0}")]
44    InvalidParams(String),
45
46    #[error("Internal error: {0}")]
47    Internal(String),
48
49    #[error("Tool not found: {0}")]
50    ToolNotFound(String),
51
52    #[error("Resource not found: {0}")]
53    ResourceNotFound(String),
54
55    #[error("Not found: {0}")]
56    PromptNotFound(String),
57}
58
59impl From<RouterError> for mcp_spec::protocol::ErrorData {
60    fn from(err: RouterError) -> Self {
61        use mcp_spec::protocol::*;
62        match err {
63            RouterError::MethodNotFound(msg) => ErrorData {
64                code: METHOD_NOT_FOUND,
65                message: msg,
66                data: None,
67            },
68            RouterError::InvalidParams(msg) => ErrorData {
69                code: INVALID_PARAMS,
70                message: msg,
71                data: None,
72            },
73            RouterError::Internal(msg) => ErrorData {
74                code: INTERNAL_ERROR,
75                message: msg,
76                data: None,
77            },
78            RouterError::ToolNotFound(msg) => ErrorData {
79                code: INVALID_REQUEST,
80                message: msg,
81                data: None,
82            },
83            RouterError::ResourceNotFound(msg) => ErrorData {
84                code: INVALID_REQUEST,
85                message: msg,
86                data: None,
87            },
88            RouterError::PromptNotFound(msg) => ErrorData {
89                code: INVALID_REQUEST,
90                message: msg,
91                data: None,
92            },
93        }
94    }
95}
96
97impl From<mcp_spec::handler::ResourceError> for RouterError {
98    fn from(err: mcp_spec::handler::ResourceError) -> Self {
99        match err {
100            mcp_spec::handler::ResourceError::NotFound(msg) => RouterError::ResourceNotFound(msg),
101            _ => RouterError::Internal("Unknown resource error".to_string()),
102        }
103    }
104}