@hk_net/pi-timestamp 0.1.8 → 0.1.9

package/README.md

@@ -1,41 +1,41 @@
-# pi-timestamp
-
-Shows timestamps for user input and agent completion timing. Requires pi `>=0.81.1`.
-
-## What it does
-
-- **User input**: Shows `Sent HH:MM:SS` as a dim status line in the chat display after each user message
-- **Agent completion**: Shows `Done at HH:MM:SS · duration` as a dim status line in the chat display after each agent turn (e.g., `Done at 14:32:05 · 3.2s`)
-- **Session/runtime summaries**: Shows accent-colored summaries when switching sessions and at final Pi exit, including start/end times and durations. The final summary includes the complete Pi process runtime and every session interval.
-
-All timestamps and summaries are **display-only** — session-switch summaries render in Pi's UI and the final runtime summary prints after Pi restores the terminal in TUI mode. They never enter the LLM context.
-
-## Display behavior
-
-Timestamps appear inline in the **chat display**, similar to Pi's built-in tool timing lines such as `Took 1.9s`. They are not appended to the session as user or custom messages, so they do not pollute the model context.
-
-## Installation
-
-```bash
-pi install npm:@hk_net/pi-timestamp
-```
-
-Or copy manually:
-
-```bash
-mkdir -p ~/.pi/agent/extensions
-cp packages/pi-timestamp/timestamp.ts ~/.pi/agent/extensions/timestamp.ts
-```
-
-## Duration measurement
-
-Captured from `agent_start` (when the model starts processing) to `agent_end` (when all tool calls and model processing are done).
-
-## Issues and feedback
-
-Found a bug or have a feature request? Please report it on
-[GitHub Issues](https://github.com/hknet/pi-extensions/issues).
-
-For security vulnerabilities, please use
-[GitHub's private vulnerability reporting](https://github.com/hknet/pi-extensions/security/advisories/new)
-instead of opening a public issue.
+# pi-timestamp
+
+Shows timestamps for user input and agent completion timing. Requires pi `>=0.84.1`.
+
+## What it does
+
+- **User input**: Shows `Sent HH:MM:SS` as a dim status line in the chat display after each user message
+- **Agent completion**: Shows `Done at HH:MM:SS · duration` as a dim status line in the chat display after each agent turn (e.g., `Done at 14:32:05 · 3.2s`)
+- **Session/runtime summaries**: Shows accent-colored summaries when switching sessions and at final Pi exit, including start/end times and durations. The final summary includes the complete Pi process runtime and every session interval.
+
+All timestamps and summaries are **display-only** — session-switch summaries render in Pi's UI and the final runtime summary prints after Pi restores the terminal in TUI mode. They never enter the LLM context.
+
+## Display behavior
+
+Timestamps appear inline in the **chat display**, similar to Pi's built-in tool timing lines such as `Took 1.9s`. They are not appended to the session as user or custom messages, so they do not pollute the model context.
+
+## Installation
+
+```bash
+pi install npm:@hk_net/pi-timestamp
+```
+
+Or copy manually:
+
+```bash
+mkdir -p ~/.pi/agent/extensions
+cp packages/pi-timestamp/timestamp.ts ~/.pi/agent/extensions/timestamp.ts
+```
+
+## Duration measurement
+
+Captured from `agent_start` (when the model starts processing) to `agent_end` (when all tool calls and model processing are done).
+
+## Issues and feedback
+
+Found a bug or have a feature request? Please report it on
+[GitHub Issues](https://github.com/hknet/pi-extensions/issues).
+
+For security vulnerabilities, please use
+[GitHub's private vulnerability reporting](https://github.com/hknet/pi-extensions/security/advisories/new)
+instead of opening a public issue.

package/package.json

@@ -1,37 +1,37 @@
-{
-  "name": "@hk_net/pi-timestamp",
-  "version": "0.1.8",
-  "description": "Pi extension that shows timestamps for user input and agent completion timing",
-  "type": "module",
-  "private": false,
-  "keywords": [
-    "pi-package",
-    "pi",
-    "extension",
-    "timestamp"
-  ],
-  "license": "EUPL-1.2",
-  "author": "hknet",
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/hknet/pi-extensions.git",
-    "directory": "packages/pi-timestamp"
-  },
-  "bugs": {
-    "url": "https://github.com/hknet/pi-extensions/issues"
-  },
-  "homepage": "https://github.com/hknet/pi-extensions/tree/main/packages/pi-timestamp#readme",
-  "files": [
-    "timestamp.ts",
-    "README.md"
-  ],
-  "peerDependencies": {
-    "@earendil-works/pi-coding-agent": ">=0.81.1",
-    "@earendil-works/pi-tui": ">=0.81.1"
-  },
-  "pi": {
-    "extensions": [
-      "./timestamp.ts"
-    ]
-  }
-}
+{
+  "name": "@hk_net/pi-timestamp",
+  "version": "0.1.9",
+  "description": "Pi extension that shows timestamps for user input and agent completion timing",
+  "type": "module",
+  "private": false,
+  "keywords": [
+    "pi-package",
+    "pi",
+    "extension",
+    "timestamp"
+  ],
+  "license": "EUPL-1.2",
+  "author": "hknet",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/hknet/pi-extensions.git",
+    "directory": "packages/pi-timestamp"
+  },
+  "bugs": {
+    "url": "https://github.com/hknet/pi-extensions/issues"
+  },
+  "homepage": "https://github.com/hknet/pi-extensions/tree/main/packages/pi-timestamp#readme",
+  "files": [
+    "timestamp.ts",
+    "README.md"
+  ],
+  "peerDependencies": {
+    "@earendil-works/pi-coding-agent": ">=0.84.1",
+    "@earendil-works/pi-tui": ">=0.84.1"
+  },
+  "pi": {
+    "extensions": [
+      "./timestamp.ts"
+    ]
+  }
+}

package/timestamp.ts

@@ -1,237 +1,237 @@
-/**
- * Timestamp Extension
- *
- * Displays timestamps for user input and agent completion timing.
- * All timestamps are display-only — they never enter the LLM context.
- *
- * - Shows `Sent HH:MM:SS` after each user message in the chat UI
- * - Shows `Done at HH:MM:SS · duration` after each agent turn in the chat UI
- * - Summarizes completed sessions and the complete Pi process runtime
- */
-
-import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
-
-export interface SessionInterval {
-    id: string;
-    startedAt: number;
-    endedAt: number;
-}
-
-export interface TimestampRuntimeState {
-    processStartedAt: number;
-    activeSession: { id: string; startedAt: number } | undefined;
-    completedSessions: SessionInterval[];
-    pendingSessionSummary: SessionInterval | undefined;
-}
-
-const RUNTIME_STATE_KEY = Symbol.for("hknet.pi-timestamp.runtime-state");
-
-export function createRuntimeState(processStartedAt: number): TimestampRuntimeState {
-    return { processStartedAt, activeSession: undefined, completedSessions: [], pendingSessionSummary: undefined };
-}
-
-function getRuntimeState(): TimestampRuntimeState {
-    const globalStore = globalThis as Record<symbol, unknown>;
-    const existing = globalStore[RUNTIME_STATE_KEY];
-    if (existing && typeof existing === "object") return existing as TimestampRuntimeState;
-
-    // The extension can load after Pi has initialized. Anchor the summary to the
-    // Node process start rather than the extension load time.
-    const state = createRuntimeState(Date.now() - process.uptime() * 1000);
-    globalStore[RUNTIME_STATE_KEY] = state;
-    return state;
-}
-
-export function beginSession(state: TimestampRuntimeState, id: string, startedAt: number): boolean {
-    if (state.activeSession?.id === id) return false;
-    state.activeSession = { id, startedAt };
-    return true;
-}
-
-export function finishSession(state: TimestampRuntimeState, endedAt: number): SessionInterval | undefined {
-    const active = state.activeSession;
-    if (!active) return undefined;
-
-    const interval = { ...active, endedAt };
-    state.completedSessions.push(interval);
-    state.activeSession = undefined;
-    return interval;
-}
-
-export function takePendingSessionSummary(state: TimestampRuntimeState): SessionInterval | undefined {
-    const pending = state.pendingSessionSummary;
-    state.pendingSessionSummary = undefined;
-    return pending;
-}
-
-function formatTime(ts: number): string {
-    const d = new Date(ts);
-    const hh = String(d.getHours()).padStart(2, "0");
-    const mm = String(d.getMinutes()).padStart(2, "0");
-    const ss = String(d.getSeconds()).padStart(2, "0");
-    return `${hh}:${mm}:${ss}`;
-}
-
-function formatDateTime(ts: number): string {
-    const d = new Date(ts);
-    const yyyy = d.getFullYear();
-    const mm = String(d.getMonth() + 1).padStart(2, "0");
-    const dd = String(d.getDate()).padStart(2, "0");
-    return `${yyyy}-${mm}-${dd} ${formatTime(ts)}`;
-}
-
-function isSameLocalDate(left: number, right: number): boolean {
-    const a = new Date(left);
-    const b = new Date(right);
-    return a.getFullYear() === b.getFullYear() && a.getMonth() === b.getMonth() && a.getDate() === b.getDate();
-}
-
-function formatSessionRange(session: SessionInterval): string {
-    if (isSameLocalDate(session.startedAt, session.endedAt)) {
-        return `${formatTime(session.startedAt)}–${formatTime(session.endedAt)}`;
-    }
-    return `${formatDateTime(session.startedAt)} → ${formatDateTime(session.endedAt)}`;
-}
-
-function formatDuration(ms: number): string {
-    if (ms < 1000) return `${ms}ms`;
-    const totalSecs = ms / 1000;
-    if (totalSecs < 60) return `${totalSecs.toFixed(1)}s`;
-    const totalMins = totalSecs / 60;
-    if (totalMins < 60) {
-        const m = Math.floor(totalMins);
-        const s = (totalSecs % 60).toFixed(1);
-        return `${m}m ${s}s`;
-    }
-    const hrs = totalMins / 60;
-    return `${Math.floor(hrs)}h ${Math.floor(totalMins % 60)}m`;
-}
-
-export function formatRuntimeDuration(ms: number): string {
-    let remainingSeconds = Math.max(0, Math.floor(ms / 1000));
-    const weeks = Math.floor(remainingSeconds / (7 * 24 * 60 * 60));
-    remainingSeconds %= 7 * 24 * 60 * 60;
-    const days = Math.floor(remainingSeconds / (24 * 60 * 60));
-    remainingSeconds %= 24 * 60 * 60;
-    const hours = Math.floor(remainingSeconds / (60 * 60));
-    remainingSeconds %= 60 * 60;
-    const minutes = Math.floor(remainingSeconds / 60);
-    const seconds = remainingSeconds % 60;
-
-    const parts = [
-        weeks > 0 ? `${weeks}w` : undefined,
-        days > 0 ? `${days}d` : undefined,
-        hours > 0 ? `${hours}h` : undefined,
-        minutes > 0 ? `${minutes}m` : undefined,
-        `${seconds}s`,
-    ];
-    return parts.filter((part): part is string => part !== undefined).join(" ");
-}
-
-function isTimeoutErrorMessage(message: string | undefined): boolean {
-    return /timed? out|timeout/i.test(message ?? "");
-}
-
-export default function (pi: ExtensionAPI) {
-    let taskStartTime: number | undefined;
-    let waitingForRetryAfterTimeout = false;
-
-    function notifyAccent(ctx: ExtensionContext, message: string): void {
-        ctx.ui.notify(ctx.ui.theme.fg("accent", message), "info");
-    }
-
-    // Track when the agent starts processing. If Pi is auto-retrying after a timeout,
-    // keep the original start time so the eventual completion covers the full task.
-    pi.on("agent_start", async () => {
-        if (waitingForRetryAfterTimeout) return;
-        taskStartTime = Date.now();
-    });
-
-    // Show "Sent HH:MM:SS" after each user message.
-    // notify(..., "info") renders a display-only status line in the TUI chat; it is not
-    // appended to the session and does not enter the LLM context.
-    pi.on("message_end", async (event, ctx) => {
-        if (event.message.role !== "user") return;
-
-        if (waitingForRetryAfterTimeout) {
-            taskStartTime = undefined;
-            waitingForRetryAfterTimeout = false;
-        }
-
-        const ts = event.message.timestamp;
-        if (!ts) return;
-
-        ctx.ui.notify(`Sent ${formatTime(ts)}`, "info");
-    });
-
-    // Show completion timing after the whole task finishes. If Pi ended this agent loop
-    // with a timeout error and will likely auto-retry, wait for the later completion.
-    pi.on("agent_end", async (event, ctx) => {
-        const lastAssistantMessage = [...event.messages].reverse().find((message) => message.role === "assistant");
-        if (
-            lastAssistantMessage?.role === "assistant" &&
-            lastAssistantMessage.stopReason === "error" &&
-            isTimeoutErrorMessage(lastAssistantMessage.errorMessage)
-        ) {
-            waitingForRetryAfterTimeout = true;
-            return;
-        }
-
-        const startTime = taskStartTime;
-        taskStartTime = undefined;
-        waitingForRetryAfterTimeout = false;
-
-        if (startTime === undefined) return;
-
-        const endTime = Date.now();
-        const duration = endTime - startTime;
-
-        ctx.ui.notify(`Done at ${formatTime(endTime)} · ${formatDuration(duration)}`, "info");
-    });
-
-    pi.on("session_start", (_event, ctx) => {
-        const state = getRuntimeState();
-        const previousSession = takePendingSessionSummary(state);
-        if (previousSession) {
-            notifyAccent(
-                ctx,
-                `Previous session complete · ${formatSessionRange(previousSession)} · ${formatRuntimeDuration(previousSession.endedAt - previousSession.startedAt)}`,
-            );
-        }
-
-        const sessionId = ctx.sessionManager.getSessionId();
-        // A reload rebinds the extension to the same session. Keep its timer running.
-        beginSession(state, sessionId, Date.now());
-    });
-
-    pi.on("session_shutdown", (event, ctx) => {
-        // Reload replaces the extension runtime but not the Pi session.
-        if (event.reason === "reload") return;
-
-        const endTime = Date.now();
-        const interval = finishSession(getRuntimeState(), endTime);
-
-        if (event.reason !== "quit") {
-            if (interval) getRuntimeState().pendingSessionSummary = interval;
-            return;
-        }
-
-        const state = getRuntimeState();
-        const sessionRows = state.completedSessions.map(
-            (session, index) =>
-                `  ${index + 1}. ${formatSessionRange(session)} · ${formatRuntimeDuration(session.endedAt - session.startedAt)}`,
-        );
-        // On normal Ctrl+D shutdown Pi has already stopped the TUI, so ui.notify()
-        // cannot render. Write after terminal restoration instead.
-        const summary = [
-            "Pi runtime complete",
-            `  Started: ${formatDateTime(state.processStartedAt)}`,
-            `  Ended:   ${formatDateTime(endTime)}`,
-            `  Total:   ${formatRuntimeDuration(endTime - state.processStartedAt)}`,
-            ...(sessionRows.length > 0 ? ["  Sessions:", ...sessionRows] : []),
-        ].join("\n");
-        if (ctx.mode === "tui") {
-            process.stdout.write(`${ctx.ui.theme.fg("accent", summary)}\n`);
-        }
-    });
-}
+/**
+ * Timestamp Extension
+ *
+ * Displays timestamps for user input and agent completion timing.
+ * All timestamps are display-only — they never enter the LLM context.
+ *
+ * - Shows `Sent HH:MM:SS` after each user message in the chat UI
+ * - Shows `Done at HH:MM:SS · duration` after each agent turn in the chat UI
+ * - Summarizes completed sessions and the complete Pi process runtime
+ */
+
+import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
+
+export interface SessionInterval {
+    id: string;
+    startedAt: number;
+    endedAt: number;
+}
+
+export interface TimestampRuntimeState {
+    processStartedAt: number;
+    activeSession: { id: string; startedAt: number } | undefined;
+    completedSessions: SessionInterval[];
+    pendingSessionSummary: SessionInterval | undefined;
+}
+
+const RUNTIME_STATE_KEY = Symbol.for("hknet.pi-timestamp.runtime-state");
+
+export function createRuntimeState(processStartedAt: number): TimestampRuntimeState {
+    return { processStartedAt, activeSession: undefined, completedSessions: [], pendingSessionSummary: undefined };
+}
+
+function getRuntimeState(): TimestampRuntimeState {
+    const globalStore = globalThis as Record<symbol, unknown>;
+    const existing = globalStore[RUNTIME_STATE_KEY];
+    if (existing && typeof existing === "object") return existing as TimestampRuntimeState;
+
+    // The extension can load after Pi has initialized. Anchor the summary to the
+    // Node process start rather than the extension load time.
+    const state = createRuntimeState(Date.now() - process.uptime() * 1000);
+    globalStore[RUNTIME_STATE_KEY] = state;
+    return state;
+}
+
+export function beginSession(state: TimestampRuntimeState, id: string, startedAt: number): boolean {
+    if (state.activeSession?.id === id) return false;
+    state.activeSession = { id, startedAt };
+    return true;
+}
+
+export function finishSession(state: TimestampRuntimeState, endedAt: number): SessionInterval | undefined {
+    const active = state.activeSession;
+    if (!active) return undefined;
+
+    const interval = { ...active, endedAt };
+    state.completedSessions.push(interval);
+    state.activeSession = undefined;
+    return interval;
+}
+
+export function takePendingSessionSummary(state: TimestampRuntimeState): SessionInterval | undefined {
+    const pending = state.pendingSessionSummary;
+    state.pendingSessionSummary = undefined;
+    return pending;
+}
+
+function formatTime(ts: number): string {
+    const d = new Date(ts);
+    const hh = String(d.getHours()).padStart(2, "0");
+    const mm = String(d.getMinutes()).padStart(2, "0");
+    const ss = String(d.getSeconds()).padStart(2, "0");
+    return `${hh}:${mm}:${ss}`;
+}
+
+function formatDateTime(ts: number): string {
+    const d = new Date(ts);
+    const yyyy = d.getFullYear();
+    const mm = String(d.getMonth() + 1).padStart(2, "0");
+    const dd = String(d.getDate()).padStart(2, "0");
+    return `${yyyy}-${mm}-${dd} ${formatTime(ts)}`;
+}
+
+function isSameLocalDate(left: number, right: number): boolean {
+    const a = new Date(left);
+    const b = new Date(right);
+    return a.getFullYear() === b.getFullYear() && a.getMonth() === b.getMonth() && a.getDate() === b.getDate();
+}
+
+function formatSessionRange(session: SessionInterval): string {
+    if (isSameLocalDate(session.startedAt, session.endedAt)) {
+        return `${formatTime(session.startedAt)}–${formatTime(session.endedAt)}`;
+    }
+    return `${formatDateTime(session.startedAt)} → ${formatDateTime(session.endedAt)}`;
+}
+
+function formatDuration(ms: number): string {
+    if (ms < 1000) return `${ms}ms`;
+    const totalSecs = ms / 1000;
+    if (totalSecs < 60) return `${totalSecs.toFixed(1)}s`;
+    const totalMins = totalSecs / 60;
+    if (totalMins < 60) {
+        const m = Math.floor(totalMins);
+        const s = (totalSecs % 60).toFixed(1);
+        return `${m}m ${s}s`;
+    }
+    const hrs = totalMins / 60;
+    return `${Math.floor(hrs)}h ${Math.floor(totalMins % 60)}m`;
+}
+
+export function formatRuntimeDuration(ms: number): string {
+    let remainingSeconds = Math.max(0, Math.floor(ms / 1000));
+    const weeks = Math.floor(remainingSeconds / (7 * 24 * 60 * 60));
+    remainingSeconds %= 7 * 24 * 60 * 60;
+    const days = Math.floor(remainingSeconds / (24 * 60 * 60));
+    remainingSeconds %= 24 * 60 * 60;
+    const hours = Math.floor(remainingSeconds / (60 * 60));
+    remainingSeconds %= 60 * 60;
+    const minutes = Math.floor(remainingSeconds / 60);
+    const seconds = remainingSeconds % 60;
+
+    const parts = [
+        weeks > 0 ? `${weeks}w` : undefined,
+        days > 0 ? `${days}d` : undefined,
+        hours > 0 ? `${hours}h` : undefined,
+        minutes > 0 ? `${minutes}m` : undefined,
+        `${seconds}s`,
+    ];
+    return parts.filter((part): part is string => part !== undefined).join(" ");
+}
+
+function isTimeoutErrorMessage(message: string | undefined): boolean {
+    return /timed? out|timeout/i.test(message ?? "");
+}
+
+export default function (pi: ExtensionAPI) {
+    let taskStartTime: number | undefined;
+    let waitingForRetryAfterTimeout = false;
+
+    function notifyAccent(ctx: ExtensionContext, message: string): void {
+        ctx.ui.notify(ctx.ui.theme.fg("accent", message), "info");
+    }
+
+    // Track when the agent starts processing. If Pi is auto-retrying after a timeout,
+    // keep the original start time so the eventual completion covers the full task.
+    pi.on("agent_start", async () => {
+        if (waitingForRetryAfterTimeout) return;
+        taskStartTime = Date.now();
+    });
+
+    // Show "Sent HH:MM:SS" after each user message.
+    // notify(..., "info") renders a display-only status line in the TUI chat; it is not
+    // appended to the session and does not enter the LLM context.
+    pi.on("message_end", async (event, ctx) => {
+        if (event.message.role !== "user") return;
+
+        if (waitingForRetryAfterTimeout) {
+            taskStartTime = undefined;
+            waitingForRetryAfterTimeout = false;
+        }
+
+        const ts = event.message.timestamp;
+        if (!ts) return;
+
+        ctx.ui.notify(`Sent ${formatTime(ts)}`, "info");
+    });
+
+    // Show completion timing after the whole task finishes. If Pi ended this agent loop
+    // with a timeout error and will likely auto-retry, wait for the later completion.
+    pi.on("agent_end", async (event, ctx) => {
+        const lastAssistantMessage = [...event.messages].reverse().find((message) => message.role === "assistant");
+        if (
+            lastAssistantMessage?.role === "assistant" &&
+            lastAssistantMessage.stopReason === "error" &&
+            isTimeoutErrorMessage(lastAssistantMessage.errorMessage)
+        ) {
+            waitingForRetryAfterTimeout = true;
+            return;
+        }
+
+        const startTime = taskStartTime;
+        taskStartTime = undefined;
+        waitingForRetryAfterTimeout = false;
+
+        if (startTime === undefined) return;
+
+        const endTime = Date.now();
+        const duration = endTime - startTime;
+
+        ctx.ui.notify(`Done at ${formatTime(endTime)} · ${formatDuration(duration)}`, "info");
+    });
+
+    pi.on("session_start", (_event, ctx) => {
+        const state = getRuntimeState();
+        const previousSession = takePendingSessionSummary(state);
+        if (previousSession) {
+            notifyAccent(
+                ctx,
+                `Previous session complete · ${formatSessionRange(previousSession)} · ${formatRuntimeDuration(previousSession.endedAt - previousSession.startedAt)}`,
+            );
+        }
+
+        const sessionId = ctx.sessionManager.getSessionId();
+        // A reload rebinds the extension to the same session. Keep its timer running.
+        beginSession(state, sessionId, Date.now());
+    });
+
+    pi.on("session_shutdown", (event, ctx) => {
+        // Reload replaces the extension runtime but not the Pi session.
+        if (event.reason === "reload") return;
+
+        const endTime = Date.now();
+        const interval = finishSession(getRuntimeState(), endTime);
+
+        if (event.reason !== "quit") {
+            if (interval) getRuntimeState().pendingSessionSummary = interval;
+            return;
+        }
+
+        const state = getRuntimeState();
+        const sessionRows = state.completedSessions.map(
+            (session, index) =>
+                `  ${index + 1}. ${formatSessionRange(session)} · ${formatRuntimeDuration(session.endedAt - session.startedAt)}`,
+        );
+        // On normal Ctrl+D shutdown Pi has already stopped the TUI, so ui.notify()
+        // cannot render. Write after terminal restoration instead.
+        const summary = [
+            "Pi runtime complete",
+            `  Started: ${formatDateTime(state.processStartedAt)}`,
+            `  Ended:   ${formatDateTime(endTime)}`,
+            `  Total:   ${formatRuntimeDuration(endTime - state.processStartedAt)}`,
+            ...(sessionRows.length > 0 ? ["  Sessions:", ...sessionRows] : []),
+        ].join("\n");
+        if (ctx.mode === "tui") {
+            process.stdout.write(`${ctx.ui.theme.fg("accent", summary)}\n`);
+        }
+    });
+}