初版gitlab合并test mcp
This commit is contained in:
zhangguoyun committed
1 parent
227d77f586
commit
0ccc8b625b
6 files changed
+2866
-35
No files matched your search
+399
@@ -0,0 +1,399 @@
|
||||
// 引入必要的模块
|
||||
import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
|
||||
import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
|
||||
import { z } from "zod";
|
||||
|
||||
// 创建 MCP 服务器实例
|
||||
// 为其设置一个名字和版本
|
||||
const server = new McpServer({
|
||||
name: "gitlab-mcp-server",
|
||||
version: "1.0.0",
|
||||
});
|
||||
|
||||
/**
|
||||
* GitLab API 客户端配置
|
||||
* 用于连接到GitLab实例并执行操作
|
||||
*/
|
||||
const gitlabApiClient = {
|
||||
baseUrl: process.env.GITLAB_API_BASE_URL || "https://gitlab.com/api/v4",
|
||||
token: process.env.GITLAB_ACCESS_TOKEN,
|
||||
|
||||
async makeRequest(endpoint, options = {}) {
|
||||
if (!this.token) {
|
||||
throw new Error("GITLAB_ACCESS_TOKEN 环境变量未设置");
|
||||
}
|
||||
|
||||
const url = `${this.baseUrl}${endpoint}`;
|
||||
const defaultOptions = {
|
||||
method: "GET",
|
||||
headers: {
|
||||
"PRIVATE-TOKEN": this.token,
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
};
|
||||
|
||||
const requestOptions = { ...defaultOptions, ...options };
|
||||
const response = await fetch(url, requestOptions);
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`GitLab API请求失败: ${response.status} ${response.statusText}`);
|
||||
}
|
||||
|
||||
return await response.json();
|
||||
},
|
||||
|
||||
/**
|
||||
* 创建Merge Request
|
||||
*/
|
||||
async createMergeRequest(projectId, sourceBranch, targetBranch, title, description = "", assigneeId, reviewerIds) {
|
||||
const body = {
|
||||
id: projectId,
|
||||
source_branch: sourceBranch,
|
||||
target_branch: targetBranch,
|
||||
title,
|
||||
description,
|
||||
...(assigneeId !== undefined && { assignee_id: assigneeId }),
|
||||
...(reviewerIds !== undefined && { reviewer_ids: reviewerIds }),
|
||||
};
|
||||
|
||||
return await this.makeRequest(
|
||||
`/projects/${projectId}/merge_requests`,
|
||||
{
|
||||
method: "POST",
|
||||
body: JSON.stringify(body),
|
||||
headers: {
|
||||
"PRIVATE-TOKEN": this.token,
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
}
|
||||
);
|
||||
},
|
||||
|
||||
/**
|
||||
* 合并Merge Request
|
||||
*/
|
||||
async mergeMergeRequest(projectId, mrIid, mergeCommitMessage = "", shouldRemoveSourceBranch = false, mergeWhenPipelineSucceeds = false) {
|
||||
const body = {
|
||||
id: projectId,
|
||||
merge_request_iid: mrIid,
|
||||
...(mergeCommitMessage && { merge_commit_message: mergeCommitMessage }),
|
||||
remove_source_branch: shouldRemoveSourceBranch,
|
||||
merge_when_pipeline_succeeds: mergeWhenPipelineSucceeds,
|
||||
};
|
||||
|
||||
return await this.makeRequest(
|
||||
`/projects/${projectId}/merge_requests/${mrIid}/merge`,
|
||||
{
|
||||
method: "PUT",
|
||||
body: JSON.stringify(body),
|
||||
headers: {
|
||||
"PRIVATE-TOKEN": this.token,
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
}
|
||||
);
|
||||
},
|
||||
|
||||
/**
|
||||
* 获取Merge Request详情
|
||||
*/
|
||||
async getMergeRequest(projectId, mrIid) {
|
||||
return await this.makeRequest(`/projects/${projectId}/merge_requests/${mrIid}`);
|
||||
},
|
||||
|
||||
/**
|
||||
* 关闭Merge Request
|
||||
*/
|
||||
async closeMergeRequest(projectId, mrIid) {
|
||||
return await this.makeRequest(
|
||||
`/projects/${projectId}/merge_requests/${mrIid}`,
|
||||
{
|
||||
method: "PUT",
|
||||
body: JSON.stringify({ state_event: "close" }),
|
||||
headers: {
|
||||
"PRIVATE-TOKEN": this.token,
|
||||
"Content-Type": "application/json",
|
||||
},
|
||||
}
|
||||
);
|
||||
}
|
||||
};
|
||||
|
||||
// /**
|
||||
// * 创建Merge Request工具
|
||||
// */
|
||||
// server.registerTool(
|
||||
// "create_merge_request",
|
||||
// {
|
||||
// description: "在指定的GitLab项目中创建一个新的Merge Request",
|
||||
// inputSchema: z.object({
|
||||
// projectId: z.string().describe("GitLab项目的ID或URL编码的路径"),
|
||||
// sourceBranch: z.string().describe("源分支名称"),
|
||||
// targetBranch: z.string().describe("目标分支名称"),
|
||||
// title: z.string().describe("Merge Request的标题"),
|
||||
// description: z.string().optional().describe("Merge Request的描述"),
|
||||
// assigneeId: z.number().optional().describe("负责人ID"),
|
||||
// reviewerIds: z.array(z.number()).optional().describe("审阅者ID数组"),
|
||||
// }),
|
||||
// },
|
||||
// async ({ projectId, sourceBranch, targetBranch, title, description = "", assigneeId, reviewerIds }) => {
|
||||
// try {
|
||||
// const result = await gitlabApiClient.createMergeRequest(
|
||||
// projectId,
|
||||
// sourceBranch,
|
||||
// targetBranch,
|
||||
// title,
|
||||
// description,
|
||||
// assigneeId,
|
||||
// reviewerIds
|
||||
// );
|
||||
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `成功创建Merge Request: ${result.web_url}\n标题: ${result.title}\n状态: ${result.state}`,
|
||||
// },
|
||||
// ],
|
||||
// };
|
||||
// } catch (error) {
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `创建Merge Request失败: ${error.message}`,
|
||||
// },
|
||||
// ],
|
||||
// isError: true,
|
||||
// };
|
||||
// }
|
||||
// }
|
||||
// );
|
||||
|
||||
// /**
|
||||
// * 合并Merge Request工具
|
||||
// */
|
||||
// server.registerTool(
|
||||
// "merge_merge_request",
|
||||
// {
|
||||
// description: "合并指定的GitLab Merge Request",
|
||||
// inputSchema: z.object({
|
||||
// projectId: z.string().describe("GitLab项目的ID或URL编码的路径"),
|
||||
// mrIid: z.number().describe("Merge Request的内部ID"),
|
||||
// mergeCommitMessage: z.string().optional().describe("合并提交的消息"),
|
||||
// shouldRemoveSourceBranch: z.boolean().optional().describe("是否在合并后删除源分支"),
|
||||
// mergeWhenPipelineSucceeds: z.boolean().optional().describe("是否在CI管道成功后合并"),
|
||||
// }),
|
||||
// },
|
||||
// async ({
|
||||
// projectId,
|
||||
// mrIid,
|
||||
// mergeCommitMessage = "",
|
||||
// shouldRemoveSourceBranch = false,
|
||||
// mergeWhenPipelineSucceeds = false
|
||||
// }) => {
|
||||
// try {
|
||||
// const result = await gitlabApiClient.mergeMergeRequest(
|
||||
// projectId,
|
||||
// mrIid,
|
||||
// mergeCommitMessage,
|
||||
// shouldRemoveSourceBranch,
|
||||
// mergeWhenPipelineSucceeds
|
||||
// );
|
||||
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `成功合并Merge Request #${mrIid}:\n状态: ${result.state}\n已合并时间: ${result.merged_at}`,
|
||||
// },
|
||||
// ],
|
||||
// };
|
||||
// } catch (error) {
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `合并Merge Request失败: ${error.message}`,
|
||||
// },
|
||||
// ],
|
||||
// isError: true,
|
||||
// };
|
||||
// }
|
||||
// }
|
||||
// );
|
||||
|
||||
// /**
|
||||
// * 获取Merge Request详情工具
|
||||
// */
|
||||
// server.registerTool(
|
||||
// "get_merge_request",
|
||||
// {
|
||||
// description: "获取指定Merge Request的详细信息",
|
||||
// inputSchema: z.object({
|
||||
// projectId: z.string().describe("GitLab项目的ID或URL编码的路径"),
|
||||
// mrIid: z.number().describe("Merge Request的内部ID"),
|
||||
// }),
|
||||
// },
|
||||
// async ({ projectId, mrIid }) => {
|
||||
// try {
|
||||
// const result = await gitlabApiClient.getMergeRequest(projectId, mrIid);
|
||||
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `Merge Request #${mrIid} 详情:\n标题: ${result.title}\n作者: ${result.author.name}\n状态: ${result.state}\n源分支: ${result.source_branch}\n目标分支: ${result.target_branch}\n创建时间: ${result.created_at}\n更新时间: ${result.updated_at}`,
|
||||
// },
|
||||
// ],
|
||||
// };
|
||||
// } catch (error) {
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `获取Merge Request详情失败: ${error.message}`,
|
||||
// },
|
||||
// ],
|
||||
// isError: true,
|
||||
// };
|
||||
// }
|
||||
// }
|
||||
// );
|
||||
|
||||
// /**
|
||||
// * 关闭Merge Request工具
|
||||
// */
|
||||
// server.registerTool(
|
||||
// "close_merge_request",
|
||||
// {
|
||||
// description: "关闭指定的GitLab Merge Request而不合并",
|
||||
// inputSchema: z.object({
|
||||
// projectId: z.string().describe("GitLab项目的ID或URL编码的路径"),
|
||||
// mrIid: z.number().describe("Merge Request的内部ID"),
|
||||
// }),
|
||||
// },
|
||||
// async ({ projectId, mrIid }) => {
|
||||
// try {
|
||||
// const result = await gitlabApiClient.closeMergeRequest(projectId, mrIid);
|
||||
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `成功关闭Merge Request #${mrIid}:\n标题: ${result.title}\n当前状态: ${result.state}`,
|
||||
// },
|
||||
// ],
|
||||
// };
|
||||
// } catch (error) {
|
||||
// return {
|
||||
// content: [
|
||||
// {
|
||||
// type: "text",
|
||||
// text: `关闭Merge Request失败: ${error.message}`,
|
||||
// },
|
||||
// ],
|
||||
// isError: true,
|
||||
// };
|
||||
// }
|
||||
// }
|
||||
// );
|
||||
|
||||
|
||||
server.registerTool(
|
||||
'merge_to_test',
|
||||
{
|
||||
description: "把源分支名合并到测试分支，并部署到测试环境，默认用当前分支作为源分支",
|
||||
inputSchema: z.object({
|
||||
projectId: z.string().describe("GitLab项目的ID"),
|
||||
sourceBranch: z.string().optional().describe("源分支名称，默认使用当前分支"),
|
||||
}),
|
||||
},
|
||||
|
||||
async ({ projectId, sourceBranch, }) => {
|
||||
if (!sourceBranch) {
|
||||
// 获取当前分支，用 git branch --show-current 命令
|
||||
const { execSync } = await import('child_process');
|
||||
try {
|
||||
sourceBranch = execSync('git branch --show-current', { encoding: 'utf-8' }).trim();
|
||||
} catch (error) {
|
||||
console.error('执行 git branch --show-current 命令失败:', error);
|
||||
return {
|
||||
content: [
|
||||
{
|
||||
type: "text",
|
||||
text: `获取当前分支失败: ${error.message}`,
|
||||
},
|
||||
],
|
||||
isError: true,
|
||||
};
|
||||
}
|
||||
}
|
||||
try {
|
||||
const result = await gitlabApiClient.createMergeRequest(
|
||||
projectId,
|
||||
sourceBranch,
|
||||
'test',
|
||||
`Merge ${sourceBranch} to test`, // 标题
|
||||
`Automated merge request from ${sourceBranch} to test branch` // 描述
|
||||
);
|
||||
|
||||
const mrIid = result.iid;
|
||||
|
||||
// 等待3秒钟，让GitLab有时间处理新创建的MR
|
||||
await new Promise(resolve => setTimeout(resolve, 3000));
|
||||
|
||||
// 尝试合并MR
|
||||
try {
|
||||
const mergeResult = await gitlabApiClient.mergeMergeRequest(
|
||||
projectId,
|
||||
mrIid
|
||||
);
|
||||
return {
|
||||
content: [
|
||||
{
|
||||
type: "text",
|
||||
text: `成功创建并合并Merge Request: ${mergeResult.web_url}\n标题: ${mergeResult.title}\n状态: ${mergeResult.state}\n合并时间: ${mergeResult.merged_at}`,
|
||||
},
|
||||
],
|
||||
};
|
||||
} catch (mergeError) {
|
||||
// 如果合并失败，返回MR创建成功的信息，但提示合并需要审核
|
||||
return {
|
||||
content: [
|
||||
{
|
||||
type: "text",
|
||||
text: `成功创建Merge Request，但自动合并失败，请手动审核: ${result.web_url}\n标题: ${result.title}\n状态: ${result.state}\n错误信息: ${mergeError.message}`,
|
||||
},
|
||||
],
|
||||
};
|
||||
}
|
||||
} catch (error) {
|
||||
return {
|
||||
content: [
|
||||
{
|
||||
type: "text",
|
||||
text: `合并到test分支失败: ${error.message}`,
|
||||
},
|
||||
],
|
||||
isError: true,
|
||||
};
|
||||
}
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
// 启动服务器，使用标准输入/输出（stdio）与客户端通信
|
||||
async function main() {
|
||||
const transport = new StdioServerTransport();
|
||||
await server.connect(transport);
|
||||
console.error("GitLab MCP 服务器已通过 stdio 启动"); // 日志输出到 stderr
|
||||
}
|
||||
|
||||
// 运行主函数，并捕获可能的错误
|
||||
main().catch((error) => {
|
||||
console.error("服务器运行出现致命错误:", error);
|
||||
process.exit(1);
|
||||
});
|
||||
|
||||
export { server };
|
||||
Reference in new issue
Block a user