ironflow-mcp 0.1.2

MCP server for Ironflow workflow orchestration
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

//! MCP server handler implementation.

use std::sync::Arc;

use async_trait::async_trait;
use rust_mcp_sdk::McpServer;
use rust_mcp_sdk::mcp_server::ServerHandler;
use rust_mcp_sdk::schema::{
    CallToolRequestParams, CallToolResult, ListToolsResult, PaginatedRequestParams, RpcError,
    schema_utils::CallToolError,
};

use crate::client::ApiClient;
use crate::tools::IronflowTools;

/// MCP server handler that dispatches tool calls to the Ironflow API.
pub struct IronflowHandler {
    client: Arc<ApiClient>,
}

impl IronflowHandler {
    /// Create a new handler with the given API client.
    pub fn new(client: Arc<ApiClient>) -> Self {
        Self { client }
    }
}

#[async_trait]
impl ServerHandler for IronflowHandler {
    async fn handle_list_tools_request(
        &self,
        _params: Option<PaginatedRequestParams>,
        _runtime: Arc<dyn McpServer>,
    ) -> Result<ListToolsResult, RpcError> {
        Ok(ListToolsResult {
            tools: IronflowTools::tools(),
            meta: None,
            next_cursor: None,
        })
    }

    async fn handle_call_tool_request(
        &self,
        params: CallToolRequestParams,
        _runtime: Arc<dyn McpServer>,
    ) -> Result<CallToolResult, CallToolError> {
        let tool = IronflowTools::try_from(params).map_err(CallToolError::new)?;

        match tool {
            IronflowTools::ListWorkflowsTool(t) => t.run(&self.client).await,
            IronflowTools::GetWorkflowTool(t) => t.run(&self.client).await,
            IronflowTools::CreateRunTool(t) => t.run(&self.client).await,
            IronflowTools::ListRunsTool(t) => t.run(&self.client).await,
            IronflowTools::GetRunTool(t) => t.run(&self.client).await,
            IronflowTools::CancelRunTool(t) => t.run(&self.client).await,
            IronflowTools::ApproveRunTool(t) => t.run(&self.client).await,
            IronflowTools::RejectRunTool(t) => t.run(&self.client).await,
            IronflowTools::RetryRunTool(t) => t.run(&self.client).await,
            IronflowTools::GetStatsTool(t) => t.run(&self.client).await,
        }
    }
}