100 lines
4.4 KiB
C++
100 lines
4.4 KiB
C++
#ifndef HUMANUS_TOOL_PUPPETEER_H
|
|
#define HUMANUS_TOOL_PUPPETEER_H
|
|
|
|
#include "base.h"
|
|
|
|
namespace humanus {
|
|
|
|
struct Puppeteer : BaseTool {
|
|
inline static const std::string name_ = "puppeteer";
|
|
inline static const std::string description_ = "A Model Context Protocol server that provides browser automation capabilities using Puppeteer.";
|
|
inline static const json parameters_ = json::parse(R"json({
|
|
"type": "object",
|
|
"properties": {
|
|
"tool": {
|
|
"type": "string",
|
|
"description": "### Tools\n\n- **navigate**\n - Navigate to any URL in the browser\n - Input: `url` (string)\n\n- **screenshot**\n - Capture screenshots of the entire page or specific elements\n - Inputs:\n - `name` (string, required): Name for the screenshot\n - `selector` (string, optional): CSS selector for element to screenshot\n - `width` (number, optional, default: 800): Screenshot width\n - `height` (number, optional, default: 600): Screenshot height\n\n- **click**\n - Click elements on the page\n - Input: `selector` (string): CSS selector for element to click\n\n- **hover**\n - Hover elements on the page\n - Input: `selector` (string): CSS selector for element to hover\n\n- **fill**\n - Fill out input fields\n - Inputs:\n - `selector` (string): CSS selector for input field\n - `value` (string): Value to fill\n\n- **select**\n - Select an element with SELECT tag\n - Inputs:\n - `selector` (string): CSS selector for element to select\n - `value` (string): Value to select\n\n- **evaluate**\n - Execute JavaScript in the browser console\n - Input: `script` (string): JavaScript code to execute",
|
|
"enum": [
|
|
"navigate",
|
|
"screenshot",
|
|
"click",
|
|
"hover",
|
|
"fill",
|
|
"select",
|
|
"evaluate"
|
|
]
|
|
},
|
|
"url": {
|
|
"type": "string",
|
|
"description": "The URL to navigate to. Required by `navigate`."
|
|
},
|
|
"name": {
|
|
"type": "string",
|
|
"description": "The name of the screenshot. Required by `screenshot`."
|
|
},
|
|
"selector": {
|
|
"type": "string",
|
|
"description": "The CSS selector for the element to interact with. Required by `click`, `hover`, `fill`, and `select`."
|
|
},
|
|
"width": {
|
|
"type": "number",
|
|
"description": "The width of the screenshot. Required by `screenshot`. Default: 800",
|
|
"default": 800
|
|
},
|
|
"height": {
|
|
"type": "number",
|
|
"description": "The height of the screenshot. Required by `screenshot`. Default: 600",
|
|
"default": 600
|
|
},
|
|
"value": {
|
|
"type": "string",
|
|
"description": "The value to fill in input fields. Required by `fill`."
|
|
},
|
|
"script": {
|
|
"type": "string",
|
|
"description": "The JavaScript code to execute. Required by `evaluate`."
|
|
}
|
|
},
|
|
"required": ["tool"]
|
|
})json");
|
|
|
|
Puppeteer() : BaseTool(name_, description_, parameters_) {}
|
|
|
|
ToolResult execute(const json& args) override {
|
|
try {
|
|
if (!_client) {
|
|
return ToolError("Failed to initialize puppeteer client");
|
|
}
|
|
|
|
// 处理命令参数
|
|
std::string tool;
|
|
if (args.contains("tool")) {
|
|
if (args["tool"].is_string()) {
|
|
tool = args["tool"].get<std::string>();
|
|
} else {
|
|
return ToolError("Invalid tool format");
|
|
}
|
|
} else {
|
|
return ToolError("Tool is required");
|
|
}
|
|
|
|
json result = _client->call_tool("puppeteer_" + tool, args);
|
|
|
|
bool is_error = result.value("isError", false);
|
|
|
|
// 根据是否有错误返回不同的ToolResult
|
|
if (is_error) {
|
|
return ToolError(result.value("content", json::array()));
|
|
} else {
|
|
return ToolResult(result.value("content", json::array()));
|
|
}
|
|
} catch (const std::exception& e) {
|
|
return ToolError(std::string(e.what()));
|
|
}
|
|
}
|
|
};
|
|
|
|
};
|
|
|
|
#endif // HUMANUS_TOOL_PUPPETEER_H
|