399 lines
11 KiB
JavaScript
399 lines
11 KiB
JavaScript
// 引入必要的模块
|
|
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 }; |