sage-mem 0.9.0 → 0.9.2

package/lib/autodream/rollback.js

@@ -8,8 +8,9 @@
 
 import { REPORT_DIR } from './config.js'
 import { readJson, safeName, snapshotEntries, stampCompact, stampHuman } from './util.js'
+import { restoreMemory } from '../memory/archive.js'
 import { randomUUID } from 'node:crypto'
-import { copyFile, mkdir, readdir, rename, stat, writeFile } from 'node:fs/promises'
+import { copyFile, mkdir, readdir, rename, rm, stat, writeFile } from 'node:fs/promises'
 import { join } from 'node:path'
 
 /** 读回滚留痕（面板也许要显示「最近谁退回过」）。 */
@@ -86,6 +87,34 @@ export async function rollback(opts = {}) {
     const protectionId = `pre-rollback-${stamp}-${randomUUID().replace(/-/g, '').slice(0, 4)}`
     const protection = await this.snapshot(protectionId, 'pre-rollback')
 
+    /**
+     * ⚠️ 保护快照不完整就**中止**，一个字都不许写。
+     *
+     * 它是「滚错了还能再退回来」这句承诺的唯一支撑。残缺时继续往下滚，等于拿一张假保险
+     * 去覆盖当前版本 —— 而当前版本恰恰是此刻最想保住的东西。残缺的回滚点比没有回滚点
+     * **更危险**，这与 `engine.js` 开工前那条（`snapshot.failed.length` 就中止）是同一条原则，
+     * 只是回滚这条路上一直漏了。
+     *
+     * 顺手把它删掉：它落进 `snapshots/` 就会被 `listSnapshots()` 当成一份可选的回滚点，
+     * 而按它恢复只会把「没存下来的那些文件」误判成「快照里没有」→ 停进 `archive/`。
+     */
+    if (protection.failed.length || protection.count !== protection.expected) {
+      await rm(protection.dir, { recursive: true, force: true }).catch(() => {})
+      const why = protection.failed.length
+        ? `有 ${protection.failed.length} 个文件没存下来（${protection.failed.slice(0, 3).join('、')}${
+            protection.failed.length > 3 ? '…' : ''
+          }）`
+        : `只存下 ${protection.count}/${protection.expected} 个文件`
+      this.log('warn', `回滚中止：保护快照不完整，${why}`)
+      this.phase = ''
+      return {
+        ok: false,
+        error:
+          `回滚前保护快照不完整（${why}）。为免滚错之后退不回来，这次回滚已中止，` +
+          '记忆目录一个字没动。请检查文件占用与磁盘空间后重试。',
+      }
+    }
+
     // ② 恢复
     this.phase = '恢复文件'
     const snapNames = await snapshotEntries(snapDir)
@@ -155,6 +184,38 @@ export async function rollback(opts = {}) {
           }
         }
       }
+
+      /**
+       * 策略自动归档也要退回来 —— 它按设计**不写进 `changes`**。
+       *
+       * `engine.js` 把两本账分开了：模型改的进 `changes`，确定性归档策略移走的进
+       * `archive.archived`（已落盘在运行记录 `manifest.json` 里）。只遍历 `changes`
+       * 就会整段漏掉，而这趟运行照样被标成「已回滚」、面板照报「恢复 0 个文件…已回滚」——
+       * 用户以为退回来了，文件其实还躺在 `archive/` 里。
+       *
+       * 恢复走 `memory/archive.js` 的共享实现（与手动恢复、模型工具同一份）：它会顺手
+       * 去掉 `archived_at` / `archived_reason` 两行留痕，否则「恢复」回来的是一份带着
+       * 「我已被归档」标记的活记忆。
+       *
+       * ⚠️ 只恢复**运行前快照里存在**的文件：策略候选是运行末尾扫出来的，理论上可能包含
+       * 这趟新造的文件，把它搬回来等于凭空激活一条运行前不存在的记忆（Q06 的形态）。
+       */
+      for (const a of runManifest.archive?.archived ?? []) {
+        if (!a?.ok || !a.file) continue
+        const name = a.file
+        if (restored.includes(name)) continue // 上面 `changes` 那段已经把它放回来了
+        if (!snapSet.has(name)) {
+          skipped.push(`${name}（运行前快照里没有，说明是这趟新造的，回滚不激活它）`)
+          continue
+        }
+        if (await stat(join(this.memoryDir, name)).catch(() => null)) {
+          skipped.push(`${name}（顶层已有同名文件，不覆盖）`)
+          continue
+        }
+        const res = await restoreMemory(this.memoryDir, name)
+        if (res.ok) restored.push(name)
+        else skipped.push(`${name}（恢复失败：${res.error}）`)
+      }
     }
 
     // ③ 留痕

package/lib/autodream/snapshot.js

@@ -36,7 +36,24 @@ export async function snapshot(id, kind = 'run') {
     if (err?.code === 'EEXIST') throw new Error(`回滚点 id 撞车：${id} 已存在，拒绝覆盖已有的回滚点`)
     throw err
   }
-  const entries = await readdir(this.memoryDir, { withFileTypes: true }).catch(() => [])
+  /**
+   * ⚠️ 枚举失败**不能**当成「目录是空的」。
+   *
+   * 旧写法是 `.catch(() => [])`：读不动时静默返回空列表，于是生成一份 `count=0 / expected=0 /
+   * failed=[]` 的快照 —— 它在任何一个下游检查里都长得像「一次成功的快照」，而实际上它
+   * 什么都没存。回滚据此认为「退得回去」，`scope:'all'` 时更会把顶层文件全部当成
+   * 「快照里没有」而停进 `archive/`。
+   *
+   * `ENOENT` 是唯一合法的「这里没有文件」（记忆目录还没建）。其余一律出声：宁可让调用方
+   * 在写任何东西之前中止，也不要交出一份假的回滚点。
+   */
+  let entries
+  try {
+    entries = await readdir(this.memoryDir, { withFileTypes: true })
+  } catch (err) {
+    if (err?.code === 'ENOENT') entries = []
+    else throw new Error(`快照失败：读不了记忆目录（${err?.message ?? err}），拒绝生成一份不完整的回滚点`)
+  }
   const files = entries.filter((e) => e.isFile() && e.name.toLowerCase().endsWith('.md'))
   const recorded = []
   /** 没存下来的文件。**不是「警告一下就继续」**：调用方要拿它决定是否中止。 */

package/lib/autodream-audit.js

@@ -28,14 +28,17 @@ import { join } from 'node:path'
 const INDEX_FILE = 'MEMORY.md'
 
 /**
- * 非记忆条目：不是四类记忆、也不该套「type 与前缀一致」这条规则。
+ * 审计**整个跳过**的文件（不读、不当被审对象）。
  *
- * session-log.md 是追加型流水，逐会话的账本而不是一条记忆；
- * 它没有 `user_/feedback_/project_/reference_` 前缀，扫进来只会白白读一大坨
- * 再报一条假问题（实测：报出 `session-log.md type=reference`）。
- * 同理 MEMORY.md 是索引本身，只作为链接来源读，不作为被审对象。
+ * session-log.md 是追加型流水，逐会话的账本而不是一条记忆；它没有
+ * `user_/feedback_/project_/reference_` 前缀，扫进来只会白白读一大坨再报假问题。
+ *
+ * ⚠️ 别和 `lib/memory/config.js` 的 `NON_ENTRY_FILES` 混为一谈 —— **同名不同义**：
+ * 那个管「列不列进设置页 / 星图 / 检索」，这个管「审计读不读」。它们的差集就是
+ * **索引 `MEMORY.md`**：它**要**被读（要当链接来源、要查行尾与链接），只是
+ * **不是一条记忆** —— 见下面 `f !== INDEX_FILE` 那两处。
  */
-const NON_ENTRY_FILES = new Set(['session-log.md'])
+const AUDIT_SKIP_FILES = new Set(['session-log.md'])
 
 /** 归档区目录名（与 autodream-tools.js 的 ARCHIVE_DIR、宿主侧 memory/scan.js 同名）。 */
 const ARCHIVE_DIR = 'archive'
@@ -95,7 +98,7 @@ async function readDirSnapshot(dir) {
   const all = entries
     .filter((e) => e.isFile() && e.name.toLowerCase().endsWith('.md'))
     .map((e) => e.name)
-  const files = all.filter((f) => !NON_ENTRY_FILES.has(f.toLowerCase()))
+  const files = all.filter((f) => !AUDIT_SKIP_FILES.has(f.toLowerCase()))
   let indexText = ''
   try {
     indexText = await readFile(join(dir, INDEX_FILE), 'utf8')
@@ -126,6 +129,33 @@ async function readDirSnapshot(dir) {
   return { files, all, skipped: all.length - files.length, indexText, archiveKeys, archiveCount }
 }
 
+/**
+ * 把**代码**从正文里拿掉，只留散文 —— 用来避免把代码里的 `[[…]]` 当成链接。
+ *
+ * 两类都要剥：**围栏代码块**（跨行，```` ``` ```` 起止）与**行内代码**
+ * （`` `x` ``、`` `` x `` `` —— 后者的内容里可以再出现反引号，所以闭合必须找
+ * **等长**的那一串，不能用「遇到下一枚反引号就算完」）。
+ *
+ * ⚠️ **按行剥行内代码**：行内代码本来就极少跨行，按行做能把「反引号不配对」时的
+ * 损失限制在一行内，不至于把后面真正的链接整段吞掉。2026-10-09 实测踩过一次：
+ * 只认单反引号时，`` `` `[[双链]]` `` `` 这种写法会被漏掉、照样报假断链。
+ * @param {string} raw — 文件全文
+ */
+function stripCode(raw) {
+  const out = []
+  let fence = null
+  for (const line of raw.split(/\r?\n/)) {
+    const m = line.match(/^\s*(`{3,})/)
+    if (fence !== null) {
+      if (m && m[1].length >= fence.length) fence = null // 围栏闭合
+      continue // 围栏内的行整行丢弃（含起止那两行）
+    }
+    if (m) { fence = m[1]; continue }
+    out.push(line.replace(/(`+)[^\n]*?\1/g, ''))
+  }
+  return out.join('\n')
+}
+
 /**
  * 七查主入口。
  * @param {string} dir — 记忆根目录
@@ -175,13 +205,28 @@ export async function auditMemoryDir(dir) {
   }
   for (const f of readable) linkKeys.add(f)
   for (const f of readable) linkKeys.add(f.replace(/\.md$/, ''))
+  /**
+   * ⚠️ 链接目标池还必须含**非条目文件**（`session-log.md` / `MEMORY.md`）。
+   *
+   * 它们不在 `readable` 里（不参与检索），但 `[[session-log]]` 指向的是**真实存在的文件** ——
+   * 只因它不参与检索就报成断链，是假阳性。这与上面 `dangling` 用 `allSet` 而非 `files`
+   * 判定悬空是同一个理由，别只在一处想到。（2026-10-09 实测：健康记忆库被报 1 条假断链）
+   */
+  for (const f of all) {
+    linkKeys.add(f)
+    linkKeys.add(f.replace(/\.md$/i, ''))
+  }
 
   for (const f of readable) {
     const raw = contents.get(f)
     if (raw.includes('\r')) crlf.push(f)
     const fmMatch = raw.match(/^---\r?\n([\s\S]*?)\r?\n---/)
     if (!fmMatch) {
-      noFrontmatter.push(f)
+      /**
+       * 索引 `MEMORY.md` 是**纯清单、不是一条记忆**（见 `lib/memory/config.js` 的
+       * `NON_ENTRY_FILES` 说明），不要求 frontmatter。报它等于给每一个健康的记忆库挂一盏常驻红灯。
+       */
+      if (f !== INDEX_FILE) noFrontmatter.push(f)
     } else {
       const meta = parseFrontmatterLite(raw)
       const prefix = f.split('_')[0]
@@ -196,7 +241,14 @@ export async function auditMemoryDir(dir) {
         }
       }
     }
-    for (const m of raw.matchAll(/\[\[([^\]]+)\]\]/g)) {
+    /**
+     * ⚠️ 只在**非代码**文本里找双链。
+     *
+     * `` `[[双链]]` `` 写在行内代码或围栏代码块里，是在**讲链接语法本身**，不是一条链接。
+     * 直接在原文上 `matchAll` 会把它报成断链（2026-10-09 实测的假阳性）。
+     */
+    const prose = stripCode(raw)
+    for (const m of prose.matchAll(/\[\[([^\]]+)\]\]/g)) {
       const target = m[1].trim()
       if (linkKeys.has(target)) continue
       // 指向归档区的链接：单列提示、不计问题 —— 归档不是删除，目标还在库里躺着。

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "sage-mem",
   "description": "File-based memory plugin for DeepSeek Harness (DSH) — cross-session memory in plain markdown files, with a memory star-map view, type/tag filtering, retrieval aliases, and an optional autodream pass that consolidates memories on demand or on a schedule — with per-run rollback points, a per-file change declaration (what changed and why), and a separately selectable model for the consolidation pass",
-  "version": "0.9.0",
+  "version": "0.9.2",
   "type": "module",
   "main": "lib/index.js",
   "exports": {
@@ -15,6 +15,11 @@
     "cordis.patch.yml",
     "NOTICE"
   ],
+  "scripts": {
+    "test": "node test/run.mjs",
+    "build:client": "node scripts/build-client.mjs",
+    "images": "node scripts/make-readme-images.mjs"
+  },
   "license": "Apache-2.0",
   "engines": {
     "node": ">=18",
@@ -85,10 +90,5 @@
     "ai-agent",
     "agent",
     "claude-code"
-  ],
-  "scripts": {
-    "test": "node test/run.mjs",
-    "build:client": "node scripts/build-client.mjs",
-    "images": "node scripts/make-readme-images.mjs"
-  }
-}
+  ]
+}