compressed-context 0.1.0 → 0.2.0

package/src/compression/babel.ts

@@ -1,85 +1,85 @@
-import type { BabelSummary } from "./schema.ts"
-
-/** Babel 首行：`T=标题|G=目标`；允许两侧空格。 */
-const HEADER_RE = /^\s*T\s*=\s*(.*?)\s*\|\s*G\s*=\s*(.*?)\s*$/i
-/** 兼容模型把 T/G 写成两行的情况。 */
-const TITLE_RE = /^\s*T\s*=\s*(.*?)\s*$/i
-const GOAL_RE = /^\s*G\s*=\s*(.*?)\s*$/i
-/** T2/T3 候选行：优先 `c=N`，兼容 `candidate: N`。 */
-const CANDIDATE_RE = /^\s*(?:c|candidate)\s*[:=]\s*(\d+)\s*$/i
-
-/** 输入：候选字符串；输出：第一个非空值，全部为空时返回空串。 */
-function firstNonEmpty(...values: readonly (string | undefined)[]): string {
-  for (const value of values) {
-    const trimmed = value?.trim()
-    if (trimmed !== undefined && trimmed.length > 0) return trimmed
-  }
-  return ""
-}
-
-/** 输入：模型原始文本；输出：去掉代码围栏和首尾空白的文本。 */
-export function cleanBabel(text: string): string {
-  return text
-    .trim()
-    .replace(/^```(?:babel|babeltele|text|md|markdown)?\s*/i, "")
-    .replace(/\s*```$/, "")
-    .trim()
-}
-
-/**
- * 输入：文本 + 可选显式 title/goal。
- * 输出：规范 Babel 摘要；缺少标题/目标/正文时返回 undefined。
- * 预期：首行优先解析 `T=...|G=...`，也接受两行 T/G；text 保留规范化首行。
- */
-export function toBabelSummary(value: unknown, title?: string, goal?: string): BabelSummary | undefined {
-  if (typeof value !== "string") return undefined
-  const clean = cleanBabel(value)
-  if (clean.length === 0) return undefined
-
-  const lines = clean.split(/\r?\n/)
-  const first = lines[0]?.trim() ?? ""
-  const header = HEADER_RE.exec(first)
-  let parsedTitle = header?.[1]?.trim()
-  let parsedGoal = header?.[2]?.trim()
-  let bodyLines = lines
-
-  if (header !== null) {
-    bodyLines = lines.slice(1)
-  } else {
-    const second = lines[1]?.trim() ?? ""
-    const titleLine = TITLE_RE.exec(first)
-    const goalLine = GOAL_RE.exec(second)
-    if (titleLine !== null && goalLine !== null) {
-      parsedTitle = titleLine[1]?.trim()
-      parsedGoal = goalLine[1]?.trim()
-      bodyLines = lines.slice(2)
-    }
-  }
-
-  const resolvedTitle = firstNonEmpty(title, parsedTitle)
-  const resolvedGoal = firstNonEmpty(goal, parsedGoal)
-  if (resolvedTitle.length === 0 || resolvedGoal.length === 0) return undefined
-
-  // 有首行时正文从首行之后开始；无首行但显式给了 title/goal 时保留整段正文。
-  const body = bodyLines.join("\n").trim()
-  if (body.length === 0) return undefined
-
-  return {
-    format: "babel",
-    title: resolvedTitle,
-    goal: resolvedGoal,
-    text: `T=${resolvedTitle}|G=${resolvedGoal}\n${body}`,
-  }
-}
-
-/** 输入：模型返回的文本；输出：首个非空行是否为 `c=N`，以及去掉该行后的正文。 */
-export function splitBabelCandidate(text: string): { readonly candidate: number; readonly summary: string } | undefined {
-  const clean = cleanBabel(text)
-  const lines = clean.split(/\r?\n/)
-  const first = lines[0]?.trim() ?? ""
-  const match = CANDIDATE_RE.exec(first)
-  if (match === null) return undefined
-  const candidate = Number(match[1])
-  if (!Number.isInteger(candidate)) return undefined
-  return { candidate, summary: lines.slice(1).join("\n").trim() }
-}
+import type { BabelSummary } from "./schema.ts"
+
+/** Babel 首行：`T=标题|G=目标`；允许两侧空格。 */
+const HEADER_RE = /^\s*T\s*=\s*(.*?)\s*\|\s*G\s*=\s*(.*?)\s*$/i
+/** 兼容模型把 T/G 写成两行的情况。 */
+const TITLE_RE = /^\s*T\s*=\s*(.*?)\s*$/i
+const GOAL_RE = /^\s*G\s*=\s*(.*?)\s*$/i
+/** T2/T3 候选行：优先 `c=N`，兼容 `candidate: N`。 */
+const CANDIDATE_RE = /^\s*(?:c|candidate)\s*[:=]\s*(\d+)\s*$/i
+
+/** 输入：候选字符串；输出：第一个非空值，全部为空时返回空串。 */
+function firstNonEmpty(...values: readonly (string | undefined)[]): string {
+  for (const value of values) {
+    const trimmed = value?.trim()
+    if (trimmed !== undefined && trimmed.length > 0) return trimmed
+  }
+  return ""
+}
+
+/** 输入：模型原始文本；输出：去掉代码围栏和首尾空白的文本。 */
+export function cleanBabel(text: string): string {
+  return text
+    .trim()
+    .replace(/^```(?:babel|babeltele|text|md|markdown)?\s*/i, "")
+    .replace(/\s*```$/, "")
+    .trim()
+}
+
+/**
+ * 输入：文本 + 可选显式 title/goal。
+ * 输出：规范 Babel 摘要；缺少标题/目标/正文时返回 undefined。
+ * 预期：首行优先解析 `T=...|G=...`，也接受两行 T/G；text 保留规范化首行。
+ */
+export function toBabelSummary(value: unknown, title?: string, goal?: string): BabelSummary | undefined {
+  if (typeof value !== "string") return undefined
+  const clean = cleanBabel(value)
+  if (clean.length === 0) return undefined
+
+  const lines = clean.split(/\r?\n/)
+  const first = lines[0]?.trim() ?? ""
+  const header = HEADER_RE.exec(first)
+  let parsedTitle = header?.[1]?.trim()
+  let parsedGoal = header?.[2]?.trim()
+  let bodyLines = lines
+
+  if (header !== null) {
+    bodyLines = lines.slice(1)
+  } else {
+    const second = lines[1]?.trim() ?? ""
+    const titleLine = TITLE_RE.exec(first)
+    const goalLine = GOAL_RE.exec(second)
+    if (titleLine !== null && goalLine !== null) {
+      parsedTitle = titleLine[1]?.trim()
+      parsedGoal = goalLine[1]?.trim()
+      bodyLines = lines.slice(2)
+    }
+  }
+
+  const resolvedTitle = firstNonEmpty(title, parsedTitle)
+  const resolvedGoal = firstNonEmpty(goal, parsedGoal)
+  if (resolvedTitle.length === 0 || resolvedGoal.length === 0) return undefined
+
+  // 有首行时正文从首行之后开始；无首行但显式给了 title/goal 时保留整段正文。
+  const body = bodyLines.join("\n").trim()
+  if (body.length === 0) return undefined
+
+  return {
+    format: "babel",
+    title: resolvedTitle,
+    goal: resolvedGoal,
+    text: `T=${resolvedTitle}|G=${resolvedGoal}\n${body}`,
+  }
+}
+
+/** 输入：模型返回的文本；输出：首个非空行是否为 `c=N`，以及去掉该行后的正文。 */
+export function splitBabelCandidate(text: string): { readonly candidate: number; readonly summary: string } | undefined {
+  const clean = cleanBabel(text)
+  const lines = clean.split(/\r?\n/)
+  const first = lines[0]?.trim() ?? ""
+  const match = CANDIDATE_RE.exec(first)
+  if (match === null) return undefined
+  const candidate = Number(match[1])
+  if (!Number.isInteger(candidate)) return undefined
+  return { candidate, summary: lines.slice(1).join("\n").trim() }
+}

package/src/compression/block.ts

@@ -1,25 +1,25 @@
-import type { Summary } from "./schema.ts"
-import { summaryChars } from "./summary.ts"
-import { nextBlockId, nextOrder, type Block } from "./state.ts"
-
-/** 输入：已有块 + 摘要 + 覆盖范围 + 原文字符数；输出：新的 T1 块。 */
-export function createT1Block(input: {
-  readonly blocks: readonly Block[]
-  readonly summary: Summary
-  readonly fromId: string
-  readonly toId: string
-  readonly sourceChars: number
-  readonly at?: number
-}): Block {
-  return {
-    id: nextBlockId(input.blocks),
-    tier: 1,
-    summary: input.summary,
-    fromId: input.fromId,
-    toId: input.toId,
-    tokensBefore: Math.ceil(input.sourceChars / 4),
-    tokensAfter: Math.ceil(summaryChars(input.summary) / 4),
-    at: input.at ?? Date.now(),
-    order: nextOrder(input.blocks),
-  }
-}
+import type { Summary } from "./schema.ts"
+import { summaryChars } from "./summary.ts"
+import { nextBlockId, nextOrder, type Block } from "./state.ts"
+
+/** 输入：已有块 + 摘要 + 覆盖范围 + 原文字符数；输出：新的 T1 块。 */
+export function createT1Block(input: {
+  readonly blocks: readonly Block[]
+  readonly summary: Summary
+  readonly fromId: string
+  readonly toId: string
+  readonly sourceChars: number
+  readonly at?: number
+}): Block {
+  return {
+    id: nextBlockId(input.blocks),
+    tier: 1,
+    summary: input.summary,
+    fromId: input.fromId,
+    toId: input.toId,
+    tokensBefore: Math.ceil(input.sourceChars / 4),
+    tokensAfter: Math.ceil(summaryChars(input.summary) / 4),
+    at: input.at ?? Date.now(),
+    order: nextOrder(input.blocks),
+  }
+}

package/src/compression/cache-history.ts

@@ -1,75 +1,75 @@
-import type { Plugin } from "@opencode/plugin"
-import type { Schema } from "effect"
-
-import {
-  appendCacheSampleFromStep,
-  appendCacheSamples,
-  cacheHistoryEquals,
-  cacheHistoryKey,
-  parseCacheHistory,
-  type CacheHistory,
-  type CacheSample,
-} from "../shared/cache.ts"
-import type { StorageLike } from "./state.ts"
-
-/** 输入：storage + sessionID；输出：该会话的历史；缺失或损坏时为空历史。 */
-export async function loadCacheHistory(storage: StorageLike, sessionID: string): Promise<CacheHistory> {
-  return parseCacheHistory(await storage.get(cacheHistoryKey(sessionID)))
-}
-
-/** 输入：storage + sessionID + 样本；输出：无；预期：写入 JSON，最多 10 条。 */
-export async function saveCacheHistory(
-  storage: StorageLike,
-  sessionID: string,
-  samples: readonly CacheSample[],
-): Promise<void> {
-  await storage.set(cacheHistoryKey(sessionID), { v: 1, samples } as unknown as Schema.Json)
-}
-
-/**
- * 输入：插件上下文 + sessionID + 单 step 的 assistant usage。
- * 输出：无。
- * 预期：tool loop 的后续 step 不触发 context/usage.updated 的完整读取，
- *       用 `session.step.ended` 事件里的 tokens 直接落一条缓存样本。
- */
-export async function recordCacheHistoryStep(
-  ctx: Plugin.Context,
-  sessionID: string,
-  input: {
-    readonly messageID: string
-    readonly tokens: {
-      readonly input: number
-      readonly output: number
-      readonly reasoning: number
-      readonly cache: { readonly read: number; readonly write: number }
-    }
-  },
-): Promise<void> {
-  try {
-    const history = await loadCacheHistory(ctx.storage, sessionID)
-    const samples = appendCacheSampleFromStep(history.samples, input.messageID, input.tokens)
-    if (!cacheHistoryEquals(history.samples, samples)) {
-      await saveCacheHistory(ctx.storage, sessionID, samples)
-    }
-  } catch {
-    // best effort：历史缺失不影响主流程
-  }
-}
-
-/**
- * 输入：插件上下文 + sessionID。
- * 输出：无；预期：把当前上下文里未记录过的 assistant usage 追加进该会话的历史。
- *       失败只影响历史，不影响压缩与主流程。
- */
-export async function recordCacheHistory(ctx: Plugin.Context, sessionID: string): Promise<void> {
-  try {
-    const messages = await ctx.session.context({ sessionID })
-    const history = await loadCacheHistory(ctx.storage, sessionID)
-    const samples = appendCacheSamples(history.samples, messages)
-    if (!cacheHistoryEquals(history.samples, samples)) {
-      await saveCacheHistory(ctx.storage, sessionID, samples)
-    }
-  } catch {
-    // best effort：历史缺失不影响主界面
-  }
-}
+import type { Plugin } from "@opencode/plugin"
+import type { Schema } from "effect"
+
+import {
+  appendCacheSampleFromStep,
+  appendCacheSamples,
+  cacheHistoryEquals,
+  cacheHistoryKey,
+  parseCacheHistory,
+  type CacheHistory,
+  type CacheSample,
+} from "../shared/cache.ts"
+import type { StorageLike } from "./state.ts"
+
+/** 输入：storage + sessionID；输出：该会话的历史；缺失或损坏时为空历史。 */
+export async function loadCacheHistory(storage: StorageLike, sessionID: string): Promise<CacheHistory> {
+  return parseCacheHistory(await storage.get(cacheHistoryKey(sessionID)))
+}
+
+/** 输入：storage + sessionID + 样本；输出：无；预期：写入 JSON，最多 10 条。 */
+export async function saveCacheHistory(
+  storage: StorageLike,
+  sessionID: string,
+  samples: readonly CacheSample[],
+): Promise<void> {
+  await storage.set(cacheHistoryKey(sessionID), { v: 1, samples } as unknown as Schema.Json)
+}
+
+/**
+ * 输入：插件上下文 + sessionID + 单 step 的 assistant usage。
+ * 输出：无。
+ * 预期：tool loop 的后续 step 不触发 context/usage.updated 的完整读取，
+ *       用 `session.step.ended` 事件里的 tokens 直接落一条缓存样本。
+ */
+export async function recordCacheHistoryStep(
+  ctx: Plugin.Context,
+  sessionID: string,
+  input: {
+    readonly messageID: string
+    readonly tokens: {
+      readonly input: number
+      readonly output: number
+      readonly reasoning: number
+      readonly cache: { readonly read: number; readonly write: number }
+    }
+  },
+): Promise<void> {
+  try {
+    const history = await loadCacheHistory(ctx.storage, sessionID)
+    const samples = appendCacheSampleFromStep(history.samples, input.messageID, input.tokens)
+    if (!cacheHistoryEquals(history.samples, samples)) {
+      await saveCacheHistory(ctx.storage, sessionID, samples)
+    }
+  } catch {
+    // best effort：历史缺失不影响主流程
+  }
+}
+
+/**
+ * 输入：插件上下文 + sessionID。
+ * 输出：无；预期：把当前上下文里未记录过的 assistant usage 追加进该会话的历史。
+ *       失败只影响历史，不影响压缩与主流程。
+ */
+export async function recordCacheHistory(ctx: Plugin.Context, sessionID: string): Promise<void> {
+  try {
+    const messages = await ctx.session.context({ sessionID })
+    const history = await loadCacheHistory(ctx.storage, sessionID)
+    const samples = appendCacheSamples(history.samples, messages)
+    if (!cacheHistoryEquals(history.samples, samples)) {
+      await saveCacheHistory(ctx.storage, sessionID, samples)
+    }
+  } catch {
+    // best effort：历史缺失不影响主界面
+  }
+}