Skip to main content

mcp_memory/
errors.rs

1use thiserror::Error;
2
3#[derive(Error, Debug)]
4pub enum MCSError {
5    #[error("Parse error: {0}")]
6    ParseError(String),
7
8    #[error("Method not found: {0}")]
9    MethodNotFound(String),
10
11    #[error("Invalid params: {0}")]
12    InvalidParams(String),
13
14    #[error("Memory error: {0}")]
15    MemoryError(String),
16
17    #[error("IO error: {0}")]
18    IoError(#[from] std::io::Error),
19
20    #[error("JSON error: {0}")]
21    JsonError(#[from] serde_json::Error),
22}
23
24impl MCSError {
25    pub const fn error_code(&self) -> i64 {
26        match self {
27            MCSError::ParseError(_) => -32700,
28            MCSError::MethodNotFound(_) => -32601,
29            MCSError::InvalidParams(_) => -32602,
30            MCSError::MemoryError(_) => -32000,
31            MCSError::IoError(_) => -32003,
32            MCSError::JsonError(_) => -32700,
33        }
34    }
35}
36
37pub type Result<T> = std::result::Result<T, MCSError>;
38
39#[cfg(test)]
40mod tests {
41    use super::*;
42
43    #[test]
44    fn test_error_codes() {
45        assert_eq!(MCSError::ParseError("".into()).error_code(), -32700);
46        assert_eq!(MCSError::MethodNotFound("".into()).error_code(), -32601);
47        assert_eq!(MCSError::InvalidParams("".into()).error_code(), -32602);
48        assert_eq!(MCSError::MemoryError("".into()).error_code(), -32000);
49        assert_eq!(MCSError::IoError(std::io::Error::other("")).error_code(), -32003);
50        assert_eq!(MCSError::JsonError(serde_json::from_str::<()>("invalid").unwrap_err()).error_code(), -32700);
51    }
52}