ai-gains 1.5.1 → 1.5.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-gains",
-  "version": "1.5.1",
+  "version": "1.5.2",
   "description": "Interactive browser dashboard for AI development session tracking",
   "main": "src/server.js",
   "bin": {

package/template/.claude/scripts/ai-gains/get-session-times.cjs

@@ -31,15 +31,44 @@ function findFirstTimestamp() {
   });
 }
 
-// Scan from the end for the last entry with a top-level timestamp
-function findLastTimestamp() {
+function getEntryText(entry) {
+  const content = entry.message?.content;
+  if (!content) return '';
+  if (typeof content === 'string') return content;
+  if (Array.isArray(content)) {
+    return content
+      .filter(c => c.type === 'text')
+      .map(c => c.text || '')
+      .join(' ');
+  }
+  return '';
+}
+
+// A user message that contains the expanded /ai-gains skill content.
+// 'estimated_human_time_minutes' is unique to SKILL.md and won't appear
+// in normal conversation or the UserPromptSubmit hook injection.
+function isAiGainsInvocation(entry) {
+  if (entry.type !== 'user' && entry.message?.role !== 'user') return false;
+  return getEntryText(entry).includes('estimated_human_time_minutes');
+}
+
+function isAssistant(entry) {
+  return entry.type === 'assistant' || entry.message?.role === 'assistant';
+}
+
+// Scan from the end. Find the last assistant timestamp that precedes the
+// most recent /ai-gains skill invocation. Falls back to the last timestamp
+// overall if no ai-gains invocation is found (e.g. during a dry-run).
+function findEndTimestamp() {
   const fd = fs.openSync(transcriptPath, 'r');
   const { size } = fs.fstatSync(fd);
   let offset = size;
   let remainder = '';
-  let lastTimestamp = null;
+  let phase = 'looking_for_aigains'; // then 'looking_for_assistant'
+  let lastTimestamp = null; // fallback
+  let result = null;
 
-  while (offset > 0) {
+  outer: while (offset > 0) {
     const readSize = Math.min(CHUNK_SIZE, offset);
     offset -= readSize;
     const buf = Buffer.alloc(readSize);
@@ -53,22 +82,34 @@ function findLastTimestamp() {
       if (!line) continue;
       try {
         const entry = JSON.parse(line);
-        if (entry.timestamp) {
+
+        // Track the absolute last timestamp as a fallback
+        if (lastTimestamp === null && entry.timestamp) {
           lastTimestamp = entry.timestamp;
-          break;
+        }
+
+        if (phase === 'looking_for_aigains') {
+          if (isAiGainsInvocation(entry)) {
+            phase = 'looking_for_assistant';
+          }
+        } else {
+          // phase === 'looking_for_assistant'
+          if (isAssistant(entry) && entry.timestamp) {
+            result = entry.timestamp;
+            break outer;
+          }
         }
       } catch {}
     }
-    if (lastTimestamp) break;
   }
 
   fs.closeSync(fd);
-  return lastTimestamp;
+  return result || lastTimestamp;
 }
 
 (async () => {
   const start_time = await findFirstTimestamp();
-  const end_time = findLastTimestamp();
+  const end_time = findEndTimestamp();
 
   if (!start_time || !end_time) {
     console.error('Could not extract timestamps from transcript');