Zig MCP 服务器

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

功能

工具

1. 代码优化 (optimize_code)

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

• Debug
• ReleaseSafe
• ReleaseFast
• ReleaseSmall

// Example usage
{
  "code": "const std = @import(\"std\");\n...",
  "optimizationLevel": "ReleaseFast"
}

2. 计算单元估计 (estimate_compute_units)

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

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

// Example usage
{
  "code": "const std = @import(\"std\");\n..."
}

3. 代码生成 (generate_code)

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

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

// Example usage
{
  "prompt": "Create a function that sorts an array of integers",
  "context": "Should handle empty arrays and use comptime when possible"
}

4. 代码建议 (get_recommendations)

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

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

// Example usage
{
  "code": "const std = @import(\"std\");\n...",
  "prompt": "Improve performance and safety"
}

资源

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

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

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

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

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

   • GitHub 上的顶级 Zig 项目
   • 社区示例和模式
   • 实际应用实现

安装

1. 克隆仓库：

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

1. 安装依赖项：

npm install

1. 构建服务器：

npm run build

1. 配置环境变量：

# Create a GitHub token for better API rate limits
# https://github.com/settings/tokens
# Required scope: public_repo
GITHUB_TOKEN=your_token_here

1. 添加到 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: "Create a thread-safe counter struct",
  context: "Should use atomic operations and handle overflow"
});

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: "performance"
});

开发

项目结构

zig-mcp-server/
├── src/
│   └── index.ts    # Main server implementation
├── build/          # Compiled JavaScript
├── package.json    # Dependencies and scripts
└── tsconfig.json   # TypeScript configuration

构建

# Development build with watch mode
npm run watch

# Production build
npm run build

测试

npm test

贡献

1. 叉出仓库
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 文件。