dsh-plugin-windows-guard 0.1.0 → 0.2.1

package/scripts/smoke-server.mjs

@@ -1,10 +1,17 @@
-// dsh-plugin-windows-guard 服务端冒烟：假 ctx 走注册全链路（纯数据插件）。
+// dsh-plugin-windows-guard 服务端冒烟：假 ctx 走全链路
+// （技能注册 → 工具注册 → pre/post hook 决策 → 工具执行 → 修复往返 → 清理）。零依赖。
 import assert from 'node:assert'
+import fs from 'node:fs'
+import { tmpdir } from 'node:os'
+import { join } from 'node:path'
 import * as mod from '../lib/index.js'
 
-// 假 ctx：收集注册与日志
+// ---- 构造假 ctx ----
 const registered = []
+const registrations = {}
+const listeners = {}
 const logs = []
+
 const ctx = {
   skills: {
     register: (def) => {
@@ -12,15 +19,17 @@ const ctx = {
       return () => {} // disposer
     }
   },
+  tools: { register: (def) => { registrations[def.name] = def; return () => {} } },
+  on: (event, fn) => { listeners[event] = fn; return () => {} },
 }
 const origLog = console.log
 console.log = (...args) => { logs.push(args.join(' ')) }
 
-mod.apply(ctx)
+const dispose = mod.apply(ctx)
 
 console.log = origLog
 
-// 断言：拆分后的两个技能
+// ---- 断言：拆分后的两个技能 ----
 const names = registered.map((s) => s.name).sort()
 assert.deepEqual(names, ['windows-enc', 'windows-sys'], '应注册 windows-enc 与 windows-sys')
 for (const skill of registered) {
@@ -34,4 +43,67 @@ const sys = registered.find((s) => s.name === 'windows-sys')
 assert.ok(sys.content.includes('Stop-Process'), 'windows-sys 正文含进程守则')
 assert.ok(logs.some((l) => l.includes('技能已注册：windows-enc')), '日志有注册记录')
 
-console.log(`smoke: 全部通过（注册 ${registered.length} 个技能：${registered.map((s) => s.name).join(', ')}）`)
+// ---- 断言：合并的主动防护层（0.2.0）----
+assert.ok(registrations.windows_encode_detect, '注册 windows_encode_detect')
+assert.ok(registrations.windows_encode_fix, '注册 windows_encode_fix')
+assert.ok(listeners['tools/pre-execute'] && listeners['tools/post-execute'], '注册 hooks')
+
+// ---- pre-execute 决策 ----
+const denyExec = { name: 'pwsh', arguments: { command: 'Set-Content -Path a.json -Value $x' } }
+const preDeny = await listeners['tools/pre-execute'](denyExec, async () => ({ kind: 'allow' }))
+assert.equal(preDeny.kind, 'deny', '危险写命令被拒')
+assert.ok(preDeny.reason.includes('[windows-guard]'), '拒绝理由带 [windows-guard] 标签')
+const preAllow = await listeners['tools/pre-execute']({ name: 'pwsh', arguments: { command: 'Get-Content x.md' } }, async () => ({ kind: 'allow' }))
+assert.equal(preAllow.kind, 'allow', '读命令放行')
+
+// ---- post-execute 决策（乱码 → 附加提示）----
+const postHit = await listeners['tools/post-execute'](
+  { name: 'pwsh' },
+  { isError: false, content: [{ type: 'text', text: '# DSH 鍚姩鏁呴殰璁板綍锛堜簨鏁呭洓鎺樼栵?' }] },
+  async () => ({ kind: 'accept', content: [{ type: 'text', text: 'fallback' }] })
+)
+assert.equal(postHit.kind, 'accept')
+assert.ok(postHit.content.some((c) => typeof c.text === 'string' && c.text.includes('[windows-guard]')), '附加 windows-guard 提示')
+assert.ok(postHit.content.some((c) => typeof c.text === 'string' && c.text.includes('windows_encode_detect')), '提示引导到新工具名')
+
+const postClean = await listeners['tools/post-execute'](
+  { name: 'pwsh' },
+  { isError: false, content: [{ type: 'text', text: 'all good here' }] },
+  async () => ({ kind: 'accept', content: [{ type: 'text', text: 'fallback' }] })
+)
+assert.equal(postClean.kind, 'accept')
+assert.ok(!postClean.content.some((c) => typeof c.text === 'string' && c.text.includes('[windows-guard]')), '正常输出不附加')
+
+// ---- 非 pwsh 工具不受影响 ----
+const postOther = await listeners['tools/post-execute'](
+  { name: 'read' },
+  { isError: false, content: [{ type: 'text', text: '鎻掍欢鏋勮瑙勮寖' }] },
+  async () => ({ kind: 'accept', content: [{ type: 'text', text: 'fallback' }] })
+)
+assert.equal(postOther.kind, 'accept')
+assert.ok(!postOther.content.some((c) => typeof c.text === 'string' && c.text.includes('[windows-guard]')), '非 pwsh 工具不附加')
+
+// ---- 工具执行 ----
+const tmp = join(tmpdir(), 'windowsguard-smoke')
+fs.mkdirSync(tmp, { recursive: true })
+const f = join(tmp, 'sample.gbk.txt')
+fs.writeFileSync(f, Buffer.from([0xb2, 0xe5, 0xbc, 0xfe, 0xb9, 0xe6, 0xb7, 0xb6])) // GBK 插件规范(近似)
+const fakeExec = { agent: { session: { header: { cwd: tmp } } } }
+const detectResult = await registrations.windows_encode_detect.execute({ path: 'sample.gbk.txt' }, fakeExec)
+assert.equal(detectResult.encoding, 'gbk', 'GBK 文件被正确检出')
+const dryRun = await registrations.windows_encode_fix.execute({ path: 'sample.gbk.txt', dryRun: true }, fakeExec)
+assert.ok(dryRun.notice.includes('dryRun'), 'dryRun 不写文件')
+assert.ok(!fs.existsSync(`${f}.windowsguard.bak`), 'dryRun 无备份')
+const fixResult = await registrations.windows_encode_fix.execute({ path: 'sample.gbk.txt' }, fakeExec)
+assert.equal(fixResult.before, 'gbk')
+assert.equal(fixResult.after, 'utf-8')
+const reread = fs.readFileSync(f, 'utf8')
+assert.ok(reread.includes('插件'), '修复后为 UTF-8 中文')
+assert.ok(fs.existsSync(`${f}.windowsguard.bak`), '备份存在（.windowsguard.bak）')
+
+// ---- 清理函数可调用（HMR 幂等）----
+dispose()
+
+fs.rmSync(tmp, { recursive: true, force: true })
+
+console.log(`smoke: 全部通过（技能 ${registered.length} + 工具 ${Object.keys(registrations).length} + hooks ${Object.keys(listeners).length}，修复往返 OK）`)

package/skills/windows-enc.md

@@ -30,7 +30,11 @@ whenToUse: 读/写文本与配置文件、处理中文输出、命令引号转
 | pwsh 7 写文件 | `Out-File -Encoding utf8`（5.1 下 = 带 BOM；7 下无 BOM） | `Out-File -Encoding utf8NoBOM`（仅 pwsh 7） |
 
 补充：
-- **查文件编码**（前 3 字节嗅探）：`Format-Hex -Path a.json -Count 3`——
+- **配套零 token 工具**（本插件注册，优先于手写嗅探/转换命令）：
+  `windows_encode_detect`（诊断 BOM/UTF-8/GBK/UTF-16 + 置信度）与
+  `windows_encode_fix`（转成 UTF-8 无 BOM，自动备份 `<path>.windowsguard.bak`，
+  支持 `dryRun`）。遇到乱码或 JSON/YAML 解析失败时，先 detect 后 fix，再复检。
+- **查文件编码**（无工具可用时的替代）：`Format-Hex -Path a.json -Count 3`——
   `EF BB BF` = UTF-8 BOM（解析器可能拒绝）；`FF FE` = UTF-16LE（必然炸 JSON）；
   无 BOM 且含 `FF FD` 替换符 = 文件本来就被写坏了。
 - **GBK→UTF-8 转换**（确证文件是 GBK 时，PowerShell 无直接 cmdlet，用 .NET）：
@@ -64,3 +68,6 @@ whenToUse: 读/写文本与配置文件、处理中文输出、命令引号转
 1. **不猜**——乱码文本语义不可信，别试图「意思理解一下」继续用；
 2. **不引用**——不要把它当答案、文件名、配置值传给下游；
 3. **先重读**——回到源头（原文件原命令）按守则重新读/重新执行。
+
+附：pwsh 工具输出出现乱码时，本插件会自动在结果后追加 `[windows-guard]` 修复提示；
+文件类乱码直接走 `windows_encode_detect` / `windows_encode_fix` 工具链。