px2cc 1.1.0 → 1.1.1

package/cli.js

@@ -13,45 +13,53 @@ import path from 'path';
 import fs from 'fs';
 import { execSync } from 'child_process';
 
-// 获取所有可用工具
-async function discoverAllTools() {
-  const tools = {
+// 发现MCP服务器
+async function discoverMCPServers() {
+  const servers = {
     defaultTools: ['Read', 'Write', 'Edit', 'Bash'],
-    mcpTools: []
+    mcpServers: []
   };
 
   try {
-    // 使用claude mcp命令获取所有可用工具
-    const mcpOutput = execSync('claude mcp list-tools', { 
+    // 使用claude mcp list获取所有MCP服务器
+    const mcpOutput = execSync('claude mcp list', { 
       encoding: 'utf8',
       timeout: 10000 
     });
     
-    // 解析输出，提取工具名称
+    // 解析输出，提取服务器信息
     const lines = mcpOutput.split('\n');
+    
     for (const line of lines) {
       const trimmedLine = line.trim();
-      // 匹配工具名称格式（通常以mcp__开头）
-      if (trimmedLine.startsWith('mcp__') || trimmedLine.includes('__')) {
-        tools.mcpTools.push(trimmedLine.split(/\s+/)[0]);
+      // 匹配服务器名称（格式：serverName: command - status）
+      const match = trimmedLine.match(/^([^:]+):\s+(.+)\s+-\s+(✓|✗)\s+(.*)$/);
+      if (match && !trimmedLine.includes('Checking MCP server health')) {
+        const [, name, command, status, statusText] = match;
+        servers.mcpServers.push({
+          name: name.trim(),
+          command: command.trim(), 
+          connected: status === '✓',
+          status: statusText.trim()
+        });
       }
     }
     
-    console.log(chalk.green(`✅ 发现 ${tools.mcpTools.length} 个MCP工具`));
+    console.log(chalk.green(`✅ 发现 ${servers.mcpServers.length} 个MCP服务器`));
     
   } catch (error) {
-    console.log(chalk.yellow('⚠️  无法获取MCP工具列表，使用默认配置'));
+    console.log(chalk.yellow('⚠️  无法获取MCP服务器列表，使用默认配置'));
     console.log(chalk.gray(`   原因: ${error.message}`));
   }
 
-  return tools;
+  return servers;
 }
 
-// 显示工具选择界面
-async function selectTools(roleName, availableTools) {
+// 显示MCP服务器选择界面
+async function selectMCPServers(roleName, availableServers) {
   const choices = [
-    new inquirer.Separator(chalk.green('🔧 默认工具（必选）')),
-    ...availableTools.defaultTools.map(tool => ({
+    new inquirer.Separator(chalk.green('🔧 默认工具（已包含）')),
+    ...availableServers.defaultTools.map(tool => ({
       name: `✓ ${tool}`,
       value: tool,
       checked: true,
@@ -59,25 +67,31 @@ async function selectTools(roleName, availableTools) {
     }))
   ];
 
-  if (availableTools.mcpTools.length > 0) {
+  if (availableServers.mcpServers.length > 0) {
     choices.push(
-      new inquirer.Separator(chalk.blue('🛠️  MCP工具（可选）')),
-      ...availableTools.mcpTools.map(tool => ({
-        name: tool,
-        value: tool,
-        checked: tool.includes('promptx') // PromptX工具默认选中
-      }))
+      new inquirer.Separator(chalk.blue('🛠️  MCP服务器（选择后将获得该服务器的所有工具权限）'))
     );
+
+    for (const server of availableServers.mcpServers) {
+      const statusIcon = server.connected ? '✓' : '✗';
+      const statusColor = server.connected ? chalk.green : chalk.red;
+      choices.push({
+        name: `${statusColor(statusIcon)} ${server.name} ${chalk.gray(`(${server.status})`)}`,
+        value: server.name,
+        checked: server.name.includes('promptx'), // PromptX服务器默认选中
+        disabled: !server.connected ? '(未连接)' : false
+      });
+    }
   }
 
   const answer = await inquirer.prompt([{
     type: 'checkbox',
-    name: 'selectedTools',
-    message: `为 ${roleName} 选择可用工具:`,
+    name: 'selectedItems',
+    message: `为 ${roleName} 选择工具和服务器:`,
     choices: choices,
     validate: (input) => {
       // 确保包含所有默认工具
-      const hasAllDefaults = availableTools.defaultTools.every(tool => input.includes(tool));
+      const hasAllDefaults = availableServers.defaultTools.every(tool => input.includes(tool));
       if (!hasAllDefaults) {
         return '必须包含所有默认工具';
       }
@@ -85,7 +99,21 @@ async function selectTools(roleName, availableTools) {
     }
   }]);
 
-  return answer.selectedTools;
+  // 分离默认工具和选中的服务器
+  const selectedTools = availableServers.defaultTools.slice(); // 始终包含默认工具
+  const selectedServers = [];
+  
+  for (const item of answer.selectedItems) {
+    if (availableServers.defaultTools.includes(item)) {
+      continue; // 默认工具已经包含
+    } else {
+      selectedServers.push(item);
+      // 为选中的服务器添加通配符权限
+      selectedTools.push(`mcp__${item}__*`);
+    }
+  }
+
+  return selectedTools;
 }
 
 // 获取PromptX角色
@@ -112,7 +140,7 @@ function showWelcome() {
 }
 
 // 显示角色选择菜单
-async function showRoleMenu(systemRoles, userRoles, availableTools) {
+async function showRoleMenu(systemRoles, userRoles, availableServers) {
   const choices = [
     ...systemRoles.map(role => ({
       name: `📦 ${role.id} ${chalk.gray('(系统角色)')}`,
@@ -166,8 +194,8 @@ async function showRoleMenu(systemRoles, userRoles, availableTools) {
 
   let selectedTools = [];
   if (typeAnswer.confirm) {
-    // 选择工具
-    selectedTools = await selectTools(roleAnswer.selectedRole.role, availableTools);
+    // 选择MCP服务器和工具
+    selectedTools = await selectMCPServers(roleAnswer.selectedRole.role, availableServers);
   }
 
   return {
@@ -263,9 +291,9 @@ async function main() {
   try {
     showWelcome();
     
-    // 发现可用工具
-    console.log(chalk.cyan('🔍 正在发现可用工具...\n'));
-    const availableTools = await discoverAllTools();
+    // 发现MCP服务器
+    console.log(chalk.cyan('🔍 正在发现MCP服务器...\n'));
+    const availableServers = await discoverMCPServers();
     
     // 加载角色
     console.log(chalk.cyan('🔍 正在从PromptX系统加载角色...\n'));
@@ -275,7 +303,7 @@ async function main() {
     console.log(`📊 发现 ${chalk.bold(systemRoles.length)} 个系统角色，${chalk.bold(userRoles.length)} 个用户角色\n`);
     
     // 显示角色选择
-    const { selectedRole, installType, confirm, selectedTools } = await showRoleMenu(systemRoles, userRoles, availableTools);
+    const { selectedRole, installType, confirm, selectedTools } = await showRoleMenu(systemRoles, userRoles, availableServers);
     
     if (!confirm) {
       console.log(chalk.yellow('\n👋 安装已取消'));

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "px2cc",
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "CLI tool to quickly install PromptX roles into Claude Code as agents or commands",
   "main": "cli.js",
   "type": "module",