pi-harness-runtime 1.1.303 → 1.1.304

package/index.ts

@@ -916,13 +916,133 @@ interface PlanEvent {
 	fileStableSince: number;
 }
 
+interface PendingPINGPONGAction {
+	token: string;
+	sessionId: string;
+	action: "code" | "review" | "code_again" | "done";
+	projectRoot: string;
+	allowedStage: string;
+	expiresAt: number;
+	used: boolean;
+}
+
 const codexSessionState = {
 	deliveredEvents: new Map<string, PlanEvent>(),
 	currentCodexSessionId: null as string | null,
 	currentProjectRoot: null as string | null,
 	currentPlanCandidate: null as { ordinal: number; text: string; timestamp: number } | null,
+	pendingActions: new Map<string, PendingPINGPONGAction>(),
 };
 
+// PING_PONG callback action type (different from generic APPROVE/REJECT)
+const PING_PONG_ACTIONS = {
+	CODE: "ping_code",
+	REVIEW: "ping_review",
+	CODE_AGAIN: "ping_code_again",
+	DONE: "ping_done",
+} as const;
+
+// Generate opaque token for PING_PONG action
+function createPINGPONGToken(): string {
+	const array = new Uint8Array(16);
+	crypto.getRandomValues(array);
+	return Array.from(array, b => b.toString(16).padStart(2, "0")).join("");
+}
+
+// Create a pending PING_PONG action and return its token (used when sending button notifications)
+function _createPendingAction(
+	sessionId: string,
+	action: "code" | "review" | "code_again" | "done",
+	projectRoot: string,
+	allowedStage: string
+): string {
+	const token = createPINGPONGToken();
+	codexSessionState.pendingActions.set(token, {
+		token,
+		sessionId,
+		action,
+		projectRoot,
+		allowedStage,
+		expiresAt: Date.now() + 30 * 60 * 1000, // 30 min expiry
+		used: false,
+	});
+	return token;
+}
+
+// Find pending action by session ID (for /task fallback)
+function findPendingActionBySession(sessionId: string): PendingPINGPONGAction | undefined {
+	for (const action of codexSessionState.pendingActions.values()) {
+		if (action.sessionId === sessionId && !action.used && Date.now() < action.expiresAt) {
+			return action;
+		}
+	}
+	return undefined;
+}
+
+// Advance Codex task - shared by both button callbacks and /task commands
+async function advanceCodexTask(
+	token: string,
+	action: "code" | "review" | "code_again" | "done",
+	pi: ExtensionAPI
+): Promise<{ success: boolean; error?: string }> {
+	const pending = codexSessionState.pendingActions.get(token);
+	
+	// Validate pending action
+	if (!pending) {
+		return { success: false, error: "Action expired or not found" };
+	}
+	if (pending.used) {
+		return { success: false, error: "Action already used" };
+	}
+	if (Date.now() > pending.expiresAt) {
+		codexSessionState.pendingActions.delete(token);
+		return { success: false, error: "Action expired" };
+	}
+	if (pending.sessionId !== codexSessionState.currentCodexSessionId) {
+		return { success: false, error: "Session no longer active" };
+	}
+	if (pending.projectRoot !== codexSessionState.currentProjectRoot) {
+		return { success: false, error: "Project root mismatch" };
+	}
+	
+	// Mark as used
+	pending.used = true;
+	
+	// Find the event
+	const event = Array.from(codexSessionState.deliveredEvents.values()).find(
+		e => e.codexSessionId === pending.sessionId
+	);
+	if (!event) {
+		return { success: false, error: "Session event not found" };
+	}
+	
+	// Check status for code action
+	if (action === "code" && event.status !== "manual_wait" && event.status !== "ready") {
+		return { success: false, error: `Plan not ready (status=${event.status})` };
+	}
+	
+	// Execute action
+	if (action === "code" || action === "code_again") {
+		await pi.sendUserMessage(
+			`Human approved. Proceed to implement: ${event.planPath}`,
+			{ deliverAs: "steer" }
+		);
+		event.status = "triggered";
+		logToFile("[codex-trigger]", `Advanced to code: ${event.planPath}`);
+	} else if (action === "review") {
+		await pi.sendUserMessage(
+			"Human requests a review of the implementation.",
+			{ deliverAs: "steer" }
+		);
+		logToFile("[codex-trigger]", "Advanced to review");
+	} else if (action === "done") {
+		event.status = "done";
+		logToFile("[codex-trigger]", "Human marked done");
+	}
+	
+	return { success: true };
+}
+
 async function initTelegram(pi: ExtensionAPI): Promise<void> {
 	addHarnessMessage("Telegram", "initializing", "wait");
 	try {
@@ -986,14 +1106,37 @@ async function initTelegram(pi: ExtensionAPI): Promise<void> {
 			return;
 		}
 
-		// Register callback handler for inline keyboard button clicks (approve/reject)
+		// Register callback handler for inline keyboard button clicks
 		const { parseCallbackData, CallbackActions } = await import(
-				"./packages/notification/dist/telegram-webhook-handler.js"
+			"./packages/notification/dist/telegram-webhook-handler.js"
 		);
-		center.setCallbackHandler(async (data: string, _query: { id: string }) => {
+		center.setCallbackHandler(async (data: string, _query: { id: string; from: { id: number } }) => {
 			const parsed = parseCallbackData(data);
 			logToFile("[TelegramCallback]", `action=${parsed.action} target=${parsed.targetId}`);
 
+			// Handle PING_PONG button callbacks
+			if (parsed.action === PING_PONG_ACTIONS.CODE || 
+				parsed.action === PING_PONG_ACTIONS.REVIEW ||
+				parsed.action === PING_PONG_ACTIONS.CODE_AGAIN ||
+				parsed.action === PING_PONG_ACTIONS.DONE) {
+				// Map action string to internal action type
+				const actionMap: Record<string, "code" | "review" | "code_again" | "done"> = {
+					[PING_PONG_ACTIONS.CODE]: "code",
+					[PING_PONG_ACTIONS.REVIEW]: "review",
+					[PING_PONG_ACTIONS.CODE_AGAIN]: "code_again",
+					[PING_PONG_ACTIONS.DONE]: "done",
+				};
+				const internalAction = actionMap[parsed.action];
+				
+				const result = await advanceCodexTask(parsed.targetId, internalAction, pi);
+				if (result.success) {
+					logToFile("[TelegramCallback]", `PING_PONG ${internalAction} succeeded`);
+				} else {
+					logToFile("[TelegramCallback]", `PING_PONG ${internalAction} failed: ${result.error}`);
+				}
+				return; // Don't fall through to generic handlers
+			}
+
 			switch (parsed.action) {
 				case CallbackActions.APPROVE:
 					// User clicked Continue - clear waiting state and resume continuation
@@ -1021,52 +1164,38 @@ async function initTelegram(pi: ExtensionAPI): Promise<void> {
 			if (taskCommandMatch) {
 				const [, sessionId, action] = taskCommandMatch;
 				
-				// Find matching Codex session
-				const matchingSession = codexSessionState.deliveredEvents.get(sessionId) || 
-					Array.from(codexSessionState.deliveredEvents.values()).find(e => e.codexSessionId === sessionId);
+				// Map action string to internal type
+				const actionMap: Record<string, "code" | "review" | "code_again" | "done"> = {
+					code: "code",
+					review: "review",
+					done: "done",
+				};
+				const internalAction = actionMap[action.toLowerCase()];
+				if (!internalAction) {
+					logToFile("[codex-trigger]", `Unknown /task action: ${action}`);
+					return;
+				}
 				
-				// Validate session matches current project
-				if (!matchingSession || matchingSession.codexSessionId !== codexSessionState.currentCodexSessionId) {
-					logToFile("[codex-trigger]", `Unknown session or session not active: ${sessionId}`);
+				// For /task, find pending action by session ID (buttons use token directly)
+				const pendingAction = findPendingActionBySession(sessionId);
+				if (!pendingAction) {
+					logToFile("[codex-trigger]", `No pending action found for session: ${sessionId}`);
 					return;
 				}
 				
-				// For code/review commands, check Pi cwd matches project root
-				if (action.toLowerCase() === "code" || action.toLowerCase() === "review") {
+				// Validate cwd for code/review
+				if (internalAction === "code" || internalAction === "review") {
 					if (_piSessionCwd && codexSessionState.currentProjectRoot && 
 					    _piSessionCwd !== codexSessionState.currentProjectRoot) {
-						logToFile("[codex-trigger]", `Cwd mismatch for /task ${action}: Pi cwd=${_piSessionCwd} vs project=${codexSessionState.currentProjectRoot}`);
-						return;
-					}
-					
-					// For code command, require manual_wait or ready status
-					if (action.toLowerCase() === "code" && 
-					    matchingSession.status !== "manual_wait" && matchingSession.status !== "ready") {
-						logToFile("[codex-trigger]", `Cannot /task code: plan not ready (status=${matchingSession.status})`);
+						logToFile("[codex-trigger]", `Cwd mismatch: Pi cwd=${_piSessionCwd} vs project=${codexSessionState.currentProjectRoot}`);
 						return;
 					}
 				}
 				
-				logToFile("[codex-trigger]", `Human command: ${action} for session ${sessionId}`);
-				
-				if (action.toLowerCase() === "code") {
-					// Advance to code stage - send a message to Pi
-					await pi.sendUserMessage(
-						`Human approved the plan. Proceed to implement: ${matchingSession.planPath}`,
-						{ deliverAs: "steer" }
-					);
-					matchingSession.status = "triggered";
-					logToFile("[codex-trigger]", "Triggered Pi for plan:", matchingSession.planPath);
-				} else if (action.toLowerCase() === "review") {
-					// Advance to review stage
-					await pi.sendUserMessage(
-						"Human requests a review of the implementation.",
-						{ deliverAs: "steer" }
-					);
-				} else if (action.toLowerCase() === "done") {
-					// Mark as done
-					matchingSession.status = "done";
-					logToFile("[codex-trigger]", "Human marked session done:", sessionId);
+				logToFile("[codex-trigger]", `Human command: /task ${sessionId} ${action}`);
+				const result = await advanceCodexTask(pendingAction.token, internalAction, pi);
+				if (!result.success) {
+					logToFile("[codex-trigger]", `/task failed: ${result.error}`);
 				}
 				return;
 			}
@@ -2435,34 +2564,51 @@ export default function (pi: ExtensionAPI) {
 							}
 				}
 
-				// Emit Telegram alert with full content
+				// Emit Telegram alert with full content (async, non-blocking)
 				const planPath = codexSessionState.currentProjectRoot ? findPlanFile(codexSessionState.currentProjectRoot) : undefined;
 
+				// Create pending actions for PING_PONG buttons
+				const sessionId = codexSessionState.currentCodexSessionId || "unknown";
+				const projectRoot = codexSessionState.currentProjectRoot || "";
+				
+				// Create tokens for each action
+				const codeToken = _createPendingAction(sessionId, "code", projectRoot, "plan");
+				const doneToken = _createPendingAction(sessionId, "done", projectRoot, "plan");
+				
 				// Use event type based on stage
 				if (hasReviewMarker) {
-							nc("CodexReviewDetected", {
-									requirement: msg.text,
-										taskTitle: codexSessionState.currentCodexSessionId || `Ordinal ${msg.ordinal}`,
-											details: {
-												ordinal: msg.ordinal,
-												stage: "review",
-												projectRoot: codexSessionState.currentProjectRoot || undefined,
-											messageLength: msg.text.length,
-										},
-								});
-							} else {
-								nc("CodexPlanDetected", {
-								requirement: msg.text, // Full text, no truncation
-								taskTitle: codexSessionState.currentCodexSessionId || `Ordinal ${msg.ordinal}`,
-								details: {
-								ordinal: msg.ordinal,
-									stage: "plan",
-									projectRoot: codexSessionState.currentProjectRoot || undefined,
-											planPath: planPath || undefined,
-											messageLength: msg.text.length,
-									},
-								});
-							}
+					// Review stage: buttons are Review and Done
+					const reviewToken = _createPendingAction(sessionId, "review", projectRoot, "review");
+					void nc.center.notifyWithButtons("CodexReviewDetected", {
+						requirement: msg.text,
+						taskTitle: sessionId,
+						details: {
+							ordinal: msg.ordinal,
+							stage: "review",
+							projectRoot,
+							messageLength: msg.text.length,
+					},
+					}, [
+						{ text: "Review", action: PING_PONG_ACTIONS.REVIEW, targetId: reviewToken },
+						{ text: "Done", action: PING_PONG_ACTIONS.DONE, targetId: doneToken },
+					]);
+				} else {
+					// Plan stage: buttons are Code and Done
+					void nc.center.notifyWithButtons("CodexPlanDetected", {
+						requirement: msg.text, // Full text, no truncation
+						taskTitle: sessionId,
+						details: {
+							ordinal: msg.ordinal,
+							stage: "plan",
+							projectRoot,
+							planPath: planPath || undefined,
+							messageLength: msg.text.length,
+					},
+					}, [
+						{ text: "Code", action: PING_PONG_ACTIONS.CODE, targetId: codeToken },
+						{ text: "Done", action: PING_PONG_ACTIONS.DONE, targetId: doneToken },
+					]);
+				}
 				}
 			},
 		});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-harness-runtime",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "[BETA] Codex-style /usage status + autonomous coding harness for pi. Not production ready — expect breaking changes.",
   "type": "module",
   "scripts": {

package/packages/a2a-adapter/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/a2a-adapter",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "description": "Google Agent-to-Agent (A2A) protocol adapter for pi-harness",
   "main": "dist/index.js",

package/packages/agent-policy/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/agent-policy",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Agent policy runtime for pi-harness - enforces mandatory rule delivery and mutation gating",
   "type": "module",
   "main": "./dist/index.js",

package/packages/architecture-generator/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/architecture-generator",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Generate ADR and Mermaid diagrams from project analysis",
   "type": "module",
   "main": "./dist/index.js",

package/packages/auth/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/auth",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "pi-harness-runtime — auth",
   "type": "module",
   "main": "./dist/index.js",

package/packages/autonomous-refactor/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/autonomous-refactor",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "description": "Autonomous refactoring engine",
   "main": "dist/index.js",

package/packages/autonomous-runtime/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/autonomous-runtime",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Autonomous Operations Runtime — task inbox, lease management, and worker lifecycle (RFC-0101)",
   "type": "module",
   "main": "./dist/index.js",

package/packages/cache-strategy/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/cache-strategy",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "description": "Smart caching with TTL and invalidation strategies (RFC-0022)",
   "main": "dist/index.js",

package/packages/capability-registry/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/capability-registry",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Capability Registry for pi-harness-runtime — tracks model capabilities and enables capability-based routing",
   "type": "module",
   "main": "./dist/index.js",

package/packages/checkpoint/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/checkpoint",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Enhanced checkpoint system with incremental snapshots, diff-based storage, and recovery strategies",
   "type": "module",
   "main": "./dist/index.js",

package/packages/cli-plugin-sdk/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/cli-plugin-sdk",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "description": "CLI Plugin SDK for pi-harness — install, list, remove, update, invoke plugins",
   "main": "dist/index.js",

package/packages/clipboard/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/clipboard",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "main": "src/index.ts",
   "exports": {

package/packages/clipboard-plugin/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/clipboard-plugin",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "main": "src/index.ts",
   "exports": {

package/packages/code-generation/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/code-generation",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Code generation pipeline for pi-harness-runtime",
   "type": "module",
   "main": "./dist/index.js",

package/packages/code-review/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/code-review",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Automated code review engine for pi-harness-runtime",
   "type": "module",
   "main": "./dist/index.js",

package/packages/codex-adapter/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/codex-adapter",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "description": "OpenAI Codex provider adapter for pi-harness",
   "main": "dist/index.js",

package/packages/codex-watcher/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/codex-watcher",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Watches Codex CLI sessions and emits visible messages for PING-PONG monitoring",
   "type": "module",
   "main": "./dist/src/index.js",

package/packages/config-capture/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@moocoding/config-capture",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Auto-capture API-fetched configuration values to AGENTS.md",
   "type": "module",
   "main": "src/index.ts",

package/packages/context-compiler/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/context-compiler",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Bounded context selection engine for pi-harness-runtime",
   "type": "module",
   "main": "./dist/index.js",

package/packages/context-discovery/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/context-discovery",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "main": "src/index.ts",
   "exports": {

package/packages/context-manager/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/context-manager",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Context Window Manager — prevents model context overflow via compaction and resume prompts (RFC-0010)",
   "type": "module",
   "main": "./dist/index.js",

package/packages/cookie-sanitizer/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/cookie-sanitizer",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "pi-harness-runtime — cookie sanitizer (provider-quota cookie file normaliser)",
   "type": "module",
   "main": "./dist/index.js",

package/packages/cost-optimizer/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/cost-optimizer",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Cost Optimizer for pi-harness-runtime — budget tracking, forecasting, and model selection optimization",
   "type": "module",
   "main": "./dist/index.js",

package/packages/dependency-analyzer/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/dependency-analyzer",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "main": "src/index.ts",
   "exports": {

package/packages/django-plugin/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/django-plugin",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Django framework analysis — models, management commands, settings",
   "type": "module",
   "main": "./dist/index.js",

package/packages/doc-generator/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/doc-generator",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Auto-Doc Generator — generate documentation from source code (RFC-0014)",
   "type": "module",
   "main": "./dist/index.js",

package/packages/e2e-tester/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/e2e-tester",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "WebMCP-powered E2E testing for pi-harness - natural language test scenarios",
   "type": "module",
   "main": "./dist/index.js",

package/packages/evaluation-engine/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/evaluation-engine",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Evaluation engine for pi-harness runtime (RFC-0057)",
   "type": "module",
   "main": "./dist/index.js",

package/packages/evaluation-runner/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/evaluation-runner",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "description": "Framework-agnostic evaluation framework for AI outputs (RFC-0020)",
   "main": "dist/index.js",

package/packages/event-bus/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/event-bus",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "main": "src/index.ts",
   "exports": {

package/packages/event-store/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/event-store",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "main": "src/index.ts",
   "exports": {

package/packages/experience-replay/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi-harness/experience-replay",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Experience replay for debugging and reproducibility (RFC-0059)",
   "type": "module",
   "main": "./dist/index.js",

package/packages/feedback-collector/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/feedback-collector",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "type": "module",
   "description": "Collect and aggregate human/automated feedback (RFC-0021)",
   "main": "dist/index.js",

package/packages/file-copy-helper/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@moocoding/file-copy-helper",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Injects cp/sudo cp rule when user asks to mimic/copy files",
   "type": "module",
   "main": "dist/index.js",

package/packages/framework-detector/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pi/framework-detector",
-  "version": "1.1.303",
+  "version": "1.1.304",
   "description": "Framework detection system for pi-harness-runtime",
   "type": "module",
   "main": "./dist/index.js",