@docyrus/docyrus 0.0.40 → 0.0.42

package/resources/pi-agent/extensions/todos.ts

@@ -65,6 +65,12 @@ const DEFAULT_TODO_SETTINGS = {
 };
 const LOCK_TTL_MS = 30 * 60 * 1000;
 
+interface ITodoCliEnvironment {
+	executable: string;
+	entryPath: string;
+	scope: "local" | "global";
+}
+
 interface TodoFrontMatter {
 	id: string;
 	title: string;
@@ -72,6 +78,7 @@ interface TodoFrontMatter {
 	status: string;
 	created_at: string;
 	assigned_to_session?: string;
+	parent_task_id?: string;
 }
 
 interface TodoRecord extends TodoFrontMatter {
@@ -111,6 +118,7 @@ const TodoParams = Type.Object({
   body: Type.Optional(
     Type.String({ description: "Long-form details (markdown). Update replaces; append adds." }),
   ),
+  parentTaskId: Type.Optional(Type.String({ description: "Optional canonical project task id for linked local subtasks" })),
   force: Type.Optional(Type.Boolean({ description: "Override another session's assignment" })),
 });
 
@@ -150,6 +158,71 @@ function formatTodoId(id: string): string {
   return `${TODO_ID_PREFIX}${id}`;
 }
 
+function readCliEnvironment(env: NodeJS.ProcessEnv = process.env): ITodoCliEnvironment {
+  const executable = env.DOCYRUS_CLI_EXECUTABLE?.trim();
+  const entryPath = env.DOCYRUS_CLI_ENTRY?.trim();
+  const scope = env.DOCYRUS_CLI_SCOPE?.trim() as "local" | "global" | undefined;
+
+  if (!executable || !entryPath || (scope !== "local" && scope !== "global")) {
+    throw new Error("Missing Docyrus CLI runtime env. Expected DOCYRUS_CLI_EXECUTABLE, DOCYRUS_CLI_ENTRY, and DOCYRUS_CLI_SCOPE.");
+  }
+
+  return {
+    executable,
+    entryPath,
+    scope,
+  };
+}
+
+async function runProjectPlanCliJson<TValue>(
+  environment: ITodoCliEnvironment,
+  pi: ExtensionAPI,
+  ctx: ExtensionContext,
+  args: string[],
+): Promise<TValue> {
+  const scopedArgs = environment.scope === "global" ? ["-g", ...args] : args;
+  const result = await pi.exec(environment.executable, [environment.entryPath, ...scopedArgs, "--json"], {
+    cwd: ctx.cwd,
+  });
+
+  const stdout = result.stdout?.toString().trim() || "";
+  const stderr = result.stderr?.toString().trim() || "";
+  const output = stdout || stderr;
+  if (result.code !== 0 || !output) {
+    throw new Error(output || `Command exited with code ${result.code ?? "unknown"}`);
+  }
+
+  return JSON.parse(output) as TValue;
+}
+
+async function validateParentTaskId(parentTaskId: string | undefined, pi: ExtensionAPI, ctx: ExtensionContext): Promise<string | undefined> {
+  const normalized = parentTaskId?.trim();
+  if (!normalized) {
+    return undefined;
+  }
+
+  const environment = readCliEnvironment();
+  const payload = await runProjectPlanCliJson<{
+    id?: string;
+    assignee?: string;
+  }>(environment, pi, ctx, [
+    "project-plan",
+    "get-task",
+    "--taskId",
+    normalized,
+  ]);
+
+  if (!payload.id) {
+    throw new Error(`Task "${normalized}" was not found.`);
+  }
+
+  if (payload.assignee !== "agent") {
+    throw new Error(`Task "${payload.id}" is assigned to "${payload.assignee}" and cannot own local agent subtasks.`);
+  }
+
+  return payload.id;
+}
+
 function normalizeTodoId(id: string): string {
   let trimmed = id.trim();
   if (trimmed.startsWith("#")) {
@@ -198,7 +271,8 @@ function sortTodos(todos: TodoFrontMatter[]): TodoFrontMatter[] {
 function buildTodoSearchText(todo: TodoFrontMatter): string {
   const tags = todo.tags.join(" ");
   const assignment = todo.assigned_to_session ? `assigned:${todo.assigned_to_session}` : "";
-  return `${formatTodoId(todo.id)} ${todo.id} ${todo.title} ${tags} ${todo.status} ${assignment}`.trim();
+  const parentTask = todo.parent_task_id ? `task:${todo.parent_task_id}` : "";
+  return `${formatTodoId(todo.id)} ${todo.id} ${todo.title} ${tags} ${todo.status} ${assignment} ${parentTask}`.trim();
 }
 
 function filterTodos(todos: TodoFrontMatter[], query: string): TodoFrontMatter[] {
@@ -375,6 +449,7 @@ class TodoSelectorComponent extends Container implements Focusable {
       const statusColor = closed ? "dim" : "success";
       const tagText = todo.tags.length ? ` [${todo.tags.join(", ")}]` : "";
       const assignmentText = renderAssignmentSuffix(this.theme, todo, this.currentSessionId);
+      const parentTaskText = todo.parent_task_id ? this.theme.fg("muted", ` ↳ ${todo.parent_task_id}`) : "";
       const line =
 				prefix +
 				this.theme.fg("accent", formatTodoId(todo.id)) +
@@ -382,6 +457,7 @@ class TodoSelectorComponent extends Container implements Focusable {
 				this.theme.fg(titleColor, todo.title || "(untitled)") +
 				this.theme.fg("muted", tagText) +
 				assignmentText +
+        parentTaskText +
 				" " +
 				this.theme.fg(statusColor, `(${todo.status || "open"})`);
       this.listContainer.addChild(new Text(line, 0, 0));
@@ -798,6 +874,7 @@ function parseFrontMatter(text: string, idFallback: string): TodoFrontMatter {
     status: "open",
     created_at: "",
     assigned_to_session: undefined,
+    parent_task_id: undefined,
   };
 
   const trimmed = text.trim();
@@ -813,6 +890,9 @@ function parseFrontMatter(text: string, idFallback: string): TodoFrontMatter {
     if (typeof parsed.assigned_to_session === "string" && parsed.assigned_to_session.trim()) {
       data.assigned_to_session = parsed.assigned_to_session;
     }
+    if (typeof parsed.parent_task_id === "string" && parsed.parent_task_id.trim()) {
+      data.parent_task_id = parsed.parent_task_id.trim();
+    }
     if (Array.isArray(parsed.tags)) {
       data.tags = parsed.tags.filter((tag): tag is string => typeof tag === "string");
     }
@@ -890,6 +970,7 @@ function parseTodoContent(content: string, idFallback: string): TodoRecord {
     status: parsed.status,
     created_at: parsed.created_at,
     assigned_to_session: parsed.assigned_to_session,
+    parent_task_id: parsed.parent_task_id,
     body: body ?? "",
   };
 }
@@ -903,6 +984,7 @@ function serializeTodo(todo: TodoRecord): string {
       status: todo.status,
       created_at: todo.created_at,
       assigned_to_session: todo.assigned_to_session || undefined,
+      parent_task_id: todo.parent_task_id || undefined,
     },
     null,
     2,
@@ -1039,6 +1121,7 @@ async function listTodos(todosDir: string): Promise<TodoFrontMatter[]> {
         status: parsed.status,
         created_at: parsed.created_at,
         assigned_to_session: parsed.assigned_to_session,
+        parent_task_id: parsed.parent_task_id,
       });
     } catch {
 			// ignore unreadable todo
@@ -1072,6 +1155,7 @@ function listTodosSync(todosDir: string): TodoFrontMatter[] {
         status: parsed.status,
         created_at: parsed.created_at,
         assigned_to_session: parsed.assigned_to_session,
+        parent_task_id: parsed.parent_task_id,
       });
     } catch {
 			// ignore
@@ -1107,7 +1191,8 @@ function renderAssignmentSuffix(
 
 function formatTodoHeading(todo: TodoFrontMatter): string {
   const tagText = todo.tags.length ? ` [${todo.tags.join(", ")}]` : "";
-  return `${formatTodoId(todo.id)} ${getTodoTitle(todo)}${tagText}${formatAssignmentSuffix(todo)}`;
+  const parentTaskText = todo.parent_task_id ? ` ↳ ${todo.parent_task_id}` : "";
+  return `${formatTodoId(todo.id)} ${getTodoTitle(todo)}${tagText}${formatAssignmentSuffix(todo)}${parentTaskText}`;
 }
 
 function buildRefinePrompt(todoId: string, title: string): string {
@@ -1248,6 +1333,7 @@ function renderTodoDetail(theme: Theme, todo: TodoRecord, expanded: boolean): st
     theme.fg("muted", `Status: ${getTodoStatus(todo)}`),
     theme.fg("muted", `Tags: ${tags}`),
     theme.fg("muted", `Created: ${createdAt}`),
+    ...(todo.parent_task_id ? [theme.fg("muted", `Parent task: ${todo.parent_task_id}`)] : []),
     "",
     theme.fg("muted", "Body:"),
     ...bodyLines.map((line) => theme.fg("text", `  ${line}`)),
@@ -1497,6 +1583,16 @@ export default function todosExtension(pi: ExtensionAPI) {
               details: { action: "create", error: "title required" },
             };
           }
+          let parentTaskId: string | undefined;
+          try {
+            parentTaskId = await validateParentTaskId(params.parentTaskId, pi, ctx);
+          } catch (error) {
+            const message = error instanceof Error ? error.message : String(error);
+            return {
+              content: [{ type: "text", text: message }],
+              details: { action: "create", error: message },
+            };
+          }
           await ensureTodosDir(todosDir);
           const id = await generateTodoId(todosDir);
           const filePath = getTodoPath(todosDir, id);
@@ -1506,6 +1602,7 @@ export default function todosExtension(pi: ExtensionAPI) {
             tags: params.tags ?? [],
             status: params.status ?? "open",
             created_at: new Date().toISOString(),
+            parent_task_id: parentTaskId,
             body: params.body ?? "",
           };
 
@@ -1559,6 +1656,9 @@ export default function todosExtension(pi: ExtensionAPI) {
             if (params.status !== undefined) {existing.status = params.status;}
             if (params.tags !== undefined) {existing.tags = params.tags;}
             if (params.body !== undefined) {existing.body = params.body;}
+            if (params.parentTaskId !== undefined) {
+              existing.parent_task_id = await validateParentTaskId(params.parentTaskId, pi, ctx);
+            }
             if (!existing.created_at) {existing.created_at = new Date().toISOString();}
             clearAssignmentIfClosed(existing);