feat: add Gitee error handling and MCP server implementation

normal-coder · normal-coder · commit abd040f5ddce · 2025-02-18T21:01:18.000+08:00
Signed-off-by: 诺墨 &lt;normal@normalcoder.com&gt;
diff --git a/common/errors.ts b/common/errors.ts
@@ -0,0 +1,47 @@
+export class GiteeError extends Error {
+  constructor(message: string) {
+    super(message);
+    this.name = "GiteeError";
+  }
+}
+
+export function isGiteeError(error: unknown): error is GiteeError {
+  return error instanceof GiteeError;
+}
+
+export function createGiteeError(status: number, responseBody: unknown): GiteeError {
+  let message = "Gitee API 请求失败";
+  let resetAt: Date | undefined;
+
+  if (typeof responseBody === "object" && responseBody !== null) {
+    const body = responseBody as Record<string, unknown>;
+
+    if (body.message && typeof body.message === "string") {
+      message = body.message;
+    }
+
+    if (body.documentation_url && typeof body.documentation_url === "string") {
+      message += ` - 文档: ${body.documentation_url}`;
+    }
+  }
+
+  switch (status) {
+    case 400:
+      return new GiteeValidationError(message, responseBody);
+    case 401:
+      return new GiteeAuthenticationError(message);
+    case 403:
+      return new GiteePermissionError(message);
+    case 404:
+      return new GiteeResourceNotFoundError(message);
+    case 409:
+      return new GiteeConflictError(message);
+    case 429:
+      return new GiteeRateLimitError(
+        message,
+        resetAt || new Date(Date.now() + 60 * 1000) // 默认 1 分钟后重置
+      );
+    default:
+      return new GiteeError(message);
+  }
+}
diff --git a/common/server.ts b/common/server.ts
@@ -0,0 +1,89 @@
+import { Server } from "@modelcontextprotocol/sdk/server/index.js";
+import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
+import { CallToolRequestSchema, ListToolsRequestSchema } from "@modelcontextprotocol/sdk/types.js";
+import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+import { isGiteeError } from "./errors.js";
+
+type MCPServerOptions = {
+  name: string;
+  version: string;
+};
+
+type ToolDefinition = {
+  name: string;
+  description: string;
+  schema: z.ZodType<any, any, any>;
+  handler: (params: any) => Promise<any>;
+};
+
+export class MCPServer {
+  private server: Server;
+  private tools: Map<string, ToolDefinition> = new Map();
+
+  constructor(options: MCPServerOptions) {
+    this.server = new Server(
+      {
+        name: options.name,
+        version: options.version,
+      },
+      {
+        capabilities: {
+          tools: {},
+        },
+      }
+    );
+
+    this.setupRequestHandlers();
+  }
+
+  private setupRequestHandlers() {
+    this.server.setRequestHandler(ListToolsRequestSchema, async () => {
+      const toolsList = Array.from(this.tools.values()).map((tool) => ({
+        name: tool.name,
+        description: tool.description,
+        inputSchema: zodToJsonSchema(tool.schema),
+      }));
+
+      return {
+        tools: toolsList,
+      };
+    });
+
+    this.server.setRequestHandler(CallToolRequestSchema, async (request) => {
+      try {
+        if (!request.params.arguments) {
+          throw new Error("Parameters are necessary.");
+        }
+
+        const tool = this.tools.get(request.params.name);
+        if (!tool) {
+          throw new Error(`Unknown tool: ${request.params.name}`);
+        }
+
+        const args = tool.schema.parse(request.params.arguments);
+        const result = await tool.handler(args);
+
+        return {
+          content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
+        };
+      } catch (error) {
+        if (error instanceof z.ZodError) {
+          throw new Error(`Invalid input: ${JSON.stringify(error.errors)}`);
+        }
+        if (isGiteeError(error)) {
+          throw error;
+        }
+        throw error;
+      }
+    });
+  }
+
+  public registerTool(tool: ToolDefinition) {
+    this.tools.set(tool.name, tool);
+  }
+
+  public async connect(transport: StdioServerTransport) {
+    await this.server.connect(transport);
+  }
+}
diff --git a/common/version.ts b/common/version.ts
@@ -0,0 +1 @@
+export const VERSION = "0.1.0";
