Modular MCP Server

模块化 MCP 服务

这是一个使用 github.com/metoro-io/mcp-golang 库实现的 Model Context Protocol（MCP）服务的 Go 版本，并经过模块化重构。

项目结构

mcp-server/
├── cmd/
│   └── mcp-server/
│       └── main.go           # Entry point
├── internal/
│   ├── config/
│   │   └── config.go         # Server configuration
│   ├── server/
│   │   ├── server.go         # MCP server implementation
│   │   └── server_test.go    # Server tests
│   ├── tools/
│   │   ├── tool.go           # Tool interface
│   │   ├── execute.go        # Execute shell command tool
│   │   ├── showfile.go       # Show file tool
│   │   ├── searchfile.go     # Search in file tool
│   │   └── writefile.go      # Write file tool
│   └── utils/
│       └── response.go       # Common response utilities
├── go.mod
├── go.sum
└── README.md

构建和运行

# Build the server
cd cmd/mcp-server
go build -o mcp-server

# Run the server
./mcp-server

安全功能

服务现在包含路径限制系统，可将文件操作限制在指定目录中。

配置允许的路径

你可以通过多种方式配置允许的路径：

1. 使用命令行标志

# Allow operations only in specific directories
./mcp-server --paths=/home/user/safe:/tmp/workspace

# Explicitly deny specific paths even if within allowed paths
./mcp-server --paths=/home/user --deny-paths=/home/user/.ssh:/home/user/credentials

2. 使用环境变量

# Set allowed paths
export MCP_ALLOWED_PATHS=/home/user/safe:/tmp/workspace

# Set denied paths
export MCP_DENIED_PATHS=/home/user/.ssh:/home/user/credentials

# Run the server
./mcp-server

3. 以编程方式配置

你也可以通过编程方式创建自定义配置：

cfg := config.DefaultConfig()
cfg.AddAllowedPath("/path/to/allow")
cfg.AddDeniedPath("/path/to/deny")

server, err := server.NewServerWithConfig(cfg)

默认行为

• 如果未指定路径，服务默认只允许访问当前工作目录。
• .git 和 .env 等常见敏感目录会自动添加到拒绝列表中。

Shell 命令安全性

对于 execute_shell_command 工具：

• 命令仅限使用常见实用程序白名单中的程序
• 会根据允许的路径配置检查自定义可执行文件路径
• 工作目录必须位于允许的路径中

添加新工具

要添加新工具：

1. 在 internal/tools 目录中创建新文件
2. 实现 Tool 接口
3. 如果工具需要访问服务配置，可选实现 ConfigAware 接口
4. 在 cmd/mcp-server/main.go 中注册该工具

示例：

// internal/tools/newtool.go
package tools

import (
    "github.com/metoro-io/mcp-golang"
    "mcp-server/internal/config"
    "mcp-server/internal/utils"
)

type NewToolArgs struct {
    // Tool arguments
}

type NewTool struct{
    config *config.ServerConfig
}

func NewNewTool() *NewTool {
    return &NewTool{}
}

// Implement ConfigAware interface
func (t *NewTool) SetConfig(cfg *config.ServerConfig) {
    t.config = cfg
}

func (t *NewTool) Name() string {
    return "new_tool"
}

func (t *NewTool) Description() string {
    return "Description of the new tool"
}

func (t *NewTool) Execute(args NewToolArgs) (*mcp.ToolResponse, error) {
    // Access configuration if needed
    if t.config != nil {
        // Use configuration for security checks
    }

    // Tool implementation
    return utils.CreateSuccessResponse(result), nil
}

测试

go test ./...