@kkelly-offical/kkcode 0.1.2 → 0.1.6

package/src/provider/sse.mjs

@@ -1,91 +1,104 @@
-/**
- * 将 fetch response.body (ReadableStream) 解析为 SSE 事件的 AsyncIterator。
- * 同时支持 OpenAI（纯 data: 行）和 Anthropic（event: + data: 对）格式。
- *
- * @param {ReadableStream} body
- * @param {AbortSignal} [signal]
- * @param {object} [options]
- * @param {number} [options.idleTimeoutMs] - per-chunk idle timeout (resets on each chunk)
- * @yields {{ event: string|null, data: string }}
- */
-export async function* parseSSE(body, signal, options = {}) {
-  const reader = body.getReader()
-  const decoder = new TextDecoder()
-  let buffer = ""
-  const idleMs = options.idleTimeoutMs || 0
-
-  try {
-    while (true) {
-      if (signal?.aborted) break
-
-      let readResult
-      if (idleMs > 0) {
-        // Race between next chunk and idle timeout
-        readResult = await Promise.race([
-          reader.read(),
-          idleTimeout(idleMs, signal)
-        ])
-      } else {
-        readResult = await reader.read()
-      }
-
-      const { done, value } = readResult
-      if (done) break
-      buffer += decoder.decode(value, { stream: true })
-
-      const parts = buffer.split("\n\n")
-      buffer = parts.pop()
-
-      for (const part of parts) {
-        const result = parsePart(part)
-        if (result === null) return // [DONE]
-        if (result) yield result
-      }
-    }
-    // flush remaining buffer
-    if (buffer.trim()) {
-      const result = parsePart(buffer)
-      if (result && result !== null) yield result
-    }
-  } finally {
-    try { reader.releaseLock() } catch { /* reader may have pending read if generator was force-closed */ }
-  }
-}
-
-function idleTimeout(ms, signal) {
-  return new Promise((resolve, reject) => {
-    const timer = setTimeout(() => {
-      const err = new Error(`stream idle timeout: no data received for ${ms}ms`)
-      err.code = "STREAM_IDLE_TIMEOUT"
-      reject(err)
-    }, ms)
-    if (signal) {
-      const onAbort = () => {
-        clearTimeout(timer)
-        const err = new Error("aborted")
-        err.code = "ABORT_ERR"
-        reject(err)
-      }
-      if (signal.aborted) { clearTimeout(timer); onAbort(); return }
-      signal.addEventListener("abort", onAbort, { once: true })
-    }
-  })
-}
-
-function parsePart(part) {
-  const trimmed = part.trim()
-  if (!trimmed) return undefined
-  let event = null
-  let data = ""
-  for (const line of trimmed.split("\n")) {
-    if (line.startsWith("event:")) {
-      event = line.slice(6).trim()
-    } else if (line.startsWith("data:")) {
-      const payload = line.slice(5).trim()
-      if (payload === "[DONE]") return null
-      data = payload
-    }
-  }
-  if (!data) return undefined
-  return { event, data }
-}
+/**
+ * 将 fetch response.body (ReadableStream) 解析为 SSE 事件的 AsyncIterator。
+ * 同时支持 OpenAI（纯 data: 行）和 Anthropic（event: + data: 对）格式。
+ *
+ * @param {ReadableStream} body
+ * @param {AbortSignal} [signal]
+ * @param {object} [options]
+ * @param {number} [options.idleTimeoutMs] - per-chunk idle timeout (resets on each chunk)
+ * @yields {{ event: string|null, data: string }}
+ */
+export async function* parseSSE(body, signal, options = {}) {
+  const reader = body.getReader()
+  const decoder = new TextDecoder()
+  let buffer = ""
+  const idleMs = options.idleTimeoutMs || 0
+  let currentTimeout = null
+
+  try {
+    while (true) {
+      if (signal?.aborted) break
+
+      let readResult
+      if (idleMs > 0) {
+        if (currentTimeout) currentTimeout.cancel()
+        currentTimeout = idleTimeout(idleMs, signal)
+        readResult = await Promise.race([
+          reader.read(),
+          currentTimeout.promise
+        ])
+      } else {
+        readResult = await reader.read()
+      }
+
+      const { done, value } = readResult
+      if (done) break
+      buffer += decoder.decode(value, { stream: true })
+
+      const parts = buffer.split("\n\n")
+      buffer = parts.pop()
+
+      for (const part of parts) {
+        const result = parsePart(part)
+        if (result === null) return // [DONE]
+        if (result) yield result
+      }
+    }
+    // flush remaining buffer
+    if (buffer.trim()) {
+      const result = parsePart(buffer)
+      if (result && result !== null) yield result
+    }
+  } finally {
+    if (currentTimeout) currentTimeout.cancel()
+    try { reader.releaseLock() } catch { /* reader may have pending read if generator was force-closed */ }
+  }
+}
+
+function idleTimeout(ms, signal) {
+  let timer = null
+  let onAbort = null
+  const promise = new Promise((resolve, reject) => {
+    timer = setTimeout(() => {
+      const err = new Error(`stream idle timeout: no data received for ${ms}ms`)
+      err.code = "STREAM_IDLE_TIMEOUT"
+      reject(err)
+    }, ms)
+    if (signal) {
+      onAbort = () => {
+        clearTimeout(timer)
+        const err = new Error("aborted")
+        err.code = "ABORT_ERR"
+        reject(err)
+      }
+      if (signal.aborted) { clearTimeout(timer); onAbort(); return }
+      signal.addEventListener("abort", onAbort, { once: true })
+    }
+  })
+  function cancel() {
+    if (timer !== null) { clearTimeout(timer); timer = null }
+    if (onAbort && signal) {
+      signal.removeEventListener("abort", onAbort)
+      onAbort = null
+    }
+  }
+  return { promise, cancel }
+}
+
+function parsePart(part) {
+  const trimmed = part.trim()
+  if (!trimmed) return undefined
+  let event = null
+  let data = ""
+  for (const line of trimmed.split("\n")) {
+    if (line.startsWith("event:")) {
+      event = line.slice(6).trim()
+    } else if (line.startsWith("data:")) {
+      const payload = line.slice(5).trim()
+      if (payload === "[DONE]") return null
+      data = payload
+    }
+  }
+  if (!data) return undefined
+  return { event, data }
+}

package/src/repl.mjs

@@ -2720,7 +2720,7 @@ function startSplash() {
     "  ╚═╝  ╚═╝ ╚═╝  ╚═╝  ╚═════╝  ╚═════╝  ╚═════╝  ╚══════╝ "
   ]
   const tagline = "AI Coding Agent"
-  const version = "v0.1.2"
+  const version = "v0.1.6"
 
   // Gradient colors for the wave animation (cyan → blue → purple → pink → back)
   const wave = [

package/src/session/checkpoint.mjs

@@ -46,6 +46,71 @@ export async function listCheckpoints(sessionId) {
     .sort()
 }
 
+// ========== Phase 7: Task 级 Checkpoint ==========
+
+export async function saveTaskCheckpoint(sessionId, stageId, taskId, data) {
+  const dir = checkpointDir(sessionId)
+  await mkdir(dir, { recursive: true })
+  const name = `task_${stageId}_${taskId}`
+  const checkpoint = {
+    sessionId,
+    stageId,
+    taskId,
+    savedAt: Date.now(),
+    ...data
+  }
+  await writeJson(checkpointFile(sessionId, name), checkpoint)
+  return checkpoint
+}
+
+export async function loadTaskCheckpoints(sessionId, stageId) {
+  const dir = checkpointDir(sessionId)
+  const files = await readdir(dir, { withFileTypes: true }).catch(() => [])
+  const prefix = `task_${stageId}_`
+  const results = {}
+  for (const entry of files) {
+    if (entry.isFile() && entry.name.startsWith(prefix) && entry.name.endsWith(".json")) {
+      const data = await readJson(path.join(dir, entry.name), null)
+      if (data?.taskId) results[data.taskId] = data
+    }
+  }
+  return results
+}
+
+// ========== Phase 10: Checkpoint 清理策略 ==========
+
+export async function cleanupCheckpoints(sessionId, options = {}) {
+  const maxKeep = options.maxKeep || 10
+  const keepStageCheckpoints = options.keepStageCheckpoints !== false
+  const dir = checkpointDir(sessionId)
+  const all = await listCheckpoints(sessionId)
+  if (all.length <= maxKeep + 1) return { removed: 0 }
+
+  const toKeep = new Set(["latest"])
+  // 保留 stage 级和 task 级 checkpoint
+  if (keepStageCheckpoints) {
+    for (const name of all) {
+      if (name.startsWith("hybrid_stage_") || name.startsWith("task_")) {
+        toKeep.add(name)
+      }
+    }
+  }
+  // 保留最近 maxKeep 个编号 checkpoint
+  const numbered = all.filter(n => n.startsWith("cp_")).sort()
+  for (const n of numbered.slice(-maxKeep)) toKeep.add(n)
+
+  let removed = 0
+  for (const name of all) {
+    if (toKeep.has(name)) continue
+    try {
+      const { unlink: unlinkFile } = await import("node:fs/promises")
+      await unlinkFile(checkpointFile(sessionId, name)).catch(() => {})
+      removed++
+    } catch { /* ignore */ }
+  }
+  return { removed }
+}
+
 // ============================================================================
 // Git Snapshot Integration - AI Agent 自动 Git 快照功能
 // ============================================================================
@@ -61,12 +126,10 @@ export async function listCheckpoints(sessionId) {
  * @returns {Promise<{ok: boolean, snapshot?: Object, skipped?: boolean, reason?: string}>}
  */
 export async function autoSnapshotBeforeEdit(sessionId, cwd, config = {}, options = {}) {
-  // 检查 Git 自动化是否启用
+  // 检查 Git 自动化是否启用（默认启用，只有显式关闭才跳过）
   if (config.git_auto?.enabled === false) {
     return { ok: true, skipped: true, reason: "git_auto_disabled" }
   }
-
-  // 检查自动快照是否启用
   if (config.git_auto?.auto_snapshot === false) {
     return { ok: true, skipped: true, reason: "auto_snapshot_disabled" }
   }