@zzp123/mcp-zentao 1.15.0 → 1.17.0

package/scripts/generate-role-versions.cjs

@@ -0,0 +1,214 @@
+const fs = require('fs');
+const path = require('path');
+
+// 角色工具配置
+const TOOL_CATEGORIES = {
+    init: ['initZentao'],
+    story: ['createStory', 'getStoryDetail', 'changeStory', 'deleteStory', 'getProductStories', 'addStoryComment'],
+    storyReadonly: ['getStoryDetail', 'getProductStories'],
+    product: ['getProducts', 'getProductDetail', 'createProduct', 'updateProduct', 'deleteProduct'],
+    plan: ['getProductPlans', 'getPlanDetail', 'createPlan', 'updatePlan', 'deletePlan', 'linkStoriesToPlan', 'unlinkStoriesFromPlan', 'linkBugsToPlan', 'unlinkBugsFromPlan'],
+    program: ['getPrograms', 'getProgramDetail', 'createProgram', 'updateProgram', 'deleteProgram'],
+    feedback: ['getFeedbacks', 'getFeedbackDetail', 'createFeedback', 'updateFeedback', 'deleteFeedback', 'assignFeedback', 'closeFeedback'],
+    bug: ['getMyBugs', 'getProductBugs', 'getBugDetail', 'createBug', 'updateBug', 'deleteBug', 'resolveBug', 'addBugComment'],
+    bugResolve: ['getMyBugs', 'getBugDetail', 'resolveBug', 'addBugComment'],
+    bugReadonly: ['getMyBugs', 'getProductBugs', 'getBugDetail'],
+    testcase: ['getProductTestCases', 'getTestCaseDetail', 'createTestCase', 'updateTestCase', 'deleteTestCase'],
+    task: ['getMyTasks', 'getTaskDetail', 'createTask', 'updateTask', 'finishTask', 'deleteTask', 'addTaskComment'],
+    taskReadonly: ['getMyTasks', 'getTaskDetail'],
+    project: ['getProjects', 'getProjectDetail', 'createProject', 'updateProject', 'deleteProject', 'getProjectReleases', 'getProjectExecutions', 'getProjectBuilds'],
+    execution: ['getExecutionDetail', 'createExecution', 'updateExecution', 'deleteExecution', 'getExecutionBuilds'],
+    build: ['getBuildDetail', 'createBuild', 'updateBuild', 'deleteBuild'],
+    ticket: ['getTickets', 'getTicketDetail', 'createTicket', 'updateTicket', 'deleteTicket'],
+    comment: ['getComments', 'getCommentDetail', 'addComment', 'updateComment', 'deleteComment', 'addTaskComment', 'addBugComment', 'addStoryComment'],
+    utility: ['getModules', 'uploadFile', 'uploadImageFromClipboard', 'downloadFile'],
+    user: ['getUsers', 'getUserDetail', 'getMyProfile', 'createUser', 'updateUser', 'deleteUser']
+};
+
+const ROLE_PERMISSIONS = {
+    pm: [
+        ...TOOL_CATEGORIES.init,
+        ...TOOL_CATEGORIES.story,
+        ...TOOL_CATEGORIES.product,
+        ...TOOL_CATEGORIES.plan,
+        ...TOOL_CATEGORIES.program,
+        ...TOOL_CATEGORIES.feedback,
+        ...TOOL_CATEGORIES.bugReadonly,
+        ...TOOL_CATEGORIES.taskReadonly,
+        ...TOOL_CATEGORIES.comment,
+        ...TOOL_CATEGORIES.ticket,
+        ...TOOL_CATEGORIES.utility
+    ],
+    qa: [
+        ...TOOL_CATEGORIES.init,
+        ...TOOL_CATEGORIES.bug,
+        ...TOOL_CATEGORIES.testcase,
+        ...TOOL_CATEGORIES.build,
+        ...TOOL_CATEGORIES.storyReadonly,
+        ...TOOL_CATEGORIES.taskReadonly,
+        ...TOOL_CATEGORIES.comment,
+        ...TOOL_CATEGORIES.ticket,
+        ...TOOL_CATEGORIES.utility
+    ],
+    dev: [
+        ...TOOL_CATEGORIES.init,
+        ...TOOL_CATEGORIES.task,
+        ...TOOL_CATEGORIES.bugResolve,
+        ...TOOL_CATEGORIES.project,
+        ...TOOL_CATEGORIES.execution,
+        ...TOOL_CATEGORIES.build,
+        ...TOOL_CATEGORIES.storyReadonly,
+        ...TOOL_CATEGORIES.comment,
+        ...TOOL_CATEGORIES.ticket,
+        ...TOOL_CATEGORIES.utility
+    ]
+};
+
+const ROLE_NAMES = {
+    pm: '产品经理',
+    qa: '测试工程师',
+    dev: '开发工程师'
+};
+
+// 读取完整的 index.ts
+const indexPath = path.join(__dirname, 'src', 'index.ts');
+const indexContent = fs.readFileSync(indexPath, 'utf8');
+
+// 提取工具注册块的函数
+function extractToolBlocks(content) {
+    const lines = content.split('\n');
+    const blocks = [];
+    let currentBlock = null;
+    let braceDepth = 0;
+    let parenDepth = 0;
+
+    for (let i = 0; i < lines.length; i++) {
+        const line = lines[i];
+        const trimmed = line.trim();
+
+        // 检测 server.tool 的开始
+        if (trimmed.startsWith('server.tool(')) {
+            // 提取工具名
+            const match = trimmed.match(/server\.tool\("([^"]+)"/);
+            if (match) {
+                currentBlock = {
+                    toolName: match[1],
+                    startLine: i,
+                    lines: [line],
+                    braceDepth: 0,
+                    parenDepth: 0
+                };
+
+                // 计算当前行的括号和大括号
+                currentBlock.parenDepth += (line.match(/\(/g) || []).length - (line.match(/\)/g) || []).length;
+                currentBlock.braceDepth += (line.match(/\{/g) || []).length - (line.match(/\}/g) || []).length;
+            }
+        } else if (currentBlock) {
+            // 在工具块内
+            currentBlock.lines.push(line);
+            currentBlock.parenDepth += (line.match(/\(/g) || []).length - (line.match(/\)/g) || []).length;
+            currentBlock.braceDepth += (line.match(/\{/g) || []).length - (line.match(/\}/g) || []).length;
+
+            // 检查是否到达块的结束（圆括号和大括号都平衡，且以 ); 结尾）
+            if (currentBlock.parenDepth === 0 && currentBlock.braceDepth === 0 && trimmed.endsWith(');')) {
+                currentBlock.endLine = i;
+                blocks.push(currentBlock);
+                currentBlock = null;
+            }
+        }
+    }
+
+    return blocks;
+}
+
+// 为每个角色生成文件
+['pm', 'qa', 'dev'].forEach(role => {
+    console.log(`\n正在生成 ${ROLE_NAMES[role]} (${role}) 版本...`);
+
+    const allowedTools = new Set(ROLE_PERMISSIONS[role]);
+    console.log(`  允许的工具: ${Array.from(allowedTools).join(', ')}`);
+
+    // 提取所有工具块
+    const toolBlocks = extractToolBlocks(indexContent);
+    console.log(`  总工具数: ${toolBlocks.length}`);
+
+    // 过滤出允许的工具
+    const allowedBlocks = toolBlocks.filter(block => allowedTools.has(block.toolName));
+    console.log(`  ${ROLE_NAMES[role]}可用工具数: ${allowedBlocks.length}`);
+
+    // 构建新文件内容
+    const lines = indexContent.split('\n');
+    const outputLines = [];
+    const includedLines = new Set();
+
+    // 标记所有要包含的行
+    allowedBlocks.forEach(block => {
+        for (let i = block.startLine; i <= block.endLine; i++) {
+            includedLines.add(i);
+        }
+        // 如果前一行是注释，也包含进来
+        if (block.startLine > 0 && lines[block.startLine - 1].trim().startsWith('//')) {
+            includedLines.add(block.startLine - 1);
+        }
+    });
+
+    // 找到第一个 server.tool 的位置
+    let firstToolLine = toolBlocks.length > 0 ? toolBlocks[0].startLine : lines.length;
+
+    // 输出文件头部（到第一个工具之前）
+    for (let i = 0; i < firstToolLine; i++) {
+        outputLines.push(lines[i]);
+    }
+
+    // 更新服务器名称和版本
+    const headerContent = outputLines.join('\n');
+    let updatedHeader = headerContent
+        .replace(/name: "Zentao API"/, `name: "Zentao API (${ROLE_NAMES[role]})"`)
+        .replace(/version: "1\.16\.0"/, `version: "1.17.0"`);
+
+    // 输出更新后的头部
+    const finalLines = updatedHeader.split('\n');
+
+    // 添加所有允许的工具块
+    let lastEndLine = firstToolLine - 1;
+    allowedBlocks.forEach((block, index) => {
+        // 添加空行分隔（如果需要）
+        if (block.startLine > lastEndLine + 1) {
+            finalLines.push('');
+        }
+
+        // 添加工具块
+        for (let i = block.startLine; i <= block.endLine; i++) {
+            finalLines.push(lines[i]);
+        }
+
+        lastEndLine = block.endLine;
+    });
+
+    // 添加文件尾部（服务器启动代码）
+    const lastToolBlock = toolBlocks[toolBlocks.length - 1];
+    if (lastToolBlock) {
+        for (let i = lastToolBlock.endLine + 1; i < lines.length; i++) {
+            finalLines.push(lines[i]);
+        }
+    }
+
+    // 写入文件
+    const output = finalLines.join('\n');
+    const outputPath = path.join(__dirname, 'src', `index-${role}.ts`);
+    fs.writeFileSync(outputPath, output, 'utf8');
+
+    console.log(`  ✓ 已生成: ${outputPath}`);
+
+    // 统计文件大小
+    const stats = fs.statSync(outputPath);
+    const originalStats = fs.statSync(indexPath);
+    const reduction = ((1 - stats.size / originalStats.size) * 100).toFixed(1);
+    console.log(`  文件大小: ${(stats.size / 1024).toFixed(1)} KB (原始: ${(originalStats.size / 1024).toFixed(1)} KB, 减少 ${reduction}%)`);
+});
+
+console.log('\n✓ 所有角色版本生成完成！');
+console.log('\n下一步：');
+console.log('1. 运行 npx tsc 编译所有版本');
+console.log('2. 配置 package.json 的构建脚本');
+console.log('3. 测试各个版本');