ai-dev-requirements 0.1.11 → 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) {
@@ -116,6 +133,26 @@ const ISSUE_TYPES_QUERY = `
     }
   }
 `;
+const PROJECTS_QUERY = `
+  query Projects($groupBy: GroupBy, $orderBy: OrderBy, $pagination: Pagination, $projectOrderBy: OrderBy, $projectFilterGroup: [Filter!]) {
+    buckets(groupBy: $groupBy, orderBy: $orderBy, pagination: $pagination) {
+      key
+      projects(limit: 10000, orderBy: $projectOrderBy, filterGroup: $projectFilterGroup) {
+        key
+        uuid
+        name
+        identifier
+      }
+    }
+  }
+`;
+const ADD_MANHOUR_MUTATION = `
+  mutation AddManhour {
+    addManhour(mode: $mode, owner: $owner, task: $task, type: $type, start_time: $start_time, hours: $hours, description: $description, customData: $customData) {
+      key
+    }
+  }
+`;
 const TASK_BY_NUMBER_QUERY = SEARCH_TASKS_QUERY;
 const RELATED_TASKS_QUERY = `
   query Task($key: Key) {
@@ -304,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);
@@ -312,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]),
@@ -320,7 +357,75 @@ function parseOnesWikiPageRoute(input) {
 	};
 }
 function isOnesWikiUrlInput(input) {
-	return input.includes("/wiki/");
+	return /\/wiki(?:\/|(?=[#?]|$))/.test(input);
+}
+function parseAuthorizeRequestId(location) {
+	try {
+		const parsed = new URL(location);
+		return parsed.searchParams.get("auth_request_id") ?? parsed.searchParams.get("id");
+	} catch {
+		const match = location.match(/[?&](?:auth_request_id|id)=([^&#]+)/);
+		return match?.[1] ? decodeURIComponent(match[1]) : null;
+	}
+}
+function parseAuthorizationCode(location) {
+	try {
+		return new URL(location).searchParams.get("code");
+	} catch {
+		const match = location.match(/[?&]code=([^&#]+)/);
+		return match?.[1] ? decodeURIComponent(match[1]) : null;
+	}
+}
+function parseDisplayId(input) {
+	const match = input.trim().match(/^([a-z]\w*)-(\d+)$/i);
+	if (!match?.[1] || !match[2]) return null;
+	return {
+		identifier: match[1],
+		number: Number.parseInt(match[2], 10)
+	};
+}
+function isValidOnesDate(value) {
+	if (!/^\d{4}-\d{2}-\d{2}$/.test(value)) return false;
+	const [yearText, monthText, dayText] = value.split("-");
+	const year = Number.parseInt(yearText, 10);
+	const month = Number.parseInt(monthText, 10);
+	const day = Number.parseInt(dayText, 10);
+	const date = new Date(Date.UTC(year, month - 1, day));
+	return date.getUTCFullYear() === year && date.getUTCMonth() === month - 1 && date.getUTCDate() === day;
+}
+function toOnesHours(hours) {
+	if (!Number.isFinite(hours) || hours <= 0) throw new Error("ONES: hours must be a positive number");
+	return Math.round(hours * 1e5);
+}
+function getTodayStartUnixSeconds() {
+	const now = /* @__PURE__ */ new Date();
+	const localStart = new Date(now.getFullYear(), now.getMonth(), now.getDate());
+	return Math.floor(localStart.getTime() / 1e3);
+}
+function toLocalDateString(date) {
+	return `${date.getFullYear()}-${String(date.getMonth() + 1).padStart(2, "0")}-${String(date.getDate()).padStart(2, "0")}`;
+}
+function getLocalStartUnixSeconds(year, month, day) {
+	return Math.floor(new Date(year, month - 1, day).getTime() / 1e3);
+}
+function parseManhourDate(input) {
+	const value = input?.trim();
+	if (!value) return {
+		date: null,
+		startTime: getTodayStartUnixSeconds()
+	};
+	const fullDateMatch = value.match(/^(\d{4})-(\d{2})-(\d{2})$/);
+	const dayOnlyMatch = value.match(/^(\d{1,2})号?$/);
+	const now = /* @__PURE__ */ new Date();
+	const year = fullDateMatch ? Number.parseInt(fullDateMatch[1], 10) : now.getFullYear();
+	const month = fullDateMatch ? Number.parseInt(fullDateMatch[2], 10) : now.getMonth() + 1;
+	const day = fullDateMatch ? Number.parseInt(fullDateMatch[3], 10) : dayOnlyMatch ? Number.parseInt(dayOnlyMatch[1], 10) : NaN;
+	const parsed = new Date(year, month - 1, day);
+	if (!(Number.isInteger(day) && parsed.getFullYear() === year && parsed.getMonth() === month - 1 && parsed.getDate() === day)) throw new Error("ONES: date must be a valid YYYY-MM-DD date or day of current month");
+	return {
+		date: toLocalDateString(parsed),
+		startTime: getLocalStartUnixSeconds(year, month, day)
+	};
 }
 function htmlToPlainText(html) {
 	return html.replace(/<br\s*\/?>/gi, "\n").replace(/<\/p>/gi, "\n").replace(/<[^>]+>/g, "").replace(/&nbsp;/g, " ").replace(/&amp;/g, "&").replace(/&lt;/g, "<").replace(/&gt;/g, ">").replace(/\n{3,}/g, "\n\n").trim();
@@ -376,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) {
@@ -516,32 +725,35 @@ var OnesAdapter = class extends BaseAdapter {
 			redirect: "manual"
 		})).headers.get("location");
 		if (!authorizeLocation) throw new Error("ONES: Authorize response missing location header");
-		const authRequestId = new URL(authorizeLocation).searchParams.get("id");
-		if (!authRequestId) throw new Error("ONES: Cannot parse auth_request_id from authorize redirect");
-		const finalizeRes = await fetch(`${baseUrl}/identity/api/auth_request/finalize`, {
-			method: "POST",
-			headers: {
-				"Content-Type": "application/json;charset=UTF-8",
-				"Cookie": cookies
-			},
-			body: JSON.stringify({
-				auth_request_id: authRequestId,
-				region_uuid: orgUser.region_uuid,
-				org_uuid: orgUser.org_uuid,
-				org_user_uuid: orgUser.org_user.org_user_uuid
-			})
-		});
-		if (!finalizeRes.ok) {
-			const text = await finalizeRes.text().catch(() => "");
-			throw new Error(`ONES: Finalize failed: ${finalizeRes.status} ${text}`);
+		let code = parseAuthorizationCode(authorizeLocation);
+		if (!code) {
+			const authRequestId = parseAuthorizeRequestId(authorizeLocation);
+			if (!authRequestId) throw new Error("ONES: Cannot parse auth_request_id from authorize redirect");
+			const finalizeRes = await fetch(`${baseUrl}/identity/api/auth_request/finalize`, {
+				method: "POST",
+				headers: {
+					"Content-Type": "application/json;charset=UTF-8",
+					"Cookie": cookies
+				},
+				body: JSON.stringify({
+					auth_request_id: authRequestId,
+					region_uuid: orgUser.region_uuid,
+					org_uuid: orgUser.org_uuid,
+					org_user_uuid: orgUser.org_user.org_user_uuid
+				})
+			});
+			if (!finalizeRes.ok) {
+				const text = await finalizeRes.text().catch(() => "");
+				throw new Error(`ONES: Finalize failed: ${finalizeRes.status} ${text}`);
+			}
+			const callbackLocation = (await fetch(`${baseUrl}/identity/authorize/callback?id=${authRequestId}&lang=zh`, {
+				method: "GET",
+				headers: { Cookie: cookies },
+				redirect: "manual"
+			})).headers.get("location");
+			if (!callbackLocation) throw new Error("ONES: Callback response missing location header");
+			code = parseAuthorizationCode(callbackLocation);
 		}
-		const callbackLocation = (await fetch(`${baseUrl}/identity/authorize/callback?id=${authRequestId}&lang=zh`, {
-			method: "GET",
-			headers: { Cookie: cookies },
-			redirect: "manual"
-		})).headers.get("location");
-		if (!callbackLocation) throw new Error("ONES: Callback response missing location header");
-		const code = new URL(callbackLocation).searchParams.get("code");
 		if (!code) throw new Error("ONES: Cannot parse authorization code from callback redirect");
 		const tokenRes = await fetch(`${baseUrl}/identity/oauth/token`, {
 			method: "POST",
@@ -611,11 +823,136 @@ 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`;
+		const response = await fetch(url, { headers: { Authorization: `Bearer ${session.accessToken}` } });
+		if (!response.ok) return null;
+		const found = ((await response.json()).datas?.task ?? []).map((item) => item.fields).find((fields) => fields?.uuid && fields.number === taskNumber);
+		if (!found?.uuid) return null;
+		return {
+			key: `task-${found.uuid}`,
+			uuid: found.uuid,
+			number: found.number ?? taskNumber,
+			name: found.summary ?? "",
+			status: {
+				uuid: "",
+				name: "",
+				category: void 0
+			},
+			issueType: found.issue_type_uuid || found.issue_type_name ? {
+				uuid: found.issue_type_uuid ?? "",
+				name: found.issue_type_name ?? ""
+			} : void 0,
+			project: found.project_uuid || found.project_name ? {
+				uuid: found.project_uuid ?? "",
+				name: found.project_name ?? ""
+			} : void 0
+		};
+	}
 	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() {
+		return (await this.graphql(PROJECTS_QUERY, {
+			projectOrderBy: {
+				isPin: "DESC",
+				namePinyin: "ASC",
+				createTime: "DESC"
+			},
+			projectFilterGroup: [{
+				visibleInProject_equal: true,
+				isArchive_equal: false
+			}],
+			groupBy: { projects: {} },
+			orderBy: null,
+			pagination: {
+				limit: 50,
+				after: "",
+				preciseCount: true
+			}
+		}, "projects-group-list-for-project-view")).data?.buckets?.flatMap((bucket) => bucket.projects ?? []) ?? [];
+	}
+	async findTaskByNumber(taskNumber, projectUuid) {
+		const filter = { number_in: [taskNumber] };
+		if (projectUuid) filter.project_in = [projectUuid];
+		const found = ((await this.graphql(TASK_BY_NUMBER_QUERY, {
+			groupBy: { tasks: {} },
+			groupOrderBy: null,
+			orderBy: { createTime: "DESC" },
+			filterGroup: [filter],
+			search: null,
+			pagination: {
+				limit: 10,
+				preciseCount: false
+			},
+			limit: 10
+		}, "group-task-data")).data?.buckets?.flatMap((b) => b.tasks ?? []) ?? []).find((task) => task.number === taskNumber && (!projectUuid || task.project?.uuid === projectUuid));
+		if (found) return found;
+		if (projectUuid) return null;
+		return this.searchTaskByNumber(taskNumber);
+	}
+	async resolveTaskRef(input) {
+		const taskId = input.trim();
+		if (!taskId) throw new Error("ONES: taskId is required");
+		const numMatch = taskId.match(/^#?(\d+)$/);
+		if (numMatch) {
+			const taskNumber = Number.parseInt(numMatch[1], 10);
+			const found = await this.findTaskByNumber(taskNumber);
+			if (!found) throw new Error(`ONES: Task #${taskNumber} not found in current team`);
+			return {
+				key: found.key ?? `task-${found.uuid}`,
+				uuid: found.uuid
+			};
+		}
+		const displayId = parseDisplayId(taskId);
+		if (displayId) {
+			const project = (await this.fetchProjects()).find((item) => item.identifier?.toLowerCase() === displayId.identifier.toLowerCase());
+			if (!project) throw new Error(`ONES: Project identifier "${displayId.identifier}" not found in current team`);
+			const found = await this.findTaskByNumber(displayId.number, project.uuid);
+			if (!found) throw new Error(`ONES: Task "${taskId}" not found in current team`);
+			return {
+				key: found.key ?? `task-${found.uuid}`,
+				uuid: found.uuid
+			};
+		}
+		const key = taskId.startsWith("task-") ? taskId : `task-${taskId}`;
+		return {
+			key,
+			uuid: key.slice(5)
+		};
+	}
 	async searchTeamUsers(keyword) {
 		const session = await this.login();
 		const url = `${this.config.apiBase}/project/api/project/team/${session.teamUuid}/users/search`;
@@ -804,27 +1141,11 @@ 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}");
-		let taskUuid = params.id;
-		const numMatch = taskUuid.match(/^#?(\d+)$/);
-		if (numMatch) {
-			const taskNumber = Number.parseInt(numMatch[1], 10);
-			const found = ((await this.graphql(TASK_BY_NUMBER_QUERY, {
-				groupBy: { tasks: {} },
-				groupOrderBy: null,
-				orderBy: { createTime: "DESC" },
-				filterGroup: [{ number_in: [taskNumber] }],
-				search: null,
-				pagination: {
-					limit: 10,
-					preciseCount: false
-				},
-				limit: 10
-			}, "group-task-data")).data?.buckets?.flatMap((b) => b.tasks ?? []) ?? []).find((t) => t.number === taskNumber);
-			if (!found) throw new Error(`ONES: Task #${taskNumber} not found in current team`);
-			taskUuid = found.uuid;
-		}
-		const task = (await this.graphql(TASK_DETAIL_QUERY, { key: `task-${taskUuid}` }, "Task")).data?.task;
-		if (!task) throw new Error(`ONES: Task "${taskUuid}" not found`);
+		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,
@@ -865,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");
@@ -895,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.
@@ -959,6 +1297,69 @@ var OnesAdapter = class extends BaseAdapter {
 			pageSize
 		};
 	}
+	async addManhour(params) {
+		const description = params.description.trim();
+		if (!description) throw new Error("ONES: description is required");
+		const taskRef = await this.resolveTaskRef(params.taskId);
+		const onesHours = toOnesHours(params.hours);
+		const workDate = parseManhourDate(params.date);
+		const key = (await this.graphql(ADD_MANHOUR_MUTATION, {
+			mode: "simple",
+			type: "recorded",
+			customData: {},
+			owner: (await this.login()).userUuid,
+			task: taskRef.uuid,
+			start_time: workDate.startTime,
+			hours: onesHours,
+			description
+		}, "add-manhour")).data?.addManhour?.key;
+		if (!key) throw new Error("ONES: Failed to add manhour");
+		return {
+			key,
+			taskUuid: taskRef.uuid,
+			hours: params.hours,
+			description,
+			date: workDate.date
+		};
+	}
+	async updateTaskPlanDates(params) {
+		const planStartDate = params.planStartDate?.trim();
+		const planEndDate = params.planEndDate?.trim();
+		if (!planStartDate && !planEndDate) throw new Error("ONES: planStartDate or planEndDate is required");
+		if (planStartDate && !isValidOnesDate(planStartDate)) throw new Error("ONES: planStartDate must be a valid YYYY-MM-DD date");
+		if (planEndDate && !isValidOnesDate(planEndDate)) throw new Error("ONES: planEndDate must be a valid YYYY-MM-DD date");
+		const taskRef = await this.resolveTaskRef(params.taskId);
+		const session = await this.login();
+		const fieldValues = [];
+		if (planStartDate) fieldValues.push({
+			field_uuid: "field027",
+			value: planStartDate
+		});
+		if (planEndDate) fieldValues.push({
+			field_uuid: "field028",
+			value: planEndDate
+		});
+		const response = await fetch(`${this.config.apiBase}/project/api/project/team/${session.teamUuid}/tasks/update3`, {
+			method: "POST",
+			headers: {
+				"Authorization": `Bearer ${session.accessToken}`,
+				"Content-Type": "application/json"
+			},
+			body: JSON.stringify({ tasks: [{
+				uuid: taskRef.uuid,
+				field_values: fieldValues
+			}] })
+		});
+		if (!response.ok) {
+			const text = await response.text().catch(() => "");
+			throw new Error(`ONES: Failed to update task plan dates: ${response.status} ${text}`);
+		}
+		return {
+			taskUuid: taskRef.uuid,
+			planStartDate: planStartDate ?? null,
+			planEndDate: planEndDate ?? null
+		};
+	}
 	async getRelatedIssues(params) {
 		const session = await this.login();
 		const taskKey = params.taskId.startsWith("task-") ? params.taskId : `task-${params.taskId}`;
@@ -1135,7 +1536,6 @@ var OnesAdapter = class extends BaseAdapter {
 		};
 	}
 };
-
 //#endregion
 //#region src/adapters/index.ts
 const ADAPTER_MAP = { ones: OnesAdapter };
@@ -1147,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", [
@@ -1305,7 +1704,41 @@ function loadConfig(startDir) {
 		configPath
 	};
 }
-
+//#endregion
+//#region src/tools/add-manhour.ts
+const AddManhourSchema = z.object({
+	taskId: z.string().min(1).describe("The task or requirement ID, key, number, or displayId (e.g. \"task-mock-uuid\", \"mock-uuid\", \"1001\", or \"DEMO-1001\")"),
+	hours: z.number().positive().describe("Work hours to record. Natural hours are converted to ONES internal units."),
+	description: z.string().min(1).describe("Work log description."),
+	date: z.string().optional().describe("Optional work date. Accepts YYYY-MM-DD, a day number like \"11\", or a day-of-month phrase like \"11号\"; day-only values use the current year and month."),
+	source: z.string().optional().describe("Source to update. If omitted, uses the default source.")
+});
+async function handleAddManhour(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(", ")}`);
+	return { content: [{
+		type: "text",
+		text: formatAddManhourResult(await adapter.addManhour({
+			taskId: input.taskId,
+			hours: input.hours,
+			description: input.description,
+			date: input.date
+		}))
+	}] };
+}
+function formatAddManhourResult(result) {
+	return [
+		"Added manhour.",
+		"",
+		`- **Key**: ${result.key}`,
+		`- **Task UUID**: ${result.taskUuid}`,
+		`- **Hours**: ${result.hours}`,
+		`- **Date**: ${result.date ?? "today"}`,
+		`- **Description**: ${result.description}`
+	].join("\n");
+}
 //#endregion
 //#region src/tools/get-issue-detail.ts
 const GetIssueDetailSchema = z.object({
@@ -1380,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({
@@ -1421,7 +1853,6 @@ function formatRelatedIssues(issues) {
 	}
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/tools/get-requirement.ts
 const GetRequirementSchema = z.object({
@@ -1500,7 +1931,6 @@ function formatRequirement(req) {
 	}
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/tools/get-testcases.ts
 const GetTestcasesSchema = z.object({
@@ -1552,7 +1982,6 @@ function formatTestcases(result) {
 	}
 	return lines.join("\n");
 }
-
 //#endregion
 //#region src/tools/list-sources.ts
 async function handleListSources(adapters, config) {
@@ -1578,7 +2007,6 @@ async function handleListSources(adapters, config) {
 		text: lines.join("\n")
 	}] };
 }
-
 //#endregion
 //#region src/tools/search-requirements.ts
 const SearchRequirementsSchema = z.object({
@@ -1619,7 +2047,39 @@ 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");
+const UpdateTaskPlanDatesSchema = z.object({
+	taskId: z.string().min(1).describe("The task or requirement ID, key, number, or displayId (e.g. \"task-mock-uuid\", \"mock-uuid\", \"1001\", or \"DEMO-1001\")"),
+	planStartDate: DateSchema.optional().describe("Plan start date in YYYY-MM-DD format."),
+	planEndDate: DateSchema.optional().describe("Plan end date in YYYY-MM-DD format."),
+	source: z.string().optional().describe("Source to update. If omitted, uses the default source.")
+});
+async function handleUpdateTaskPlanDates(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(", ")}`);
+	return { content: [{
+		type: "text",
+		text: formatUpdateTaskPlanDatesResult(await adapter.updateTaskPlanDates({
+			taskId: input.taskId,
+			planStartDate: input.planStartDate,
+			planEndDate: input.planEndDate
+		}))
+	}] };
+}
+function formatUpdateTaskPlanDatesResult(result) {
+	const lines = [
+		"Updated task plan dates.",
+		"",
+		`- **Task UUID**: ${result.taskUuid}`
+	];
+	if (result.planStartDate) lines.push(`- **Plan Start Date**: ${result.planStartDate}`);
+	if (result.planEndDate) lines.push(`- **Plan End Date**: ${result.planEndDate}`);
+	return lines.join("\n");
+}
 //#endregion
 //#region src/index.ts
 /**
@@ -1667,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) {
@@ -1680,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) {
@@ -1693,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) {
@@ -1706,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) {
@@ -1719,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) {
@@ -1732,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) {
@@ -1745,6 +2220,38 @@ async function main() {
 			};
 		}
 	});
+	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) {
+			return {
+				content: [{
+					type: "text",
+					text: `Error: ${err.message}`
+				}],
+				isError: true
+			};
+		}
+	});
+	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) {
+			return {
+				content: [{
+					type: "text",
+					text: `Error: ${err.message}`
+				}],
+				isError: true
+			};
+		}
+	});
 	const transport = new StdioServerTransport();
 	await server.connect(transport);
 }
@@ -1752,7 +2259,7 @@ main().catch((err) => {
 	console.error("[requirements-mcp] Fatal error:", err);
 	process.exit(1);
 });
-
 //#endregion
-export {  };
+export {};
+
 //# sourceMappingURL=index.mjs.map