"添加前端模板和运行代码模块"

qiming-run_code_rmcp/src/mcp/mcp_server.rs

@@ -0,0 +1,164 @@
+use anyhow::Result;
+use rmcp::{
+    ErrorData as McpError, ServerHandler,
+    handler::server::wrapper::Parameters,
+    model::{
+        CallToolResult, Content, Implementation, ProtocolVersion, ServerCapabilities, ServerInfo,
+    },
+    tool, tool_handler, tool_router,
+};
+use serde::Deserialize;
+use serde_json::json;
+
+use crate::model::{CodeExecutor, LanguageScript};
+
+/// 代码执行请求参数
+#[derive(Debug, Deserialize, schemars::JsonSchema)]
+pub struct CodeRunRequest {
+    #[schemars(description = "要执行的代码")]
+    pub code: String,
+
+    #[schemars(description = "可选的执行参数")]
+    pub params: Option<serde_json::Value>,
+}
+
+/// 代码执行工具服务
+#[derive(Debug, Clone, Default)]
+pub struct CodeRunnerService;
+
+#[tool_router]
+impl CodeRunnerService {
+    #[tool(description = "执行JavaScript代码并返回结果")]
+    async fn run_javascript(
+        &self,
+        request: Parameters<CodeRunRequest>,
+    ) -> Result<CallToolResult, McpError> {
+        let request = request.0;
+        match CodeExecutor::execute_with_params_compat(
+            &request.code,
+            LanguageScript::Js,
+            request.params,
+        )
+        .await
+        {
+            Ok(result) => {
+                if result.success {
+                    let content = Content::json(json!({
+                        "result": result.result,
+                        "logs": result.logs,
+                        "success": true
+                    }))?;
+                    Ok(CallToolResult::success(vec![content]))
+                } else {
+                    let content = Content::json(json!({
+                        "success": false,
+                        "error": result.error,
+                        "logs": result.logs
+                    }))?;
+                    Ok(CallToolResult::success(vec![content]))
+                }
+            }
+            Err(err) => {
+                let content = Content::json(json!({
+                    "success": false,
+                    "error": err.to_string(),
+                    "logs": []
+                }))?;
+                Ok(CallToolResult::success(vec![content]))
+            }
+        }
+    }
+
+    #[tool(description = "执行TypeScript代码并返回结果")]
+    async fn run_typescript(
+        &self,
+        request: Parameters<CodeRunRequest>,
+    ) -> Result<CallToolResult, McpError> {
+        let request = request.0;
+        match CodeExecutor::execute_with_params_compat(
+            &request.code,
+            LanguageScript::Ts,
+            request.params,
+        )
+        .await
+        {
+            Ok(result) => {
+                if result.success {
+                    let content = Content::json(json!({
+                        "result": result.result,
+                        "logs": result.logs,
+                        "success": true
+                    }))?;
+                    Ok(CallToolResult::success(vec![content]))
+                } else {
+                    let content = Content::json(json!({
+                        "success": false,
+                        "error": result.error,
+                        "logs": result.logs
+                    }))?;
+                    Ok(CallToolResult::success(vec![content]))
+                }
+            }
+            Err(err) => {
+                let content = Content::json(json!({
+                    "success": false,
+                    "error": err.to_string(),
+                    "logs": []
+                }))?;
+                Ok(CallToolResult::success(vec![content]))
+            }
+        }
+    }
+
+    #[rmcp::tool(description = "执行Python代码并返回结果")]
+    async fn run_python(
+        &self,
+        request: Parameters<CodeRunRequest>,
+    ) -> Result<CallToolResult, McpError> {
+        let request = request.0;
+        match CodeExecutor::execute_with_params_compat(
+            &request.code,
+            LanguageScript::Python,
+            request.params,
+        )
+        .await
+        {
+            Ok(result) => {
+                if result.success {
+                    let content = Content::json(json!({
+                        "result": result.result,
+                        "logs": result.logs,
+                        "success": true
+                    }))?;
+                    Ok(CallToolResult::success(vec![content]))
+                } else {
+                    let content = Content::json(json!({
+                        "success": false,
+                        "error": result.error,
+                        "logs": result.logs
+                    }))?;
+                    Ok(CallToolResult::success(vec![content]))
+                }
+            }
+            Err(err) => {
+                let content = Content::json(json!({
+                    "success": false,
+                    "error": err.to_string(),
+                    "logs": []
+                }))?;
+                Ok(CallToolResult::success(vec![content]))
+            }
+        }
+    }
+}
+
+impl ServerHandler for CodeRunnerService {
+    fn get_info(&self) -> ServerInfo {
+        ServerInfo {
+            protocol_version: ProtocolVersion::V_2024_11_05,
+            capabilities: ServerCapabilities::builder().enable_tools().build(),
+            server_info: Implementation::from_build_env(),
+            instructions: Some("一个支持执行JavaScript、TypeScript和Python代码的服务".to_string()),
+        }
+    }
+}

qiming-run_code_rmcp/src/mcp/mod.rs

@@ -0,0 +1,3 @@
+mod mcp_server;
+
+pub use mcp_server::{CodeRunnerService,CodeRunRequest};