aira-mcp 1.0.0

package/.vscode/launch.json

@@ -0,0 +1,29 @@
+{
+    // 使用 IntelliSense 了解相关属性。 
+    // 悬停以查看现有属性的描述。
+    // 欲了解更多信息，请访问: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "Launch Program",
+            "program": "${workspaceFolder}/index.js",
+            "args": [
+                "--aria2-host=http://localhost:16800"
+            ],
+            "request": "launch",
+            "skipFiles": [
+                "<node_internals>/**"
+            ],
+            "type": "node"
+        },
+        {
+            "type": "node",
+            "request": "launch",
+            "name": "index",
+            "skipFiles": [
+                "<node_internals>/**"
+            ],
+            "program": "node ${workspaceFolder}\\index.js --aria2-host=http://localhost:16800"
+        }
+    ]
+}

package/README.md

@@ -0,0 +1,256 @@
+# Aira MCP Service
+
+使用JavaScript构建的MCP服务，能够通过aria2进行下载，aria服务器地址可以通过npx参数配置传入。
+
+## 安装
+
+```bash
+npm install
+```
+
+## 运行
+
+### 基本运行
+
+```bash
+npm start
+```
+
+### 通过npx参数配置aria服务器地址
+
+```bash
+npx aira-mcp --aria2-host=localhost --aria2-port=6800 --aria2-secret=your-secret
+```
+
+### 支持的命令行参数
+
+| 参数名 | 描述 | 默认值 |
+|-------|------|--------|
+| --aria2-host | aria2服务器地址或完整URL | localhost |
+| --aria2-port | aria2服务器端口 | 6800 |
+| --aria2-secret | aria2服务器密钥 | 无 |
+| --aria2-secure | 是否使用HTTPS | false |
+| --aria2-path | aria2 JSON-RPC路径 | /jsonrpc |
+
+### 示例
+
+使用完整URL配置：
+```bash
+npx aira-mcp --aria2-host=http://localhost:6800/jsonrpc
+```
+
+使用独立参数配置：
+```bash
+npx aira-mcp --aria2-host=192.168.1.100 --aria2-port=6801 --aria2-secret=mysecret --aria2-secure=false
+```
+
+## 接口说明
+
+### 健康检查
+
+```
+GET /health
+```
+
+响应：
+```json
+{
+  "status": "ok",
+  "message": "MCP Service is running"
+}
+```
+
+### 下载文件
+
+```
+POST /download
+```
+
+请求体：
+```json
+{
+  "uri": "http://example.com/file.zip",
+  "options": {
+    "dir": "/downloads",
+    "max-connection-per-server": 5
+  }
+}
+```
+
+响应：
+```json
+{
+  "success": true,
+  "gid": "2089b05ecca3d829",
+}
+```
+
+### Magnet下载
+
+```
+POST /download/magnet
+```
+
+请求体：
+```json
+{
+  "hashkey": "88594AAACBDE40EF3E2510C47374EC0AA396C08E",
+  "options": {
+    "dir": "/downloads"
+  }
+}
+```
+
+响应：
+```json
+{
+  "success": true,
+  "gid": "2089b05ecca3d829",
+}
+```
+
+### 获取下载状态
+
+```
+GET /download/:gid
+```
+
+响应：
+```json
+{
+  "success": true,
+  "status": {
+    "gid": "2089b05ecca3d829",
+    "status": "active",
+    "totalLength": "1000000",
+    "completedLength": "500000",
+    "uploadLength": "0",
+    "bitfield": "eeeeeeeeeeeeeeee",
+    "downloadSpeed": "100000",
+    "uploadSpeed": "0",
+    "infoHash": "",
+    "numSeeders": "0",
+    "seeder": false,
+    "connections": "5",
+    "errorCode": "0",
+    "errorMessage": "",
+    "followedBy": [],
+    "following": "",
+    "belongsTo": "",
+    "dir": "/downloads",
+    "files": [
+      {
+        "index": "1",
+        "path": "/downloads/file.zip",
+        "length": "1000000",
+        "completedLength": "500000",
+        "selected": true,
+        "uris": [
+          {
+            "uri": "http://example.com/file.zip",
+            "status": "used"
+          }
+        ]
+      }
+    ],
+    "bittorrent": null,
+    "verifiedLength": "0",
+    "verifyIntegrityPending": false
+  }
+}
+```
+
+### 暂停下载
+
+```
+POST /download/:gid/pause
+```
+
+响应：
+```json
+{
+  "success": true,
+  "message": "Download paused"
+}
+```
+
+### 恢复下载
+
+```
+POST /download/:gid/resume
+```
+
+响应：
+```json
+{
+  "success": true,
+  "message": "Download resumed"
+}
+```
+
+### 删除下载
+
+```
+DELETE /download/:gid
+```
+
+响应：
+```json
+{
+  "success": true,
+  "message": "Download removed"
+}
+```
+
+## 配置参数
+
+| 参数名 | 描述 | 默认值 |
+|-------|------|--------|
+| --aria2-host | aria2服务器地址 | http://localhost:6800 |
+| --aria2-secret | aria2服务器密钥 | 无 |
+
+## MCP配置方式
+
+在MCP配置文件中，可以按照以下方式配置Aira MCP Service：
+
+```json
+{
+  "mcpServers": {
+    "aira-mcp": {
+      "command": "npx",
+      "args": [
+        "aira-mcp",
+        "--aria2-host=localhost",
+        "--aria2-port=6800",
+        "--aria2-secret=your-secret"
+      ]
+    }
+  }
+}
+```
+
+### 配置说明
+
+- `command`: 使用`npx`命令来执行包
+- `args`: 命令行参数列表
+  - `aira-mcp`: 包名
+  - `--aria2-host`: aria2服务器地址
+  - `--aria2-port`: aria2服务器端口
+  - `--aria2-secret`: aria2服务器密钥（可选）
+
+### 完整配置示例
+
+```json
+{
+  "mcpServers": {
+    "aira-mcp": {
+      "command": "npx",
+      "args": [
+        "aira-mcp",
+        "--aria2-host=http://192.168.1.100:6801/jsonrpc",
+        "--aria2-secret=mysecret"
+      ]
+    }
+  }
+}
+```

package/index.js

@@ -0,0 +1,158 @@
+#!/usr/bin/env node
+const express = require('express');
+const Aria2 = require('aria2');
+const app = express();
+const port = 23121;
+
+// 解析命令行参数获取aria2服务器配置
+const args = process.argv.slice(2);
+let aria2Host = 'localhost';
+let aria2Port = 6800;
+let aria2Secret = '';
+let aria2Secure = false;
+let aria2Path = '/jsonrpc';
+
+for (let i = 0; i < args.length; i++) {
+  if (args[i].startsWith('--aria2-host=')) {
+    const value = args[i].split('=')[1];
+    // 检查是否是完整URL格式
+    if (value.startsWith('http://') || value.startsWith('https://')) {
+      const url = new URL(value);
+      aria2Secure = url.protocol === 'https:';
+      aria2Host = url.hostname;
+      aria2Port = parseInt(url.port) || (aria2Secure ? 443 : 80);
+      // 如果path为空或为"/"，使用默认值
+      aria2Path = (url.pathname && url.pathname !== '/') ? url.pathname : '/jsonrpc';
+    } else {
+      // 单独的host参数
+      aria2Host = value;
+    }
+  } else if (args[i].startsWith('--aria2-port=')) {
+    aria2Port = parseInt(args[i].split('=')[1]) || 6800;
+  } else if (args[i].startsWith('--aria2-secret=')) {
+    aria2Secret = args[i].split('=')[1];
+  } else if (args[i].startsWith('--aria2-secure=')) {
+    aria2Secure = args[i].split('=')[1] === 'true';
+  } else if (args[i].startsWith('--aria2-path=')) {
+    const value = args[i].split('=')[1];
+    aria2Path = value || '/jsonrpc';
+  }
+}
+
+// 创建aria2客户端
+const aria2 = new Aria2({
+  secure: aria2Secure,
+  host: aria2Host,
+  port: aria2Port,
+  secret: aria2Secret,
+  path: aria2Path
+});
+
+// 中间件
+app.use(express.json());
+
+// 健康检查接口
+app.get('/health', (req, res) => {
+  res.json({ status: 'ok', message: 'MCP Service is running' });
+});
+
+// 下载接口
+app.post('/download', async (req, res) => {
+  try {
+    const { uri, options } = req.body;
+    if (!uri) {
+      return res.status(400).json({ error: 'URI is required' });
+    }
+
+    const result = await aria2.call('addUri', [uri], options || {});
+    res.json({ success: true, gid: result });
+  } catch (error) {
+    console.error('Error adding URI to aria2:', error);
+    res.status(500).json({ error: `Failed to add URI to aria2: ${error.message}` });
+  }
+});
+
+// Magnet下载接口
+app.post('/download/magnet', async (req, res) => {
+  try {
+    const { hashkey, options } = req.body;
+    if (!hashkey) {
+      return res.status(400).json({ error: 'Hashkey is required' });
+    }
+
+    // 构建magnet URI
+    const magnetUri = `magnet:?xt=urn:btih:${hashkey}`;
+    const result = await aria2.call('addUri', [magnetUri], options || {});
+    res.json({ success: true, gid: result });
+  } catch (error) {
+    console.error('Error adding magnet to aria2:', error);
+    res.status(500).json({ error: `Failed to add magnet to aria2: ${error.message}` });
+  }
+});
+
+// 获取下载状态接口
+app.get('/download/:gid', async (req, res) => {
+  try {
+    const { gid } = req.params;
+    const result = await aria2.call('tellStatus', gid);
+    res.json({ success: true, status: result });
+  } catch (error) {
+    console.error('Error getting download status from aria2:', error);
+    res.status(500).json({ error: `Failed to get download status from aria2: ${error.message}` });
+  }
+});
+
+// 暂停下载接口
+app.post('/download/:gid/pause', async (req, res) => {
+  try {
+    const { gid } = req.params;
+    await aria2.call('pause', gid);
+    res.json({ success: true, message: 'Download paused' });
+  } catch (error) {
+    console.error('Error pausing download in aria2:', error);
+    res.status(500).json({ error: `Failed to pause download in aria2: ${error.message}` });
+  }
+});
+
+// 恢复下载接口
+app.post('/download/:gid/resume', async (req, res) => {
+  try {
+    const { gid } = req.params;
+    await aria2.call('unpause', gid);
+    res.json({ success: true, message: 'Download resumed' });
+  } catch (error) {
+    console.error('Error resuming download in aria2:', error);
+    res.status(500).json({ error: `Failed to resume download in aria2: ${error.message}` });
+  }
+});
+
+// 删除下载接口
+app.delete('/download/:gid', async (req, res) => {
+  try {
+    const { gid } = req.params;
+    await aria2.call('remove', gid);
+    res.json({ success: true, message: 'Download removed' });
+  } catch (error) {
+    console.error('Error removing download from aria2:', error);
+    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 () => {
+      try {
+        // 尝试连接aria2服务器
+        await aria2.open();
+        console.log('Successfully connected to aria2 server');
+        await aria2.close(); // 关闭初始连接
+      } 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.');
+      }
+    }, 1000); // 延迟1秒后尝试连接
+  });

package/package.json

@@ -0,0 +1,28 @@
+{
+  "name": "aira-mcp",
+  "version": "1.0.0",
+  "main": "index.js",
+  "bin": {
+    "aira-mcp": "index.js"
+  },
+  "scripts": {
+    "start": "node index.js",
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [
+    "aria2",
+    "download",
+    "cli",
+    "mcp"
+  ],
+  "author": "",
+  "license": "ISC",
+  "description": "Aria2 MCP (Media Control Protocol) Service CLI",
+  "dependencies": {
+    "aria2": "4.1.0",
+    "express": "^5.2.1"
+  },
+  "engines": {
+    "node": ">=14.0.0"
+  }
+}