aira-mcp 1.0.1 → 1.0.2

package/index.js

@@ -138,22 +138,92 @@ 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`);
+  
+  // 简单的MCP响应处理
+  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.2",
   "main": "index.js",
   "bin": {
     "aira-mcp": "index.js"