Zig MCP 服务器

一个模型上下文协议 (MCP) 服务器，提供 Zig 语言工具、代码分析和文档访问。该服务器通过 Zig 相关的代码优化、计算单元评估、代码生成和最佳实践建议等功能增强 AI 能力。

功能

工具

1. 代码优化 (optimize_code)

分析和优化 Zig 代码，支持不同的优化级别：

• Debug (调试)
• ReleaseSafe (发布安全)
• ReleaseFast (发布快速)
• ReleaseSmall (发布小尺寸)

// 示例用法
{
  "code": "const std = @import(\"std\");\n...",
  "optimizationLevel": "ReleaseFast"
}

2. 计算单元评估 (estimate_compute_units)

评估 Zig 代码的计算复杂度和资源使用情况：

• 内存使用分析
• 时间复杂度估计
• 内存分配模式检测

// 示例用法
{
  "code": "const std = @import(\"std\");\n..."
}

3. 代码生成 (generate_code)

从自然语言描述生成 Zig 代码，支持：

• 错误处理
• 测试
• 性能优化
• 文档

// 示例用法
{
  "prompt": "创建一个对整数数组进行排序的函数",
  "context": "应处理空数组并在可能的情况下使用 comptime"
}

4. 代码建议 (get_recommendations)

提供代码改进建议和最佳实践：

• 风格和约定
• 设计模式
• 安全考虑
• 性能见解

// 示例用法
{
  "code": "const std = @import(\"std\");\n...",
  "prompt": "提高性能和安全性"
}

资源

1. 语言参考 (zig://docs/language-reference)

   • 官方 Zig 语言文档
   • 语法和特性指南
   • 最佳实践

2. 标准库文档 (zig://docs/std-lib)

   • 完整的 std 库参考
   • 函数签名和用法
   • 示例和注释

3. 热门仓库 (zig://repos/popular)

   • GitHub 上热门的 Zig 项目
   • 社区示例和模式
   • 真实世界的实现

安装

1. 克隆仓库：

git clone [repository-url]
cd zig-mcp-server

2. 安装依赖：

npm install

3. 构建服务器：

npm run build

4. 配置环境变量：

# 创建一个 GitHub token 以获得更好的 API 速率限制
# https://github.com/settings/tokens
# 所需权限：public_repo
GITHUB_TOKEN=your_token_here

5. 添加到 MCP 设置：

{
  "mcpServers": {
    "zig": {
      "command": "node",
      "args": ["/path/to/zig-mcp-server/build/index.js"],
      "env": {
        "GITHUB_TOKEN": "your_token_here",
        "NODE_OPTIONS": "--experimental-vm-modules"
      },
      "restart": true
    }
  }
}

使用示例

1. 优化代码

const result = await useMcpTool("zig", "optimize_code", {
  code: `
    pub fn fibonacci(n: u64) u64 {
        if (n <= 1) return n;
        return fibonacci(n - 1) + fibonacci(n - 2);
    }
  `,
  optimizationLevel: "ReleaseFast"
});

2. 评估计算单元

const result = await useMcpTool("zig", "estimate_compute_units", {
  code: `
    pub fn bubbleSort(arr: []i32) void {
        var i: usize = 0;
        while (i < arr.len) : (i += 1) {
            var j: usize = 0;
            while (j < arr.len - 1) : (j += 1) {
                if (arr[j] > arr[j + 1]) {
                    const temp = arr[j];
                    arr[j] = arr[j + 1];
                    arr[j + 1] = temp;
                }
            }
        }
    }
  `
});

3. 生成代码

const result = await useMcpTool("zig", "generate_code", {
  prompt: "创建一个线程安全的计数器结构体",
  "context": "应使用原子操作并处理溢出"
});

4. 获取建议

const result = await useMcpTool("zig", "get_recommendations", {
  code: `
    pub fn main() !void {
        var list = std.ArrayList(u8).init(allocator);
        var i: u32 = 0;
        while (true) {
            if (i >= 100) break;
            try list.append(@intCast(u8, i));
            i += 1;
        }
    }
  `,
  prompt: "性能"
});

开发

项目结构

zig-mcp-server/
├── src/
│   └── index.ts    # 主要服务器实现
├── build/          # 编译后的 JavaScript
├── package.json    # 依赖和脚本
└── tsconfig.json   # TypeScript 配置

构建

# 带有 watch 模式的开发构建
npm run watch

# 生产构建
npm run build

测试

npm test

贡献

1. Fork 仓库
2. 创建您的特性分支 (git checkout -b feature/amazing-feature)
3. 提交您的更改 (git commit -m 'Add some amazing feature')
4. 推送到分支 (git push origin feature/amazing-feature)
5. 打开一个 Pull Request

许可证

MIT 许可证 - 有关详细信息，请参阅 LICENSE 文件。