southesk 0.0.8

A Rust client library for the Montrose MCP API.
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81

// Copyright 2026 Thomas Axelsson
// SPDX-License-Identifier: MIT

//! Errors that are not tied to specific modules.

use crate::auth_handler;

/// Errors that occur when doing calls to the MCP API.
#[derive(Debug, thiserror::Error)]
pub enum ClientCallError {
    /// Problems when communicating with the MCP server.
    #[error("MCP service communication error")]
    CommunicationError(#[from] rmcp::ServiceError),
    /// The MCP service returned an error response.
    #[error("error response from MCP service: {0}")]
    McpError(String),
    /// Failure to parse the response from the MCP service.
    #[error("failed to parse response: {msg}")]
    ParseError {
        msg: String,
        source: Option<Box<dyn std::error::Error + Send + Sync>>,
    },
    /// The arguments provided to the tool function are invalid.
    #[error("invalid arguments")]
    InvalidArguments(String),
}

impl ClientCallError {
    pub(crate) fn parse_err(msg: impl Into<String>) -> ClientCallError {
        Self::ParseError {
            msg: msg.into(),
            source: None,
        }
    }
}

/// Errors that occur when connecting the client.
#[derive(Debug, thiserror::Error)]
pub enum ClientConnectError {
    /// A step of the authentication process failed.
    #[error("authentication failed: {msg}")]
    AuthError {
        msg: String,
        source: Option<Box<dyn std::error::Error + Send + Sync>>,
    },
    /// A step of the authentication process failed, inside the
    /// [authentication handler](auth_handler).
    ///
    /// [`AuthFlowError::Aborted`](auth_handler::AuthFlowError::Aborted)
    ///  can indicate that the user cancelled or failed the authentication
    /// steps.
    #[error("authentication handler failed")]
    AuthHandlerError(#[from] auth_handler::AuthFlowError),
    /// Failed to set up a connection to the MCP servic.
    #[error("connection failed: {msg}")]
    ConnectionError {
        msg: String,
        source: Option<Box<dyn std::error::Error + Send + Sync>>,
    },
}

pub(crate) trait MapAuthToConnectError<T> {
    fn to_connect_auth_err(self, msg: impl Into<String>) -> Result<T, ClientConnectError>;
}

impl<T> MapAuthToConnectError<T> for Result<T, rmcp::transport::AuthError> {
    fn to_connect_auth_err(self, msg: impl Into<String>) -> Result<T, ClientConnectError> {
        self.map_err(|e| ClientConnectError::AuthError {
            msg: msg.into(),
            source: Some(Box::new(e)),
        })
    }
}

/// An error occured when building the client.
#[derive(Debug, thiserror::Error)]
#[error("failed to build client: {msg}")]
pub struct ClientBuildError {
    pub msg: String,
    pub source: Option<Box<dyn std::error::Error + Send + Sync>>,
}