dsh-fish-memory 1.4.4 → 1.4.5

package/lib/client.js

@@ -3949,6 +3949,36 @@ const 洗键进度 = { 已: 0, 总: 0, 在跑: false, 干跑: false }
               : '全关着 —— 哪个会话都不记也不读，工具列表里也不会出现它。',
           ),
         ),
+        // ⭐ 1.4.5：记忆按什么归类 —— 按会话 / 按工作区
+        React.createElement(
+          'div',
+          { style: Object.assign({}, S.row, S.条, { alignItems: 'center' }) },
+          React.createElement(
+            'button',
+            {
+              style: Object.assign({}, S.btn, {
+                flex: 'none',
+                color: d.模式 === 'workspace' ? '#8fd0a0' : 'inherit',
+                borderColor: d.模式 === 'workspace' ? 'rgba(143,208,160,.6)' : 'rgba(128,128,128,.35)',
+              }),
+              disabled: busy,
+              onClick: () => {
+                setBusy(true)
+                post('/mode', { 模式: d.模式 === 'workspace' ? 'session' : 'workspace' })
+                  .then(() => { setBusy(false); load() })
+                  .catch(() => setBusy(false))
+              },
+            },
+            (d.模式 === 'workspace' ? '按工作区' : '按会话'),
+          ),
+          React.createElement(
+            'div',
+            { style: Object.assign({}, S.dim, { flex: 1, minWidth: 0 }) },
+            d.模式 === 'workspace'
+              ? '整个工作区共用一个记忆库 —— 新开的会话自动接上，归档旧会话记忆也不会跟着走。'
+              : '每个会话各记各的。会话开久了变卡、要归档换新会话时，切成「按工作区」就接得上了。',
+          ),
+        ),
         (d.sessions || []).map((s) =>
           React.createElement(
             'div',

package/lib/index.js

@@ -39,6 +39,50 @@ const CFG_DIR = process.env.DSH_HOME || join(homedir(), '.dsh')
 const STORE_DIR = join(CFG_DIR, 'fish-memory')
 const BACKUP_DIR = join(CFG_DIR, 'fish-memory-backup')
 const CFG_PATH = join(CFG_DIR, 'fish-memory-config.json')
+
+/**
+ * ⭐ 1.4.5：工作区标识 —— 跟 DSH 自己的 `projectKey(cwd)` **一字不差**。
+ *
+ * DSH 把会话存在 `~/.dsh/sessions/<工作区目录名>/<会话id>/`，那层目录名就是它算出来的键。
+ * 规则（跟着 DSH 抄，别自己发明）：
+ *   · 分隔符（`/` `\` `:`）压成一个 `-`（连续的分隔符只留一个）；
+ *   · 只有 `[A-Za-z0-9._-]` 原样保留 —— 其余字符（含 `~`、`'`、中文）一律写成
+ *     `~` + 4 位大写十六进制（按 UTF-16 码元）；
+ *   · 去掉开头连续的 `-`（空了就是 `root`），截到 251 个字符，两头各包一个 `-`。
+ * 例：`C:\Users\me\Desktop\项目目录\dsh`
+ *   → `--C-Users-me-Desktop-~9879~76EE~76EE~5F55-dsh--`
+ *
+ * ⚠ 算法必须跟 DSH 完全一致：差一个字符，同一个工作区就会裂成两份。
+ *   单引号 / `~` 以前被简化成 `-` 就踩过这个坑（心情插件那边修过一次）。
+ */
+export function 工作区目录名(p) {
+  const s = String(p == null ? '' : p).trim()
+  if (!s) return ''
+  if (/^--.*--$/.test(s)) return s          // 给的本来就是目录名
+  let 可读 = ''
+  let 连续分隔 = false
+  for (let i = 0; i < s.length; i++) {
+    const ch = s[i]
+    if (ch === '/' || ch === '\\' || ch === ':') {
+      if (!连续分隔) 可读 += '-'
+      连续分隔 = true
+    } else if (ch !== '~' && /^[A-Za-z0-9._-]$/.test(ch)) {
+      可读 += ch
+      连续分隔 = false
+    } else {
+      可读 += '~' + ch.charCodeAt(0).toString(16).toUpperCase().padStart(4, '0')
+      连续分隔 = false
+    }
+  }
+  可读 = 可读.replace(/^-+/, '') || 'root'
+  return '--' + 可读.slice(0, 251) + '--'
+}
+
+/** 工作区的可读路径：从会话目录名反查不回来，就直接给目录名 */
+function 工作区可读名(键) {
+  return String(键 || '').replace(/^--/, '').replace(/--$/, '') || '（未分组）'
+}
+
 const DAY = 86400000
 // ⭐⭐ 2026-09-21（反馈：「别人导入会踩什么坑」时改的）：**时区不能再写死北京** ✗
 //   原来写死 +8。实测：换纽约、换东京，日期显示一模一样 ——
@@ -225,9 +269,93 @@ function diag(msg) {
 // ---------- 每个会话一份库 ----------
 
 function storePath(sid) {
-  return join(STORE_DIR, String(sid).replace(/[^a-zA-Z0-9_-]/g, '_') + '.json')
+  // ⭐ 1.4.5：工作区模式下，同一工作区的所有会话指向**同一个库文件**。
+  //   放在最底层改（而不是改每个调用点）—— 少一处漏改就少一个「这个库怎么没带上记忆」。
+  //   模式从配置里读一次（用户可能在设置页切）；读不到 / 算不出工作区都退回按会话，
+  //   绝不因为工作区算不出来就把记忆弄丢。
+  const 用名 = String(库名(sid, { 模式: 读模式() }) || sid)
+  return join(STORE_DIR, 用名.replace(/[^a-zA-Z0-9_-]/g, '_') + '.json')
+}
+
+// --------------------------------------------------------------------------
+// ⭐ 1.4.5：工作区模式
+//   DSH 的会话都躺在 `~/.dsh/sessions/<工作区目录名>/<会话 id>/`，所以：
+//     · 工作区标识 = 中间那层目录名（跨机器可复制、不依赖绝对路径）
+//     · 工作区模式下：**同一工作区的会话共用一个记忆库**
+//     · 跟「按会话」二选一：同时开等于一份记忆被两套归属抢，必出 bug
+//
+//   为什么需要：一个会话久了会大到打开都卡，为了流畅只能归档、开新会话。
+//   可记忆是按会话存的，新会话里什么都不记得 —— 工作区模式就是治这个的。
+// --------------------------------------------------------------------------
+const 工作区缓存 = { at: 0, 数据: null }
+
+function 目录们(p) {
+  try {
+    return readdirSync(p, { withFileTypes: true }).filter((e) => e.isDirectory()).map((e) => e.name)
+  } catch (error) {
+    return []
+  }
+}
+
+/**
+ * 扫一遍会话根目录：会话 → 工作区目录名、工作区目录名 → 会话 ids。
+ * 结果缓存 5 秒（心跳每 3 秒一次，别每次都读盘）。
+ * ⚠ 跟心情插件一个写法：**只认 CFG_DIR**，不给测试留参数（测试靠先设 DSH_HOME 再 import）。
+ */
+export function 扫会话目录(强制 = false) {
+  const 现在 = Date.now()
+  if (!强制 && 工作区缓存.数据 && 现在 - 工作区缓存.at < 5000) return 工作区缓存.数据
+  const 根 = join(CFG_DIR, 'sessions')
+  const bySession = new Map()    // 会话 id -> 工作区目录名
+  const byWorkspace = new Map()  // 工作区目录名 -> [会话 id]
+  for (const ws of 目录们(根)) {
+    const 会话们 = []
+    for (const sid of 目录们(join(根, ws))) {
+      const k = String(sid || '').trim()
+      if (!k) continue
+      会话们.push(sid)
+      if (!bySession.has(k)) bySession.set(k, ws)
+      // 老会话可能带 `session-` 前缀，两种写法都记上
+      if (k.startsWith('session-')) {
+        const 短 = k.slice('session-'.length)
+        if (短 && !bySession.has(短)) bySession.set(短, ws)
+      } else if (!bySession.has('session-' + k)) {
+        bySession.set('session-' + k, ws)
+      }
+    }
+    if (会话们.length) byWorkspace.set(ws, 会话们)
+  }
+  const 数据 = { bySession, byWorkspace, at: 现在 }
+  工作区缓存.data = 数据
+  工作区缓存.at = 现在
+  return 数据
 }
 
+/** 这个会话属于哪个工作区（认不出来就空串 → 退回按会话） */
+export function 会话的工作区(sid) {
+  const k = String(sid == null ? '' : sid).trim()
+  if (!k) return ''
+  const 表 = 扫会话目录()
+  return 表.bySession.get(k) || ''
+}
+
+/**
+ * ⭐ 1.4.5：工作区模式下，一份记忆归**哪个库**。
+ *
+ * 用工作区目录名当库名（加个前缀，别跟会话 id 撞）。这样：
+ *   · 同一个工作区里开多少新会话，读写的都是同一份；
+ *   · 归档旧会话不会把记忆带走 —— 新会话一开就接上了。
+ */
+export function 库名(sid, 配置) {
+  // ⚠ 不在这里读全局配置 —— 调用方把「模式」传进来。
+  //   自己读配置会让这个函数没法单测（而且 storePath 在模块加载早期就会被调用）。
+  const 模式 = String((配置 && 配置.模式) || 'session')
+  if (模式 !== 'workspace') return String(sid)
+  const ws = 会话的工作区(sid)
+  return ws ? 'ws_' + ws : String(sid)
+}
+
+
 // ⭐ 1.2.7（issue #7）：**全局参数**
 //   以前参数只存在每个记忆库（memory-store/<会话>.json）里 —— 一个库都没有时，
 //   保存等于没保存，F5 一刷就回默认值。
@@ -546,6 +674,17 @@ function setGlobalOn(on) {
   }
 }
 
+// ⭐ 1.4.5：记忆按什么归类 —— 'session'（默认，跟以前一样）/ 'workspace'（同一工作区共用一份）。
+//   两种模式二选一：同时开等于一份记忆被两套归属抢，必出 bug。
+function 读模式() {
+  try {
+    const c = loadCfg()
+    return String(c.模式 || 'session') === 'workspace' ? 'workspace' : 'session'
+  } catch (error) {
+    return 'session'
+  }
+}
+
 // 取消一个注册（DSH 那套返回的是**函数**，个别地方是 { off }）
 function 卸掉效果(d) {
   try {
@@ -3643,7 +3782,7 @@ function 分支来处Of(ctx, sid) {
 }
 
 function buildMemoryList(ctx) {
-  const out = { ok: true, sessions: [], policy: null, ghosts: 0, 全局开关: globalOn() }
+  const out = { ok: true, sessions: [], policy: null, ghosts: 0, 全局开关: globalOn(), 模式: 读模式() }
   // ⭐ 1.2.7：设置页显示「全局参数」—— 以前是「碰到的第一个库的参数」，没有库就什么都没有
   try {
     const c0 = loadCfg()
@@ -4544,8 +4683,25 @@ export function apply(ctx) {
       res.end(JSON.stringify({ ok: true, on }))
     })
 
-    mroute('/dsh-fish-memory/toggle', async (req, res) => {
+    // ⭐ 1.4.5：切换「按会话 / 按工作区」。切换不会搬动已有记忆 ——
+    //   只是此后读写的库文件换了，旧的还在（想合并得单独做，不在这条路上）。
+    mroute('/dsh-fish-memory/mode', async (req, res) => {
       let body = ''
+      for await (const chunk of req) body += chunk
+      let b = {}
+      try { b = JSON.parse(body || '{}') } catch (error) {}
+      const 模式 = String(b.模式 || '') === 'workspace' ? 'workspace' : 'session'
+      try {
+        const c = loadCfg()
+        c.模式 = 模式
+        saveCfg(c)
+        diag('记忆归类模式 → ' + 模式)
+      } catch (error) {}
+      res.writeHead(200, { 'content-type': 'application/json; charset=utf-8' })
+      res.end(JSON.stringify({ ok: true, 模式 }))
+    })
+
+    mroute('/dsh-fish-memory/toggle', async (req, res) => {      let body = ''
       for await (const chunk of req) body += chunk
       let b = {}
       try {
@@ -6620,3 +6776,15 @@ export {
   标作废,
   真重复门槛,
 }
+
+/**
+ * ⭐ 1.4.5：给自测用的出口（跟心情插件一个写法）。
+ *   ⚠ 只给测试用，不要在生产代码里调；里面都是纯函数或内部状态读取。
+ */
+export const __测试 = {
+  工作区目录名,
+  扫会话目录,
+  会话的工作区,
+  库名,
+  读模式,
+}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-fish-memory",
-  "version": "1.4.4",
+  "version": "1.4.5",
   "description": "给 DSH 里的会话一份会淡忘的长期记忆，每条都带时间。",
   "license": "MIT",
   "type": "module",