clideck 1.22.7 → 1.23.2

package/transcript.js

@@ -1,4 +1,4 @@
-const { appendFile, mkdirSync, existsSync, readdirSync, readFileSync, unlinkSync } = require('fs');
+const { appendFile, writeFileSync, mkdirSync, existsSync, readdirSync, readFileSync, unlinkSync } = require('fs');
 const { join, basename } = require('path');
 const { DATA_DIR } = require('./paths');
 
@@ -9,11 +9,19 @@ const MAX_CACHE = 50 * 1024;
 const inputBuf = {};
 const outputBuf = {};
 const cache = {};
+const prefixes = {};
 let broadcast = null;
+let notifyPlugin = null;
 
-function init(bc, validIds) {
+function init(bc, validIds, pluginNotify) {
   broadcast = bc;
+  notifyPlugin = pluginNotify || null;
   if (!existsSync(DIR)) mkdirSync(DIR, { recursive: true });
+  // Clean up orphaned .screen files (no matching .jsonl or session)
+  for (const file of readdirSync(DIR).filter(f => f.endsWith('.screen'))) {
+    const id = basename(file, '.screen');
+    if (!existsSync(fpath(id)) && (!validIds || !validIds.has(id))) { try { unlinkSync(join(DIR, file)); } catch {} }
+  }
   for (const file of readdirSync(DIR).filter(f => f.endsWith('.jsonl'))) {
     const id = basename(file, '.jsonl');
     if (validIds && !validIds.has(id)) { try { unlinkSync(join(DIR, file)); } catch {} continue; }
@@ -26,13 +34,16 @@ function init(bc, validIds) {
 }
 
 function fpath(id) { return join(DIR, `${id}.jsonl`); }
+function setPrefix(id, prefix) { prefixes[id] = prefix; }
 
 function store(id, role, text) {
-  appendFile(fpath(id), JSON.stringify({ ts: Date.now(), role, text }) + '\n', () => {});
+  const prefix = prefixes[id] || '';
+  appendFile(fpath(id), JSON.stringify({ ts: Date.now(), role, text, ...(prefix && { prefix }) }) + '\n', () => {});
   if (!cache[id]) cache[id] = '';
   cache[id] += '\n' + text;
   if (cache[id].length > MAX_CACHE) cache[id] = cache[id].slice(-MAX_CACHE);
-  if (broadcast) broadcast({ type: 'transcript.append', id, text });
+  if (broadcast) broadcast({ type: 'transcript.append', id, role, text });
+  if (notifyPlugin) notifyPlugin(id, role, text);
 }
 
 function trackInput(id, data) {
@@ -58,6 +69,7 @@ function trackInput(id, data) {
   }
 }
 
+// Server-side fallback: captures raw PTY output (noisy but always available)
 function trackOutput(id, data) {
   if (!outputBuf[id]) outputBuf[id] = { text: '', timer: null };
   const buf = outputBuf[id];
@@ -75,6 +87,155 @@ function flush(id) {
   if (lines.length) store(id, 'agent', lines.join('\n'));
 }
 
+// Browser-side clean snapshot: overwrites a per-session file with the full
+// xterm buffer as rendered by the browser. No diffing, no JSONL — just the
+// clean screen content. Mobile reads this for "last agent message".
+function storeBuffer(id, lines) {
+  const isChrome = t => !t || /^[─━═\u2500-\u257f]+$/.test(t) || /^[❯>$%#]\s*$/.test(t) || t === 'esc to interrupt' || t === '? for shortcuts';
+  const filtered = lines.filter(l => !isChrome(l.trim()));
+  while (filtered.length && !filtered[filtered.length - 1].trim()) filtered.pop();
+  const screenPath = join(DIR, `${id}.screen`);
+  if (filtered.length) writeFileSync(screenPath, filtered.join('\n'));
+}
+
+// Read the clean screen snapshot for a session (if available).
+function getScreen(id) {
+  const file = join(DIR, `${id}.screen`);
+  if (!existsSync(file)) return null;
+  try { return readFileSync(file, 'utf8'); } catch { return null; }
+}
+
+// Per-agent screen parsers. Each returns [{role, text}] from .screen content.
+const agentParsers = {
+  'claude-code': (lines) => {
+    const turns = [];
+    let current = null;
+    for (const line of lines) {
+      const m = line.match(/^(?:[│ ]\s*)?([❯›]|[⏺•])\s(.*)$/);
+      if (m) {
+        if (current) { current.text = current.text.replace(/\n+$/, ''); turns.push(current); }
+        current = { role: m[1] === '❯' || m[1] === '›' ? 'user' : 'agent', text: m[2] };
+        continue;
+      }
+      if (!current) continue;
+      let cont = line;
+      if (cont.startsWith('│ ')) cont = cont.slice(2);
+      else if (cont.startsWith('  ')) cont = cont.slice(2);
+      current.text += '\n' + cont;
+    }
+    if (current) { current.text = current.text.replace(/\n+$/, ''); turns.push(current); }
+    return turns.length >= 2 ? turns : null;
+  },
+  'codex': (lines) => {
+    const turns = [];
+    let current = null;
+    for (const line of lines) {
+      const m = line.match(/^(?:│\s*)?([›•])\s(.*)$/);
+      if (m) {
+        if (current) { current.text = current.text.replace(/\n+$/, ''); turns.push(current); }
+        current = { role: m[1] === '›' ? 'user' : 'agent', text: m[2] };
+        continue;
+      }
+      if (!current) continue;
+      let cont = line;
+      if (cont.startsWith('│ ')) cont = cont.slice(2);
+      else if (cont.startsWith('  ')) cont = cont.slice(2);
+      current.text += '\n' + cont;
+    }
+    if (current) { current.text = current.text.replace(/\n+$/, ''); turns.push(current); }
+    return turns.length >= 2 ? turns : null;
+  },
+  'gemini-cli': (lines) => {
+    const turns = [];
+    let current = null;
+    for (const line of lines) {
+      const isUser = line.startsWith(' > ');
+      const isAgent = line.startsWith('✦ ');
+      if (isUser || isAgent) {
+        if (current) { current.text = current.text.replace(/\n+$/, ''); turns.push(current); }
+        current = { role: isUser ? 'user' : 'agent', text: isUser ? line.slice(3) : line.slice(2) };
+        continue;
+      }
+      if (!current) continue;
+      current.text += '\n' + line;
+    }
+    if (current) { current.text = current.text.replace(/\n+$/, ''); turns.push(current); }
+    return turns.length >= 2 ? turns : null;
+  },
+};
+
+// Fallback anchor parser for agents without a dedicated parser.
+// Uses JSONL user inputs to find prompt lines in the screen.
+function anchorParse(id, lines) {
+  const file = fpath(id);
+  if (!existsSync(file)) return null;
+  const users = [];
+  try {
+    for (const l of readFileSync(file, 'utf8').trim().split('\n')) {
+      try { const e = JSON.parse(l); if (e.role === 'user') users.push(e.text); } catch {}
+    }
+  } catch { return null; }
+  if (!users.length) return null;
+
+  const isPrompt = (line, text) => { const t = line.trim(); return t.endsWith(text) && t.length - text.length <= 6; };
+
+  const lastUser = users[users.length - 1];
+  let lastIdx = -1;
+  for (let i = lines.length - 1; i >= 0; i--) {
+    if (isPrompt(lines[i], lastUser)) { lastIdx = i; break; }
+  }
+  if (lastIdx < 0) return null;
+
+  const anchors = [{ idx: lastIdx, text: lastUser }];
+  for (let u = users.length - 2; u >= 0 && anchors.length < 3; u--) {
+    for (let i = anchors[anchors.length - 1].idx - 1; i >= 0; i--) {
+      if (isPrompt(lines[i], users[u])) { anchors.push({ idx: i, text: users[u] }); break; }
+    }
+  }
+  anchors.reverse();
+
+  const turns = [];
+  for (let p = 0; p < anchors.length; p++) {
+    turns.push({ role: 'user', text: anchors[p].text });
+    const start = anchors[p].idx + 1;
+    const end = p + 1 < anchors.length ? anchors[p + 1].idx : lines.length;
+    const agentLines = lines.slice(start, end).filter(l => l.trim());
+    if (agentLines.length) turns.push({ role: 'agent', text: agentLines.join('\n') });
+  }
+  if (turns.length < 2 || turns[turns.length - 1].role !== 'agent') return null;
+  return turns;
+}
+
+// Parse .screen into structured turns — use agent-specific parser if available, else anchor fallback.
+function getScreenTurns(id, agent) {
+  const screen = getScreen(id);
+  if (!screen) return null;
+  const lines = screen.split('\n');
+  const parser = agentParsers[agent];
+  return parser ? parser(lines) : anchorParse(id, lines);
+}
+
+function getLastTurns(id, n) {
+  n = n || 4;
+  const file = fpath(id);
+  if (!existsSync(file)) return [];
+  try {
+    const lines = readFileSync(file, 'utf8').trim().split('\n');
+    const turns = [];
+    for (let i = lines.length - 1; i >= 0; i--) {
+      let entry;
+      try { entry = JSON.parse(lines[i]); } catch { continue; }
+      if (turns.length && turns[turns.length - 1].role === entry.role) {
+        turns[turns.length - 1].text = entry.text + '\n' + turns[turns.length - 1].text;
+      } else {
+        turns.push({ role: entry.role, text: entry.text });
+        if (turns.length >= n) break;
+      }
+    }
+    return turns.reverse();
+  } catch { return []; }
+}
+
 function getCache() { return { ...cache }; }
 
 function clear(id) {
@@ -85,6 +246,8 @@ function clear(id) {
     delete outputBuf[id];
   }
   delete cache[id];
+  delete prefixes[id];
+  try { unlinkSync(join(DIR, `${id}.screen`)); } catch {}
 }
 
-module.exports = { init, trackInput, trackOutput, getCache, clear };
+module.exports = { init, trackInput, trackOutput, storeBuffer, getScreen, getScreenTurns, getLastTurns, getCache, clear, setPrefix };