Zig MCP 服务器

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

功能特性

工具集

1. 代码优化 (optimize_code)

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

• 调试模式
• 安全发布模式
• 快速发布模式
• 小型发布模式

// 使用示例
{
  "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)

   • 完整的标准库参考
   • 函数签名和使用说明
   • 示例和注意事项

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

   • GitHub 上最受欢迎的 Zig 项目
   • 社区示例和模式
   • 实际应用实现

安装指南

1. 克隆仓库：

git clone [仓库地址]
cd zig-mcp-server

2. 安装依赖：

npm install

3. 构建服务器：

npm run build

4. 配置环境变量：

# 创建 GitHub 令牌以获得更好的 API 速率限制
# https://github.com/settings/tokens
# 所需范围：public_repo
GITHUB_TOKEN=你的令牌

5. 添加到 MCP 设置：

{
  "mcpServers": {
    "zig": {
      "command": "node",
      "args": ["/path/to/zig-mcp-server/build/index.js"],
      "env": {
        "GITHUB_TOKEN": "你的令牌",
        "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 配置

构建命令

# 开发构建（带监视模式）
npm run watch

# 生产构建
npm run build

测试命令

npm test

贡献指南

1. 叉取仓库
2. 创建你的功能分支 (git checkout -b feature/awesome-feature)
3. 提交你的更改 (git commit -m '添加一些很棒的功能')
4. 推送到分支 (git push origin feature/awesome-feature)
5. 创建 Pull Request

许可证

MIT 许可证 - 详情见 LICENSE 文件。