ai-dev-requirements 0.1.7 → 0.1.9

package/dist/index.mjs

@@ -95,7 +95,7 @@ const SEARCH_TASKS_QUERY = `
       key
       tasks(filterGroup: $filterGroup, orderBy: $orderBy, limit: $limit, includeAncestors: { pathField: "path" }) {
         key uuid number name
-        issueType { uuid name }
+        issueType { uuid name detailType }
         status { uuid name category }
         priority { value }
         assign { uuid name }
@@ -104,6 +104,15 @@ const SEARCH_TASKS_QUERY = `
     }
   }
 `;
+const ISSUE_TYPES_QUERY = `
+  query IssueTypes($orderBy: OrderBy) {
+    issueTypes(orderBy: $orderBy) {
+      uuid
+      name
+      detailType
+    }
+  }
+`;
 const TASK_BY_NUMBER_QUERY = SEARCH_TASKS_QUERY;
 const RELATED_TASKS_QUERY = `
   query Task($key: Key) {
@@ -165,6 +174,117 @@ const DEFAULT_STATUS_NOT_IN = [
 	"Dn3k8ffK",
 	"TbmY2So5"
 ];
+const TESTCASE_LIBRARY_LIST_QUERY = `
+  query Q {
+    testcaseLibraries {
+      uuid name key
+      testcaseCaseCount
+    }
+  }
+`;
+const TESTCASE_MODULE_SEARCH_QUERY = `
+  query Q($filter: Filter) {
+    testcaseModules(filter: $filter) {
+      uuid name key
+      parent { uuid name }
+    }
+  }
+`;
+const TESTCASE_LIST_PAGED_QUERY = `
+  query PAGED_LIBRARY_TESTCASE_LIST($testCaseFilter: Filter, $pagination: Pagination) {
+    buckets(groupBy: {testcaseCases: {}}, pagination: $pagination) {
+      testcaseCases(filterGroup: $testCaseFilter, limit: 10000) {
+        uuid name key id
+        priority { uuid value }
+        type { uuid value }
+        assign { uuid name }
+        testcaseModule { uuid }
+      }
+      key
+      pageInfo { count totalCount hasNextPage endCursor }
+    }
+  }
+`;
+const TESTCASE_DETAIL_QUERY = `
+  query QUERY_TESTCASES_DETAIL($testCaseFilter: Filter, $stepFilter: Filter) {
+    testcaseCases(filter: $testCaseFilter) {
+      uuid name key id condition desc path
+      assign { uuid name }
+      priority { uuid value }
+      type { uuid value }
+      testcaseLibrary { uuid }
+      testcaseModule { uuid }
+      relatedTasks { uuid name number }
+    }
+    testcaseCaseSteps(filter: $stepFilter, orderBy: { index: ASC }) {
+      key uuid
+      testcaseCase { uuid }
+      desc result index
+    }
+  }
+`;
+function parseOnesSearchIntent(query) {
+	if (!query) return "keyword";
+	const normalized = query.toLowerCase();
+	if (query.includes("缺陷") || normalized.includes("bug")) return "all_bugs";
+	if (query.includes("任务")) return "all_tasks";
+	return "keyword";
+}
+function extractAssigneeName(query, intent) {
+	if (intent === "keyword") return null;
+	const trimmed = query.trim();
+	if (!trimmed) return null;
+	const ownerStyleMatch = trimmed.match(/\u8D1F\u8D23\u4EBA\u4E3A(.+?)\u7684?(?:\u7F3A\u9677|bug)$/i);
+	if (ownerStyleMatch?.[1]) return ownerStyleMatch[1].trim();
+	const candidate = trimmed.match(/^(查询)?(.+?)的(?:缺陷|bug|任务)$/i)?.[2]?.trim();
+	if (!candidate || candidate.includes("我")) return null;
+	return candidate;
+}
+function extractNamedAssignee(query, intent) {
+	if (intent === "keyword") return null;
+	const compact = query.replace(/\s+/g, "").trim();
+	if (!compact) return null;
+	const ownerStyleMatch = compact.match(/(?:\u8D1F\u8D23\u4EBA\u4E3A|\u8D1F\u8D23\u4EBA\u662F|\u6307\u6D3E\u7ED9|\u5206\u914D\u7ED9)(.+?)\u7684?(?:\u7F3A\u9677|bug|\u4EFB\u52A1)$/i);
+	if (ownerStyleMatch?.[1]) return ownerStyleMatch[1].trim();
+	const candidate = compact.match(/^(?:\u67E5\u8BE2|\u67E5\u627E|\u641C\u7D22)?(.+?)\u7684?(?:\u7F3A\u9677|bug|\u4EFB\u52A1)$/i)?.[1]?.trim();
+	if (!candidate || candidate.startsWith("我") || /^(?:\u6211|\u6211\u7684|\u6211\u6240\u6709|\u6211\u5168\u90E8|\u672C\u4EBA|\u5F53\u524D\u7528\u6237)$/.test(candidate)) return null;
+	return candidate;
+}
+function getBugStatusPriority(task) {
+	if (task.status?.category === "to_do") return 0;
+	if (task.status?.category === "in_progress") return 1;
+	return Number.POSITIVE_INFINITY;
+}
+function isOpenOrInProgressBug(task) {
+	const category = task.status?.category;
+	return category === "to_do" || category === "in_progress";
+}
+function extractTeamUsers(payload) {
+	const record = payload && typeof payload === "object" ? payload : null;
+	if (!record) return [];
+	const rawUsers = [
+		record.users,
+		record.items,
+		record.list,
+		record.results,
+		record.data?.users,
+		record.data?.items,
+		record.data?.list,
+		record.data?.results
+	].find(Array.isArray);
+	if (!rawUsers) return [];
+	return rawUsers.map((item) => {
+		const user = item && typeof item === "object" ? item : null;
+		if (!user) return null;
+		const uuid = user.uuid ?? user.user?.uuid ?? user.orgUser?.uuid ?? user.orgUserUuid ?? user.org_user_uuid ?? user.org_user?.org_user_uuid;
+		const name = user.name ?? user.user?.name ?? user.orgUser?.name ?? user.org_user?.name;
+		if (!uuid || !name) return null;
+		return {
+			uuid,
+			name
+		};
+	}).filter((item) => item !== null);
+}
 function base64Url(buffer) {
 	return buffer.toString("base64").replace(/\+/g, "-").replace(/\//g, "_").replace(/=+$/g, "");
 }
@@ -195,6 +315,7 @@ function toRequirement(task, description = "") {
 }
 var OnesAdapter = class extends BaseAdapter {
 	session = null;
+	issueTypesCache = null;
 	constructor(sourceType, config, resolvedAuth) {
 		super(sourceType, config, resolvedAuth);
 	}
@@ -355,6 +476,42 @@ var OnesAdapter = class extends BaseAdapter {
 		}
 		return response.json();
 	}
+	async fetchIssueTypes() {
+		if (this.issueTypesCache) return this.issueTypesCache;
+		this.issueTypesCache = (await this.graphql(ISSUE_TYPES_QUERY, { orderBy: { namePinyin: "ASC" } }, "issueTypes")).data?.issueTypes ?? [];
+		return this.issueTypesCache;
+	}
+	async searchTeamUsers(keyword) {
+		const session = await this.login();
+		const url = `${this.config.apiBase}/project/api/project/team/${session.teamUuid}/users/search`;
+		const response = await fetch(url, {
+			method: "POST",
+			headers: {
+				"Authorization": `Bearer ${session.accessToken}`,
+				"Content-Type": "application/json"
+			},
+			body: JSON.stringify({
+				keyword,
+				status: [1],
+				team_member_status: [1, 4],
+				types: [1, 10]
+			})
+		});
+		if (!response.ok) {
+			const text = await response.text().catch(() => "");
+			throw new Error(`ONES user search error: ${response.status} ${text}`);
+		}
+		return extractTeamUsers(await response.json());
+	}
+	async resolveAssigneeUuid(name) {
+		const trimmed = name.trim();
+		if (!trimmed) return null;
+		const users = await this.searchTeamUsers(trimmed);
+		const exactMatch = users.find((user) => user.name === trimmed);
+		if (exactMatch) return exactMatch.uuid;
+		const normalizedTarget = trimmed.toLowerCase();
+		return users.find((user) => user.name.toLowerCase().includes(normalizedTarget))?.uuid ?? null;
+	}
 	/**
 	* Fetch task info via REST API (includes description/rich fields not available in GraphQL).
 	* Reference: ones/packages/core/src/tasks.ts → fetchTaskInfo
@@ -519,6 +676,27 @@ var OnesAdapter = class extends BaseAdapter {
 	async searchRequirements(params) {
 		const page = params.page ?? 1;
 		const pageSize = params.pageSize ?? 50;
+		const intent = parseOnesSearchIntent(params.query);
+		const assigneeName = extractNamedAssignee(params.query, intent) ?? extractAssigneeName(params.query, intent);
+		const assigneeUuid = assigneeName ? await this.resolveAssigneeUuid(assigneeName) : null;
+		if (assigneeName && !assigneeUuid) return {
+			items: [],
+			total: 0,
+			page,
+			pageSize
+		};
+		let bugTypeUuids = [];
+		let taskTypeUuids = [];
+		if (intent === "all_bugs" || intent === "all_tasks") {
+			const issueTypes = await this.fetchIssueTypes();
+			bugTypeUuids = issueTypes.filter((item) => item.detailType === 3).map((item) => item.uuid);
+			taskTypeUuids = issueTypes.filter((item) => item.detailType === 2).map((item) => item.uuid);
+		}
+		const filter = { status_notIn: DEFAULT_STATUS_NOT_IN };
+		if (assigneeName) filter.assign_in = [assigneeUuid];
+		else filter.assign_in = ["${currentUser}"];
+		if (intent === "all_bugs") filter.issueType_in = bugTypeUuids;
+		if (intent === "all_tasks") filter.issueType_in = taskTypeUuids;
 		let tasks = (await this.graphql(SEARCH_TASKS_QUERY, {
 			groupBy: { tasks: {} },
 			groupOrderBy: null,
@@ -526,10 +704,7 @@ var OnesAdapter = class extends BaseAdapter {
 				position: "ASC",
 				createTime: "DESC"
 			},
-			filterGroup: [{
-				assign_in: ["${currentUser}"],
-				status_notIn: DEFAULT_STATUS_NOT_IN
-			}],
+			filterGroup: [filter],
 			search: null,
 			pagination: {
 				limit: pageSize * page,
@@ -537,8 +712,11 @@ var OnesAdapter = class extends BaseAdapter {
 			},
 			limit: 1e3
 		}, "group-task-data")).data?.buckets?.flatMap((b) => b.tasks ?? []) ?? [];
-		if (params.query) {
-			const keyword = params.query;
+		if (intent === "all_bugs") tasks = tasks.filter((task) => task.issueType?.uuid ? bugTypeUuids.includes(task.issueType.uuid) : false).filter((task) => isOpenOrInProgressBug(task)).sort((a, b) => getBugStatusPriority(a) - getBugStatusPriority(b));
+		if (intent === "all_tasks") tasks = tasks.filter((task) => task.issueType?.uuid ? taskTypeUuids.includes(task.issueType.uuid) : false);
+		if (assigneeUuid) tasks = tasks.filter((task) => task.assign?.uuid === assigneeUuid);
+		if (intent === "keyword" && params.query) {
+			const keyword = params.query.trim();
 			const lower = keyword.toLowerCase();
 			const numMatch = keyword.match(/^#?(\d+)$/);
 			if (numMatch) tasks = tasks.filter((t) => t.number === Number.parseInt(numMatch[1], 10));
@@ -626,6 +804,108 @@ var OnesAdapter = class extends BaseAdapter {
 			raw: task
 		};
 	}
+	async getTestcases(params) {
+		let libraryUuid = params.libraryUuid ?? this.config.options?.testcaseLibraryUuid;
+		if (!libraryUuid) {
+			const libs = (await this.graphql(TESTCASE_LIBRARY_LIST_QUERY, {}, "library-select")).data?.testcaseLibraries ?? [];
+			if (libs.length === 0) throw new Error("ONES: No testcase libraries found for this team");
+			libs.sort((a, b) => b.testcaseCaseCount - a.testcaseCaseCount);
+			libraryUuid = libs[0].uuid;
+		}
+		const task = ((await this.graphql(SEARCH_TASKS_QUERY, {
+			groupBy: { tasks: {} },
+			groupOrderBy: null,
+			orderBy: { createTime: "DESC" },
+			filterGroup: [{ number_in: [params.taskNumber] }],
+			search: null,
+			pagination: {
+				limit: 10,
+				preciseCount: false
+			},
+			limit: 10
+		}, "group-task-data")).data?.buckets?.flatMap((b) => b.tasks ?? []) ?? []).find((t) => t.number === params.taskNumber);
+		if (!task) throw new Error(`ONES: Task #${params.taskNumber} not found`);
+		const modules = (await this.graphql(TESTCASE_MODULE_SEARCH_QUERY, { filter: {
+			testcaseLibrary_in: [libraryUuid],
+			name_match: `#${params.taskNumber}`
+		} }, "find-testcase-module")).data?.testcaseModules ?? [];
+		if (modules.length === 0) throw new Error(`ONES: No testcase module matching "#${params.taskNumber}" in library ${libraryUuid}`);
+		const mod = modules[0];
+		const caseList = [];
+		let cursor = "";
+		let totalCount = 0;
+		while (true) {
+			const bucket = (await this.graphql(TESTCASE_LIST_PAGED_QUERY, {
+				testCaseFilter: [{
+					testcaseLibrary_in: [libraryUuid],
+					path_match: mod.uuid
+				}],
+				pagination: {
+					limit: 50,
+					after: cursor,
+					preciseCount: true
+				}
+			}, "testcase-list-paged")).data?.buckets?.[0];
+			if (!bucket) break;
+			caseList.push(...bucket.testcaseCases ?? []);
+			totalCount = bucket.pageInfo.totalCount;
+			if (!bucket.pageInfo.hasNextPage) break;
+			cursor = bucket.pageInfo.endCursor;
+		}
+		if (caseList.length === 0) return {
+			taskNumber: params.taskNumber,
+			taskName: task.name,
+			moduleName: mod.name,
+			moduleUuid: mod.uuid,
+			totalCount: 0,
+			cases: []
+		};
+		const allCases = [];
+		const BATCH_SIZE = 20;
+		for (let i = 0; i < caseList.length; i += BATCH_SIZE) {
+			const uuids = caseList.slice(i, i + BATCH_SIZE).map((c) => c.uuid);
+			const detailData = await this.graphql(TESTCASE_DETAIL_QUERY, {
+				testCaseFilter: { uuid_in: [...uuids, null] },
+				stepFilter: { testcaseCase_in: uuids }
+			}, "library-testcase-detail");
+			const cases = detailData.data?.testcaseCases ?? [];
+			const steps = detailData.data?.testcaseCaseSteps ?? [];
+			const stepsByCase = /* @__PURE__ */ new Map();
+			for (const step of steps) {
+				const caseUuid = step.testcaseCase.uuid;
+				if (!stepsByCase.has(caseUuid)) stepsByCase.set(caseUuid, []);
+				stepsByCase.get(caseUuid).push({
+					uuid: step.uuid,
+					index: step.index,
+					desc: step.desc ?? "",
+					result: step.result ?? ""
+				});
+			}
+			for (const c of cases) {
+				const freshDesc = c.desc ? await this.refreshImageUrls(c.desc) : "";
+				allCases.push({
+					uuid: c.uuid,
+					id: c.id,
+					name: c.name,
+					priority: c.priority?.value ?? "N/A",
+					type: c.type?.value ?? "Unknown",
+					assignName: c.assign?.name ?? null,
+					condition: c.condition ?? "",
+					desc: freshDesc,
+					steps: (stepsByCase.get(c.uuid) ?? []).sort((a, b) => a.index - b.index),
+					modulePath: c.path ?? ""
+				});
+			}
+		}
+		return {
+			taskNumber: params.taskNumber,
+			taskName: task.name,
+			moduleName: mod.name,
+			moduleUuid: mod.uuid,
+			totalCount,
+			cases: allCases
+		};
+	}
 };
 
 //#endregion
@@ -727,6 +1007,11 @@ function loadConfigFromEnv() {
 	const account = process.env.ONES_ACCOUNT;
 	const password = process.env.ONES_PASSWORD;
 	if (!apiBase || !account || !password) return null;
+	let options;
+	const configPath = findConfigFile(process.cwd());
+	if (configPath) try {
+		options = JSON.parse(readFileSync(configPath, "utf-8"))?.sources?.ones?.options;
+	} catch {}
 	return {
 		sources: { ones: {
 			enabled: true,
@@ -735,7 +1020,8 @@ function loadConfigFromEnv() {
 				type: "ones-pkce",
 				emailEnv: "ONES_ACCOUNT",
 				passwordEnv: "ONES_PASSWORD"
-			}
+			},
+			options
 		} },
 		defaultSource: "ones"
 	};
@@ -948,6 +1234,58 @@ function formatRequirement(req) {
 	return lines.join("\n");
 }
 
+//#endregion
+//#region src/tools/get-testcases.ts
+const GetTestcasesSchema = z.object({
+	taskNumber: z.string().describe("Task number (e.g. \"302\" or \"#302\"). Finds all testcases in the matching module."),
+	libraryUuid: z.string().optional().describe("Testcase library UUID. If omitted, uses configured default."),
+	source: z.string().optional().describe("Source to fetch from. If omitted, uses the default source.")
+});
+async function handleGetTestcases(input, adapters, defaultSource) {
+	const sourceType = input.source ?? defaultSource;
+	if (!sourceType) throw new Error("No source specified and no default source configured");
+	const adapter = adapters.get(sourceType);
+	if (!adapter) throw new Error(`Source "${sourceType}" is not configured. Available: ${[...adapters.keys()].join(", ")}`);
+	const numMatch = input.taskNumber.match(/^#?(\d+)$/);
+	if (!numMatch) throw new Error(`Invalid task number: "${input.taskNumber}". Expected a number like "302" or "#302".`);
+	return { content: [{
+		type: "text",
+		text: formatTestcases(await adapter.getTestcases({
+			taskNumber: Number.parseInt(numMatch[1], 10),
+			libraryUuid: input.libraryUuid
+		}))
+	}] };
+}
+function formatTestcases(result) {
+	const lines = [
+		`# ${result.taskName} — 测试用例`,
+		"",
+		`- **模块**: ${result.moduleName}`,
+		`- **共 ${result.totalCount} 个用例**（已加载 ${result.cases.length} 个）`,
+		""
+	];
+	for (const tc of result.cases) {
+		lines.push(`## ${tc.id} ${tc.name}`);
+		lines.push("");
+		lines.push(`- 优先级: ${tc.priority} | 类型: ${tc.type}`);
+		if (tc.assignName) lines.push(`- 维护人: ${tc.assignName}`);
+		if (tc.condition) lines.push(`- 前置条件: ${tc.condition}`);
+		if (tc.desc) lines.push(`- 备注: ${tc.desc}`);
+		if (tc.steps.length > 0) {
+			lines.push("");
+			lines.push("| 步骤 | 操作描述 | 预期结果 |");
+			lines.push("|------|----------|----------|");
+			for (const step of tc.steps) {
+				const desc = step.desc.replace(/\n/g, "<br>");
+				const res = step.result.replace(/\n/g, "<br>");
+				lines.push(`| ${step.index + 1} | ${desc} | ${res} |`);
+			}
+		}
+		lines.push("");
+	}
+	return lines.join("\n");
+}
+
 //#endregion
 //#region src/tools/list-sources.ts
 async function handleListSources(adapters, config) {
@@ -982,6 +1320,9 @@ const SearchRequirementsSchema = z.object({
 	page: z.number().int().min(1).optional().describe("Page number (default: 1)"),
 	pageSize: z.number().int().min(1).max(50).optional().describe("Results per page (default: 20, max: 50)")
 });
+function formatStatusMarker(status) {
+	return `[${status.toUpperCase()}]`;
+}
 async function handleSearchRequirements(input, adapters, defaultSource) {
 	const sourceType = input.source ?? defaultSource;
 	if (!sourceType) throw new Error("No source specified and no default source configured");
@@ -992,15 +1333,18 @@ async function handleSearchRequirements(input, adapters, defaultSource) {
 		page: input.page,
 		pageSize: input.pageSize
 	});
-	const lines = [`Found **${result.total}** results (page ${result.page}/${Math.ceil(result.total / result.pageSize) || 1}):`, ""];
+	const lines = [`Found **${result.total}** items (page ${result.page}/${Math.ceil(result.total / result.pageSize) || 1}):`, ""];
+	if (/\u6211.*\u7F3A\u9677|bug|\u6211.*\u4EFB\u52A1/i.test(input.query)) {
+		lines.push(`Query: ${input.query}`);
+		lines.push("Use an item ID or number in the next step to fetch detail.");
+		lines.push("");
+	}
 	for (const item of result.items) {
-		lines.push(`### ${item.id}: ${item.title}`);
+		lines.push(`### ${formatStatusMarker(item.status)} ${item.id}: ${item.title}`);
 		lines.push(`- Status: ${item.status} | Priority: ${item.priority} | Type: ${item.type}`);
 		lines.push(`- Assignee: ${item.assignee ?? "Unassigned"}`);
-		if (item.description) {
-			const desc = item.description.length > 200 ? `${item.description.slice(0, 200)}...` : item.description;
-			lines.push(`- ${desc}`);
-		}
+		const desc = item.description ? item.description.length > 200 ? `${item.description.slice(0, 200)}...` : item.description : "(empty)";
+		lines.push(`- Content: ${desc}`);
 		lines.push("");
 	}
 	return { content: [{
@@ -1121,6 +1465,19 @@ async function main() {
 			};
 		}
 	});
+	server.tool("get_testcases", "Get all test cases for a task by its number (e.g. 302). Searches the testcase library for a matching module and returns all cases with steps.", GetTestcasesSchema.shape, async (params) => {
+		try {
+			return await handleGetTestcases(params, adapters, config.config.defaultSource);
+		} catch (err) {
+			return {
+				content: [{
+					type: "text",
+					text: `Error: ${err.message}`
+				}],
+				isError: true
+			};
+		}
+	});
 	const transport = new StdioServerTransport();
 	await server.connect(transport);
 }