agenterra 0.2.2

Generate production-ready MCP (Model Context Protocol) servers and clients 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
77
78
79
80
81
82
83
84
85
86
87
88

//! Business rules for the generation domain

use crate::generation::{GenerationError, Language};
use crate::protocols::{Protocol, Role};

/// Validates language support for a given protocol and role
pub fn validate_language_support(
    protocol: Protocol,
    role: &Role,
    language: Language,
) -> Result<(), GenerationError> {
    // Define supported languages per protocol/role combination
    match (protocol, role) {
        (Protocol::Mcp, Role::Server) => match language {
            Language::Rust => Ok(()),
            _ => Err(GenerationError::UnsupportedLanguageForProtocol { language, protocol }),
        },
        (Protocol::Mcp, Role::Client) => match language {
            Language::Rust => Ok(()),
            _ => Err(GenerationError::UnsupportedLanguageForProtocol { language, protocol }),
        },
        _ => {
            // For now, only MCP is implemented
            Err(GenerationError::ValidationError(format!(
                "Protocol {protocol} is not yet implemented"
            )))
        }
    }
}

/// Validates project name format
pub fn validate_project_name(name: &str) -> Result<(), GenerationError> {
    if name.is_empty() {
        return Err(GenerationError::ValidationError(
            "Project name cannot be empty".to_string(),
        ));
    }

    if !name
        .chars()
        .all(|c| c.is_alphanumeric() || c == '-' || c == '_')
    {
        return Err(GenerationError::ValidationError(
            "Project name must contain only alphanumeric characters, dashes, and underscores"
                .to_string(),
        ));
    }

    if name.starts_with('-') || name.starts_with('_') {
        return Err(GenerationError::ValidationError(
            "Project name cannot start with a dash or underscore".to_string(),
        ));
    }

    Ok(())
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_validate_language_support() {
        // MCP Server supports Rust, Python, TypeScript
        assert!(validate_language_support(Protocol::Mcp, &Role::Server, Language::Rust).is_ok());
        assert!(validate_language_support(Protocol::Mcp, &Role::Server, Language::Python).is_err());
        assert!(
            validate_language_support(Protocol::Mcp, &Role::Server, Language::TypeScript).is_err()
        );
        assert!(validate_language_support(Protocol::Mcp, &Role::Server, Language::Go).is_err());

        // MCP Client supports more languages
        assert!(validate_language_support(Protocol::Mcp, &Role::Client, Language::Go).is_err());
    }

    #[test]
    fn test_validate_project_name() {
        assert!(validate_project_name("my-project").is_ok());
        assert!(validate_project_name("my_project").is_ok());
        assert!(validate_project_name("project123").is_ok());

        assert!(validate_project_name("").is_err());
        assert!(validate_project_name("-project").is_err());
        assert!(validate_project_name("_project").is_err());
        assert!(validate_project_name("my project").is_err());
        assert!(validate_project_name("my@project").is_err());
    }
}