123 lines
3.5 KiB
C++
123 lines
3.5 KiB
C++
/**
|
||
* @file mcp_server_main.cpp
|
||
* @brief OpenManus MCP服务器实现
|
||
*
|
||
* 这个文件实现了OpenManus的MCP服务器，提供工具调用功能。
|
||
* 当前实现了PythonExecute工具。
|
||
*/
|
||
|
||
#include "mcp/include/mcp_server.h"
|
||
#include "mcp/include/mcp_tool.h"
|
||
#include "mcp/include/mcp_resource.h"
|
||
|
||
#include <iostream>
|
||
#include <fstream>
|
||
#include <string>
|
||
#include <memory>
|
||
#include <filesystem>
|
||
#include <cstdio>
|
||
#include <array>
|
||
#include <stdexcept>
|
||
|
||
// 执行系统命令并返回输出
|
||
std::string exec_command(const std::string& cmd) {
|
||
std::array<char, 128> buffer;
|
||
std::string result;
|
||
|
||
// 打开管道，执行命令
|
||
FILE* pipe = popen(cmd.c_str(), "r");
|
||
if (!pipe) {
|
||
throw std::runtime_error("popen() failed!");
|
||
}
|
||
|
||
// 读取命令输出
|
||
while (fgets(buffer.data(), buffer.size(), pipe) != nullptr) {
|
||
result += buffer.data();
|
||
}
|
||
|
||
// 关闭管道
|
||
int status = pclose(pipe);
|
||
if (status == -1) {
|
||
throw std::runtime_error("pclose() failed!");
|
||
}
|
||
|
||
return result;
|
||
}
|
||
|
||
// PythonExecute工具处理函数
|
||
mcp::json python_execute_handler(const mcp::json& args) {
|
||
if (!args.contains("code")) {
|
||
std::cout << args.dump() << std::endl;
|
||
throw mcp::mcp_exception(mcp::error_code::invalid_params, "缺少'code'参数");
|
||
}
|
||
|
||
std::string code = args["code"];
|
||
|
||
// 创建临时Python文件
|
||
std::string temp_file = "/tmp/openmanus_temp.py";
|
||
|
||
// 写入Python代码到临时文件
|
||
std::ofstream out(temp_file);
|
||
if (!out) {
|
||
throw mcp::mcp_exception(mcp::error_code::internal_error, "无法创建临时Python文件");
|
||
}
|
||
out << code;
|
||
out.close();
|
||
|
||
try {
|
||
// 执行Python代码
|
||
std::string cmd = "python3 " + temp_file + " 2>&1";
|
||
std::string output = exec_command(cmd);
|
||
|
||
// 删除临时文件
|
||
std::filesystem::remove(temp_file);
|
||
|
||
if (output.empty()) {
|
||
output = "Executed successfully but w/o output. Maybe you should print more information.";
|
||
}
|
||
|
||
return {{
|
||
{"type", "text"},
|
||
{"text", output}
|
||
}};
|
||
} catch (const std::exception& e) {
|
||
// 删除临时文件
|
||
std::filesystem::remove(temp_file);
|
||
|
||
throw mcp::mcp_exception(mcp::error_code::internal_error,
|
||
"执行Python代码失败: " + std::string(e.what()));
|
||
}
|
||
}
|
||
|
||
int main() {
|
||
// 创建并配置服务器
|
||
mcp::server server("localhost", 8080);
|
||
server.set_server_info("OpenManusMCPServer", "2024-11-05");
|
||
|
||
// 设置服务器能力
|
||
mcp::json capabilities = {
|
||
{"tools", {{"listChanged", true}}},
|
||
{"resources", {{"listChanged", true}}}
|
||
};
|
||
server.set_capabilities(capabilities);
|
||
|
||
// 注册方法处理函数
|
||
server.register_method("ping", [](const mcp::json& params) {
|
||
return mcp::json{{"pong", true}};
|
||
});
|
||
|
||
// 注册PythonExecute工具
|
||
mcp::tool python_tool = mcp::tool_builder("PythonExecute")
|
||
.with_description("执行Python代码并返回结果")
|
||
.with_string_param("code", "要执行的Python代码", true)
|
||
.build();
|
||
|
||
server.register_tool(python_tool, python_execute_handler);
|
||
|
||
// 启动服务器
|
||
std::cout << "启动OpenManus MCP服务器，地址: localhost:8080..." << std::endl;
|
||
std::cout << "按Ctrl+C停止服务器" << std::endl;
|
||
server.start(true); // 阻塞模式
|
||
|
||
return 0;
|
||
}
|