pi-ui-extend 0.1.34 → 0.1.36

package/external/pi-tools-suite/src/todo/todo.ts

@@ -13,6 +13,7 @@
  */
 
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+import { resolve } from "node:path";
 import { TODO_TOOL_DESCRIPTION } from "../tool-descriptions.js";
 import {
 	disablePersistence,
@@ -25,7 +26,7 @@ import { AUTO_CLEAR_COMPLETED_MESSAGE, autoClearCompletedTodos } from "./state/a
 import { replayFromBranch } from "./state/replay.js";
 import { isTaskBlocked, selectTasksByStatus, selectTodoCounts } from "./state/selectors.js";
 import { applyTaskMutation } from "./state/state-reducer.js";
-import { commitState, getState, replaceState } from "./state/store.js";
+import { activateStateScope, commitState, getState, replaceState } from "./state/store.js";
 import { buildToolResult, formatContent } from "./tool/response-envelope.js";
 import {
 	COMMAND_NAME,
@@ -75,7 +76,16 @@ const PERSIST_ARGUMENT_COMPLETIONS: CommandCompletion[] = [
 ];
 
 interface TodoToolHooks {
-	afterCommit?: (state: ReturnType<typeof getState>, ctx: ExtensionContext, info: { action: TaskAction; params: TaskMutationParams }) => void | Promise<void>;
+	prepareMutation?: (
+		state: ReturnType<typeof getState>,
+		ctx: ExtensionContext,
+		info: { action: TaskAction; params: TaskMutationParams },
+	) => TaskMutationParams | Promise<TaskMutationParams>;
+	afterCommit?: (
+		state: ReturnType<typeof getState>,
+		ctx: ExtensionContext,
+		info: { action: TaskAction; params: TaskMutationParams; committedState: ReturnType<typeof getState> },
+	) => void | Promise<void>;
 }
 
 interface TodoToolRegistrationOptions extends TodoToolHooks {
@@ -83,7 +93,7 @@ interface TodoToolRegistrationOptions extends TodoToolHooks {
 	promptGuidelines?: string[];
 }
 
-type TodoStateEventContext = { sessionManager?: { getSessionFile?: () => unknown } };
+type TodoStateEventContext = { sessionManager?: { getSessionFile?: () => unknown; getSessionId?: () => unknown } };
 type TodoStateEventEmitter = { events?: { emit?: (channel: string, data: unknown) => void } };
 
 interface TodosCommandOptions {
@@ -183,14 +193,34 @@ function sessionFileFromContext(ctx: unknown): string | undefined {
 	return typeof sessionFile === "string" && sessionFile.trim() ? sessionFile : undefined;
 }
 
+function sessionIdFromContext(ctx: unknown): string | undefined {
+	const sessionId = (ctx as TodoStateEventContext | undefined)?.sessionManager?.getSessionId?.();
+	return typeof sessionId === "string" && sessionId.trim() ? sessionId : undefined;
+}
+
+function todoStateScopeFromContext(ctx: unknown): string | undefined {
+	const sessionFile = sessionFileFromContext(ctx);
+	if (sessionFile) return `file:${resolve(sessionFile)}`;
+	const sessionId = sessionIdFromContext(ctx);
+	if (sessionId) return `id:${sessionId}`;
+	const cwd = (ctx as { cwd?: unknown } | undefined)?.cwd;
+	return typeof cwd === "string" && cwd.trim() ? `cwd:${resolve(cwd)}` : undefined;
+}
+
+export function activateTodoStateScope(ctx: unknown): void {
+	activateStateScope(todoStateScopeFromContext(ctx));
+}
+
 export function publishTodoState(
 	pi: TodoStateEventEmitter,
 	ctx: unknown,
 	action: TaskAction = "list",
 	params: Record<string, unknown> = {},
 ): void {
+	activateTodoStateScope(ctx);
 	const state = getState();
 	const sessionFile = sessionFileFromContext(ctx);
+	const sessionId = sessionIdFromContext(ctx);
 	pi.events?.emit?.(TODO_STATE_EVENT, {
 		version: 1,
 		details: {
@@ -200,6 +230,7 @@ export function publishTodoState(
 			nextId: state.nextId,
 		},
 		...(sessionFile ? { sessionFile } : {}),
+		...(sessionId ? { sessionId } : {}),
 		checkedAt: Date.now(),
 	});
 }
@@ -363,15 +394,24 @@ export function registerTodoTool(pi: ExtensionAPI, hooks: TodoToolRegistrationOp
 		parameters: TodoParamsSchema,
 
 		async execute(_toolCallId, params, _signal, _onUpdate, _ctx) {
-			const result = applyTaskMutation(getState(), params.action, params as TaskMutationParams);
+			activateTodoStateScope(_ctx);
+			const preparedParams = await hooks.prepareMutation?.(getState(), _ctx as ExtensionContext, {
+				action: params.action,
+				params: params as TaskMutationParams,
+			}) ?? params as TaskMutationParams;
+			const result = applyTaskMutation(getState(), params.action, preparedParams);
 			if (result.op.kind === "error") {
 				throw new Error(result.op.message);
 			}
 			const autoClear = autoClearCompletedTodos(result.state);
 			commitState(autoClear.state);
 			publishTodoState(pi as TodoStateEventEmitter, _ctx, params.action, params as Record<string, unknown>);
-			await hooks.afterCommit?.(autoClear.state, _ctx as ExtensionContext, { action: params.action, params: params as TaskMutationParams });
-			const toolResult = buildToolResult(params.action, params as TaskMutationParams, autoClear.state, result.op);
+			await hooks.afterCommit?.(result.state, _ctx as ExtensionContext, {
+				action: params.action,
+				params: preparedParams,
+				committedState: autoClear.state,
+			});
+			const toolResult = buildToolResult(params.action, preparedParams, autoClear.state, result.op);
 			if (!autoClear.cleared) return toolResult;
 			return {
 				...toolResult,
@@ -391,6 +431,7 @@ export function registerTodosCommand(pi: ExtensionAPI): void {
 	description: "Show todos on the current branch. Flags: --active, --ready, --blocked, --tree, --status <status>, --export [json|markdown]. Commands: persist on|off|clear|status, scope <id...>",
 		getArgumentCompletions: (prefix) => completeCommandArguments(String(prefix ?? ""), TODOS_ARGUMENT_COMPLETIONS),
 		handler: async (args, ctx) => {
+			activateTodoStateScope(ctx);
 			if (handlePersistCommand(args, ctx)) return;
 			if (handleScopeCommand(args, ctx, () => publishTodoState(pi as TodoStateEventEmitter, ctx))) return;
 			if (!ctx.hasUI) {
@@ -459,6 +500,7 @@ export function registerTodosCommand(pi: ExtensionAPI): void {
 		description: "Enable, disable, clear, or inspect project todo persistence. Args: on|off|clear|status.",
 		getArgumentCompletions: (prefix) => completeCommandArguments(String(prefix ?? ""), PERSIST_ARGUMENT_COMPLETIONS),
 		handler: async (args, ctx) => {
+			activateTodoStateScope(ctx);
 			handlePersistCommand(`persist ${getCommandText(args)}`, ctx);
 		},
 	});
@@ -466,6 +508,7 @@ export function registerTodosCommand(pi: ExtensionAPI): void {
 	pi.registerCommand(SCOPE_COMMAND_NAME, {
 		description: "Select todo ids to continue from a persisted plan; pending/in_progress items outside the scope become deferred.",
 		handler: async (args, ctx) => {
+			activateTodoStateScope(ctx);
 			handleScopeCommand(`scope ${getCommandText(args)}`, ctx, () => publishTodoState(pi as TodoStateEventEmitter, ctx));
 		},
 	});

package/external/pi-tools-suite/src/tool-descriptions.ts

@@ -27,10 +27,10 @@ export const COMPRESS_TOOL_DESCRIPTION: ToolDescription = {
 	name: "compress",
 	label: "Compress Context",
 	description: COMPRESS_RANGE_DESCRIPTION,
-	promptSnippet: "Compress closed conversation slices as steady context housekeeping; after implementation + verification, compress before replying or starting a new task unless exact raw logs/diffs are still needed.",
+	promptSnippet: "Compress closed conversation slices as steady context housekeeping; before any new search/read/test/web lookup, compress older completed work unless exact raw text is still needed.",
 	promptGuidelines: [
-		"Treat completed implementation + verification slices as compression candidates immediately, not only at context-limit warnings.",
-		"Do not carry large stale tool outputs, diffs, or exploration logs across task boundaries; summarize their actionable result instead.",
+		"Treat completed implementation, verification, config/doc edits, answered exploration, dead ends, and log inspection as compression candidates immediately.",
+		"Do not carry large stale shell/read/repo/web outputs, diffs, or passing logs across task boundaries; summarize their actionable result instead.",
 		"Keep active, still-needed context raw; compress only closed ranges whose exact content is unlikely to be needed next.",
 	],
 };
@@ -329,7 +329,7 @@ export const CODEX_ALIAS_TOOL_DESCRIPTIONS = {
 	shellCommand: {
 		name: "shell",
 		label: "shell",
-		description: "Run shell commands for builds, tests, package managers, git, and project CLIs. Prefer tail for long test output. Set workdir/cwd instead of cd; prefer read for simple file reads.",
+		description: "Run shell commands for builds, tests, package managers, git, and project CLIs. For long/verification output, redirect to a log and show only bounded tail on failure; summarize passing logs. Set workdir/cwd instead of cd; prefer read for simple file reads.",
 	},
 	applyPatch: {
 		name: "apply_patch",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-ui-extend",
-  "version": "0.1.34",
+  "version": "0.1.36",
   "private": false,
   "type": "module",
   "bin": {
@@ -43,12 +43,14 @@
     "watch:pix": "npm run link:pix --silent && mise exec node@24.16.0 -- tsc -p tsconfig.json --watch --preserveWatchOutput --noEmitOnError",
     "check": "mise exec node@24.16.0 -- npm run check:inner",
     "precheck:inner": "npm run generate-schemas --silent",
-    "check:inner": "tsc --noEmit && npm run test:inner",
+    "check:inner": "npm run sync:sdk-pin:check --silent && tsc --noEmit && npm run test:inner",
     "test": "mise exec node@24.16.0 -- npm run test:inner",
     "test:inner": "node --import tsx --test \"tests/**/*.test.ts\"",
     "test:tools-suite": "npm --prefix external/pi-tools-suite test",
     "sync:pi-tools-suite": "node scripts/sync-pi-tools-suite.mjs",
     "sync:pi-tools-suite:check": "node scripts/sync-pi-tools-suite.mjs --check",
+    "sync:sdk-pin": "node scripts/sync-sdk-pin.mjs",
+    "sync:sdk-pin:check": "node scripts/sync-sdk-pin.mjs --check",
     "test:coverage": "mise exec node@24.16.0 -- node --import tsx --test --experimental-test-coverage --test-coverage-include=src/**/*.ts --test-coverage-exclude=src/main.ts --test-coverage-lines=95 --test-coverage-branches=80 --test-coverage-functions=95 \"tests/**/*.test.ts\"",
     "update-sdk-references": "mise exec node@24.16.0 -- node .pi/skills/pi-sdk/scripts/update-references.mjs",
     "prepack": "npm run generate-schemas --silent",
@@ -62,9 +64,9 @@
     "prepublishOnly": "npm run check && npm run build:pix && npm run generate-schemas"
   },
   "dependencies": {
-    "@earendil-works/pi-ai": "0.79.3",
-    "@earendil-works/pi-coding-agent": "0.79.3",
-    "@earendil-works/pi-tui": "0.79.3",
+    "@earendil-works/pi-ai": "0.79.4",
+    "@earendil-works/pi-coding-agent": "0.79.4",
+    "@earendil-works/pi-tui": "0.79.4",
     "@mariozechner/clipboard": "^0.3.9",
     "jsonc-parser": "3.3.1",
     "typebox": "1.1.38",