mta-mcp 2.9.0 → 2.11.0

package/bin/mta.cjs

@@ -5,7 +5,7 @@
  * 最低要求: Node.js 16+ (由 @modelcontextprotocol/sdk 决定)
  */
 
-const { existsSync } = require('fs');
+const { existsSync, readdirSync, readFileSync } = require('fs');
 const { join } = require('path');
 const { pathToFileURL } = require('url');
 
@@ -23,6 +23,99 @@ if (majorVersion < 16) {
 
 // 获取包根目录
 const packageRoot = join(__dirname, '..');
+
+// 处理 CLI 参数
+const args = process.argv.slice(2);
+
+// --get-agent <id>: 获取指定 Agent 的完整内容
+const getAgentIndex = args.indexOf('--get-agent');
+if (getAgentIndex !== -1) {
+  const agentId = args[getAgentIndex + 1];
+  
+  if (!agentId) {
+    console.log(JSON.stringify({ error: '请提供 Agent ID', content: null }));
+    process.exit(1);
+  }
+  
+  const agentsDir = join(packageRoot, 'agents');
+  const agentFile = join(agentsDir, `${agentId}.agent.md`);
+  
+  if (!existsSync(agentFile)) {
+    console.log(JSON.stringify({ error: `Agent 不存在: ${agentId}`, content: null }));
+    process.exit(1);
+  }
+  
+  try {
+    const content = readFileSync(agentFile, 'utf-8');
+    console.log(JSON.stringify({ content }));
+  } catch (e) {
+    console.log(JSON.stringify({ error: `读取 Agent 失败: ${e.message}`, content: null }));
+    process.exit(1);
+  }
+  
+  process.exit(0);
+}
+
+// --list-agents: 列出所有可用的 Agents
+if (args.includes('--list-agents')) {
+  const agentsDir = join(packageRoot, 'agents');
+  
+  if (!existsSync(agentsDir)) {
+    console.error(JSON.stringify({ error: 'Agents 目录不存在', agents: [] }));
+    process.exit(1);
+  }
+  
+  const agents = [];
+  const files = readdirSync(agentsDir).filter(f => f.endsWith('.agent.md') && !f.endsWith('.bak'));
+  
+  for (const file of files) {
+    try {
+      const content = readFileSync(join(agentsDir, file), 'utf-8');
+      const lines = content.split('\n');
+      
+      // 提取 ID
+      const id = file.replace('.agent.md', '');
+      
+      // 提取标题
+      let title = id;
+      const titleLine = lines.find(l => l.trim().startsWith('# '));
+      if (titleLine) {
+        title = titleLine.replace(/^#\s*/, '').trim();
+      }
+      
+      // 提取描述
+      let description = '暂无描述';
+      const descLine = lines.find(l => l.trim().startsWith('> '));
+      if (descLine) {
+        description = descLine.replace(/^>\s*/, '').trim();
+      }
+      
+      // 提取标签
+      const tags = [];
+      const tagsLine = lines.find(l => 
+        l.includes('**标签**') || 
+        l.includes('**关键词**') ||
+        l.includes('**Tags**')
+      );
+      if (tagsLine) {
+        const tagsMatch = tagsLine.match(/[:：]\s*(.+)/);
+        if (tagsMatch) {
+          tags.push(...tagsMatch[1].split(/[,，、]/).map(t => t.trim()));
+        }
+      }
+      
+      agents.push({ id, title, description, tags });
+    } catch (e) {
+      // 忽略解析失败的文件
+    }
+  }
+  
+  // 输出 JSON
+  console.log(JSON.stringify({ agents }, null, 2));
+  process.exit(0);
+}
+
+// 默认行为：启动 MCP 服务器
 const esmPath = join(packageRoot, 'dist', 'index.js');
 
 if (!existsSync(esmPath)) {