1
import { z } from 'zod';
2
import { Server, ListToolsRequestSchema, CallToolRequestSchema, StdioServerTransport } from 'mcp-sdk';
3
import * as repository from './repository';
4
import * as issues from './issues';
5
import * as search from './search';
6

7
const VERSION = '1.0.0';
8

9
const server = new Server(
10
  { name: "github-mcp-server", version: VERSION },
11
  { capabilities: { tools: {} } }
12
);
13

14
// 定义工具列表，告知 AI Agent 可用的功能
15
server.setRequestHandler(ListToolsRequestSchema, async () => {
16
  return {
17
    tools: [
18
      {
19
        name: "search_repositories",
20
        description: "搜索 GitHub 仓库",
21
        inputSchema: repository.SearchRepositoriesSchema,
22
      },
23
      {
24
        name: "create_issue",
25
        description: "在 GitHub 仓库中创建 Issue",
26
        inputSchema: issues.CreateIssueSchema,
27
      },
28
      {
29
        name: "search_issues",
30
        description: "搜索 GitHub Issue 和 Pull Request",
31
        inputSchema: search.SearchIssuesSchema,
32
      }
33
    ],
34
  };
35
});
36

37
// 处理具体工具调用请求
38
server.setRequestHandler(CallToolRequestSchema, async (request) => {
39
  if (!request.params.arguments) {
40
    throw new Error("缺少必要的参数");
41
  }
42
  switch (request.params.name) {
43
    case "search_repositories": {
44
      const args = repository.SearchRepositoriesSchema.parse(request.params.arguments);
45
      const results = await repository.searchRepositories(args.query, args.page, args.perPage);
46
      return { content: [{ type: "text", text: JSON.stringify(results, null, 2) }] };
47
    }
48
    case "create_issue": {
49
      const args = issues.CreateIssueSchema.parse(request.params.arguments);
50
      const issue = await issues.createIssue(args.owner, args.repo, args.options);
51
      return { content: [{ type: "text", text: JSON.stringify(issue, null, 2) }] };
52
    }
53
    case "search_issues": {
54
      const args = search.SearchIssuesSchema.parse(request.params.arguments);
55
      const results = await search.searchIssues(args);
56
      return { content: [{ type: "text", text: JSON.stringify(results, null, 2) }] };
57
    }
58
    default:
59
      throw new Error(`未知的工具：${request.params.name}`);
60
  }
61
});
62

63
async function runServer() {
64
  const transport = new StdioServerTransport();
65
  await server.connect(transport);
66
  console.error("GitHub MCP Server 已启动，通过 stdio 进行通信");
67
}
68

69
runServer().catch((error) => {
70
  console.error("启动服务器时发生致命错误：", error);
71
  process.exit(1);
72
});