pi-conversation-timer 1.0.0 → 1.0.1

package/index.ts

@@ -1,3 +1,6 @@
+import { existsSync, readFileSync, writeFileSync } from "node:fs";
+import { homedir } from "node:os";
+import { join } from "node:path";
 import type { ExtensionAPI, ExtensionContext } from "@earendil-works/pi-coding-agent";
 
 /**
@@ -28,6 +31,30 @@ export default function (pi: ExtensionAPI) {
   let taskCount = 0; // 任务计数
   let activeContext: ExtensionContext | null = null;
 
+  const cachePath = join(homedir(), ".pi", "agent", "conversation-timer-cache.json");
+
+  /**
+   * 保存当前会话的状态到本地缓存
+   */
+  function saveStateToCache(ctx: ExtensionContext) {
+    try {
+      const sessionKey = ctx.sessionManager?.getSessionFile?.() ?? "default";
+      let allData: Record<string, any> = {};
+      if (existsSync(cachePath)) {
+        allData = JSON.parse(readFileSync(cachePath, "utf-8"));
+      }
+      allData[sessionKey] = {
+        totalActiveTimeMs,
+        lastTaskDuration,
+        taskCount,
+        updatedAt: Date.now(),
+      };
+      writeFileSync(cachePath, JSON.stringify(allData, null, 2), "utf-8");
+    } catch {
+      // ignore
+    }
+  }
+
   /**
    * 生成状态栏文本
    */
@@ -119,6 +146,18 @@ export default function (pi: ExtensionAPI) {
       totalActiveTimeMs += duration;
       lastTaskDuration = duration;
       currentTaskStartTime = null;
+
+      // 持久化当前状态到 session 记录和本地缓存，确保 reload 绝不丢失
+      try {
+        pi.appendEntry("conversation-timer-state", {
+          totalActiveTimeMs,
+          lastTaskDuration,
+          taskCount,
+        });
+      } catch {
+        // ignore
+      }
+      saveStateToCache(ctx);
     }
 
     // 刷新为静止状态，等待下一次任务发出
@@ -126,34 +165,94 @@ export default function (pi: ExtensionAPI) {
   }
 
   // 会话启动或恢复
-  pi.on("session_start", async (_event, ctx) => {
+  pi.on("session_start", async (event, ctx) => {
     activeContext = ctx;
     stopActiveTimer();
-    totalActiveTimeMs = 0;
-    taskCount = 0;
-    lastTaskDuration = null;
-    currentTaskStartTime = null;
 
-    // 尝试从历史消息中恢复交互耗时
+    // 只有在明确新开会话（/new）时才归零；reload、resume 等情况必须保留历史时间
+    if (event.reason === "new") {
+      totalActiveTimeMs = 0;
+      taskCount = 0;
+      lastTaskDuration = null;
+      currentTaskStartTime = null;
+      updateStatus(ctx);
+      return;
+    }
+
+    let restored = false;
+    const sessionKey = ctx.sessionManager?.getSessionFile?.() ?? "default";
+
+    // 1. 优先从 session 自定义持久化记录中恢复
     try {
       const entries = ctx.sessionManager?.getEntries?.() ?? [];
-      let lastUserTs: number | null = null;
       for (const entry of entries) {
-        if ((entry as any).type === "message") {
-          const msg = (entry as any).message;
-          if (msg?.role === "user" && typeof msg.timestamp === "number") {
-            lastUserTs = msg.timestamp;
-          } else if (msg?.role === "assistant" && typeof msg.timestamp === "number" && lastUserTs !== null) {
-            const duration = Math.max(0, msg.timestamp - lastUserTs);
-            totalActiveTimeMs += duration;
-            lastTaskDuration = duration;
-            taskCount++;
-            lastUserTs = null;
+        if ((entry as any).type === "custom" && (entry as any).customType === "conversation-timer-state") {
+          const data = (entry as any).data;
+          if (data && typeof data.totalActiveTimeMs === "number") {
+            totalActiveTimeMs = data.totalActiveTimeMs;
+            lastTaskDuration = data.lastTaskDuration ?? null;
+            taskCount = data.taskCount ?? 0;
+            restored = true;
           }
         }
       }
     } catch {
-      // 默认从 0 开始
+      // ignore
+    }
+
+    // 2. 如果 session 记录中没有，尝试从本地缓存文件中读取
+    if (!restored) {
+      try {
+        if (existsSync(cachePath)) {
+          const allData = JSON.parse(readFileSync(cachePath, "utf-8"));
+          const cached = allData[sessionKey];
+          if (cached && typeof cached.totalActiveTimeMs === "number") {
+            totalActiveTimeMs = cached.totalActiveTimeMs;
+            lastTaskDuration = cached.lastTaskDuration ?? null;
+            taskCount = cached.taskCount ?? 0;
+            restored = true;
+          }
+        }
+      } catch {
+        // ignore
+      }
+    }
+
+    // 3. 兜底：如果未曾记录过（旧会话），从历史消息精准计算
+    if (!restored) {
+      try {
+        const entries = ctx.sessionManager?.getEntries?.() ?? [];
+        let curStart: number | null = null;
+        let accTime = 0;
+        let count = 0;
+        let lastDur: number | null = null;
+
+        for (const entry of entries) {
+          if ((entry as any).type === "message") {
+            const msg = (entry as any).message;
+            if (msg?.role === "user" && typeof msg.timestamp === "number") {
+              curStart = msg.timestamp;
+            } else if (msg?.role === "assistant" && typeof msg.timestamp === "number" && curStart !== null) {
+              // 只有当不是 toolUse 中断时，才算整个任务真正完成
+              if (msg.stopReason !== "toolUse") {
+                const dur = Math.max(0, msg.timestamp - curStart);
+                accTime += dur;
+                lastDur = dur;
+                count++;
+                curStart = null;
+              }
+            }
+          }
+        }
+
+        if (accTime > 0) {
+          totalActiveTimeMs = accTime;
+          lastTaskDuration = lastDur;
+          taskCount = count;
+        }
+      } catch {
+        // ignore
+      }
     }
 
     updateStatus(ctx);
@@ -177,6 +276,9 @@ export default function (pi: ExtensionAPI) {
   // 会话关闭或重置
   pi.on("session_shutdown", async (_event, ctx) => {
     stopActiveTimer();
+    if (activeContext) {
+      saveStateToCache(activeContext);
+    }
     if (ctx && ctx.hasUI) {
       try {
         ctx.ui.setStatus("conversation-timer", undefined);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-conversation-timer",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "Active task timer extension for Pi coding agent",
   "author": "Agonieler",
   "license": "MIT",