dsh-plugin-memory-share 0.3.1 → 0.4.0

package/README.md

@@ -143,6 +143,8 @@ dsh plugin --profile desktop add "file:C:/path/to/dsh-plugin-MemoryShare"
 | `relevance` | `true` | 按当前会话最近发言给 workspace/session 条目打分，0 分不注入；`user` 条目永远注入 |
 | `relevanceTopK` | `4` | 筛选后最多保留几条非 user 条目 |
 | `relevanceKeepRecent` | `0` | 全都不相关时退回最近 n 条；`0` = 一条不留 |
+| `bootSnapshot` | `live` | `live` 每步按最新记忆渲染；`session` 本会话首帧冻结，记忆再变也不失效前缀缓存 |
+| `disabledSessions` | `[]` | 这些会话 id 不注入；也可用 `ctx.memoryShare.setSessionEnabled(id, bool)` 运行时开关 |
 | `injectInPrompt` | `true` | 关掉后只剩工具，不再被动注入 |
 | `guidance` | `true` | 是否附带那段用法指引 |
 | `refreshIntervalMs` | `30000` | 跨进程同步的轮询间隔；`0` = 只在启动/建会话时读 |
@@ -288,6 +290,15 @@ function ensureService(ctx) {
 
 ---
 
+## 状态检查
+
+```sh
+dsh-plugin-memory-share          # 人读
+dsh-plugin-memory-share --json   # 给脚本用
+```
+
+输出磁盘版本、**运行中版本**（插件启动时写的 `runtime.json`）、条目数与作用域分布。两边版本不一致时退出码为 1——用来回答「我装的和跑的是不是同一个版本」。
+
 ## 开发
 
 ```bash

package/bin/status.mjs

@@ -0,0 +1,79 @@
+#!/usr/bin/env node
+// dsh-plugin-memory-share 状态检查：磁盘版本 vs 运行中版本 vs 记忆库概况。
+// 用法：dsh-plugin-memory-share [--json]
+// 退出码：0 一致 / 1 版本不一致或库损坏 / 2 无法读取
+import { readFile, stat } from 'node:fs/promises'
+import { join } from 'node:path'
+import { homedir } from 'node:os'
+
+const asJson = process.argv.includes('--json')
+const dataDir = process.env.MEMORY_SHARE_DIR
+  || join(process.env.DSH_HOME || join(homedir(), '.dsh'), 'memory-share')
+
+async function readJson(path) {
+  try {
+    return JSON.parse(await readFile(path, 'utf8'))
+  } catch {
+    return undefined
+  }
+}
+
+const pkg = await readJson(new URL('../package.json', import.meta.url))
+const runtime = await readJson(join(dataDir, 'runtime.json'))
+const store = await readJson(join(dataDir, 'memories.json'))
+
+let storeStat
+try {
+  storeStat = await stat(join(dataDir, 'memories.json'))
+} catch {
+  storeStat = undefined
+}
+
+const entries = Array.isArray(store?.entries) ? store.entries : []
+const counts = { user: 0, workspace: 0, session: 0 }
+for (const entry of entries) {
+  if (counts[entry.scope] !== undefined) counts[entry.scope] += 1
+}
+
+const diskVersion = pkg?.version ?? 'unknown'
+const runVersion = runtime?.version
+const consistent = runVersion === undefined ? null : runVersion === diskVersion
+
+const report = {
+  dataDir,
+  diskVersion,
+  running: runtime === undefined ? null : {
+    version: runVersion,
+    pid: runtime.pid,
+    startedAt: runtime.startedAt,
+    node: runtime.node,
+  },
+  consistent,
+  memories: {
+    total: entries.length,
+    counts,
+    bytes: storeStat?.size ?? 0,
+    updatedAt: storeStat?.mtime?.toISOString() ?? null,
+  },
+}
+
+if (asJson) {
+  process.stdout.write(`${JSON.stringify(report, null, 2)}\n`)
+} else {
+  const line = (label, value) => process.stdout.write(`${label.padEnd(14)}${value}\n`)
+  process.stdout.write('dsh-plugin-memory-share\n')
+  process.stdout.write('─'.repeat(46) + '\n')
+  line('数据目录', dataDir)
+  line('磁盘版本', diskVersion)
+  if (runtime === undefined) {
+    line('运行中版本', '（没有 runtime.json —— 插件未在此机器上启动过，或从未重启过后加载）')
+  } else {
+    line('运行中版本', `${runVersion}${consistent ? '  ✓ 一致' : '  ✗ 与磁盘不一致，需要重启 DSH'}`)
+    line('进程', `pid ${runtime.pid}　启动于 ${runtime.startedAt}`)
+  }
+  line('记忆条目', `${entries.length} 条　全局 ${counts.user} · 工作区 ${counts.workspace} · 会话 ${counts.session}`)
+  line('文件大小', `${(report.memories.bytes / 1024).toFixed(1)} KiB`)
+  line('最后写入', report.memories.updatedAt ?? '（还没有记忆库）')
+}
+
+process.exit(consistent === false ? 1 : 0)

package/lib/index.js

@@ -10,6 +10,8 @@
 // 这样插件能装在任意 profile 下而不需要 DSH 的内部模块解析（与 dsh-agentmemory 的做法一致，但它多依赖了 dsh-tools）。
 // License: MIT
 import { MemoryStore, defaultDataDir, normalizeKey, SCOPES } from './store.js'
+import { mkdir, readFile, writeFile } from 'node:fs/promises'
+import { join } from 'node:path'
 import { renderMemoryBlock } from './render.js'
 
 export const name = 'memoryShare'
@@ -19,6 +21,8 @@ export const inject = ['systemPrompt', 'tools']
 /** 提示词段的名称与位置：100 = persona 之后、工具指引之前。 */
 const SECTION_NAME = 'memory-share'
 const SECTION_ORDER = 100
+/** bootSnapshot=session 时最多冻结多少会话的块，防止无界增长。 */
+const FROZEN_LIMIT = 200
 
 const DEFAULTS = {
   dataDir: '',
@@ -34,6 +38,10 @@ const DEFAULTS = {
   relevanceTopK: 4,
   /** 全都不相关时退回最近 n 条；0 = 一条都不留（严格模式）。 */
   relevanceKeepRecent: 0,
+  /** 'live' = 每步按最新记忆渲染；'session' = 本会话首次渲染后冻结，保护前缀缓存。 */
+  bootSnapshot: 'live',
+  /** 这些会话 id 不注入（服务方法可运行时开关）。 */
+  disabledSessions: [],
   guidance: true,
   refreshIntervalMs: 30000,
 }
@@ -128,11 +136,35 @@ function normalizeOptions(config) {
   options.relevance = options.relevance !== false
   options.relevanceTopK = Math.max(0, Math.floor(Number(options.relevanceTopK) || 0))
   options.relevanceKeepRecent = Math.max(0, Math.floor(Number(options.relevanceKeepRecent) || 0))
+  options.bootSnapshot = options.bootSnapshot === 'session' ? 'session' : 'live'
+  options.disabledSessions = Array.isArray(options.disabledSessions)
+    ? options.disabledSessions.map((id) => String(id)).filter(Boolean)
+    : []
   options.guidance = options.guidance !== false
   options.refreshIntervalMs = Math.max(0, Math.floor(Number(options.refreshIntervalMs) || 0))
   return options
 }
 
+/**
+ * 写一份「运行中实例」标记，供 CLI 对比磁盘版本与进程版本。
+ * 失败不影响插件 —— 最多少一条诊断信息。
+ */
+async function writeRuntimeMarker(dir) {
+  try {
+    const pkg = JSON.parse(await readFile(new URL('../package.json', import.meta.url), 'utf8'))
+    await mkdir(dir, { recursive: true })
+    await writeFile(join(dir, 'runtime.json'), JSON.stringify({
+      name: pkg.name,
+      version: pkg.version,
+      pid: process.pid,
+      startedAt: new Date().toISOString(),
+      node: process.version,
+    }, null, 2))
+  } catch {
+    /* 忽略 */
+  }
+}
+
 export function apply(ctx, config = {}) {
   const options = normalizeOptions(config)
   const store = new MemoryStore({
@@ -141,6 +173,11 @@ export function apply(ctx, config = {}) {
     logger: ctx.logger,
   })
 
+  // 会话级开关与 boot 快照（bootSnapshot: 'session' 时按会话冻结渲染结果）
+  const sessionOff = new Set(options.disabledSessions)
+  const frozenBlocks = new Map()
+  void writeRuntimeMarker(options.dataDir)
+
   const warn = (message) => {
     try {
       (ctx.logger?.warn ?? console.warn)(`[memory-share] ${message}`)
@@ -184,6 +221,11 @@ export function apply(ctx, config = {}) {
       if (options.injectFor === 'top' && isDerivedAgent(context?.agent)) return ''
       try {
         const { sessionId, cwd } = identityOf(context?.agent)
+        if (sessionOff.has(sessionId)) return ''
+        if (options.bootSnapshot === 'session') {
+          const frozen = frozenBlocks.get(sessionId)
+          if (frozen !== undefined) return frozen
+        }
         const visible = store.selectFor({ cwd, sessionId })
         const queryText = options.relevance ? recentText.get(sessionId) : undefined
         const entries = queryText === undefined || queryText.trim() === ''
@@ -194,12 +236,17 @@ export function apply(ctx, config = {}) {
             topK: options.relevanceTopK,
             keepRecent: options.relevanceKeepRecent,
           })
-        return renderMemoryBlock({
+        const block = renderMemoryBlock({
           entries,
           maxChars: options.maxChars,
           entryChars: options.entryChars,
           guidance: options.guidance,
         })
+        if (options.bootSnapshot === 'session' && sessionId !== '') {
+          if (frozenBlocks.size >= FROZEN_LIMIT) frozenBlocks.delete(frozenBlocks.keys().next().value)
+          frozenBlocks.set(sessionId, block)
+        }
+        return block
       } catch (error) {
         warn(`渲染记忆失败：${error?.message ?? error}`)
         return ''
@@ -437,6 +484,19 @@ export function apply(ctx, config = {}) {
     async forget(id) {
       return { removed: (await store.remove(id)) ? 1 : 0 }
     },
+    /** 会话级开关：false = 该会话不再注入；下次 assemble 生效。 */
+    async setSessionEnabled(sessionId, enabled) {
+      const id = String(sessionId ?? '')
+      if (id === '') throw new Error('memory-share: sessionId 不能为空')
+      if (enabled === false) sessionOff.add(id)
+      else sessionOff.delete(id)
+      frozenBlocks.delete(id)
+      return { sessionId: id, enabled: enabled !== false }
+    },
+    /** 当前被关掉的会话、以及已冻结的会话。 */
+    async listSessions() {
+      return { disabled: [...sessionOff], frozen: [...frozenBlocks.keys()] }
+    },
   }
   // Typert 反射绑定：网页客户端据此把这三个方法挂到 ctx.remote.memoryShare 上。
   service.typertRemote = Object.freeze({ service, serviceKey: 'memoryShare', namespace: 'memoryShare' })

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-plugin-memory-share",
-  "version": "0.3.1",
+  "version": "0.4.0",
   "description": "dsh-plugin-MemoryShare — 跨会话共享的长期记忆：被动注入系统提示词，配 memory_save / memory_recall / memory_forget 三个工具，以及「设置 → 插件 → 记忆」管理页。",
   "type": "module",
   "license": "MIT",
@@ -27,6 +27,7 @@
   },
   "files": [
     "lib",
+    "bin",
     "cordis.patch.yml",
     "README.md",
     "LICENSE"
@@ -75,5 +76,8 @@
         "@deepseek-ai/dsh-api-gateway"
       ]
     }
+  },
+  "bin": {
+    "dsh-plugin-memory-share": "bin/status.mjs"
   }
 }