neva 0.3.2

MCP SDK for Rust
Documentation 
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

//! Middleware factory functions

use crate::{
    middleware::{Middleware, MwContext, Next},
    types::{Message, Response},
};
use std::{future::Future, sync::Arc};

/// Turns a closure into middleware
#[inline]
pub(crate) fn make_mw<F, R>(f: F) -> Middleware
where
    F: Fn(MwContext, Next) -> R + Clone + Send + Sync + 'static,
    R: Future<Output = Response> + Send + 'static,
{
    Arc::new(move |ctx: MwContext, next: Next| Box::pin(f(ctx, next)))
}

/// Turns a closure into middleware that runs only
/// if the MCP server received a message that satisfies the condition.
#[inline]
pub(super) fn make_on<F, P, R>(f: F, p: P) -> Middleware
where
    F: Fn(MwContext, Next) -> R + Clone + Send + Sync + 'static,
    P: Fn(&Message) -> bool + Clone + Send + Sync + 'static,
    R: Future<Output = Response> + Send + 'static,
{
    let mw = move |ctx: MwContext, next: Next| {
        let f = f.clone();
        let p = p.clone();
        async move {
            if p(&ctx.msg) {
                f(ctx, next).await
            } else {
                next(ctx).await
            }
        }
    };
    make_mw(mw)
}

/// Turns a closure into middleware that runs only
/// if the MCP server received a message that satisfies the condition.
#[inline]
pub(super) fn make_on_command<F, R>(f: F, command: &'static str) -> Middleware
where
    F: Fn(MwContext, Next) -> R + Clone + Send + Sync + 'static,
    R: Future<Output = Response> + Send + 'static,
{
    make_on(f, move |msg| {
        if let Message::Request(req) = msg {
            req.method == command
        } else {
            false
        }
    })
}