pi-interactive-shell 0.9.0 → 0.10.1

package/handoff-utils.ts

@@ -0,0 +1,92 @@
+import { mkdirSync, writeFileSync } from "node:fs";
+import { join } from "node:path";
+import { getAgentDir } from "@mariozechner/pi-coding-agent";
+import type { InteractiveShellConfig } from "./config.js";
+import type { InteractiveShellOptions, InteractiveShellResult } from "./types.js";
+import type { PtyTerminalSession } from "./pty-session.js";
+
+export function captureCompletionOutput(
+	session: PtyTerminalSession,
+	config: InteractiveShellConfig,
+): InteractiveShellResult["completionOutput"] {
+	const result = session.getTailLines({
+		lines: config.completionNotifyLines,
+		ansi: false,
+		maxChars: config.completionNotifyMaxChars,
+	});
+	return {
+		lines: result.lines,
+		totalLines: result.totalLinesInBuffer,
+		truncated: result.lines.length < result.totalLinesInBuffer || result.truncatedByChars,
+	};
+}
+
+export function captureTransferOutput(
+	session: PtyTerminalSession,
+	config: InteractiveShellConfig,
+): InteractiveShellResult["transferred"] {
+	const result = session.getTailLines({
+		lines: config.transferLines,
+		ansi: false,
+		maxChars: config.transferMaxChars,
+	});
+	return {
+		lines: result.lines,
+		totalLines: result.totalLinesInBuffer,
+		truncated: result.lines.length < result.totalLinesInBuffer || result.truncatedByChars,
+	};
+}
+
+export function maybeBuildHandoffPreview(
+	session: PtyTerminalSession,
+	when: "exit" | "detach" | "kill" | "timeout" | "transfer",
+	config: InteractiveShellConfig,
+	overrides?: Pick<InteractiveShellOptions, "handoffPreviewEnabled" | "handoffPreviewLines" | "handoffPreviewMaxChars">,
+): InteractiveShellResult["handoffPreview"] | undefined {
+	const enabled = overrides?.handoffPreviewEnabled ?? config.handoffPreviewEnabled;
+	if (!enabled) return undefined;
+	const lines = overrides?.handoffPreviewLines ?? config.handoffPreviewLines;
+	const maxChars = overrides?.handoffPreviewMaxChars ?? config.handoffPreviewMaxChars;
+	if (lines <= 0 || maxChars <= 0) return undefined;
+	const result = session.getTailLines({ lines, ansi: false, maxChars });
+	return { type: "tail", when, lines: result.lines };
+}
+
+export function maybeWriteHandoffSnapshot(
+	session: PtyTerminalSession,
+	when: "exit" | "detach" | "kill" | "timeout" | "transfer",
+	config: InteractiveShellConfig,
+	context: { command: string; cwd?: string },
+	overrides?: Pick<InteractiveShellOptions, "handoffSnapshotEnabled" | "handoffSnapshotLines" | "handoffSnapshotMaxChars">,
+): InteractiveShellResult["handoff"] | undefined {
+	const enabled = overrides?.handoffSnapshotEnabled ?? config.handoffSnapshotEnabled;
+	if (!enabled) return undefined;
+	const lines = overrides?.handoffSnapshotLines ?? config.handoffSnapshotLines;
+	const maxChars = overrides?.handoffSnapshotMaxChars ?? config.handoffSnapshotMaxChars;
+	if (lines <= 0 || maxChars <= 0) return undefined;
+
+	const baseDir = join(getAgentDir(), "cache", "interactive-shell");
+	mkdirSync(baseDir, { recursive: true });
+	const timestamp = new Date().toISOString().replace(/[:.]/g, "-");
+	const pid = session.pid;
+	const filename = `snapshot-${timestamp}-pid${pid}.log`;
+	const transcriptPath = join(baseDir, filename);
+	const tailResult = session.getTailLines({
+		lines,
+		ansi: config.ansiReemit,
+		maxChars,
+	});
+	const header = [
+		`# interactive-shell snapshot (${when})`,
+		`time: ${new Date().toISOString()}`,
+		`command: ${context.command}`,
+		`cwd: ${context.cwd ?? ""}`,
+		`pid: ${pid}`,
+		`exitCode: ${session.exitCode ?? ""}`,
+		`signal: ${session.signal ?? ""}`,
+		`lines: ${tailResult.lines.length} (requested ${lines}, maxChars ${maxChars})`,
+		"",
+	].join("\n");
+	writeFileSync(transcriptPath, header + tailResult.lines.join("\n") + "\n", { encoding: "utf-8" });
+	return { type: "snapshot", when, transcriptPath, linesWritten: tailResult.lines.length };
+}

package/headless-monitor.ts

@@ -2,13 +2,17 @@ import { stripVTControlCharacters } from "node:util";
 import type { PtyTerminalSession } from "./pty-session.js";
 import type { InteractiveShellConfig } from "./config.js";
 
+/** Runtime options for monitoring a headless dispatch session. */
 export interface HeadlessMonitorOptions {
 	autoExitOnQuiet: boolean;
 	quietThreshold: number;
 	gracePeriod?: number;
 	timeout?: number;
+	/** Original session start time in ms since epoch, preserved when a foreground session moves headless. */
+	startedAt?: number;
 }
 
+/** Completion payload emitted when a headless dispatch session finishes. */
 export interface HeadlessCompletionInfo {
 	exitCode: number | null;
 	signal?: number;
@@ -22,7 +26,7 @@ export interface HeadlessCompletionInfo {
 }
 
 export class HeadlessDispatchMonitor {
-	readonly startTime = Date.now();
+	readonly startTime: number;
 	private _disposed = false;
 	private quietTimer: ReturnType<typeof setTimeout> | null = null;
 	private timeoutTimer: ReturnType<typeof setTimeout> | null = null;
@@ -39,6 +43,7 @@ export class HeadlessDispatchMonitor {
 		private options: HeadlessMonitorOptions,
 		private onComplete: (info: HeadlessCompletionInfo) => void,
 	) {
+		this.startTime = options.startedAt ?? Date.now();
 		this.subscribe();
 
 		if (options.autoExitOnQuiet) {