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}