aira-mcp 1.0.0 → 1.0.2

package/index.js

@@ -67,7 +67,7 @@ app.post('/download', async (req, res) => {
     const result = await aria2.call('addUri', [uri], options || {});
     res.json({ success: true, gid: result });
   } catch (error) {
-    console.error('Error adding URI to aria2:', error);
+    process.stderr.write(`Error adding URI to aria2: ${error}\n`);
     res.status(500).json({ error: `Failed to add URI to aria2: ${error.message}` });
   }
 });
@@ -85,7 +85,7 @@ app.post('/download/magnet', async (req, res) => {
     const result = await aria2.call('addUri', [magnetUri], options || {});
     res.json({ success: true, gid: result });
   } catch (error) {
-    console.error('Error adding magnet to aria2:', error);
+    process.stderr.write(`Error adding magnet to aria2: ${error}\n`);
     res.status(500).json({ error: `Failed to add magnet to aria2: ${error.message}` });
   }
 });
@@ -97,7 +97,7 @@ app.get('/download/:gid', async (req, res) => {
     const result = await aria2.call('tellStatus', gid);
     res.json({ success: true, status: result });
   } catch (error) {
-    console.error('Error getting download status from aria2:', error);
+    process.stderr.write(`Error getting download status from aria2: ${error}\n`);
     res.status(500).json({ error: `Failed to get download status from aria2: ${error.message}` });
   }
 });
@@ -109,7 +109,7 @@ app.post('/download/:gid/pause', async (req, res) => {
     await aria2.call('pause', gid);
     res.json({ success: true, message: 'Download paused' });
   } catch (error) {
-    console.error('Error pausing download in aria2:', error);
+    process.stderr.write(`Error pausing download in aria2: ${error}\n`);
     res.status(500).json({ error: `Failed to pause download in aria2: ${error.message}` });
   }
 });
@@ -121,7 +121,7 @@ app.post('/download/:gid/resume', async (req, res) => {
     await aria2.call('unpause', gid);
     res.json({ success: true, message: 'Download resumed' });
   } catch (error) {
-    console.error('Error resuming download in aria2:', error);
+    process.stderr.write(`Error resuming download in aria2: ${error}\n`);
     res.status(500).json({ error: `Failed to resume download in aria2: ${error.message}` });
   }
 });
@@ -133,26 +133,97 @@ app.delete('/download/:gid', async (req, res) => {
     await aria2.call('remove', gid);
     res.json({ success: true, message: 'Download removed' });
   } catch (error) {
-    console.error('Error removing download from aria2:', error);
+    process.stderr.write(`Error removing download from aria2: ${error}\n`);
     res.status(500).json({ error: `Failed to remove download from aria2: ${error.message}` });
   }
 });
 
-// 启动服务器
-  app.listen(port, () => {
-    console.log(`MCP Service running at http://localhost:${port}`);
-    console.log(`Configured to connect to aria2 at ${aria2Secure ? 'https' : 'http'}://${aria2Host}:${aria2Port}${aria2Path}`);
-    
-    // 稍后尝试连接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();
-        console.log('Successfully connected to aria2 server');
-        await aria2.close(); // 关闭初始连接
+        const request = JSON.parse(line);
+        const response = handleMCPRequest(request);
+        process.stdout.write(JSON.stringify(response) + '\n');
       } catch (error) {
-        console.warn(`Warning: Could not connect to aria2 server at ${aria2Secure ? 'https' : 'http'}://${aria2Host}:${aria2Port}${aria2Path}:`, error.message);
-        console.log('MCP Service will still run, but aria2 operations will fail until a valid aria2 server is available.');
+        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.0",
+  "version": "1.0.2",
   "main": "index.js",
   "bin": {
     "aira-mcp": "index.js"