auto-api-mcp
Version:
Auto API - YApi接口信息获取工具，支持MCP协议
205 lines (204 loc) • 7.3 kB
JavaScript
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
import { z } from "zod";
const server = new McpServer({
name: "auto-api-mcp",
version: "1.0.0",
description: "Auto API - YApi接口信息获取工具",
});
// YApi接口获取工具
server.tool("yapi_get_interfaces", {
url: z
.string()
.describe("YApi分类页面URL，格式如：https://xxxxx.com/project/810/interface/api/cat_2783，用于获取该分类下的所有接口列表，需要拿到ID通过调用yapi_get_interface_detail获取接口详情。"),
}, async ({ url }) => {
try {
// 参数验证
if (!url || typeof url !== "string") {
return {
content: [
{
type: "text",
text: "❌ 参数错误：缺少有效的URL参数" + url,
},
],
};
}
// 1. 解析URL提取分类ID
const urlPattern = /\/cat_(\d+)$/;
const match = url.match(urlPattern);
if (!match) {
return {
content: [
{
type: "text",
text: "❌ URL格式错误，请提供正确的YApi分类页面URL，格式如：https://xxxxx.com/project/810/interface/api/cat_2783",
},
],
};
}
const catId = match[1];
// 2. 构建API请求URL
const apiUrl = `${process.env.BASE_URL}/api/interface/list_cat?token=${process.env.YAPI_TOKEN}&catid=${catId}&page=1&limit=100`;
// 3. 发起请求
const response = await fetch(apiUrl, {
method: "GET",
});
if (!response.ok) {
return {
content: [
{
type: "text",
text: `❌ 请求失败：HTTP ${response.status} ${response.statusText}`,
},
],
};
}
const data = await response.json();
// 5. 格式化返回结果
if (data.errcode !== 0) {
return {
content: [
{
type: "text",
text: `❌ API返回错误：${data.errmsg || "未知错误"}`,
},
],
};
}
const interfaces = data.data?.list || [];
let result = `📋 分类ID ${catId} 的接口列表 (共${data.data?.total || 0}个):\n\n`;
if (interfaces.length === 0) {
result += "该分类下暂无接口";
}
else {
interfaces.forEach((item, index) => {
result += `${index + 1}. **${item.title}**\n`;
result += ` • ID: \`${item._id}\` (用于获取详情)\n`;
result += ` • 路径: ${item.method.toUpperCase()} ${item.path}\n`;
result += ` • 链接: ${process.env.BASE_URL}/project/810/interface/api/${item._id}\n\n`;
});
}
return {
content: [{ type: "text", text: result }],
};
}
catch (error) {
return {
content: [
{
type: "text",
text: `❌ 处理过程中发生错误：${error instanceof Error ? error.message : "未知错误"}`,
},
],
};
}
});
// YApi接口详情获取工具
server.tool("yapi_get_interface_detail", {
id: z
.string()
.describe("接口ID，来自接口列表中的_id字段，或者来自直接传入的id用于获取该接口的详情。"),
}, async ({ id }) => {
try {
// 参数验证
if (!id || typeof id !== "string") {
return {
content: [
{
type: "text",
text: "❌ 参数错误：缺少有效的ID参数",
},
],
};
}
// 1. 构建详情API请求URL
const apiUrl = `${process.env.BASE_URL}/api/interface/get?token=${process.env.YAPI_TOKEN}&id=${id}`;
// 3. 发起请求
const response = await fetch(apiUrl, {
method: "GET",
});
if (!response.ok) {
return {
content: [
{
type: "text",
text: `❌ 请求失败：HTTP ${response.status} ${response.statusText}`,
},
],
};
}
const data = await response.json();
// 4. 格式化返回结果
if (data.errcode !== 0) {
return {
content: [
{
type: "text",
text: `❌ API返回错误：${data.errmsg || "未知错误"}`,
},
],
};
}
const interfaceData = data.data;
if (!interfaceData) {
return {
content: [
{
type: "text",
text: "❌ 未找到接口详情数据",
},
],
};
}
// 5. 生成详细信息
let result = `📋 **接口详情**: ${interfaceData.title}\n\n`;
// 请求体
if (interfaceData.req_body_other) {
result += `## 📝 请求体 (req_body_other)\n`;
result += `\`\`\`json\n${interfaceData.req_body_other}\n\`\`\`\n\n`;
}
else {
result += `## 📝 请求体 (req_body_other)\n`;
result += `暂无请求体数据\n\n`;
}
// 响应体
if (interfaceData.res_body) {
result += `## 📥 响应体 (res_body)\n`;
result += `\`\`\`json\n${interfaceData.res_body}\n\`\`\`\n\n`;
}
else {
result += `## 📥 响应体 (res_body)\n`;
result += `暂无响应体数据\n\n`;
}
if (interfaceData.method) {
result += `## 📥 请求方法 (method)\n`;
result += `\`\`\`${interfaceData.method}\n\`\`\`\n\n`;
}
else {
result += `## 📥 请求方法 (method)\n`;
result += `暂无请求方法数据\n\n`;
}
// 接口链接
result += `## 🔗 相关链接\n`;
result += `- **在线文档**: ${process.env.BASE_URL}/project/810/interface/api/${interfaceData._id}\n`;
result += `- **Mock地址**: ${process.env.BASE_URL}/mock/810${interfaceData.path}\n`;
return {
content: [{ type: "text", text: result }],
};
}
catch (error) {
return {
content: [
{
type: "text",
text: `❌ 处理过程中发生错误：${error instanceof Error ? error.message : "未知错误"}`,
},
],
};
}
});
// Start receiving messages on stdin and sending messages on stdout
const transport = new StdioServerTransport();
await server.connect(transport);