agentquad 0.3.0

package/src/codex-event-emitter.js

@@ -0,0 +1,111 @@
+import { watch, watchFile, unwatchFile, openSync, readSync, closeSync, statSync } from 'node:fs'
+
+const ABORT_DEDUP_MS = 100
+
+/**
+ * 监听 codex rollout-*.jsonl 文件增量并把关键事件抛给上层。
+ * - task_complete → Stop
+ * - turn_aborted  → TurnAborted（与 <turn_aborted> 用户消息 100ms 内去重）
+ * - error         → Error
+ * 同时记录最新一条 assistant 文本，给 getLatestAssistantContent() 用。
+ */
+export function createCodexEventEmitter({ filePath, nativeId, onEvent, logger = console } = {}) {
+  if (!filePath || !nativeId || !onEvent) throw new Error('filePath, nativeId, onEvent required')
+
+  let pos = 0
+  let watcher = null
+  let pollTimer = null
+  let buffer = ''
+  let latestAssistantText = ''
+  let lastAbortTs = 0
+
+  function readNew() {
+    let stat
+    try { stat = statSync(filePath) } catch { return }
+    if (stat.size <= pos) return
+    const fd = openSync(filePath, 'r')
+    try {
+      const buf = Buffer.alloc(stat.size - pos)
+      readSync(fd, buf, 0, buf.length, pos)
+      pos = stat.size
+      buffer += buf.toString('utf8')
+    } finally {
+      closeSync(fd)
+    }
+    let idx
+    while ((idx = buffer.indexOf('\n')) >= 0) {
+      const line = buffer.slice(0, idx)
+      buffer = buffer.slice(idx + 1)
+      if (!line.trim()) continue
+      try { handleLine(JSON.parse(line)) }
+      catch (e) { logger.warn?.(`[codex-emitter] bad jsonl line ignored: ${e.message}`) }
+    }
+  }
+
+  function handleLine(j) {
+    const t = j?.type
+    const p = j?.payload
+    if (t === 'event_msg') {
+      const pt = p?.type
+      if (pt === 'task_complete') {
+        onEvent({ event: 'Stop', nativeId, rawEventPayload: p })
+      } else if (pt === 'turn_aborted') {
+        lastAbortTs = Date.now()
+        onEvent({ event: 'TurnAborted', nativeId, rawEventPayload: p })
+      } else if (pt === 'error') {
+        onEvent({ event: 'Error', nativeId, rawEventPayload: p })
+      }
+    } else if (t === 'response_item') {
+      const pt = p?.type
+      if (pt === 'message' && p?.role === 'assistant' && Array.isArray(p?.content)) {
+        const text = p.content.map(c => c?.text || '').join('')
+        if (text) latestAssistantText = text
+      } else if (pt === 'message' && p?.role === 'user' && Array.isArray(p?.content)) {
+        // Dedup with sibling event_msg/turn_aborted
+        const txt = p.content.map(c => c?.text || '').join('')
+        if (txt.includes('<turn_aborted>') && Date.now() - lastAbortTs < ABORT_DEDUP_MS) {
+          // suppress
+        }
+      }
+    }
+  }
+
+  function start() {
+    // 注意：开始时不要把 pos 推到当前文件末尾，否则 fs.watch 在 macOS 上对刚 append
+    // 进来的内容偶尔不触发，会丢首批事件。从 0 起读 + dedup 不必要 —— 文件是会话私有的。
+    pos = 0
+    readNew()
+    try { watcher = watch(filePath, () => readNew()) } catch {}
+    // fs.watch 在 APFS / 网络盘上偶尔不触发，watchFile 30ms 轮询做兜底
+    watchFile(filePath, { interval: 30, persistent: false }, () => readNew())
+    // setInterval 自轮询：watchFile 用 Node 的中央 polling 线程，并发跑很多 emitter
+    // 时会被压垮（实测 vitest fork pool 全套 batch 下 watchFile 3 秒不触发）；这里
+    // 自己每 50ms statSync 一次，跟 OS 通知双保险，加 stat 廉价。
+    pollTimer = setInterval(readNew, 50)
+    if (pollTimer.unref) pollTimer.unref()
+  }
+
+  function stop() {
+    if (watcher) {
+      try { watcher.close() } catch {}
+      watcher = null
+    }
+    if (pollTimer) {
+      clearInterval(pollTimer)
+      pollTimer = null
+    }
+    try { unwatchFile(filePath) } catch {}
+  }
+
+  function getLatestAssistantContent() {
+    return latestAssistantText
+  }
+
+  // 给 PtyManager.onExit 用：codex 自己不会在 jsonl 里写 SessionEnd（它就只是
+  // 进程结束），所以由外层合成一条事件触发"会话整体结束"分支。
+  function emitSynthetic(evt) {
+    onEvent({ ...evt, nativeId: evt.nativeId ?? nativeId })
+  }
+
+  return { start, stop, getLatestAssistantContent, emitSynthetic }
+}

package/src/codex-prompt-detector.js

@@ -0,0 +1,53 @@
+const DEFAULT_DEBOUNCE_MS = 1500
+const RING_MAX = 32
+
+const PATTERNS = [
+  /(approve|allow|continue|proceed)\??\s*\(\s*y\/n\s*\)\s*$/i,
+  /\?\s*\[\s*y\/N\s*\]\s*$/i,
+  /\?\s*\[\s*Y\/n\s*\]\s*$/i,
+  /(允许|批准|授权).*\?\s*[（(]\s*[yYnN][\/／][nNyY][)）]\s*$/,
+  /run this command\?\s*\[[^\]]*\]\s*$/i,
+  /apply patch\?\s*\[[^\]]*\]\s*$/i,
+]
+
+function stripAnsi(s) {
+  return String(s || '').replace(/\x1b\[[0-9;?]*[A-Za-z~]/g, '').replace(/\x1b\][^\x07]*\x07/g, '')
+}
+
+export function createCodexPromptDetector({ pty, onMatch, debounceMs = DEFAULT_DEBOUNCE_MS, emitter = null } = {}) {
+  if (!pty || !onMatch) throw new Error('pty, onMatch required')
+  const ring = []
+  let timer = null
+  let stopped = false
+
+  function onData(chunk) {
+    if (stopped) return
+    ring.push({ ts: Date.now(), text: stripAnsi(String(chunk)) })
+    while (ring.length > RING_MAX) ring.shift()
+    if (timer) clearTimeout(timer)
+    timer = setTimeout(maybeMatch, debounceMs)
+  }
+
+  function maybeMatch() {
+    const tail = ring.slice(-4).map(c => c.text).join('')
+    let matchedPattern = null
+    for (const re of PATTERNS) {
+      if (re.test(tail)) { matchedPattern = re.source; break }
+    }
+    if (!matchedPattern) return
+    const resolvedEmitter = typeof emitter === 'function' ? emitter() : emitter
+    if (resolvedEmitter?.getLatestAssistantContent) {
+      const ai = resolvedEmitter.getLatestAssistantContent() || ''
+      const trimmed = tail.slice(-200).trim()
+      if (trimmed && (ai.includes(trimmed) || ai.endsWith(trimmed))) {
+        return  // AI self-quoted prompt; not a real Codex permission ask
+      }
+    }
+    onMatch({ promptText: tail.slice(-200), matchedPattern })
+  }
+
+  function start() { pty.onData(onData) }
+  function stop() { stopped = true; if (timer) clearTimeout(timer) }
+
+  return { start, stop }
+}

package/src/codex-sidecar.js

@@ -0,0 +1,52 @@
+import { mkdirSync, writeFileSync, readFileSync, readdirSync, unlinkSync, existsSync } from 'node:fs'
+import { join } from 'node:path'
+import { DEFAULT_ROOT_DIR } from './config.js'
+
+const DEFAULT_DIR = join(DEFAULT_ROOT_DIR, 'codex-sessions')
+
+export function createCodexSidecar({ baseDir = DEFAULT_DIR } = {}) {
+  mkdirSync(baseDir, { recursive: true })
+  const memory = new Map()
+
+  function fileFor(nativeId) {
+    return join(baseDir, `${nativeId}.json`)
+  }
+
+  function lookup(nativeId) {
+    if (!nativeId) return null
+    if (memory.has(nativeId)) return memory.get(nativeId)
+    const path = fileFor(nativeId)
+    if (!existsSync(path)) return null
+    try {
+      const j = JSON.parse(readFileSync(path, 'utf8'))
+      const v = { quadtodoSessionId: j.quadtodoSessionId, todoId: j.todoId, cwd: j.cwd }
+      memory.set(nativeId, v)
+      return v
+    } catch { return null }
+  }
+
+  async function write({ nativeId, quadtodoSessionId, todoId, cwd }) {
+    if (!nativeId) throw new Error('nativeId_required')
+    memory.set(nativeId, { quadtodoSessionId, todoId, cwd })
+    const payload = { nativeId, quadtodoSessionId, todoId, cwd, ts: Date.now() }
+    writeFileSync(fileFor(nativeId), JSON.stringify(payload), 'utf8')
+  }
+
+  function restoreFromDisk() {
+    if (!existsSync(baseDir)) return
+    for (const name of readdirSync(baseDir)) {
+      if (!name.endsWith('.json')) continue
+      try {
+        const j = JSON.parse(readFileSync(join(baseDir, name), 'utf8'))
+        if (j.nativeId) memory.set(j.nativeId, { quadtodoSessionId: j.quadtodoSessionId, todoId: j.todoId, cwd: j.cwd })
+      } catch {}
+    }
+  }
+
+  function clear(nativeId) {
+    memory.delete(nativeId)
+    try { unlinkSync(fileFor(nativeId)) } catch {}
+  }
+
+  return { write, lookup, restoreFromDisk, clear }
+}

package/src/codex-transcript.js

@@ -0,0 +1,74 @@
+import { readFileSync } from 'node:fs'
+
+function parseLines(filePath) {
+  return readFileSync(filePath, 'utf8').split('\n').filter(l => l.trim())
+}
+
+function blockText(content) {
+  if (!Array.isArray(content)) return ''
+  return content.map(c => c?.text || '').filter(Boolean).join('')
+}
+
+export function readLatestCodexTurn(filePath) {
+  const lines = parseLines(filePath)
+  for (let i = lines.length - 1; i >= 0; i--) {
+    try {
+      const j = JSON.parse(lines[i])
+      if (j.type !== 'response_item') continue
+      const p = j.payload
+      if (p?.type !== 'message' || p?.role !== 'assistant') continue
+      const text = blockText(p.content)
+      if (!text) continue
+      return { text, raw: p, timestamp: j.timestamp || null }
+    } catch {}
+  }
+  return null
+}
+
+export async function readLatestCodexTurnFresh(filePath, lastSeenText, { retries = 3, retryMs = 200 } = {}) {
+  for (let i = 0; i <= retries; i++) {
+    const turn = readLatestCodexTurn(filePath)
+    if (turn && turn.text !== lastSeenText) return turn
+    if (i < retries) await new Promise(r => setTimeout(r, retryMs))
+  }
+  return null
+}
+
+export function buildFullCodexTranscript(filePath) {
+  const lines = parseLines(filePath)
+  const out = []
+  let turnCount = 0
+  for (const line of lines) {
+    let j
+    try { j = JSON.parse(line) } catch { continue }
+    if (j.type !== 'response_item' || j.payload?.type !== 'message') continue
+    const role = j.payload.role
+    const text = blockText(j.payload.content)
+    if (!text) continue
+    if (role === 'assistant') turnCount++
+    out.push(`### ${role}\n\n${text}\n`)
+  }
+  const header = `# Codex Session Transcript\n\n_Generated: ${new Date().toISOString()}_\n_Source: ${filePath}_\n_Turns: ${turnCount}_\n\n---\n\n`
+  return { markdown: header + out.join('\n'), turnCount }
+}
+
+export function extractCodexTurnUsageFromLines(lines) {
+  let last = null
+  for (const line of lines) {
+    if (!line || !line.trim()) continue
+    try {
+      const j = JSON.parse(line)
+      if (j.type === 'event_msg' && j.payload?.type === 'token_count') {
+        const info = j.payload.info
+        if (info?.last_token_usage) last = info.last_token_usage
+      }
+    } catch {}
+  }
+  if (!last) return null
+  return {
+    input: Number(last.input_tokens) || 0,
+    output: Number(last.output_tokens) || 0,
+    cacheRead: Number(last.cached_input_tokens || last.cache_read_input_tokens) || 0,
+    cacheCreation: Number(last.cache_creation_input_tokens) || 0,
+  }
+}