feat: add auto-run continuation with epic context and auto-close (#56)

* feat(ticket): add auto-run continuation with epic context and auto-close

* feat(status-bar): add periodic polling to detect pr status changes

* docs: update changelog

Author: kostyay

CHANGELOG.md

@@ -95,6 +95,11 @@ All notable changes to agent-stuff are documented here.
 
 
 
+
+
+## feat/ticket-auto-run-continuation
+
+Auto-run continuation for ticket processing (#56): tickets can now be processed sequentially across epic boundaries with automatic context compaction, where the agent receives the next ready task after closing the current one without manual intervention. Status-bar PR polling now refreshes every 60 seconds to detect merge status changes in real-time. Parent epics auto-close when all child tasks are completed, with transition context automatically included when moving between epics to maintain narrative continuity. A new `/ticket-run-stop` command allows halting the auto-run loop gracefully after the current ticket finishes.
 
 ## feat/add-pr-extension-command
 

README.md

@@ -35,7 +35,7 @@ All extensions live in [`pi-extensions/`](pi-extensions). Each file is a self-co
 | [`git-rebase-master.ts`](pi-extensions/git-rebase-master.ts) | `/git-rebase-master` command — fetches latest main/master and rebases current branch with automatic LLM conflict resolution |
 | [`claude-import.ts`](pi-extensions/claude-import.ts) | Loads commands, skills, and agents from `.claude/` directories (project + global) and registers them as `/claude:*` commands |
 | [`kbrainstorm.ts`](pi-extensions/kbrainstorm.ts) | `ask_question` tool — interactive TUI for brainstorming with multiple-choice and freeform answers |
-| [`ticket/`](pi-extensions/ticket) | `ticket` tool — git-backed ticket tracker storing tickets as markdown files in `.tickets/` with hierarchy, dependencies, and status workflow |
+| [`ticket/`](pi-extensions/ticket) | `ticket` tool — git-backed ticket tracker storing tickets as markdown files in `.tickets/` with hierarchy, dependencies, status workflow, auto-closing of completed epics, and auto-run continuation across epics with context compaction |
 | [`loop.ts`](pi-extensions/loop.ts) | `/loop` command — runs a follow-up prompt loop with a breakout condition for iterative coding |
 | [`notify.ts`](pi-extensions/notify.ts) | Desktop notifications (OSC 777) when the agent finishes and is waiting for input |
 | [`op-timer.ts`](pi-extensions/op-timer.ts) | Live elapsed-time counter above the editor while the agent works — shows total operation and current tool execution duration |

pi-extensions/status-bar.ts

@@ -220,10 +220,11 @@ export default function statusBarExtension(pi: ExtensionAPI): void {
 	let diffStats: GitDiffStats | null = null;
 	let diffStatsTimer: ReturnType<typeof setTimeout> | null = null;
 
-	// PR status for current branch (refreshed on branch change)
+	// PR status for current branch (refreshed on branch change + every 60s)
 	let prStatus: PrStatus | null = null;
 	let prStatusTimer: ReturnType<typeof setTimeout> | null = null;
 	let prStatusBranch: string | null = null;
+	let prPollInterval: ReturnType<typeof setInterval> | null = null;
 
 	async function refreshDiffStats(): Promise<void> {
 		try {
@@ -280,14 +281,14 @@ export default function statusBarExtension(pi: ExtensionAPI): void {
 	}
 
 	/** Fetch PR status for the current branch via gh CLI. */
-	async function refreshPrStatus(branch?: string | null): Promise<void> {
+	async function refreshPrStatus(branch?: string | null, force = false): Promise<void> {
 		const currentBranch = branch ?? prStatusBranch;
 		if (!currentBranch || currentBranch === "main" || currentBranch === "master") {
 			prStatus = null;
 			prStatusBranch = currentBranch ?? null;
 			return;
 		}
-		if (currentBranch === prStatusBranch && prStatus !== null) return;
+		if (!force && currentBranch === prStatusBranch && prStatus !== null) return;
 		prStatusBranch = currentBranch;
 		try {
 			const result = await pi.exec(
@@ -310,6 +311,23 @@ export default function statusBarExtension(pi: ExtensionAPI): void {
 		prStatusTimer = setTimeout(() => refreshPrStatus(branch), 600);
 	}
 
+	/** Start a 60s polling interval to keep PR status up-to-date (e.g. detect merges). */
+	function startPrPoll(): void {
+		stopPrPoll();
+		prPollInterval = setInterval(async () => {
+			if (!prStatusBranch) return;
+			await refreshPrStatus(prStatusBranch, true);
+			tuiRef?.requestRender();
+		}, 60_000);
+	}
+
+	function stopPrPoll(): void {
+		if (prPollInterval) {
+			clearInterval(prPollInterval);
+			prPollInterval = null;
+		}
+	}
+
 	// ── Event handlers ─────────────────────────────────────────────────────
 
 	pi.on("message_start", async (event) => {
@@ -364,14 +382,15 @@ export default function statusBarExtension(pi: ExtensionAPI): void {
 
 		ctx.ui.setFooter((tui, theme, footerData) => {
 			refreshPrStatus(footerData.getGitBranch());
+			startPrPoll();
 			tuiRef = tui;
 			const unsub = footerData.onBranchChange(() => {
 				schedulePrRefresh(footerData.getGitBranch());
 				tui.requestRender();
 			});
 
 			return {
-				dispose() { unsub(); stopRenderTimer(); tuiRef = null; },
+				dispose() { unsub(); stopRenderTimer(); stopPrPoll(); tuiRef = null; },
 				invalidate() {},
 				render(width: number): string[] {
 					return renderFooter(width, ctx, theme, footerData);
@@ -391,6 +410,7 @@ export default function statusBarExtension(pi: ExtensionAPI): void {
 			currentBytesPerSec = 0;
 			isStreaming = false;
 			stopRenderTimer();
+			stopPrPoll();
 			refreshDiffStats();
 		}
 	});

pi-extensions/ticket/index.ts

@@ -10,7 +10,11 @@
  * - `ticket` tool with 9 actions (create, show, update, delete, start, close, reopen, list, add-note)
  * - `/ticket` TUI browser with fuzzy search and action menu
  * - `/ticket-create` prompt injection for epic + task breakdown
- * - `/ticket-run-all` automated ticket processing loop with session forking
+ * - `/ticket-run-all` automated ticket processing with auto-run continuation across epics
+ * - `/ticket-run-stop` to halt the auto-run loop after the current ticket
+ * - Auto-close of parent epics when all child tasks are closed
+ * - Auto-continuation: after closing a task, compacts context and sends next task prompt
+ * - Epic transition context: mentions the previously completed epic when moving to a new one
  * - Widget showing current in-progress ticket
  * - Status line with ticket counts
  * - Auto-nudge on agent_end when tickets remain in-progress
@@ -56,13 +60,18 @@ import {
 	LOCK_TTL_MS,
 	VALID_STATUSES,
 	VALID_TYPES,
+	allChildrenClosed,
+	buildAutoRunPrompt,
+	buildEpicContextLine,
 	buildRefinePrompt,
 	buildWorkPrompt,
+	clearAutoRun,
 	ensureDir,
 	filterTickets,
 	formatTicketLine,
 	garbageCollect,
 	generateId,
+	getChildren,
 	getLockPath,
 	getProjectPrefix,
 	getReadyTickets,
@@ -71,12 +80,14 @@ import {
 	isError,
 	listTickets,
 	listTicketsSync,
+	readAutoRun,
 	readSettings,
 	readTicketFile,
 	resolveId,
 	serializeForAgent,
 	serializeListForAgent,
 	statusIcon,
+	writeAutoRun,
 	writeTicketFile,
 } from "./ticket-core.ts";
 
@@ -581,12 +592,47 @@ export default function ticketExtension(pi: ExtensionAPI) {
 	pi.on("session_fork", async (_event, ctx) => refreshUI(ctx));
 	pi.on("session_tree", async (_event, ctx) => refreshUI(ctx));
 
-	// ── Auto-nudge ─────────────────────────────────────────────────────
+	// ── Auto-run continuation + nudge ──────────────────────────────────
 
 	pi.on("agent_end", async (_event, ctx) => {
 		const dir = getTicketsDir(ctx.cwd);
 		const tickets = await listTickets(dir);
 		const inProgress = tickets.filter((t) => t.status === "in_progress");
+
+		// Auto-run: when no tickets are in-progress, advance to the next ready task
+		const autoRun = await readAutoRun(dir);
+		if (autoRun?.active && !inProgress.length) {
+			const ready = getReadyTickets(tickets).filter((t) => t.status === "open" && t.type !== "epic");
+
+			if (!ready.length) {
+				await clearAutoRun(dir);
+				refreshUI(ctx);
+				pi.sendMessage({
+					customType: "ticket-run-all-done",
+					content: "🎉 All tickets processed! Auto-run complete.",
+					display: true,
+				});
+				return;
+			}
+
+			const nextTask = ready[0];
+			const nextRecord = await readTicketFile(getTicketPath(dir, nextTask.id), nextTask.id);
+
+			let prefix = "";
+			if (autoRun.lastCompletedEpicId) {
+				prefix += `✅ Previously completed epic ${autoRun.lastCompletedEpicId} "${autoRun.lastCompletedEpicTitle ?? ""}".\n\n`;
+			}
+			prefix += buildEpicContextLine(tickets, nextTask);
+
+			const prompt = prefix + buildAutoRunPrompt(nextTask, nextRecord, ready.length);
+
+			ctx.compact();
+			refreshUI(ctx);
+			pi.sendUserMessage(prompt);
+			return;
+		}
+
+		// Nudge: remind about in-progress tickets
 		if (!inProgress.length || nudgedThisCycle) return;
 
 		nudgedThisCycle = true;
@@ -681,6 +727,50 @@ export default function ticketExtension(pi: ExtensionAPI) {
 		return resolved;
 	}
 
+	/**
+	 * Auto-close the parent epic when all its children are closed.
+	 * Updates the auto-run state with the completed epic for transition prompts.
+	 * Returns the closed epic and child count, or undefined if nothing was closed.
+	 */
+	async function tryAutoCloseEpic(
+		dir: string,
+		closedTicket: TicketRecord,
+		ctx: ExtensionContext,
+	): Promise<{ epic: TicketRecord; childCount: number } | undefined> {
+		if (!closedTicket.parent) return undefined;
+
+		const allTickets = await listTickets(dir);
+		const parent = allTickets.find((t) => t.id === closedTicket.parent);
+		if (parent?.type !== "epic" || parent.status === "closed") return undefined;
+		if (!allChildrenClosed(allTickets, parent.id)) return undefined;
+
+		const epicResult = await withLock(dir, parent.id, ctx, async () => {
+			const fp = getTicketPath(dir, parent.id);
+			if (!existsSync(fp)) return { error: "not found" } as const;
+			const epic = await readTicketFile(fp, parent.id);
+			epic.status = "closed";
+			epic.assignee = undefined;
+			await writeTicketFile(fp, epic);
+			return epic;
+		});
+		if (isError(epicResult)) return undefined;
+
+		const epic = epicResult as TicketRecord;
+		const childCount = getChildren(allTickets, epic.id).length;
+
+		// Record completed epic in auto-run state for transition prompts
+		const autoRun = await readAutoRun(dir);
+		if (autoRun?.active) {
+			await writeAutoRun(dir, {
+				active: true,
+				lastCompletedEpicId: epic.id,
+				lastCompletedEpicTitle: epic.title,
+			});
+		}
+
+		return { epic, childCount };
+	}
+
 	// ── ticket tool ────────────────────────────────────────────────────
 
 	pi.registerTool({
@@ -863,8 +953,18 @@ export default function ticketExtension(pi: ExtensionAPI) {
 					});
 					if (isError(result)) return errorResult("close", result.error);
 
+					const closedTicket = result as TicketRecord;
+					const epicClose = await tryAutoCloseEpic(dir, closedTicket, ctx);
 					refreshUI(ctx);
-					return ticketResult("close", result as TicketRecord);
+
+					if (epicClose) {
+						const extra = `\n\n✅ Epic ${epicClose.epic.id} "${epicClose.epic.title}" auto-closed — all ${epicClose.childCount} tasks complete.`;
+						return {
+							content: [{ type: "text" as const, text: serializeForAgent(closedTicket) + extra }],
+							details: { action: "close", ticket: closedTicket },
+						};
+					}
+					return ticketResult("close", closedTicket);
 				}
 
 				// ── reopen ─────────────────────────────────────────────
@@ -1214,7 +1314,7 @@ export default function ticketExtension(pi: ExtensionAPI) {
 		handler: async (_args, ctx) => {
 			const dir = getTicketsDir(ctx.cwd);
 			const tickets = await listTickets(dir);
-			const ready = getReadyTickets(tickets).filter((t) => t.status === "open");
+			const ready = getReadyTickets(tickets).filter((t) => t.status === "open" && t.type !== "epic");
 
 			if (!ready.length) {
 				ctx.ui.notify("No ready tickets to process", "info");
@@ -1230,7 +1330,7 @@ export default function ticketExtension(pi: ExtensionAPI) {
 
 			// Ask user how to proceed
 			const forkOptions = [
-				"Fork a new session for each ticket",
+				"Auto-run: work through all, each task compacted",
 				"Work through all in this session",
 				"Cancel",
 			];
@@ -1239,7 +1339,7 @@ export default function ticketExtension(pi: ExtensionAPI) {
 			if (!forkChoice || forkChoice === "Cancel") return;
 
 			if (forkChoice === "Work through all in this session") {
-				// Inject prompt to work through all tickets
+				await clearAutoRun(dir);
 				const prompt =
 					`Work through these tickets in order, one at a time. For each ticket:\n` +
 					`1. Use \`ticket start <id>\` to begin\n` +
@@ -1251,29 +1351,43 @@ export default function ticketExtension(pi: ExtensionAPI) {
 				return;
 			}
 
-			// Fork-each mode: process first ticket, inject context for next
-			const firstTicket = ready[0];
-			const record = await readTicketFile(getTicketPath(dir, firstTicket.id), firstTicket.id);
+			// Auto-run mode: set marker, send first task, agent_end handles continuation
+			await writeAutoRun(dir, { active: true });
+
+			const firstTask = ready[0];
+			const record = await readTicketFile(getTicketPath(dir, firstTask.id), firstTask.id);
 
 			const prompt =
-				`Work on ticket ${firstTicket.id} "${firstTicket.title}".\n\n` +
-				(record.description ? `Description: ${record.description}\n\n` : "") +
-				(record.design ? `Design: ${record.design}\n\n` : "") +
-				(record.acceptance ? `Acceptance: ${record.acceptance}\n\n` : "") +
-				(record.tests ? `Tests: ${record.tests}\n\n` : "") +
-				`Steps:\n1. \`ticket start ${firstTicket.id}\`\n2. Implement the work\n3. \`ticket close ${firstTicket.id}\`${record.tests ? " (with tests_confirmed=true after verifying tests)" : ""}\n\n` +
-				`After closing, there are ${ready.length - 1} more tickets to process.`;
-
-			// Fork a new session, then send the prompt to trigger execution
+				buildEpicContextLine(tickets, firstTask) +
+				buildAutoRunPrompt(firstTask, record, ready.length) +
+				". Next task will be sent automatically after closing.";
+
 			const result = await ctx.newSession({
 				parentSession: ctx.sessionManager.getSessionFile(),
 			});
 
 			if (result.cancelled) {
+				await clearAutoRun(dir);
 				ctx.ui.notify("Session creation cancelled", "info");
 			} else {
 				pi.sendUserMessage(prompt);
 			}
 		},
 	});
+
+	// ── /ticket-run-stop command ────────────────────────────────────────
+
+	pi.registerCommand("ticket-run-stop", {
+		description: "Stop the auto-run loop after the current ticket",
+		handler: async (_args, ctx) => {
+			const dir = getTicketsDir(ctx.cwd);
+			const autoRun = await readAutoRun(dir);
+			if (!autoRun?.active) {
+				ctx.ui.notify("Auto-run is not active", "info");
+				return;
+			}
+			await clearAutoRun(dir);
+			ctx.ui.notify("Auto-run stopped. Current ticket will finish normally.", "info");
+		},
+	});
 }