dsh-recall-plugin 1.0.2 → 1.0.4

package/README.md

@@ -11,7 +11,6 @@
 ---
 **在任意一条你发过的消息下方**，**点「↶ 撤回」**，**工作区文件和对话历史一起回到那条消息发出之前的状态**。
 
-
 ## 界面预览
 - 撤回按钮位置
 
@@ -35,8 +34,9 @@
 
 - 快照在**消息发送时**创建，插件启用前的历史消息没有快照，不显示撤回按钮。
 - 会话第一条用户消息无法回退对话（仅文件回退），因为 fork 需要更早的 turn 边界。
-- 仅支持 Windows（脚本用 PowerShell + git CLI）；Linux/macOS 下插件不报错，但快照不工作、不显示撤回按钮。
+- 仅支持 Windows（脚本用 PowerShell + git CLI）；Linux/macOS 下页面会弹一次性「仅支持 Windows」提示，快照不可用、不显示撤回按钮，也不会产生任何文件副作用。
 - 工作区内嵌套的其他 git 仓库（子目录自带 `.git`）不进快照，其内容不参与回退。
+- MacOS和Linux环境未进行测试
 
 ## 安装
 

package/lib/client.js

@@ -118,19 +118,29 @@ window.__ModuleLoader__.load({
           initedSessionId = sessionId
           initDone = api('init', { sessionId }).then((res) => {
             const notice = res && res.notice
+            if (notice && notice.unsupported) {
+              showNotice('unsupported', '撤回插件仅支持 Windows，当前平台的快照不可用。')
+            }
             if (notice && notice.gitMissing) {
               showNotice('git', '未检测到 git CLI，撤回功能不可用（快照引擎依赖 git）。安装 git 并重启 DSH 后即可使用。')
             }
             if (notice && notice.homeFallback) {
               showNotice('home', 'home 目录不可写，快照已降级存储到项目内 .dsh-recall-snapshots 目录。')
             }
-          }).catch(() => {})
+          }).catch(() => {
+            // init 失败（如页面先于 Host API 就绪加载）时清掉标记：否则本会话
+            // 内被判定“已初始化”，撤回按钮永不出现；清掉后下一条消息挂载会重试
+            if (initedSessionId === sessionId) initedSessionId = null
+          })
           return initDone
         }
 
         // 消息时间：当天只显示时分，跨天显示月/日 时分
         function clockText(ms) {
           try {
+            // time 字段缺失或非法时返回空串：Invalid Date 不会 throw，
+            // 不拦会渲染出 "NaN/NaN NaN:NaN" 这样的坏时间戳
+            if (!ms || isNaN(new Date(ms).getTime())) return ''
             const d = new Date(ms)
             const now = new Date()
             const sameDay = d.getFullYear() === now.getFullYear() && d.getMonth() === now.getMonth() && d.getDate() === now.getDate()

package/lib/index.js

@@ -19,6 +19,13 @@ export const inject = ['shell', 'sessions', 'webServer']
 const MAX_FILE_BYTES = 104857600
 const HOME_RETRY_MS = 300000
 
+// 统一 UTF-8 输出前导：中文等非 ASCII 机器的默认代码页（如 GBK）下，
+// PowerShell 重定向 stdout 按 [Console]::OutputEncoding 编码，而 DSH 按
+// UTF-8 解码——不强制时 homeDirFor/resolveGit 输出里含中文的用户名/路径
+// 会变乱码，store 会指向错误目录。diff/rollback 脚本里早有同款设置，
+// 这里提到 runShell 全局注入一次，其余脚本不改自动受益（PS 5.1 / 7 均支持）。
+const UTF8_PRELUDE = '$OutputEncoding = [Text.UTF8Encoding]::new($false)\ntry { [Console]::OutputEncoding = [Text.UTF8Encoding]::new($false) } catch {}'
+
 export function apply(ctx) {
   const shell = ctx.shell
   const sessions = ctx.sessions
@@ -36,6 +43,12 @@ export function apply(ctx) {
     gitExe: null
   }
 
+  // 非 Windows 平台干净降级：脚本体系（反斜杠路径、Windows git 安装候选、
+  // Expand-Archive）绑定 Windows，硬跑不报错但会在项目里创建名字带反斜杠的
+  // 垃圾目录。这里整体短路：init 返回 unsupported，Client 弹一次性提示；
+  // 其余端点因无快照自然返回「没有可用快照」，全程零文件副作用。
+  const supported = process.platform === 'win32'
+
   function psq(value) {
     return "'" + String(value).replace(/'/g, "''") + "'"
   }
@@ -52,7 +65,7 @@ export function apply(ctx) {
   async function runShell(command, opts) {
     const sp = ctx.get('sandboxPolicy')
     const spec = shell.resolve({
-      command,
+      command: UTF8_PRELUDE + '\n' + command,
       timeoutMs: (opts && opts.timeoutMs) || 300000,
       stdoutMaxBytes: (opts && opts.stdoutMaxBytes) || 4194304,
       sandboxPolicy: { mode: 'danger-full-access', workspaceRoot: (sp && sp.workspaceRoot) || process.cwd() }
@@ -83,16 +96,35 @@ export function apply(ctx) {
       const sp = ctx.get('sandboxPolicy')
       if (sp && sp.workspaceRoot) root = sp.workspaceRoot
     }
+    // 规范化尾部反斜杠（保留 "D:\" 三字符盘根形态，去掉会变成无效的 "D:"）：
+    // cwd 是否带尾斜杠由上游决定，不归一会让哈希输入不一致（换 store 目录），
+    // 也会让大文件排除扫描的 Substring($root.Length + 1) 相对路径错一位。
+    if (root && root.length > 3) root = root.replace(/\\+$/, '')
     if (root) state.roots.set(key, root)
     return root
   }
 
   // 解析 git 可执行文件路径：DSH 进程 PATH 可能不含 git，
   // 求值一次并缓存，脚本里用绝对路径调用，避免每条命令依赖 PATH。
+  // 候选覆盖 PATH / 64 位 / 32 位 / 用户级（LocalAppData）四类安装位置，
+  // 并用 -PathType Leaf 挡掉 PATH 里恰有名为 git 的目录这种极端情形。
   async function resolveGit() {
     if (state.gitExe !== null) return state.gitExe
     try {
-      let path = stripBom(await runShell("$g = (Get-Command git -ErrorAction SilentlyContinue).Source; if (-not $g -and (Test-Path -LiteralPath 'C:\\Program Files\\Git\\cmd\\git.exe')) { $g = 'C:\\Program Files\\Git\\cmd\\git.exe' }; if ($g) { Write-Output $g }", { stdoutMaxBytes: 4096 })).trim()
+      // 逐项判空再 Join-Path：个别 env 在特殊环境（32 位系统无
+      // ProgramFiles(x86)）取到 null，EAP=Stop 下 Join-Path 抛错会
+      // 让整个探测失败、误报 gitMissing。
+      const script = [
+        '$candidates = @()',
+        '$g = (Get-Command git -ErrorAction SilentlyContinue).Source',
+        'if ($g) { $candidates += $g }',
+        "if (${env:ProgramFiles}) { $candidates += (Join-Path ${env:ProgramFiles} 'Git\\cmd\\git.exe') }",
+        "if (${env:ProgramFiles(x86)}) { $candidates += (Join-Path ${env:ProgramFiles(x86)} 'Git\\cmd\\git.exe') }",
+        "if (${env:LocalAppData}) { $candidates += (Join-Path ${env:LocalAppData} 'Programs\\Git\\cmd\\git.exe') }",
+        "$g = $candidates | Where-Object { $_ -and (Test-Path -LiteralPath $_ -PathType Leaf) } | Select-Object -First 1",
+        'if ($g) { Write-Output $g }'
+      ].join('\n')
+      const path = stripBom(await runShell(script, { stdoutMaxBytes: 4096 })).trim()
       state.gitExe = path || ''
     } catch (error) {
       state.gitExe = ''
@@ -105,9 +137,13 @@ export function apply(ctx) {
   // 后两者是 .NET 5+（仅 PS 7）API，别人机器的 shell 若是 Windows PowerShell
   // 5.1 会抛错，导致 home 存储永远降级到项目内；前者两个版本都可用。
   async function homeDirFor(root, sessionId) {
+    // DSH 的 pwsh 子进程按白名单重建 env，会话级导出的 DSH_HOME 传不进去；
+    // 主进程（node）仍能看到它，作为字面量回退注入，保证「DSH_HOME 指到哪、
+    // 快照就存哪」在任意导出层级下都成立。
+    const envHome = (process.env && process.env.DSH_HOME) || ''
     const dirScript = [
       '$r = ' + psq(root),
-      '$h = if ($env:DSH_HOME) { $env:DSH_HOME } else { Join-Path $env:USERPROFILE ".dsh" }',
+      "$h = if ($env:DSH_HOME) { $env:DSH_HOME } elseif (" + psq(envHome) + ") { " + psq(envHome) + " } else { Join-Path $env:USERPROFILE \".dsh\" }",
       '$sha = [Security.Cryptography.SHA256]::Create()',
       "$hex = ([BitConverter]::ToString($sha.ComputeHash([Text.Encoding]::UTF8.GetBytes($r))) -replace '-','').ToLower()",
       "Write-Output (Join-Path $h ('dsh-recall-snapshots\\' + $hex))"
@@ -117,6 +153,9 @@ export function apply(ctx) {
     // PS 的 Join-Path 可能带出连续反斜杠（旧版脚本遗留 "…\\<hex>" 形态）；
     // 折叠成单反斜杠。Windows 把中间的重复分隔符视为同一个目录，
     // 所以与既有快照数据（index.json、影子 git 仓库）路径完全兼容。
+    // 开头的双反斜杠是 UNC 前缀（DSH_HOME/用户主目录指到网络盘），
+    // 折叠掉会把 \\server\share 变成无效的 \server\share，必须原样保留。
+    if (/^\\\\/.test(text)) return '\\\\' + text.slice(2).replace(/\\{2,}/g, '\\')
     return text.replace(/\\{2,}/g, '\\')
   }
 
@@ -260,7 +299,10 @@ export function apply(ctx) {
       dropGitlinksScript(),
       '& $git --git-dir=$g --work-tree=$root add -A',
       dropGitlinksScript(),
-      'Get-ChildItem -LiteralPath $root -Recurse -File -Force | Where-Object { $_.Length -gt ' + MAX_FILE_BYTES + ' } | ForEach-Object {',
+      // 扫描加 SilentlyContinue：EAP=Stop 下个别不可访问子目录（杀软锁定、
+      // 异常 ACL、损坏 junction）的非致命错误会被升级为终止，整条快照作废；
+      // 本扫描只用于排除超大文件，漏看个别文件是 fail-open，可接受。
+      'Get-ChildItem -LiteralPath $root -Recurse -File -Force -ErrorAction SilentlyContinue | Where-Object { $_.Length -gt ' + MAX_FILE_BYTES + ' } | ForEach-Object {',
       "  $rel = $_.FullName.Substring($root.Length + 1).Replace('\\','/')",
       '  & $git --literal-pathspecs --git-dir=$g update-index --force-remove -- $rel',
       '}',
@@ -290,7 +332,7 @@ export function apply(ctx) {
       dropGitlinksScript(),
       '& $git --git-dir=$g --work-tree=$root add -A',
       dropGitlinksScript(),
-      'Get-ChildItem -LiteralPath $root -Recurse -File -Force | Where-Object { $_.Length -gt ' + MAX_FILE_BYTES + ' } | ForEach-Object {',
+      'Get-ChildItem -LiteralPath $root -Recurse -File -Force -ErrorAction SilentlyContinue | Where-Object { $_.Length -gt ' + MAX_FILE_BYTES + ' } | ForEach-Object {',
       "  $rel = $_.FullName.Substring($root.Length + 1).Replace('\\','/')",
       '  & $git --literal-pathspecs --git-dir=$g update-index --force-remove -- $rel',
       '}',
@@ -342,7 +384,7 @@ export function apply(ctx) {
       dropGitlinksScript(),
       '& $git --git-dir=$g --work-tree=$root add -A',
       dropGitlinksScript(),
-      'Get-ChildItem -LiteralPath $root -Recurse -File -Force | Where-Object { $_.Length -gt ' + MAX_FILE_BYTES + ' } | ForEach-Object {',
+      'Get-ChildItem -LiteralPath $root -Recurse -File -Force -ErrorAction SilentlyContinue | Where-Object { $_.Length -gt ' + MAX_FILE_BYTES + ' } | ForEach-Object {',
       "  $rel = $_.FullName.Substring($root.Length + 1).Replace('\\','/')",
       '  & $git --literal-pathspecs --git-dir=$g update-index --force-remove -- $rel',
       '}',
@@ -390,7 +432,11 @@ export function apply(ctx) {
     ].join('\n')
   }
 
-  // 索引写入合并为单次 shell 调用：pwsh 进程启动是主要耗时，能省一次是一次
+  // 索引写入合并为单次 shell 调用：pwsh 进程启动是主要耗时，能省一次是一次。
+  // base64 以内联字面量传递时受 Windows 命令行 32767 字符硬上限约束（DSH 的
+  // pwsh 执行器把命令串作为 -Command 的单个 argv 元素 spawn），快照攒到几百条
+  // 就会超限 spawn 失败——按 20000 字符分块，首块 Set-Content、续块 Add-Content，
+  // 常规体量仍是单次调用，超限后自动多写几块。
   async function saveIndex(root, sessionId) {
     const store = state.stores.get(root)
     if (!store) return
@@ -400,7 +446,16 @@ export function apply(ctx) {
     const json = JSON.stringify(entries)
     try {
       const b64 = Buffer.from(json, 'utf8').toString('base64')
-      await runShell("New-Item -ItemType Directory -Force -Path " + psq(store.dir) + " | Out-Null; [Text.Encoding]::UTF8.GetString([Convert]::FromBase64String('" + b64 + "')) | Set-Content -LiteralPath " + psq(store.dir + '\\index.json') + ' -Encoding utf8 -NoNewline', { stdoutMaxBytes: 4096 })
+      const file = psq(store.dir + '\\index.json')
+      let first = true
+      for (let i = 0; i < b64.length; i += 20000) {
+        const piece = "[Text.Encoding]::UTF8.GetString([Convert]::FromBase64String('" + b64.slice(i, i + 20000) + "')) | "
+        const cmd = first
+          ? 'New-Item -ItemType Directory -Force -Path ' + psq(store.dir) + ' | Out-Null; ' + piece + 'Set-Content -LiteralPath ' + file + ' -Encoding utf8 -NoNewline'
+          : piece + 'Add-Content -LiteralPath ' + file + ' -Encoding utf8 -NoNewline'
+        await runShell(cmd, { stdoutMaxBytes: 4096 })
+        first = false
+      }
     } catch (error) {
       console.error('recall saveIndex failed:', String(error))
     }
@@ -568,6 +623,10 @@ export function apply(ctx) {
       try {
         const args = await readJsonBody(req)
         if (name === 'init') {
+          if (!supported) {
+            sendJson(res, 200, { ok: false, root: null, notice: { unsupported: true } })
+            return
+          }
           const sessionId = args && args.sessionId ? String(args.sessionId) : null
           const root = await resolveRoot(sessionId)
           let notice = null
@@ -631,6 +690,9 @@ export function apply(ctx) {
     }
   }))
 
+  // 快照事件与启动预热仅在受支持平台注册（见上方 supported 短路说明）
+  if (!supported) return
+
   // 每条用户消息触发快照（子代理会话跳过）
   ctx.on('session/event', (session, event) => {
     if (!event || event.type !== 'user/message') return

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "dsh-recall-plugin",
-  "version": "1.0.2",
+  "version": "1.0.4",
   "description": "DSH 消息撤回插件：在用户消息气泡旁加「撤回」按钮，把项目文件（独立影子 git 仓库快照）与对话历史（官方 fork）一并回退到该消息发送之前。",
   "type": "module",
   "repository": {