rustyroad 1.8.4

Rusty Road is a framework written in Rust that is based on Ruby on Rails. It is designed to provide the familiar conventions and ease of use of Ruby on Rails, while also taking advantage of the performance and efficiency of Rust.
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

//! MCP `rustyroad_enums` tool: enum types and their allowed values.

use super::{McpServer, Tool};
use serde_json::{json, Value};
use std::env;

/// Returns the `rustyroad_enums` tool catalog entry.
pub(super) fn tool_definition() -> Tool {
    Tool {
        name: "rustyroad_enums".to_string(),
        description: "Get the database enum types and their allowed values. Use this before writing queries or validations that involve enum columns.".to_string(),
        input_schema: json!({
            "type": "object",
            "properties": {
                "env": {
                    "type": "string",
                    "description": "Environment to use (dev, prod, test)",
                    "enum": ["dev", "prod", "test"]
                }
            }
        }),
    }
}

impl McpServer {
    /// Handles `tools/call` for `rustyroad_enums`.
    ///
    /// Reuses the library's enum reader so the MCP tool and the `db enums`
    /// CLI report the same data from one implementation.
    pub(super) async fn handle_enums(&self, args: Value) -> Result<Value, String> {
        let env = args
            .get("env")
            .and_then(|v| v.as_str())
            .unwrap_or(&self.environment);

        env::set_var("ENVIRONMENT", env);
        let _guard = self.change_to_project_dir()?;

        let output = rustyroad::database::enums::read_enums()
            .await
            .map_err(|e| format!("Failed to read enums: {}", e))?;

        let enums: Vec<Value> = output
            .enums
            .iter()
            .map(|e| {
                json!({
                    "name": e.name,
                    "values": e.values,
                    "table": e.table,
                    "column": e.column,
                })
            })
            .collect();

        Ok(json!({
            "success": true,
            "environment": env,
            "database": output.database_type,
            "enums": enums
        }))
    }
}