mcp-forge 0.1.5

Generate Rust MCP servers from OpenAPI specs
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

#![cfg(test)]

use crate::ir::{
    ApiSpec, AuthMethod, EnumVariant, FieldDef, HttpMethod, Operation, RustType, TypeDef,
};
use heck::{ToSnakeCase, ToUpperCamelCase};

#[must_use]
pub fn make_field(name: &str, rust_type: RustType, required: bool) -> FieldDef {
    FieldDef {
        name: name.into(),
        rust_name: name.to_snake_case(),
        rust_type,
        required,
        description: None,
        default_value: None,
    }
}

#[must_use]
pub fn make_struct(name: &str, fields: Vec<FieldDef>) -> TypeDef {
    TypeDef {
        name: name.into(),
        rust_name: name.to_upper_camel_case(),
        fields,
        is_enum: false,
        enum_variants: Vec::new(),
        description: None,
    }
}

#[must_use]
pub fn make_enum(name: &str, variants: Vec<&str>) -> TypeDef {
    TypeDef {
        name: name.into(),
        rust_name: name.to_upper_camel_case(),
        fields: Vec::new(),
        is_enum: true,
        enum_variants: variants
            .into_iter()
            .map(|v| EnumVariant {
                name: v.into(),
                rust_name: v.to_upper_camel_case(),
            })
            .collect(),
        description: None,
    }
}

#[must_use]
pub fn make_spec_with(types: Vec<TypeDef>, operations: Vec<Operation>) -> ApiSpec {
    ApiSpec {
        name: "TestApi".into(),
        description: None,
        version: "1.0.0".into(),
        base_url: None,
        auth: AuthMethod::None,
        operations,
        types,
    }
}

#[must_use]
pub fn make_get_op(id: &str, response_type: RustType) -> Operation {
    Operation {
        id: id.into(),
        method: HttpMethod::Get,
        path: format!("/{id}"),
        summary: Some(format!("Get {id}")),
        description: None,
        parameters: vec![],
        request_body: None,
        response_type: Some(response_type),
        errors: vec![],
    }
}