aira-mcp 1.0.1 → 1.0.3

package/index.js

@@ -138,22 +138,129 @@ app.delete('/download/:gid', async (req, res) => {
   }
 });
 
-// 启动服务器
-  app.listen(port, () => {
-    // 使用stderr输出日志信息，避免干扰MCP的JSON输出
-    process.stderr.write(`MCP Service running at http://localhost:${port}\n`);
-    process.stderr.write(`Configured to connect to aria2 at ${aria2Secure ? 'https' : 'http'}://${aria2Host}:${aria2Port}${aria2Path}\n`);
-    
-    // 稍后尝试连接aria2服务器
-    setTimeout(async () => {
+// 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'
+            }
+          },
+          aria2: {
+            supportedMethods: ['addUri', 'tellStatus', 'pause', 'unpause', 'remove']
+          }
+        },
+        serverInfo: {
+          name: 'Aira MCP Service',
+          version: '1.0.3',
+          description: 'Aria2 MCP (Model Context Protocol) Service CLI',
+          author: '',
+          license: 'ISC'
+        }
+      }
+    };
+    return response;
+  } else {
+    // 处理其他请求
+    const response = {
+      jsonrpc: '2.0',
+      id: request.id,
+      result: {
+        message: 'MCP Service is running',
+        httpPort: port,
+        status: 'ok'
+      }
+    };
+    return response;
+  }
+}
+
+// 启动HTTP服务器
+app.listen(port, () => {
+  process.stderr.write(`MCP Service running at http://localhost:${port}\n`);
+  process.stderr.write(`Configured to connect to aria2 at ${aria2Secure ? 'https' : 'http'}://${aria2Host}:${aria2Port}${aria2Path}\n`);
+  
+  // 稍后尝试连接aria2服务器
+  setTimeout(async () => {
+    try {
+      // 尝试连接aria2服务器
+      await aria2.open();
+      process.stderr.write('Successfully connected to aria2 server\n');
+      await aria2.close(); // 关闭初始连接
+    } catch (error) {
+      process.stderr.write(`Warning: Could not connect to aria2 server at ${aria2Secure ? 'https' : 'http'}://${aria2Host}:${aria2Port}${aria2Path}: ${error.message}\n`);
+      process.stderr.write('MCP Service will still run, but aria2 operations will fail until a valid aria2 server is available.\n');
+    }
+  }, 1000); // 延迟1秒后尝试连接
+});
+
+// 处理MCP通信：读取stdin，写入stdout
+let buffer = '';
+
+process.stdin.on('data', (data) => {
+  buffer += data.toString();
+  
+  // 简单的JSON-RPC解析，假设每个请求是完整的一行
+  const lines = buffer.split('\n');
+  buffer = lines.pop(); // 保存不完整的行
+  
+  for (const line of lines) {
+    if (line.trim()) {
       try {
-        // 尝试连接aria2服务器
-        await aria2.open();
-        process.stderr.write('Successfully connected to aria2 server\n');
-        await aria2.close(); // 关闭初始连接
+        const request = JSON.parse(line);
+        const response = handleMCPRequest(request);
+        process.stdout.write(JSON.stringify(response) + '\n');
       } catch (error) {
-        process.stderr.write(`Warning: Could not connect to aria2 server at ${aria2Secure ? 'https' : 'http'}://${aria2Host}:${aria2Port}${aria2Path}: ${error.message}\n`);
-        process.stderr.write('MCP Service will still run, but aria2 operations will fail until a valid aria2 server is available.\n');
+        process.stderr.write(`Error handling MCP request: ${error}\n`);
       }
-    }, 1000); // 延迟1秒后尝试连接
-  });
+    }
+  }
+});
+
+// 确保进程不会退出
+process.stdin.resume();
+
+// 处理进程结束信号
+process.on('SIGINT', () => {
+  process.stderr.write('Received SIGINT, exiting...\n');
+  process.exit(0);
+});
+
+process.on('SIGTERM', () => {
+  process.stderr.write('Received SIGTERM, exiting...\n');
+  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);

package/package.json

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