ai-dev-requirements 0.1.12 → 0.1.13

package/dist/index.mjs

@@ -5,7 +5,6 @@ import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import crypto from "node:crypto";
 import { z } from "zod/v4";
-
 //#region src/utils/map-status.ts
 const ONES_STATUS_MAP = {
 	to_do: "open",
@@ -41,7 +40,6 @@ function mapOnesPriority(priority) {
 function mapOnesType(type) {
 	return ONES_TYPE_MAP[type.toLowerCase()] ?? "task";
 }
-
 //#endregion
 //#region src/adapters/base.ts
 /**
@@ -58,7 +56,6 @@ var BaseAdapter = class {
 		this.resolvedAuth = resolvedAuth;
 	}
 };
-
 //#endregion
 //#region src/adapters/ones.ts
 const TASK_DETAIL_QUERY = `
@@ -92,6 +89,26 @@ const TASK_DETAIL_QUERY = `
     }
   }
 `;
+const RELATED_ACTIVITIES_QUERY = `
+  query Task($key: Key) {
+    task(key: $key) {
+      key
+      ...RelatedActivities_task1
+    }
+  }
+
+  fragment RelatedActivities_task1 on Task {
+    relatedActivities {
+      uuid
+      name
+      projectUUID
+      project_uuid: projectUUID
+      relatedChild
+      related_child_uuid: relatedChild
+    }
+    relatedActivitiesCount
+  }
+`;
 const SEARCH_TASKS_QUERY = `
   query GROUP_TASK_DATA($groupBy: GroupBy, $groupOrderBy: OrderBy, $orderBy: OrderBy, $filterGroup: [Filter!], $search: Search, $pagination: Pagination, $limit: Int) {
     buckets(groupBy: $groupBy, orderBy: $groupOrderBy, pagination: $pagination, filter: $search) {
@@ -324,7 +341,7 @@ function extractWikiPageUuidsFromText(text) {
 	return [...uuids];
 }
 function parseOnesWikiPageRoute(input) {
-	if (!input.includes("/wiki/")) return null;
+	if (!isOnesWikiUrlInput(input)) return null;
 	const match = (() => {
 		try {
 			const parsed = new URL(input);
@@ -332,7 +349,7 @@ function parseOnesWikiPageRoute(input) {
 		} catch {
 			return input;
 		}
-	})().match(/\/team\/([^/?#]+)\/space\/[^/?#]+\/page\/([^/?#]+)/);
+	})().match(/\/team\/([^/?#]+)\/(?:space\/[^/?#]+\/)?page\/([^/?#]+)/);
 	if (!match?.[1] || !match[2]) return null;
 	return {
 		teamUuid: decodeURIComponent(match[1]),
@@ -340,7 +357,7 @@ function parseOnesWikiPageRoute(input) {
 	};
 }
 function isOnesWikiUrlInput(input) {
-	return input.includes("/wiki/");
+	return /\/wiki(?:\/|(?=[#?]|$))/.test(input);
 }
 function parseAuthorizeRequestId(location) {
 	try {
@@ -464,22 +481,126 @@ function renderWikiEmbed(block, context) {
 function escapeWikiTableCell(value) {
 	return value.replace(/\|/g, "\\|").replace(/[ \t]*\n+[ \t]*/g, " ").trim();
 }
+function escapeWikiHtml(value) {
+	return value.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;").replace(/"/g, "&quot;").replace(/'/g, "&#39;");
+}
+function parseWikiTableSpan(value) {
+	if (typeof value !== "number" || !Number.isFinite(value) || value <= 0) return 1;
+	return Math.max(Math.trunc(value), 1);
+}
+function buildWikiTableLayout(block) {
+	const columnCount = typeof block.cols === "number" && block.cols > 0 ? Math.trunc(block.cols) : 0;
+	const children = Array.isArray(block.children) ? block.children.filter((child) => typeof child === "string") : [];
+	if (!columnCount || !children.length) return null;
+	const hasDeclaredRows = typeof block.rows === "number" && block.rows > 0;
+	const initialRowCount = hasDeclaredRows ? Math.trunc(block.rows) : Math.max(Math.ceil(children.length / columnCount), 1);
+	const occupied = [];
+	const rows = [];
+	const ensureRowCount = (count) => {
+		while (occupied.length < count) {
+			occupied.push(Array.from({ length: columnCount }).fill(false));
+			rows.push([]);
+		}
+	};
+	ensureRowCount(initialRowCount);
+	let cursor = 0;
+	let hasMergedCells = false;
+	for (const childId of children) {
+		while (true) {
+			const row = Math.floor(cursor / columnCount);
+			const column = cursor % columnCount;
+			ensureRowCount(row + 1);
+			if (!occupied[row][column]) break;
+			cursor += 1;
+		}
+		const row = Math.floor(cursor / columnCount);
+		const column = cursor % columnCount;
+		const requestedRowSpan = parseWikiTableSpan(block[`${childId}_rowSpan`]);
+		if (!hasDeclaredRows) ensureRowCount(row + requestedRowSpan);
+		const rowSpan = Math.min(requestedRowSpan, occupied.length - row);
+		const colSpan = Math.min(parseWikiTableSpan(block[`${childId}_colSpan`]), columnCount - column);
+		hasMergedCells ||= rowSpan > 1 || colSpan > 1;
+		rows[row].push({
+			childId,
+			row,
+			column,
+			rowSpan,
+			colSpan
+		});
+		for (let rowOffset = 0; rowOffset < rowSpan; rowOffset += 1) for (let columnOffset = 0; columnOffset < colSpan; columnOffset += 1) occupied[row + rowOffset][column + columnOffset] = true;
+		cursor += 1;
+	}
+	return {
+		columnCount,
+		rows,
+		hasMergedCells
+	};
+}
+function wikiCellContainsTable(value) {
+	return asWikiBlocks(value).some((block) => block.type === "table");
+}
+function renderWikiTextRunsHtml(value) {
+	if (!Array.isArray(value)) return "";
+	return value.map((run) => {
+		if (!isRecord(run)) return "";
+		const attributes = isRecord(run.attributes) ? run.attributes : {};
+		let content = escapeWikiHtml(typeof run.insert === "string" ? run.insert.replace(/\u00A0/g, " ") : "").replace(/\n/g, "<br>");
+		if (attributes.code) content = `<code>${content}</code>`;
+		if (attributes.bold) content = `<strong>${content}</strong>`;
+		if (attributes.italic) content = `<em>${content}</em>`;
+		if (attributes.underline) content = `<u>${content}</u>`;
+		if (attributes.strike) content = `<s>${content}</s>`;
+		const link = typeof attributes.link === "string" ? attributes.link : "";
+		return link ? `<a href="${escapeWikiHtml(link)}">${content}</a>` : content;
+	}).join("");
+}
+function renderWikiCellHtml(value, document, context) {
+	return asWikiBlocks(value).map((block) => renderWikiBlockHtml(block, document, context)).filter(Boolean).join("");
+}
+function renderWikiBlockHtml(block, document, context) {
+	if (block.type === "table") {
+		const layout = buildWikiTableLayout(block);
+		return layout ? renderWikiTableHtml(layout, document, context) : "";
+	}
+	if (block.type === "embed") return `<p>${escapeWikiHtml(renderWikiEmbed(block, context))}</p>`;
+	const text = renderWikiTextRunsHtml(block.text);
+	if (!text) return "";
+	if (block.type === "list") {
+		const tag = block.ordered ? "ol" : "ul";
+		return `<${tag}><li>${text}</li></${tag}>`;
+	}
+	if (block.heading) {
+		const level = Math.min(Math.max(Math.trunc(block.heading), 1), 6);
+		return `<h${level}>${text}</h${level}>`;
+	}
+	return `<p>${text}</p>`;
+}
+function renderWikiTableHtml(layout, document, context) {
+	return `<table>\n<tbody>\n${layout.rows.map((row) => {
+		return `<tr>\n${row.map((cell) => {
+			const attributes = [cell.rowSpan > 1 ? `rowspan="${cell.rowSpan}"` : "", cell.colSpan > 1 ? `colspan="${cell.colSpan}"` : ""].filter(Boolean);
+			const content = renderWikiCellHtml(document[cell.childId], document, context);
+			return `<td${attributes.length ? ` ${attributes.join(" ")}` : ""}>${content}</td>`;
+		}).join("\n")}\n</tr>`;
+	}).join("\n")}\n</tbody>\n</table>`;
+}
 function renderWikiCell(value, document, context) {
 	const blocks = asWikiBlocks(value);
 	if (!blocks.length) return "";
 	return blocks.map((block) => renderWikiBlock(block, document, context)).filter(Boolean).join(" ").replace(/[ \t]*\n+[ \t]*/g, " ").trim();
 }
 function renderWikiTable(block, document, context) {
-	const cols = typeof block.cols === "number" && block.cols > 0 ? Math.trunc(block.cols) : 0;
-	const children = Array.isArray(block.children) ? block.children.filter((child) => typeof child === "string") : [];
-	if (!cols || !children.length) return "";
+	const layout = buildWikiTableLayout(block);
+	if (!layout) return "";
+	const hasNestedTable = layout.rows.some((row) => row.some((cell) => wikiCellContainsTable(document[cell.childId])));
+	if (layout.hasMergedCells || hasNestedTable) return renderWikiTableHtml(layout, document, context);
 	const rows = [];
-	for (let index = 0; index < children.length; index += cols) {
-		const cells = children.slice(index, index + cols).map((childId) => escapeWikiTableCell(renderWikiCell(document[childId], document, context)));
-		while (cells.length < cols) cells.push("");
+	for (const row of layout.rows) {
+		const cells = Array.from({ length: layout.columnCount }).fill("");
+		for (const cell of row) cells[cell.column] = escapeWikiTableCell(renderWikiCell(document[cell.childId], document, context));
 		rows.push(`| ${cells.join(" | ")} |`);
 	}
-	if (rows.length > 1) rows.splice(1, 0, `| ${Array.from({ length: cols }, () => "---").join(" | ")} |`);
+	if (rows.length > 1) rows.splice(1, 0, `| ${Array.from({ length: layout.columnCount }).fill("---").join(" | ")} |`);
 	return rows.join("\n");
 }
 function renderWikiBlock(block, document, context) {
@@ -702,6 +823,34 @@ var OnesAdapter = class extends BaseAdapter {
 		}
 		return response.json();
 	}
+	async onesql(query, variables, workItemType) {
+		const session = await this.login();
+		const url = `${this.config.apiBase}/project/api/ones-project/team/${session.teamUuid}/workitems/onesql`;
+		const response = await fetch(url, {
+			method: "POST",
+			headers: {
+				"Authorization": `Bearer ${session.accessToken}`,
+				"Content-Type": "application/json"
+			},
+			body: JSON.stringify({
+				query,
+				variables: [
+					variables,
+					workItemType,
+					null,
+					null
+				]
+			})
+		});
+		if (!response.ok) {
+			const text = await response.text().catch(() => "");
+			throw new Error(`ONES OneSQL error: ${response.status} ${text}`);
+		}
+		return response.json();
+	}
+	async fetchRelatedActivities(taskKey) {
+		return (await this.onesql(RELATED_ACTIVITIES_QUERY, { key: taskKey }, "Task")).data?.task?.relatedActivities ?? [];
+	}
 	async searchTaskByNumber(taskNumber) {
 		const session = await this.login();
 		const url = `${this.config.apiBase}/project/api/project/team/${session.teamUuid}/search?q=${encodeURIComponent(String(taskNumber))}&start=0&limit=10&types=task`;
@@ -731,7 +880,8 @@ var OnesAdapter = class extends BaseAdapter {
 	}
 	async fetchIssueTypes() {
 		if (this.issueTypesCache) return this.issueTypesCache;
-		this.issueTypesCache = (await this.graphql(ISSUE_TYPES_QUERY, { orderBy: { namePinyin: "ASC" } }, "issueTypes")).data?.issueTypes ?? [];
+		const data = await this.graphql(ISSUE_TYPES_QUERY, { orderBy: { namePinyin: "ASC" } }, "issueTypes");
+		this.issueTypesCache = data.data?.issueTypes ?? [];
 		return this.issueTypesCache;
 	}
 	async fetchProjects() {
@@ -992,8 +1142,10 @@ var OnesAdapter = class extends BaseAdapter {
 		}
 		if (isOnesWikiUrlInput(params.id)) throw new Error("ONES: Unsupported wiki page URL. Expected /wiki/#/team/{teamUuid}/space/{spaceUuid}/page/{wikiUuid}");
 		const taskRef = await this.resolveTaskRef(params.id);
+		const shouldFetchRelatedActivities = parseDisplayId(params.id.trim()) !== null;
 		const task = (await this.graphql(TASK_DETAIL_QUERY, { key: taskRef.key }, "Task")).data?.task;
 		if (!task) throw new Error(`ONES: Task "${params.id}" not found`);
+		const relatedActivities = shouldFetchRelatedActivities ? await this.fetchRelatedActivities(taskRef.key) : [];
 		const wikiRefs = /* @__PURE__ */ new Map();
 		for (const wiki of task.relatedWikiPages ?? []) if (!wiki.errorMessage) wikiRefs.set(wiki.uuid, {
 			title: wiki.title,
@@ -1034,6 +1186,18 @@ var OnesAdapter = class extends BaseAdapter {
 				parts.push(`- #${related.number} ${related.name} [${related.issueType?.name}] (${related.status?.name}) — ${assignee}`);
 			}
 		}
+		if (relatedActivities.length) {
+			parts.push("");
+			parts.push("## Related Work Items");
+			for (const activity of relatedActivities) {
+				const details = [
+					`UUID: ${activity.uuid}`,
+					activity.projectUUID ? `Project: ${activity.projectUUID}` : null,
+					activity.relatedChild ? `Relation: ${activity.relatedChild}` : null
+				].filter(Boolean);
+				parts.push(`- ${activity.name} (${details.join(", ")})`);
+			}
+		}
 		if (task.parent?.uuid) {
 			parts.push("");
 			parts.push("## Parent Task");
@@ -1064,7 +1228,12 @@ var OnesAdapter = class extends BaseAdapter {
 			parts.push(detailText);
 		}
 		const wikiAttachments = wikiContents.flatMap((wiki) => wiki.attachments);
-		return toRequirement(task, parts.join("\n"), wikiAttachments);
+		const req = toRequirement(task, parts.join("\n"), wikiAttachments);
+		req.raw = {
+			...req.raw,
+			relatedActivities
+		};
+		return req;
 	}
 	/**
 	* Search tasks assigned to current user via GraphQL.
@@ -1367,7 +1536,6 @@ var OnesAdapter = class extends BaseAdapter {
 		};
 	}
 };
-
 //#endregion
 //#region src/adapters/index.ts
 const ADAPTER_MAP = { ones: OnesAdapter };
@@ -1379,7 +1547,6 @@ function createAdapter(sourceType, config, resolvedAuth) {
 	if (!AdapterClass) throw new Error(`Unsupported source type: "${sourceType}". Supported: ${Object.keys(ADAPTER_MAP).join(", ")}`);
 	return new AdapterClass(sourceType, config, resolvedAuth);
 }
-
 //#endregion
 //#region src/config/loader.ts
 const AuthSchema = z.discriminatedUnion("type", [
@@ -1537,7 +1704,6 @@ function loadConfig(startDir) {
 		configPath
 	};
 }
-
 //#endregion
 //#region src/tools/add-manhour.ts
 const AddManhourSchema = z.object({
@@ -1573,7 +1739,6 @@ function formatAddManhourResult(result) {
 		`- **Description**: ${result.description}`
 	].join("\n");
 }
-
 //#endregion
 //#region src/tools/get-issue-detail.ts
 const GetIssueDetailSchema = z.object({
@@ -1648,7 +1813,6 @@ function formatIssueDetail(detail) {
 	else lines.push("_No description_");
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/tools/get-related-issues.ts
 const GetRelatedIssuesSchema = z.object({
@@ -1689,7 +1853,6 @@ function formatRelatedIssues(issues) {
 	}
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/tools/get-requirement.ts
 const GetRequirementSchema = z.object({
@@ -1768,7 +1931,6 @@ function formatRequirement(req) {
 	}
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/tools/get-testcases.ts
 const GetTestcasesSchema = z.object({
@@ -1820,7 +1982,6 @@ function formatTestcases(result) {
 	}
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/tools/list-sources.ts
 async function handleListSources(adapters, config) {
@@ -1846,7 +2007,6 @@ async function handleListSources(adapters, config) {
 		text: lines.join("\n")
 	}] };
 }
-
 //#endregion
 //#region src/tools/search-requirements.ts
 const SearchRequirementsSchema = z.object({
@@ -1887,7 +2047,6 @@ async function handleSearchRequirements(input, adapters, defaultSource) {
 		text: lines.join("\n")
 	}] };
 }
-
 //#endregion
 //#region src/tools/update-task-plan-dates.ts
 const DateSchema = z.string().regex(/^\d{4}-\d{2}-\d{2}$/, "Expected YYYY-MM-DD");
@@ -1921,7 +2080,6 @@ function formatUpdateTaskPlanDatesResult(result) {
 	if (result.planEndDate) lines.push(`- **Plan End Date**: ${result.planEndDate}`);
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/index.ts
 /**
@@ -1969,7 +2127,10 @@ async function main() {
 		name: "ai-dev-requirements",
 		version: "0.1.0"
 	});
-	server.tool("get_requirement", "Fetch a single requirement/issue by its ID from a configured source (ONES)", GetRequirementSchema.shape, async (params) => {
+	server.registerTool("get_requirement", {
+		description: "Fetch a single requirement/issue by its ID from a configured source (ONES)",
+		inputSchema: GetRequirementSchema.shape
+	}, async (params) => {
 		try {
 			return await handleGetRequirement(params, adapters, config.config.defaultSource);
 		} catch (err) {
@@ -1982,7 +2143,10 @@ async function main() {
 			};
 		}
 	});
-	server.tool("search_requirements", "Search for requirements/issues by keywords across a configured source", SearchRequirementsSchema.shape, async (params) => {
+	server.registerTool("search_requirements", {
+		description: "Search for requirements/issues by keywords across a configured source",
+		inputSchema: SearchRequirementsSchema.shape
+	}, async (params) => {
 		try {
 			return await handleSearchRequirements(params, adapters, config.config.defaultSource);
 		} catch (err) {
@@ -1995,7 +2159,7 @@ async function main() {
 			};
 		}
 	});
-	server.tool("list_sources", "List all configured requirement sources and their status", {}, async () => {
+	server.registerTool("list_sources", { description: "List all configured requirement sources and their status" }, async () => {
 		try {
 			return await handleListSources(adapters, config.config);
 		} catch (err) {
@@ -2008,7 +2172,10 @@ async function main() {
 			};
 		}
 	});
-	server.tool("get_related_issues", "Get pending defect issues (bugs) related to a requirement task. Returns all pending defects grouped by assignee (current user first).", GetRelatedIssuesSchema.shape, async (params) => {
+	server.registerTool("get_related_issues", {
+		description: "Get pending defect issues (bugs) related to a requirement task. Returns all pending defects grouped by assignee (current user first).",
+		inputSchema: GetRelatedIssuesSchema.shape
+	}, async (params) => {
 		try {
 			return await handleGetRelatedIssues(params, adapters, config.config.defaultSource);
 		} catch (err) {
@@ -2021,7 +2188,10 @@ async function main() {
 			};
 		}
 	});
-	server.tool("get_issue_detail", "Get detailed information about a specific issue/defect including description, rich text, and images", GetIssueDetailSchema.shape, async (params) => {
+	server.registerTool("get_issue_detail", {
+		description: "Get detailed information about a specific issue/defect including description, rich text, and images",
+		inputSchema: GetIssueDetailSchema.shape
+	}, async (params) => {
 		try {
 			return await handleGetIssueDetail(params, adapters, config.config.defaultSource);
 		} catch (err) {
@@ -2034,7 +2204,10 @@ 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) => {
+	server.registerTool("get_testcases", {
+		description: "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.",
+		inputSchema: GetTestcasesSchema.shape
+	}, async (params) => {
 		try {
 			return await handleGetTestcases(params, adapters, config.config.defaultSource);
 		} catch (err) {
@@ -2047,7 +2220,10 @@ async function main() {
 			};
 		}
 	});
-	server.tool("add_manhour", "Add a work-hour record to a ONES task, bug, or requirement. Supports task key, uuid, number, or displayId.", AddManhourSchema.shape, async (params) => {
+	server.registerTool("add_manhour", {
+		description: "Add a work-hour record to a ONES task, bug, or requirement. Supports task key, uuid, number, or displayId.",
+		inputSchema: AddManhourSchema.shape
+	}, async (params) => {
 		try {
 			return await handleAddManhour(params, adapters, config.config.defaultSource);
 		} catch (err) {
@@ -2060,7 +2236,10 @@ async function main() {
 			};
 		}
 	});
-	server.tool("update_task_plan_dates", "Update plan start and/or plan end dates for a ONES task, bug, or requirement. Supports task key, uuid, number, or displayId.", UpdateTaskPlanDatesSchema.shape, async (params) => {
+	server.registerTool("update_task_plan_dates", {
+		description: "Update plan start and/or plan end dates for a ONES task, bug, or requirement. Supports task key, uuid, number, or displayId.",
+		inputSchema: UpdateTaskPlanDatesSchema.shape
+	}, async (params) => {
 		try {
 			return await handleUpdateTaskPlanDates(params, adapters, config.config.defaultSource);
 		} catch (err) {
@@ -2080,7 +2259,7 @@ main().catch((err) => {
 	console.error("[requirements-mcp] Fatal error:", err);
 	process.exit(1);
 });
-
 //#endregion
-export {  };
+export {};
+
 //# sourceMappingURL=index.mjs.map