enc-guard

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register } from 'claude-code'
3
4import type { EncEntry, EncRule } from '../types'
5import {
6  LOG_MAX,
7  RULE,
8  barePyWrites,
9  bytesOf,
10  hasBarePsWrite,
11  hasBom,
12  hasHighByte,
13  hasNonAscii,
14  isBat,
15  isPs1,
16  isPy,
17  noteText,
18  paneLines,
19  paneSvg,
20  runsPython,
21} from './enc'
22import { followTheme, readTheme, short } from './style'
23
24const PANE = 'enc-guard'
25const TITLE = '文字コードの見張り'
26const DESCRIPTION = '文字コード・改行の罠 (Set-Content・.ps1 の BOM・.bat・Python の open) の数と記録のペインを開く'
27
28const LOG = atom({ plugin: 'enc-guard', key: 'log' } as const, [])
29const COUNTS = atom({ plugin: 'enc-guard', key: 'counts' } as const, { 'ps-encoding': 0, 'ps1-bom': 0, 'bat-ascii': 0, 'py-open': 0 })
30
31type $ = EngineInterface
32type Found = { rule: EncRule; where: string }
33
34// 設定は register で userConfig から入れる
35const cfg = { isBlocking: false, isPython: true }
36let ownsMain = false
37
38const SHELL_TOOLS = ['Bash', 'PowerShell']
39const EDIT_TOOLS = ['Write', 'Edit', 'MultiEdit']
40
41/** ファイルの頭が BOM か。読めなければ (まだ無い) 無し */
42async function bomOf($: $, path: string): Promise<boolean | undefined> {
43  const got = await $.fs.read(path, { as: 'bytes' }).catch(() => undefined)
44  return got === undefined ? undefined : hasBom(bytesOf(got.base64))
45}
46
47/** 編集で入る文 (Write は全文、Edit は new_string、MultiEdit は全部の new_string) */
48function incomingText(tool: string, args: Record<string, unknown>) {
49  if (tool === 'Write') return String(args.content ?? '')
50  if (tool === 'Edit') return String(args.new_string ?? '')
51  const edits = Array.isArray(args.edits) ? (args.edits as { new_string?: unknown }[]) : []
52  return edits.map(x => String(x.new_string ?? '')).join('\n')
53}
54
55/** 実行の前に分かる罠 */
56async function before($: $, tool: string, args: Record<string, unknown>): Promise<Found[]> {
57  const found: Found[] = []
58  if (SHELL_TOOLS.includes(tool) && typeof args.command === 'string') {
59    const command = args.command
60    const where = short(command.trim().split('\n')[0] ?? '', 60)
61    if (hasBarePsWrite(command, tool === 'PowerShell')) found.push({ rule: 'ps-encoding', where })
62    if (cfg.isPython && runsPython(command) && (await isWindows($)) && barePyWrites(command).length > 0) found.push({ rule: 'py-open', where })
63    return found
64  }
65  if (!EDIT_TOOLS.includes(tool)) return found
66  const path = String(args.file_path ?? '')
67  const text = incomingText(tool, args)
68  if (isBat(path) && hasNonAscii(text)) found.push({ rule: 'bat-ascii', where: path })
69  if (isPy(path) && cfg.isPython && (await isWindows($)) && barePyWrites(text).length > 0) found.push({ rule: 'py-open', where: path })
70  // .ps1 を Write で書く時は、全文と今のファイルの BOM で前もって分かる (Edit は書いた後に読む)
71  if (tool === 'Write' && isPs1(path) && hasNonAscii(text) && !text.startsWith('\uFEFF') && (await bomOf($, path)) !== true) found.push({ rule: 'ps1-bom', where: path })
72  return found
73}
74
75/** 書いた後に読んで分かる罠 (.ps1 の BOM) */
76async function after($: $, tool: string, args: Record<string, unknown>, known: Found[]): Promise<Found[]> {
77  if (!EDIT_TOOLS.includes(tool)) return []
78  const path = String(args.file_path ?? '')
79  if (!isPs1(path) || known.some(f => f.rule === 'ps1-bom')) return []
80  const got = await $.fs.read(path, { as: 'bytes' }).catch(() => undefined)
81  if (got === undefined) return []
82  const bytes = bytesOf(got.base64)
83  return hasHighByte(bytes) && !hasBom(bytes) ? [{ rule: 'ps1-bom', where: path }] : []
84}
85
86// ponytail: Windows かどうかはセッションの場所の形で決める (C:\ で始まる)。一度読んだら持つ
87let windows: boolean | undefined
88async function isWindows($: $) {
89  windows ??= /^[A-Za-z]:/.test(await $.session.cwd().catch(() => ''))
90  return windows
91}
92
93async function record($: $, found: Found[], action: EncEntry['action']) {
94  const at = await $.clock.now()
95  await update($, LOG, log => [...log, ...found.map(f => ({ at, rule: f.rule, where: f.where, action }))].slice(-LOG_MAX))
96  await update($, COUNTS, c => {
97    const next = { ...c }
98    for (const f of found) next[f.rule] += 1
99    return next
100  })
101  const names = [...new Set(found.map(f => RULE[f.rule].name))].join('・')
102  $.ui.toast(`enc-guard: ${names}${action === 'blocked' ? '(止めました)' : '(Claude に伝えました)'}`, { timeoutMs: 6000 })
103}
104
105const openPane = async ($: $) => {
106  await $.ui.open({ id: PANE, title: TITLE })
107  return { text: '文字コードの見張りのペインを開きました。' }
108}
109
110export const register: Register = (on, options) => {
111  cfg.isBlocking = options.mode === '止める'
112  cfg.isPython = options.checkPython !== false
113
114  followTheme(on)
115  on('session.start', async ($, e, next) => {
116    const started = await next(e)
117    await readTheme($)
118    windows = undefined
119    const got = await $.command.register({ name: 'enc', description: DESCRIPTION }).catch(() => undefined)
120    ownsMain = got?.command === 'enc'
121    if (!ownsMain) await $.command.register({ name: 'enc-guard', description: DESCRIPTION }).catch(() => {})
122    return started
123  })
124
125  on('command.run', { command: 'enc' }, ($, e, next) => (ownsMain ? openPane($) : next(e))).catch(($, e, next) => next(e))
126  on('command.run', { command: 'enc-guard' }, openPane)
127
128  // 前に分かる物は、止める設定なら止める。知らせる設定と書いた後に分かる物は、結果の後ろで Claude に伝える
129  on('tool.call', async ($, e, next) => {
130    const tool = String(e.tool)
131    const args = e as unknown as Record<string, unknown>
132    const pre = await before($, tool, args)
133    if (cfg.isBlocking && pre.length > 0) {
134      await record($, pre, 'blocked')
135      return { deny: pre.map(f => noteText(f.rule, f.where)).join('\n') }
136    }
137    const r = await next(e)
138    if (r.deny !== undefined || r.isError === true) return r
139    const found = [...pre, ...(await after($, tool, args, pre))]
140    if (found.length === 0) return r
141    await record($, found, 'told')
142    return { ...r, context: [...(r.context ?? []), ...found.map(f => noteText(f.rule, f.where))] }
143  }).catch(($, e, next) => next(e))
144
145  on('ui.render', { component: 'Pane', requestId: PANE }, async ($, e) => {
146    const t = $.ui.resolve(e)
147    const { Box, Text } = t
148    // terminal の表にも Svg の名前があるので、画面の種類でも分ける
149    const Svg = e.surface !== 'terminal' && 'Svg' in t ? t.Svg : undefined
150    const input = { log: await read($, LOG), counts: await read($, COUNTS), isBlocking: cfg.isBlocking }
151    const lines = paneLines(input)
152    return (
153      <Box flexDirection="column" gap={1}>
154        {Svg ? (
155          <Svg source={paneSvg(input)} alt={lines.map(l => l.text).join('\n')} />
156        ) : (
157          <Box key="lines" flexDirection="column">
158            {lines.map(l => (
159              <Text color={l.color} dimColor={l.isDim} bold={l.isBold}>
160                {l.text}
161              </Text>
162            ))}
163          </Box>
164        )}
165        <Text dimColor>決まりは「UTF-8 (BOM なし)・LF が既定。PowerShell 5.1 の .ps1 だけ BOM 付き、.bat は ASCII だけ」です。</Text>
166      </Box>
167    )
168  })
169}
170

1// 文字コード・改行の罠の見分けと、ペインの SVG と文。どれも純関数
2import type { EncEntry, EncRule } from '../types'
3import { C, HEADER_H, esc, frame, growBar, header, panel, short } from './style'
4
5export const LOG_MAX = 30
6export const RULES: readonly EncRule[] = ['ps-encoding', 'ps1-bom', 'bat-ascii', 'py-open']
7
8/** 罠の名前と、Claude に伝える直し方 */
9export const RULE: Record<EncRule, { name: string; fix: string }> = {
10  'ps-encoding': {
11    name: 'Set-Content に -Encoding が無い',
12    fix: 'PowerShell 5.1 の Set-Content / Add-Content は -Encoding が無いと ANSI (cp932) で書きます。-Encoding utf8 (BOM 付き) を付けるか、BOM なしが要る所では [IO.File]::WriteAllText($p, $s, [Text.UTF8Encoding]::new($false)) を使ってください',
13  },
14  'ps1-bom': {
15    name: '日本語入りの .ps1 に BOM が無い',
16    fix: 'PowerShell 5.1 は BOM の無い .ps1 を cp932 として読み、日本語の所で構文エラーになります。ASCII だけで書き直すか、UTF-8 の BOM 付きで保存し直してください (例: $t = Get-Content -Raw -Encoding UTF8 $p; [IO.File]::WriteAllText($p, $t, [Text.UTF8Encoding]::new($true)))',
17  },
18  'bat-ascii': {
19    name: '.bat / .cmd に ASCII 以外の字',
20    fix: 'cmd は .bat / .cmd を cp932 で読むので、日本語などは化けます。.bat / .cmd は ASCII だけで書いてください (日本語の表示が要るなら .ps1 に分ける)',
21  },
22  'py-open': {
23    name: 'Python の open に encoding= / newline= が無い',
24    fix: "Windows の Python は open(p, 'w') を cp932 で書き、LF を CRLF に変えます。open(p, 'w', encoding='utf-8', newline='') のように両方を指定してください (Path.write_text も encoding='utf-8' を付ける)",
25  },
26}
27
28// ── コマンドの読み方 (ship-guard・compact-cue と同じ形。mod は別々に入れるので写して持つ)
29
30/** 引用符の中身とヒアドキュメントの本文を Q に置き換える (文字列の中の言葉を拾わない) */
31export function unquote(command: string) {
32  return command
33    .replace(/<<-?\s*(['"]?)(\w+)\1[^\n]*\n[\s\S]*?\n\s*\2\s*(?=\n|$)/g, ' Q')
34    .replace(/'[^']*'|"(?:[^"\\]|\\.)*"/g, 'Q')
35}
36
37/** コマンドの区切り (&& ; | など) で分けた 1 つずつ。頭の VAR=値 と & は落とす */
38export function segments(command: string) {
39  return unquote(command)
40    .split(/&&|\|\||[;|\n`{}]|\$\(|\(/)
41    .map(s => s.trim().replace(/^(?:[\w.]+=\S*\s+)*(?:&\s*)?/, ''))
42    .filter(s => s !== '')
43}
44
45const WRITES = /^(?:Microsoft\.PowerShell\.Management\\)?(?:Set-Content|Add-Content)\b/i
46const HAS_ENCODING = /\s-En(?:c(?:o(?:d(?:i(?:n(?:g)?)?)?)?)?)?\b/i
47
48/** PowerShell の文の中の、-Encoding の無い Set-Content / Add-Content の数 */
49const countBare = (parts: string[]) => parts.filter(s => WRITES.test(s) && !HAS_ENCODING.test(s)).length
50
51/**
52 * -Encoding の無い Set-Content / Add-Content があるか。PowerShell のツールは引用符を外して区切りで分ける。
53 * Bash から powershell / pwsh を呼ぶ時は、引用符の中にコマンドがあるので、そのまま ; | で分ける
54 */
55export function hasBarePsWrite(command: string, isPowerShellTool: boolean) {
56  if (isPowerShellTool) return countBare(segments(command)) > 0
57  if (!/\b(?:powershell|pwsh)(?:\.exe)?\b/i.test(command)) return false
58  // 引用符の中の Set-Content から、文の終わり (; | 改行 ") までを 1 つとして見る
59  return [...command.matchAll(/(?:Set|Add)-Content\b[^;|\n"]*/gi)].some(m => !HAS_ENCODING.test(m[0]))
60}
61
62/** Python を走らせるコマンドか (python・python3・py・パス付きの python.exe・uv run python) */
63export const runsPython = (command: string) => segments(command).some(s => /^(?:\S*[\\/])?(?:python\d*(?:\.\d+)?|py|uv\s+run\s+python\d*)(?:\.exe)?\b/i.test(s))
64
65/** open( の後ろの、対応する ) までの引数 (入れ子の括弧を数える)。閉じていなければ行の終わりまで */
66function argsAfter(text: string, at: number) {
67  let depth = 1
68  for (let i = at; i < text.length; i++) {
69    const c = text[i]
70    if (c === '(') depth++
71    else if (c === ')' && --depth === 0) return text.slice(at, i)
72  }
73  return text.slice(at, text.indexOf('\n', at) < 0 ? undefined : text.indexOf('\n', at))
74}
75
76/**
77 * Python のコードの中の、encoding= の無い書き込み。open(..., 'w'/'a'/'x') で encoding= か newline= が無い物と、
78 * write_text( で encoding= が無い物。バイナリ ('wb') と読むだけの open は見ない
79 */
80export function barePyWrites(code: string) {
81  const hits: string[] = []
82  // x.open( (Image.open・zipfile など) は別物なので、組み込みの open( と .write_text( だけ
83  for (const m of code.matchAll(/(?<![\w.])open\(|\.write_text\(/g)) {
84    const args = argsAfter(code, (m.index ?? 0) + m[0].length)
85    const hasEncoding = /\bencoding\s*=/.test(args)
86    if (m[0].includes('write_text')) {
87      if (!hasEncoding) hits.push(`write_text(${short(args.trim(), 30)})`)
88      continue
89    }
90    const mode = /(?:^|,)\s*(?:mode\s*=\s*)?[rbuRBU]?(['"])([rwaxbt+]+)\1/.exec(args)?.[2] ?? 'r'
91    if (!/[wax]/.test(mode) || mode.includes('b')) continue
92    if (!hasEncoding || !/\bnewline\s*=/.test(args)) hits.push(`open(${short(args.trim(), 30)})`)
93  }
94  return hits
95}
96
97// ── ファイルの読み方
98
99const EXT = (path: string) => /\.([^.\\/]+)$/.exec(path)?.[1]?.toLowerCase() ?? ''
100export const isPs1 = (path: string) => ['ps1', 'psm1', 'psd1'].includes(EXT(path))
101export const isBat = (path: string) => ['bat', 'cmd'].includes(EXT(path))
102export const isPy = (path: string) => ['py', 'pyw'].includes(EXT(path))
103
104export const hasNonAscii = (text: string) => /[^\x00-\x7f]/.test(text)
105
106const B64 = 'ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/'
107
108/** base64 を中身のバイトに戻す ($.fs.read の bytes の答え) */
109export function bytesOf(base64: string) {
110  const clean = base64.replace(/[^A-Za-z0-9+/]/g, '')
111  const out = new Uint8Array(Math.floor((clean.length * 3) / 4))
112  let n = 0
113  for (let i = 0; i + 1 < clean.length; i += 4) {
114    const v = [0, 1, 2, 3].map(k => Math.max(0, B64.indexOf(clean[i + k] ?? 'A')))
115    const word = ((v[0] ?? 0) << 18) | ((v[1] ?? 0) << 12) | ((v[2] ?? 0) << 6) | (v[3] ?? 0)
116    out[n++] = (word >> 16) & 255
117    if (i + 2 < clean.length) out[n++] = (word >> 8) & 255
118    if (i + 3 < clean.length) out[n++] = word & 255
119  }
120  return out.slice(0, n)
121}
122
123export const hasBom = (bytes: Uint8Array) => bytes[0] === 0xef && bytes[1] === 0xbb && bytes[2] === 0xbf
124export const hasHighByte = (bytes: Uint8Array) => bytes.some(b => b > 0x7f)
125
126// ── ペイン
127
128const clock = (ms: number) => {
129  const d = new Date(ms)
130  return [d.getHours(), d.getMinutes()].map(n => String(n).padStart(2, '0')).join(':')
131}
132
133const RULE_COLOR = (r: EncRule) => ({ 'ps-encoding': C.blue, 'ps1-bom': C.red, 'bat-ascii': C.yellow, 'py-open': C.green })[r]
134const RULE_TERM = (r: EncRule) => ({ 'ps-encoding': 'suggestion', 'ps1-bom': 'error', 'bat-ascii': 'warning', 'py-open': 'success' })[r]
135const ACTION = { told: 'Claude に伝えた', blocked: '止めた' } as const
136
137export type PaneInput = { log: EncEntry[]; counts: Record<EncRule, number>; isBlocking: boolean }
138
139const modeWord = (i: PaneInput) => (i.isBlocking ? '見つけたら止めます' : '見つけたら Claude に伝えます')
140
141/** ペインの SVG 1 枚 */
142export function paneSvg(i: PaneInput) {
143  const total = RULES.reduce((a, r) => a + i.counts[r], 0)
144  const accent = total === 0 ? C.green : C.yellow
145  let s = panel(8, 8, 364, 52)
146  s += `<text x="20" y="32" font-size="15" font-weight="bold" fill="${accent}">${total === 0 ? '罠は踏んでいません' : `このセッションで ${total} 回`}</text>`
147  s += `<text x="20" y="50" font-size="12" fill="${C.dim}">${modeWord(i)}</text>`
148  const max = Math.max(1, ...RULES.map(r => i.counts[r]))
149  const y0 = 68
150  s += panel(8, y0, 364, RULES.length * 34 + 16)
151  RULES.forEach((r, n) => {
152    const y = y0 + 14 + n * 34
153    s += `<text x="20" y="${y + 12}" font-size="12" fill="${C.text}">${esc(short(RULE[r].name, 26))}</text>`
154    s += `<text x="360" y="${y + 12}" font-size="12" font-weight="bold" text-anchor="end" fill="${C.text}">${i.counts[r]} 回</text>`
155    s += `<rect x="20" y="${y + 18}" width="340" height="6" rx="3" fill="${C.line}"/>`
156    if (i.counts[r] > 0) s += growBar(20, y + 18, Math.max(6, (340 * i.counts[r]) / max), 6, RULE_COLOR(r), n * 0.08)
157  })
158  const rows = i.log.slice(-6).reverse()
159  const y1 = y0 + RULES.length * 34 + 24
160  const h = Math.max(1, rows.length) * 36 + 36
161  s += panel(8, y1, 364, h)
162  s += `<text x="20" y="${y1 + 22}" font-size="13" font-weight="bold" fill="${C.text}">最近の記録</text>`
163  if (rows.length === 0) s += `<text x="20" y="${y1 + 50}" font-size="12" fill="${C.dim}">まだありません</text>`
164  rows.forEach((e, n) => {
165    const y = y1 + 36 + n * 36
166    s += `<rect x="20" y="${y}" width="4" height="28" rx="2" fill="${RULE_COLOR(e.rule)}"/>`
167    s += `<text x="32" y="${y + 12}" font-size="12" font-weight="bold" fill="${C.text}">${esc(short(RULE[e.rule].name, 24))}</text>`
168    s += `<text x="360" y="${y + 12}" font-size="11" text-anchor="end" fill="${C.dim}">${clock(e.at)} ${ACTION[e.action]}</text>`
169    s += `<text x="32" y="${y + 26}" font-size="11" fill="${C.dim}">${esc(short(e.where, 40))}</text>`
170  })
171  return frame(HEADER_H + y1 + h + 8, header('文字コードの見張り', `${total} 回`, accent) + `<g transform="translate(0 ${HEADER_H})">${s}</g>`)
172}
173
174export type Line = { text: string; color?: string; isDim?: boolean; isBold?: boolean }
175
176/** Svg の alt と terminal で使う文の行 */
177export function paneLines(i: PaneInput): Line[] {
178  const total = RULES.reduce((a, r) => a + i.counts[r], 0)
179  const lines: Line[] = [
180    { text: total === 0 ? '罠は踏んでいません' : `このセッションで ${total} 回`, isBold: true, color: total === 0 ? 'success' : 'warning' },
181    { text: modeWord(i), isDim: true },
182  ]
183  for (const r of RULES) lines.push({ text: `${RULE[r].name}: ${i.counts[r]} 回` })
184  if (i.log.length > 0) lines.push({ text: '最近の記録', isBold: true })
185  for (const e of i.log.slice(-6).reverse()) lines.push({ text: `${clock(e.at)} ${RULE[e.rule].name} ─ ${ACTION[e.action]} (${short(e.where, 40)})`, color: RULE_TERM(e.rule) })
186  return lines
187}
188
189/** Claude に伝える文 */
190export const noteText = (rule: EncRule, where: string) => `enc-guard: ${RULE[rule].name} (${short(where, 80)})。${RULE[rule].fix}`
191

1// 全 mod で共通の見た目(色・見出し・動き)と小物。
2// 正本は shared/style.ts。各 mod の hooks/style.ts はその写しなので、直すのは正本だけにして
3// `node tools/copy_shared.mts` で写す(`--check` で写しのずれを見つける)。
4import type { EngineInterface, On } from 'claude-code'
5
6export type Palette = {
7  bg: string
8  face: string
9  line: string
10  text: string
11  dim: string
12  weak: string
13  blue: string
14  green: string
15  yellow: string
16  red: string
17  purple: string
18  orange: string
19}
20
21/** GitHub Dark 系(docs/mod-guide.md の 7 節) */
22export const DARK: Palette = {
23  bg: '#0d1117',
24  face: '#161b22',
25  line: '#30363d',
26  text: '#c9d1d9',
27  dim: '#8b949e',
28  weak: '#8b949e',
29  blue: '#58a6ff',
30  green: '#3fb950',
31  yellow: '#d29922',
32  red: '#f85149',
33  purple: '#bc8cff',
34  orange: '#f78166',
35}
36
37/** GitHub Light 系 */
38export const LIGHT: Palette = {
39  bg: '#ffffff',
40  face: '#f6f8fa',
41  line: '#d0d7de',
42  text: '#1f2328',
43  dim: '#656d76',
44  weak: '#656d76',
45  blue: '#0969da',
46  green: '#1a7f37',
47  yellow: '#9a6700',
48  red: '#cf222e',
49  purple: '#8250df',
50  orange: '#bc4c00',
51}
52
53// ponytail: 今の色はモジュールの変数 1 個(この mod の中だけ)。SVG を作る関数は C.bg などを呼ぶたびに読むので、
54// 引数を増やさずにテーマが変わる。描く途中で切り替わることは無い(JS は 1 本で走る)
55/** 今の色。setTheme で中身が入れ替わる(同じ物を指し続けるので、import した C のままでよい) */
56export const C: Palette = { ...DARK }
57let isLightNow = false
58
59export const isLight = () => isLightNow
60
61/** /config の theme の値が明るい系か('light'・'light-daltonized'・'light-ansi' など) */
62export const isLightTheme = (value: unknown) => typeof value === 'string' && value.startsWith('light')
63
64export function setTheme(light: boolean) {
65  isLightNow = light
66  Object.assign(C, light ? LIGHT : DARK)
67}
68
69/**
70 * 今の /config の theme を読んで色を合わせる。mod 自身の session.start の中で呼ぶ
71 * (同じイベントに matcher 無しのフックを 2 つ置くと読み込みを拒まれるので、ここでは on しない)。
72 * VS Code の配色は mod から読めないので、Claude Code の theme が light 系の時だけ明るくなる
73 */
74export async function readTheme($: EngineInterface) {
75  const rows = await $.config.list().catch(() => [])
76  setTheme(isLightTheme(rows.find(r => r.key === 'theme')?.value))
77}
78
79/** /config で theme が変わったら色を替えて描き直す。register の中で 1 回呼ぶ */
80export function followTheme(on: On) {
81  on('config.set', { key: 'theme' }, async ($, e, next) => {
82    const r = await next(e)
83    if (r.deny === undefined) {
84      setTheme(isLightTheme(r.value))
85      $.ui.invalidate('ui.render')
86    }
87    return r
88  })
89}
90
91// ---- 小物 ----
92
93/** 外から来る文字列の制御文字を落とす。エンジンは制御文字を含む Text の子や alt を拒み、ペインが閉じる */
94export const clean = (s: string) => s.replace(/[\t\n\r]+/g, ' ').replace(/[\x00-\x1f\x7f-\x9f]/g, '').trim()
95
96/** SVG の文字に入れる前のエスケープ(制御文字も落とす) */
97export const esc = (s: string) =>
98  clean(s).replace(/&/g, '&amp;').replace(/</g, '&lt;').replace(/>/g, '&gt;').replace(/"/g, '&quot;')
99
100// ---- 時間帯と季節 (日本時間 JST = UTC+9 固定。動いている環境のタイムゾーンには頼らない) ----
101
102export type DayPart = '朝' | '昼' | '夕' | '夜'
103export type Season = '春' | '夏' | '秋' | '冬'
104
105const jstDate = (now: number) => new Date(now + 9 * 3_600_000)
106
107/** 時間帯。朝 5-10 時 / 昼 10-16 時 / 夕 16-19 時 / 夜 19-5 時 */
108export const dayPart = (now: number): DayPart => {
109  const h = jstDate(now).getUTCHours()
110  return h >= 5 && h < 10 ? '朝' : h >= 10 && h < 16 ? '昼' : h >= 16 && h < 19 ? '夕' : '夜'
111}
112
113/** 季節。春 3-5 月 / 夏 6-8 月 / 秋 9-11 月 / 冬 12-2 月 */
114export const season = (now: number): Season => {
115  const m = jstDate(now).getUTCMonth() + 1
116  return m >= 3 && m <= 5 ? '春' : m >= 6 && m <= 8 ? '夏' : m >= 9 && m <= 11 ? '秋' : '冬'
117}
118
119/** n 文字を超えたら … で切る */
120export const short = (s: string, n: number) => (s.length > n ? `${s.slice(0, Math.max(0, n - 1))}…` : s)
121
122// ---- 記録を守る(docs/mod-guide.md の 9 節) ----
123
124/** JST の YYYY-MM-DD */
125export const jstDay = (now: number) => jstDate(now).toISOString().slice(0, 10)
126
127export type Backup = { day: string; value: unknown }
128
129// ponytail: $.fs に消す手が無いので、曜日ごとの 7 つの枠を上書きして 7 日分だけ残す。もっと残すなら枠を日付にして process で消す
130const backupPath = (dir: string, plugin: string, key: string, wd: number) => `${dir.replace(/[\\/]+$/, '')}/${plugin}/${key}-${wd}.json`
131
132/**
133 * $.store の key の値を、その日はじめて呼ばれた時だけ dir に控える(曜日ごとの枠で 7 日分)。
134 * dir が空なら何もしない(公開の既定は空。入れた人が backupDir を設定する)。書いたら true
135 */
136export async function backupDaily($: EngineInterface, dir: string, key: string, value: unknown, now: number): Promise<boolean> {
137  if (!dir.trim() || value === undefined) return false
138  const day = jstDay(now)
139  const path = backupPath(dir, $.plugin.name, key, jstDate(now).getUTCDay())
140  // 壊れた枠(途中まで・null)は無いのと同じに扱って上書きする
141  const old = await $.fs.read(path).then(t => parseBackup(t)?.day ?? '', () => '')
142  if (old === day) return false
143  await $.fs.write(path, JSON.stringify({ day, value } satisfies Backup))
144  return true
145}
146
147/** 控えの 1 枠を読む。壊れていれば undefined */
148export function parseBackup(text: string): Backup | undefined {
149  try {
150    const b = JSON.parse(text) as Partial<Backup> | null
151    return b && typeof b.day === 'string' && /^\d{4}-\d{2}-\d{2}$/.test(b.day) && b.value !== undefined ? (b as Backup) : undefined
152  } catch {
153    return undefined
154  }
155}
156
157/** 控えの一覧。新しい日から順。読めない枠・壊れた枠は飛ばす */
158export async function listBackups($: EngineInterface, dir: string, key: string): Promise<Backup[]> {
159  if (!dir.trim()) return []
160  const all = await Promise.all(
161    [0, 1, 2, 3, 4, 5, 6].map(wd =>
162      $.fs.read(backupPath(dir, $.plugin.name, key, wd)).then(parseBackup, () => undefined),
163    ),
164  )
165  return all.filter((b): b is Backup => b !== undefined).sort((a, b) => b.day.localeCompare(a.day))
166}
167
168let loadedPrint: string | undefined
169let staleAt = 0
170let isStaleNow = false
171let toldStale = false
172
173const normPath = (p: string) => p.replace(/[\\/]+/g, '/').replace(/\/$/, '').toLowerCase()
174
175/** FNV-1a 32bit。中身が変わったかを見るだけ */
176function fnv(s: string) {
177  let h = 0x811c9dc5
178  for (let i = 0; i < s.length; i++) h = Math.imul(h ^ s.charCodeAt(i), 0x01000193)
179  return (h >>> 0).toString(16)
180}
181
182/**
183 * 今ディスクにある、この mod の指紋。plugin.json と hooks の .ts / .tsx の中身から作る
184 * (版の数字を上げずに中身だけ変えた時も変わる)。plugin.json が読めなければ undefined
185 */
186async function diskPrint($: EngineInterface): Promise<string | undefined> {
187  const root = $.plugin.root
188  let manifest: string | undefined
189  for (const p of [`${root}/.claude-plugin/plugin.json`, `${root}/plugin.json`]) {
190    manifest = await $.fs.read(p).then(t => t, () => undefined)
191    if (manifest !== undefined) break
192  }
193  if (manifest === undefined) return undefined
194  const names = await $.fs.list(`${root}/hooks`).then(
195    es => es.filter(e => e.kind === 'file' && /\.tsx?$/.test(e.name)).map(e => e.name).sort(),
196    () => [] as string[],
197  )
198  const parts = await Promise.all(names.map(n => $.fs.read(`${root}/hooks/${n}`).then(t => `${n}\n${t}`, () => n)))
199  return fnv([manifest, ...parts].join('\0'))
200}
201
202/**
203 * マーケットプレイスから入れた形(<plugins>/cache/<market>/<name>/<版>)なら、installed_plugins.json に
204 * 今入っている場所が自分と違うか。違えば新しい版が入っている。分からなければ false
205 */
206async function replacedInCache($: EngineInterface): Promise<boolean> {
207  const root = normPath($.plugin.root)
208  const at = root.lastIndexOf('/plugins/cache/')
209  if (at < 0) return false
210  const raw = await $.fs.read(`${$.plugin.root.slice(0, at + '/plugins'.length)}/installed_plugins.json`).then(t => t, () => '')
211  try {
212    const plugins = (JSON.parse(raw) as { plugins?: Record<string, { installPath?: unknown }[]> }).plugins ?? {}
213    const mine = Object.entries(plugins)
214      .filter(([k]) => k.startsWith(`${$.plugin.name}@`))
215      .flatMap(([, v]) => (Array.isArray(v) ? v : []))
216      .map(e => (typeof e.installPath === 'string' ? normPath(e.installPath) : ''))
217    return mine.length > 0 && !mine.includes(root)
218  } catch {
219    return false
220  }
221}
222
223/** 読み込んだ時の指紋を覚える。mod の session.start で 1 回呼ぶ */
224export async function rememberVersion($: EngineInterface) {
225  loadedPrint ??= await diskPrint($)
226}
227
228/**
229 * 新しい版が入ったのに、このセッションはまだ古い版で動いているか。
230 * フォルダから読ませている時は中身の指紋が読み込んだ時と違えば、マーケットプレイスから入れた時は
231 * 入っている場所が自分でなくなれば true。読み直すのは 60 秒に 1 回まで。分からない時は false
232 */
233export async function isStale($: EngineInterface, now: number): Promise<boolean> {
234  if (isStaleNow || now - staleAt < 60_000) return isStaleNow
235  staleAt = now
236  if (await replacedInCache($)) return (isStaleNow = true)
237  if (loadedPrint === undefined) return false
238  const p = await diskPrint($)
239  isStaleNow = p !== undefined && p !== loadedPrint
240  return isStaleNow
241}
242
243/**
244 * $.store に書いてよいか。古い版のままなら書かず、1 回だけトーストで開き直しを頼む
245 * (古い版は新しい版の欄を知らないので、書くと消してしまう)
246 */
247export async function mayWrite($: EngineInterface, now: number): Promise<boolean> {
248  if (!(await isStale($, now))) return true
249  if (!toldStale) {
250    toldStale = true
251    $.ui.toast(`${$.plugin.name}: 新しい版が入りました。VS Code を開き直すまで記録は書きません`)
252  }
253  return false
254}
255
256// ---- SVG の部品 ----
257
258export const W = 380
259export const HEADER_H = 44
260
261/** SVG 全体の枠。地を塗って body を入れる。幅は W にそろえる */
262export const frame = (h: number, body: string, w = W) =>
263  `<svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 ${w} ${h}" font-family="sans-serif">` +
264  `<rect width="${w}" height="${h}" rx="10" fill="${C.bg}"/>${body}</svg>`
265
266/** 面(カード)。角丸 8、線 1 */
267export const panel = (x: number, y: number, w: number, h: number, fill = C.face) =>
268  `<rect x="${x}" y="${y}" width="${w}" height="${h}" rx="8" fill="${fill}" stroke="${C.line}"/>`
269
270/**
271 * 共通の見出し帯(高さ HEADER_H)。左に色の縦線、左に題、右に添え書き。
272 * 下の細い線が左から伸びる
273 */
274export function header(title: string, sub = '', accent = C.blue, y = 0, w = W) {
275  const pad = 14
276  return [
277    `<rect x="0" y="${y}" width="${w}" height="${HEADER_H}" rx="10" fill="${C.face}"/>`,
278    `<rect x="0" y="${y + 8}" width="4" height="${HEADER_H - 16}" rx="2" fill="${accent}"/>`,
279    `<text x="${pad}" y="${y + 28}" font-size="16" font-weight="bold" fill="${C.text}">${esc(title)}</text>`,
280    sub === '' ? '' : `<text x="${w - pad}" y="${y + 28}" font-size="12" fill="${C.dim}" text-anchor="end">${esc(sub)}</text>`,
281    growBar(pad, y + HEADER_H - 3, w - pad * 2, 2, accent, 0, 0.35),
282  ].join('')
283}
284
285/** 横に伸びる棒。幅 0 から w まで伸びて止まる(shots.mts は終わりの値で撮る) */
286export const growBar = (x: number, y: number, w: number, h: number, fill: string, delay = 0, opacity = 1) =>
287  `<rect x="${x}" y="${y}" width="${Math.max(0, w)}" height="${h}" rx="${Math.min(3, h / 2)}" fill="${fill}"${opacity === 1 ? '' : ` opacity="${opacity}"`}>` +
288  `<animate attributeName="width" from="0" to="${Math.max(0, w)}" begin="${delay}s" dur="0.6s" fill="freeze"/></rect>`
289
290/** 下から伸びる縦の棒(下端 bottom・高さ h) */
291export const growColumn = (x: number, bottom: number, w: number, h: number, fill: string, delay = 0) =>
292  `<rect x="${x}" y="${bottom - h}" width="${w}" height="${Math.max(0, h)}" rx="${Math.min(3, w / 2)}" fill="${fill}">` +
293  `<animate attributeName="y" from="${bottom}" to="${bottom - h}" begin="${delay}s" dur="0.6s" fill="freeze"/>` +
294  `<animate attributeName="height" from="0" to="${Math.max(0, h)}" begin="${delay}s" dur="0.6s" fill="freeze"/></rect>`
295
296/** ふわっと出る。中身は SVG の断片 */
297export const fadeIn = (body: string, delay = 0) =>
298  `<g opacity="1"><animate attributeName="opacity" from="0" to="1" begin="${delay}s" dur="0.4s" fill="freeze"/>${body}</g>`
299
300/**
301 * 0 から value まで数字が増えて止まる。コマ送りの文字を重ね、最後のコマだけが残る。
302 * attrs は <text> に付ける属性(x・y・font-size・fill など)。format で桁区切りや単位を付ける
303 */
304export function countUp(value: number, attrs: string, format: (n: number) => string = n => String(n), delay = 0) {
305  const STEPS = 8
306  const d = 0.08
307  const last = `<text ${attrs} opacity="0">${esc(format(value))}<animate attributeName="opacity" from="0" to="1" begin="${(delay + (STEPS - 1) * d).toFixed(2)}s" dur="0.01s" fill="freeze"/></text>`
308  if (!Number.isFinite(value) || value === 0) return `<text ${attrs}>${esc(format(value))}</text>`
309  const frames = Array.from({ length: STEPS - 1 }, (_, k) => {
310    const n = Number.isInteger(value) ? Math.round((value * k) / (STEPS - 1)) : (value * k) / (STEPS - 1)
311    return `<text ${attrs} opacity="0">${esc(format(n))}<animate attributeName="opacity" values="1;1" begin="${(delay + k * d).toFixed(2)}s" dur="${d}s"/></text>`
312  })
313  return frames.join('') + last
314}
315

1/**
2 * 見ている罠。ps-encoding は -Encoding の無い Set-Content / Add-Content、ps1-bom は日本語入りで BOM の無い .ps1、
3 * bat-ascii は ASCII 以外の字が入った .bat / .cmd、py-open は encoding= / newline= の無い Python の書き込み
4 */
5export type EncRule = 'ps-encoding' | 'ps1-bom' | 'bat-ascii' | 'py-open'
6
7/** 記録の 1 件。古い順、最大 30 */
8export type EncEntry = {
9  at: number
10  rule: EncRule
11  /** ファイルのパスか、コマンドの頭 */
12  where: string
13  /** told は Claude に伝えた、blocked は止めた */
14  action: 'told' | 'blocked'
15}
16
17declare module 'claude-code' {
18  interface PluginState {
19    'enc-guard': {
20      log: EncEntry[]
21      /** このセッションで見つけた数 */
22      counts: Record<EncRule, number>
23    }
24  }
25}
26