aira-mcp 1.0.2 → 1.0.4

package/index.js

@@ -138,22 +138,271 @@ app.delete('/download/:gid', async (req, res) => {
   }
 });
 
+// 定义服务提供的工具列表
+const tools = [
+  {
+    name: 'aria2-download',
+    description: '使用aria2下载文件',
+    parameters: {
+      type: 'object',
+      properties: {
+        uri: {
+          type: 'string',
+          description: '要下载的文件URI'
+        },
+        options: {
+          type: 'object',
+          description: 'aria2下载选项',
+          additionalProperties: true
+        }
+      },
+      required: ['uri']
+    },
+    returnType: {
+      type: 'object',
+      properties: {
+        success: {
+          type: 'boolean'
+        },
+        gid: {
+          type: 'string',
+          description: 'aria2下载任务ID'
+        }
+      }
+    }
+  },
+  {
+    name: 'aria2-download-magnet',
+    description: '使用aria2下载magnet链接',
+    parameters: {
+      type: 'object',
+      properties: {
+        hashkey: {
+          type: 'string',
+          description: 'Magnet链接的hashkey'
+        },
+        options: {
+          type: 'object',
+          description: 'aria2下载选项',
+          additionalProperties: true
+        }
+      },
+      required: ['hashkey']
+    },
+    returnType: {
+      type: 'object',
+      properties: {
+        success: {
+          type: 'boolean'
+        },
+        gid: {
+          type: 'string',
+          description: 'aria2下载任务ID'
+        }
+      }
+    }
+  },
+  {
+    name: 'aria2-get-status',
+    description: '获取aria2下载任务状态',
+    parameters: {
+      type: 'object',
+      properties: {
+        gid: {
+          type: 'string',
+          description: 'aria2下载任务ID'
+        }
+      },
+      required: ['gid']
+    },
+    returnType: {
+      type: 'object',
+      properties: {
+        success: {
+          type: 'boolean'
+        },
+        status: {
+          type: 'object',
+          description: '下载任务状态信息'
+        }
+      }
+    }
+  },
+  {
+    name: 'aria2-pause',
+    description: '暂停aria2下载任务',
+    parameters: {
+      type: 'object',
+      properties: {
+        gid: {
+          type: 'string',
+          description: 'aria2下载任务ID'
+        }
+      },
+      required: ['gid']
+    },
+    returnType: {
+      type: 'object',
+      properties: {
+        success: {
+          type: 'boolean'
+        },
+        message: {
+          type: 'string'
+        }
+      }
+    }
+  },
+  {
+    name: 'aria2-resume',
+    description: '恢复aria2下载任务',
+    parameters: {
+      type: 'object',
+      properties: {
+        gid: {
+          type: 'string',
+          description: 'aria2下载任务ID'
+        }
+      },
+      required: ['gid']
+    },
+    returnType: {
+      type: 'object',
+      properties: {
+        success: {
+          type: 'boolean'
+        },
+        message: {
+          type: 'string'
+        }
+      }
+    }
+  },
+  {
+    name: 'aria2-remove',
+    description: '删除aria2下载任务',
+    parameters: {
+      type: 'object',
+      properties: {
+        gid: {
+          type: 'string',
+          description: 'aria2下载任务ID'
+        }
+      },
+      required: ['gid']
+    },
+    returnType: {
+      type: 'object',
+      properties: {
+        success: {
+          type: 'boolean'
+        },
+        message: {
+          type: 'string'
+        }
+      }
+    }
+  }
+];
+
 // MCP协议处理
 function handleMCPRequest(request) {
   process.stderr.write(`Received MCP request: ${JSON.stringify(request)}\n`);
   
-  // 简单的MCP响应处理
-  const response = {
-    jsonrpc: '2.0',
-    id: request.id,
-    result: {
-      message: 'MCP Service is running',
-      httpPort: port,
-      status: 'ok'
-    }
-  };
-  
-  return response;
+  // 根据请求方法返回不同的响应
+  switch (request.method) {
+    case 'initialize':
+      // 处理initialize请求，返回符合MCP协议规范的响应
+      return {
+        jsonrpc: '2.0',
+        id: request.id,
+        result: {
+          protocolVersion: '2025-06-18',
+          capabilities: {
+            // 描述服务支持的功能
+            httpServer: {
+              port: port,
+              endpoints: {
+                '/health': 'GET',
+                '/download': 'POST',
+                '/download/magnet': 'POST',
+                '/download/{gid}': 'GET',
+                '/download/{gid}/pause': 'POST',
+                '/download/{gid}/resume': 'POST',
+                '/download/{gid}': 'DELETE'
+              }
+            },
+            aria2: {
+              supportedMethods: ['addUri', 'tellStatus', 'pause', 'unpause', 'remove']
+            }
+          },
+          serverInfo: {
+          name: 'Aira MCP Service',
+          version: '1.0.4',
+          description: 'Aria2 MCP (Model Context Protocol) Service CLI',
+          author: '',
+          license: 'ISC'
+        },
+          tools: tools // 添加工具信息
+        }
+      };
+    
+    case 'listTools':
+      // 处理listTools请求，返回工具列表
+      return {
+        jsonrpc: '2.0',
+        id: request.id,
+        result: {
+          tools: tools
+        }
+      };
+    
+    case 'getTool':
+      // 处理getTool请求，返回特定工具的详细信息
+      const toolName = request.params?.name;
+      if (!toolName) {
+        return {
+          jsonrpc: '2.0',
+          id: request.id,
+          error: {
+            code: -32602,
+            message: 'Invalid params: tool name is required'
+          }
+        };
+      }
+      
+      const tool = tools.find(t => t.name === toolName);
+      if (!tool) {
+        return {
+          jsonrpc: '2.0',
+          id: request.id,
+          error: {
+            code: -32602,
+            message: `Tool not found: ${toolName}`
+          }
+        };
+      }
+      
+      return {
+        jsonrpc: '2.0',
+        id: request.id,
+        result: {
+          tool: tool
+        }
+      };
+    
+    default:
+      // 处理其他请求
+      return {
+        jsonrpc: '2.0',
+        id: request.id,
+        result: {
+          message: 'MCP Service is running',
+          httpPort: port,
+          status: 'ok'
+        }
+      };
+  }
 }
 
 // 启动HTTP服务器

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aira-mcp",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "main": "index.js",
   "bin": {
     "aira-mcp": "index.js"