lds
/
ai-demo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

git_mcp_server

This commit is contained in:
lids 2026-05-08 14:47:03 +08:00
parent 245aabb862
commit 61d2974c78
7 changed files with 328 additions and 186 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
git_mcp_server/base_mcp_server.py
View File

@ -19,6 +19,7 @@ MCP Server 基类
await server.run_stdio()
"""
import asyncio
import logging
from typing import Any, Dict, List, Optional
from mcp.server import Server
@ -31,6 +32,8 @@ from mcp.types import (
ServerCapabilities,
)
logger = logging.getLogger(__name__)
class BaseMcpServer:
"""
@ -61,7 +64,8 @@ class BaseMcpServer:
async def _handle_list_tools(self) -> list[Tool]:
"""列出所有可用的工具"""
return [
try:
tools = [
Tool(
name=tool.name,
description=tool.description,
@ -69,6 +73,11 @@ class BaseMcpServer:
)
for tool in self.tools
]
logger.debug(f"返回工具列表: {[t.name for t in tools]}")
return tools
except Exception as e:
logger.error(f"列出工具列表时发生异常: {str(e)}")
return []
async def _handle_call_tool(
self, name: str, arguments: Optional[dict[str, Any]] = None
@ -77,22 +86,33 @@ class BaseMcpServer:
if not arguments:
arguments = {}
logger.info(f"调用工具: {name}, 参数: {arguments}")
# 查找工具
tool = next((t for t in self.tools if t.name == name), None)
if not tool:
return [TextContent(type="text", text=f"未知工具：{name}")]
error_msg = f"未知工具：{name}"
logger.error(error_msg)
return [TextContent(type="text", text=f"❌ {error_msg}")]
try:
result = tool.execute(**arguments)
response_text = self._format_tool_result(result)
logger.info(f"工具 {name} 执行成功: {response_text[:100]}...")
return [TextContent(type="text", text=response_text)]
except ValueError as e:
return [TextContent(type="text", text=f"❌ 参数错误：{str(e)}")]
error_msg = f"参数错误：{str(e)}"
logger.error(f"工具 {name} 参数错误: {str(e)}")
return [TextContent(type="text", text=f"❌ {error_msg}")]
except RuntimeError as e:
return [TextContent(type="text", text=f"❌ 执行失败：{str(e)}")]
error_msg = f"执行失败：{str(e)}"
logger.error(f"工具 {name} 执行失败: {str(e)}")
return [TextContent(type="text", text=f"❌ {error_msg}")]
except Exception as e:
return [TextContent(type="text", text=f"❌ 未知错误：{str(e)}")]
error_msg = f"未知错误：{str(e)}"
logger.error(f"工具 {name} 未知错误: {str(e)}", exc_info=True)
return [TextContent(type="text", text=f"❌ {error_msg}")]
@staticmethod
def _format_tool_result(result: Any) -> str:
@ -138,7 +158,12 @@ class BaseMcpServer:
"""
from mcp.server.stdio import stdio_server
logger.info(f"启动 MCP 服务器: {self.server_name} v{self.version}")
logger.info(f"可用工具: {[t.name for t in self.tools]}")
try:
async with stdio_server() as (read_stream, write_stream):
logger.info("stdio_server 连接已建立")
await self.server.run(
read_stream,
write_stream,
@ -150,3 +175,8 @@ class BaseMcpServer:
),
),
)
except Exception as e:
logger.error(f"MCP 服务器运行异常: {str(e)}", exc_info=True)
raise
finally:
logger.info("MCP 服务器已关闭")

23
git_mcp_server/git_add_tool.py
View File

@ -21,6 +21,11 @@ class GitAddTool(BaseTool):
"file_path": {
"type": "string",
"description": "要添加的文件或目录路径"
},
"timeout": {
"type": "integer",
"description": "超时时间（秒）",
"default": 30
}
},
"required": ["repo_path", "file_path"]
@ -29,6 +34,7 @@ class GitAddTool(BaseTool):
def execute(self,
repo_path: str,
file_path: str,
timeout: int = 30,
**kwargs) -> ToolResult:
"""
添加文件到暂存区
@ -36,6 +42,7 @@ class GitAddTool(BaseTool):
Args:
repo_path: 仓库路径
file_path: 文件或目录路径
timeout: 超时时间（秒）
Returns:
ToolResult: 包含添加结果的工具返回对象
@ -51,13 +58,15 @@ class GitAddTool(BaseTool):
logger.error(error_msg)
raise ValueError(error_msg)
try:
# 执行 git add
add_cmd = ["git", "add", file_path]
result = subprocess.run(
add_cmd,
capture_output=True,
encoding='utf-8',
cwd=repo_path
cwd=repo_path,
timeout=timeout
)
if result.returncode != 0:
@ -75,3 +84,15 @@ class GitAddTool(BaseTool):
},
message=f"文件已添加到暂存区: {file_path}"
)
except subprocess.TimeoutExpired:
error_msg = f"git add 超时（{timeout}秒）"
logger.error(error_msg)
raise RuntimeError(error_msg)
except FileNotFoundError:
error_msg = "未找到 git 命令，请确认已安装 git"
logger.error(error_msg)
raise RuntimeError(error_msg)
except Exception as e:
error_msg = f"git add 执行异常: {str(e)}"
logger.error(error_msg)
raise RuntimeError(error_msg)

23
git_mcp_server/git_checkout_tool.py
View File

@ -27,6 +27,11 @@ class GitCheckoutTool(BaseTool):
"type": "boolean",
"description": "是否创建新分支",
"default": True
},
"timeout": {
"type": "integer",
"description": "超时时间（秒）",
"default": 30
}
},
"required": ["repo_path", "branch_name"]
@ -36,6 +41,7 @@ class GitCheckoutTool(BaseTool):
repo_path: str,
branch_name: str,
create_new: bool = True,
timeout: int = 30,
**kwargs) -> ToolResult:
"""
切换或创建分支
@ -44,6 +50,7 @@ class GitCheckoutTool(BaseTool):
repo_path: 仓库路径
branch_name: 分支名称
create_new: 是否创建新分支
timeout: 超时时间（秒）
Returns:
ToolResult: 包含切换结果的工具返回对象
@ -59,6 +66,7 @@ class GitCheckoutTool(BaseTool):
logger.error(error_msg)
raise ValueError(error_msg)
try:
# 构建命令
if create_new:
checkout_cmd = ["git", "checkout", "-b", branch_name]
@ -69,7 +77,8 @@ class GitCheckoutTool(BaseTool):
checkout_cmd,
capture_output=True,
encoding='utf-8',
cwd=repo_path
cwd=repo_path,
timeout=timeout
)
if result.returncode != 0:
@ -88,3 +97,15 @@ class GitCheckoutTool(BaseTool):
},
message=f"已切换到分支: {branch_name}"
)
except subprocess.TimeoutExpired:
error_msg = f"git checkout 超时（{timeout}秒）"
logger.error(error_msg)
raise RuntimeError(error_msg)
except FileNotFoundError:
error_msg = "未找到 git 命令，请确认已安装 git"
logger.error(error_msg)
raise RuntimeError(error_msg)
except Exception as e:
error_msg = f"git checkout 执行异常: {str(e)}"
logger.error(error_msg)
raise RuntimeError(error_msg)

19
git_mcp_server/git_clone_tool.py
View File

@ -63,9 +63,9 @@ class GitCloneTool(BaseTool):
branch: str,
clone_dir: str,
auth_type: str = "password",
username: str = "lds",
password: str = "jrfd123456",
timeout: int = 30,
username: str = "",
password: str = "",
timeout: int = 120,
depth: int = 1,
**kwargs) -> ToolResult:
"""
@ -100,6 +100,7 @@ class GitCloneTool(BaseTool):
logger.error(error_msg)
raise ValueError(error_msg)
try:
# 如果目录已存在，先删除
if os.path.exists(clone_dir):
logger.info(f"清理已存在的目录: {clone_dir}")
@ -139,6 +140,18 @@ class GitCloneTool(BaseTool):
},
message=f"仓库克隆成功: {clone_dir}"
)
except subprocess.TimeoutExpired:
error_msg = f"Git 克隆超时（{timeout}秒），请检查网络连接或增加超时时间"
logger.error(error_msg)
raise RuntimeError(error_msg)
except FileNotFoundError:
error_msg = "未找到 git 命令，请确认已安装 git"
logger.error(error_msg)
raise RuntimeError(error_msg)
except Exception as e:
error_msg = f"Git 克隆执行异常: {str(e)}"
logger.error(error_msg)
raise RuntimeError(error_msg)
def _remove_directory(self, dir_path: str):
"""删除目录（处理 Windows 下只读文件）"""

23
git_mcp_server/git_commit_tool.py
View File

@ -23,6 +23,11 @@ class GitCommitTool(BaseTool):
"type": "string",
"description": "提交信息",
"default": "Auto commit"
},
"timeout": {
"type": "integer",
"description": "超时时间（秒）",
"default": 30
}
},
"required": ["repo_path"]
@ -31,6 +36,7 @@ class GitCommitTool(BaseTool):
def execute(self,
repo_path: str,
message: str = "Auto commit",
timeout: int = 30,
**kwargs) -> ToolResult:
"""
提交更改
@ -38,6 +44,7 @@ class GitCommitTool(BaseTool):
Args:
repo_path: 仓库路径
message: 提交信息
timeout: 超时时间（秒）
Returns:
ToolResult: 包含提交结果的工具返回对象
@ -53,13 +60,15 @@ class GitCommitTool(BaseTool):
logger.error(error_msg)
raise ValueError(error_msg)
try:
# 执行 git commit
commit_cmd = ["git", "commit", "-m", message]
result = subprocess.run(
commit_cmd,
capture_output=True,
encoding='utf-8',
cwd=repo_path
cwd=repo_path,
timeout=timeout
)
if result.returncode != 0:
@ -88,3 +97,15 @@ class GitCommitTool(BaseTool):
},
message=f"代码已提交: {message}"
)
except subprocess.TimeoutExpired:
error_msg = f"git commit 超时（{timeout}秒）"
logger.error(error_msg)
raise RuntimeError(error_msg)
except FileNotFoundError:
error_msg = "未找到 git 命令，请确认已安装 git"
logger.error(error_msg)
raise RuntimeError(error_msg)
except Exception as e:
error_msg = f"git commit 执行异常: {str(e)}"
logger.error(error_msg)
raise RuntimeError(error_msg)

13
git_mcp_server/git_mcp_server.py
View File

@ -8,6 +8,7 @@ Git MCP 服务器
- 工具定义在各自的文件中，git_server 负责组装
"""
import asyncio
import logging
from base_mcp_server import BaseMcpServer
from git_clone_tool import GitCloneTool
from git_checkout_tool import GitCheckoutTool
@ -15,6 +16,17 @@ from git_add_tool import GitAddTool
from git_commit_tool import GitCommitTool
from git_push_tool import GitPushTool
# 配置日志
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
handlers=[
logging.StreamHandler() # 输出到 stderr
]
)
logger = logging.getLogger(__name__)
class GitMcpServer(BaseMcpServer):
"""
@ -39,6 +51,7 @@ class GitMcpServer(BaseMcpServer):
async def main():
"""启动 Git MCP 服务器"""
logger.info("正在启动 Git MCP 服务器...")
server = GitMcpServer()
await server.run_stdio()

29
git_mcp_server/git_push_tool.py
View File

@ -43,6 +43,11 @@ class GitPushTool(BaseTool):
"type": "string",
"description": "Git 密码或 Token",
"default": ""
},
"timeout": {
"type": "integer",
"description": "超时时间（秒）",
"default": 60
}
},
"required": ["repo_path", "branch_name"]
@ -55,6 +60,7 @@ class GitPushTool(BaseTool):
auth_type: str = "password",
username: str = "",
password: str = "",
timeout: int = 60,
**kwargs) -> ToolResult:
"""
推送到远程仓库
@ -66,6 +72,7 @@ class GitPushTool(BaseTool):
auth_type: 认证类型（password/token/ssh）
username: 用户名
password: 密码/Token
timeout: 超时时间（秒）
Returns:
ToolResult: 包含推送结果的工具返回对象
@ -81,6 +88,7 @@ class GitPushTool(BaseTool):
logger.error(error_msg)
raise ValueError(error_msg)
try:
# 如果提供了认证信息，需要更新远程URL
if auth_type.lower() in ["password", "token"] and username and password:
# 获取当前远程URL
@ -89,7 +97,8 @@ class GitPushTool(BaseTool):
get_url_cmd,
capture_output=True,
encoding='utf-8',
cwd=repo_path
cwd=repo_path,
timeout=10
)
if result.returncode == 0:
@ -104,7 +113,8 @@ class GitPushTool(BaseTool):
set_url_cmd,
capture_output=True,
encoding='utf-8',
cwd=repo_path
cwd=repo_path,
timeout=10
)
logger.info("已更新远程URL，添加认证信息")
@ -119,7 +129,8 @@ class GitPushTool(BaseTool):
push_cmd,
capture_output=True,
encoding='utf-8',
cwd=repo_path
cwd=repo_path,
timeout=timeout
)
if result.returncode != 0:
@ -138,3 +149,15 @@ class GitPushTool(BaseTool):
},
message=f"已推送到分支: {branch_name}"
)
except subprocess.TimeoutExpired:
error_msg = f"git push 超时（{timeout}秒）"
logger.error(error_msg)
raise RuntimeError(error_msg)
except FileNotFoundError:
error_msg = "未找到 git 命令，请确认已安装 git"
logger.error(error_msg)
raise RuntimeError(error_msg)
except Exception as e:
error_msg = f"git push 执行异常: {str(e)}"
logger.error(error_msg)
raise RuntimeError(error_msg)