aira-mcp 1.0.3 → 1.0.5

package/index.js

@@ -138,58 +138,270 @@ 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`);
   
   // 根据请求方法返回不同的响应
-  if (request.method === 'initialize') {
-    // 处理initialize请求，返回符合MCP协议规范的响应
-    const response = {
-      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'
+  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']
             }
           },
-          aria2: {
-            supportedMethods: ['addUri', 'tellStatus', 'pause', 'unpause', 'remove']
-          }
-        },
-        serverInfo: {
+          serverInfo: {
           name: 'Aira MCP Service',
-          version: '1.0.3',
+          version: '1.0.5',
           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'
+          }
+        };
       }
-    };
-    return response;
-  } else {
-    // 处理其他请求
-    const response = {
-      jsonrpc: '2.0',
-      id: request.id,
-      result: {
-        message: 'MCP Service is running',
-        httpPort: port,
-        status: 'ok'
+      
+      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 response;
+      
+      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'
+        }
+      };
   }
 }
 
@@ -249,18 +461,5 @@ process.on('SIGTERM', () => {
   process.exit(0);
 });
 
-// 初始化MCP服务就绪通知
-const initResponse = {
-  jsonrpc: '2.0',
-  id: null,
-  result: {
-    message: 'MCP Service initialized',
-    httpPort: port,
-    status: 'ready'
-  }
-};
-
-// 发送初始化响应
-setTimeout(() => {
-  process.stdout.write(JSON.stringify(initResponse) + '\n');
-}, 500);
+// MCP服务不需要发送自定义的初始化响应
+// 客户端会通过调用initialize方法来获取服务信息

package/package.json

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