screen-share

1import { atom, read, update } from 'claude-code'
2import type { EngineInterface, Register } from 'claude-code'
3
4import { flagOf } from './flag'
5import { maskDeep, namesPattern } from './mask'
6
7const MOD = 'screen-share'
8const isOn = atom({ plugin: 'screen-share', key: 'isOn' } as const, false)
9
10// The strip's eye button bumps this counter; only this mod may flip its own switch.
11const TOGGLES = { plugin: 'switchboard', key: 'toggles' } as const
12let seenToggle = -1
13
14async function setShare($: EngineInterface, force?: boolean): Promise<boolean> {
15  const wanted = force ?? !(await read($, isOn))
16  await update($, isOn, () => wanted)
17  $.ui.status(wanted ? '👁 屏幕共享：打码中' : undefined)
18
19  return wanted
20}
21
22async function pollToggle($: EngineInterface) {
23  const pressed = ((await $.state.get(TOGGLES)).value ?? {})['screen-share'] ?? 0
24  if (seenToggle < 0) seenToggle = pressed
25  if (pressed !== seenToggle) {
26    seenToggle = pressed
27    if ((await flagOf($, MOD)).isOn) await setShare($)
28  }
29}
30
31async function isMasking($: EngineInterface): Promise<boolean> {
32  const flag = await flagOf($, MOD)
33
34  return (await read($, isOn)) && flag.isOn
35}
36
37export const register: Register = (on, options) => {
38  const names = namesPattern(String(options.names ?? ''))
39
40  on('session.start', async ($, e, next) => {
41    await $.command.register({
42      name: 'screen',
43      description: '屏幕共享打码模式：/screen share 开 / 关',
44      argumentHint: 'share',
45    })
46    $.clock.every(500, () => void pollToggle($).catch(() => undefined))
47
48    return next(e)
49  })
50
51  on('command.run', { command: 'screen' }, async $ => {
52    const wanted = await setShare($)
53
54    return {
55      text: wanted
56        ? '屏幕共享模式已开启：邮箱、电话、金额、密钥、人名公司名已遮成灰条。再输入 /screen share 关闭。'
57        : '屏幕共享模式已关闭，恢复明文。',
58    }
59  })
60
61  // The transcript rows keep their real text; only what is drawn is rewritten.
62  on('ui.render', { component: 'UserMessage' }, async ($, e, next) =>
63    (await isMasking($))
64      ? next({ ...e, props: maskDeep(e.props, names) } as typeof e)
65      : next(e),
66  )
67  on('ui.render', { component: 'AssistantMessage' }, async ($, e, next) =>
68    (await isMasking($))
69      ? next({ ...e, props: maskDeep(e.props, names) } as typeof e)
70      : next(e),
71  )
72  on('ui.render', { component: 'ToolUse' }, async ($, e, next) =>
73    (await isMasking($))
74      ? next({ ...e, props: maskDeep(e.props, names) } as typeof e)
75      : next(e),
76  )
77  on('ui.render', { component: 'ToolResult' }, async ($, e, next) =>
78    (await isMasking($))
79      ? next({ ...e, props: maskDeep(e.props, names) } as typeof e)
80      : next(e),
81  )
82  on('ui.render', { component: 'CommandOutput' }, async ($, e, next) =>
83    (await isMasking($))
84      ? next({ ...e, props: maskDeep(e.props, names) } as typeof e)
85      : next(e),
86  )
87  on('ui.render', { component: 'InfoNotice' }, async ($, e, next) =>
88    (await isMasking($))
89      ? next({ ...e, props: maskDeep(e.props, names) } as typeof e)
90      : next(e),
91  )
92}
93

1import type { EngineInterface } from 'claude-code'
2
3const FLAGS = { plugin: 'switchboard', key: 'flags' } as const
4
5export type Flag = { isOn: boolean; isHidden: boolean }
6
7// Switchboard owns the flags. A mod it has never heard of counts as on, shown.
8export async function flagOf($: EngineInterface, mod: string): Promise<Flag> {
9  const { value } = await $.state.get(FLAGS)
10  const own = value?.[mod]
11
12  return { isOn: own?.isOn ?? true, isHidden: own?.isHidden ?? false }
13}
14

1// Pure text -> text. Everything that looks like private data becomes a gray
2// shade bar. Only what is drawn changes; the transcript the model reads does not.
3
4const BAR = '▒'
5
6const CURRENCY =
7  '(?:RM|MYR|USD|US\\$|SGD|S\\$|CNY|RMB|\\$|¥|￥|€|£)\\s?\\d[\\d,]*(?:\\.\\d+)?(?:\\s?[kKmMbB]\\b)?'
8const CURRENCY_AFTER =
9  '\\d[\\d,]*(?:\\.\\d+)?\\s?(?:元|块钱|块|万元|万|令吉|美金|美元|RM|MYR|USD)'
10const KEYS = [
11  'sk-[A-Za-z0-9_-]{16,}',
12  'AKIA[0-9A-Z]{16}',
13  '(?:ghp|gho|ghu|ghs|ghr)_[A-Za-z0-9]{20,}',
14  'github_pat_[A-Za-z0-9_]{20,}',
15  'xox[baprs]-[A-Za-z0-9-]{10,}',
16  'AIza[0-9A-Za-z_-]{30,}',
17  'eyJ[A-Za-z0-9_-]{10,}\\.[A-Za-z0-9_-]{10,}\\.[A-Za-z0-9_-]+',
18  'Bearer\\s+[A-Za-z0-9._-]{16,}',
19]
20const COMPANY =
21  "(?:[A-Z][\\w&.'-]*\\s+){0,3}[A-Z][\\w&.'-]*\\s+(?:Sdn\\.?\\s?Bhd\\.?|Bhd\\.?|Inc\\.?|LLC|Ltd\\.?|Pte\\.?\\s?Ltd\\.?|Corp\\.?|GmbH)|[\\u4e00-\\u9fa5]{2,12}(?:有限公司|股份公司|集团|工作室)"
22const HONORIFIC = '[\\u4e00-\\u9fa5]{1,3}(?:先生|小姐|女士|老师|经理|老板)'
23const NAME_LABEL =
24  "(?:客户|顾客|买家|收件人|联系人|姓名|customer|client|buyer|recipient|name)\\s*[:：]\\s*([A-Z][\\w'-]+(?:\\s+[A-Z][\\w'-]+){0,2}|[\\u4e00-\\u9fa5]{2,4})"
25
26const FIXED: RegExp[] = [
27  /[\w.+-]+@[\w-]+(?:\.[\w-]+)+/g,
28  /\+\d[\d ()-]{7,}\d/g,
29  /(?<![\w.])0\d{1,2}[ -]?\d{3,4}[ -]?\d{4}(?![\w])/g,
30  /(?<![\w.])1[3-9]\d{9}(?![\w])/g,
31  new RegExp(CURRENCY, 'g'),
32  new RegExp(CURRENCY_AFTER, 'g'),
33  ...KEYS.map(k => new RegExp(k, 'g')),
34  /(?:api[_-]?key|token|secret|passw(?:or)?d)["'\s:=]+([A-Za-z0-9._\-/+]{8,})/gi,
35  new RegExp(COMPANY, 'g'),
36  new RegExp(HONORIFIC, 'g'),
37  new RegExp(NAME_LABEL, 'gi'),
38]
39
40const escape = (text: string) => text.replace(/[.*+?^${}()|[\]\\]/g, '\\$&')
41
42export const namesPattern = (names: string): RegExp | undefined => {
43  const list = names
44    .split(/[,，\n]/)
45    .map(one => one.trim())
46    .filter(one => one.length > 0)
47
48  return list.length === 0
49    ? undefined
50    : new RegExp(list.map(escape).join('|'), 'gi')
51}
52
53const bar = (length: number) =>
54  BAR.repeat(Math.min(14, Math.max(4, Math.round(length * 0.8))))
55
56// A capture group, when the pattern has one, is the only part that is hidden
57// (`api_key: <secret>` keeps its label; `name: <Name>` keeps its label).
58const hide = (text: string, pattern: RegExp) =>
59  text.replace(pattern, (...args: unknown[]) => {
60    const whole = String(args[0])
61    const group = typeof args[1] === 'string' ? args[1] : undefined
62
63    return group === undefined
64      ? bar(whole.length)
65      : whole.replace(group, bar(group.length))
66  })
67
68export const maskText = (text: string, names?: RegExp): string => {
69  let out = text
70  for (const pattern of FIXED) out = hide(out, new RegExp(pattern.source, pattern.flags))
71  if (names !== undefined) out = hide(out, new RegExp(names.source, names.flags))
72
73  return out
74}
75
76export const maskDeep = (value: unknown, names?: RegExp): unknown => {
77  if (typeof value === 'string') return maskText(value, names)
78  if (Array.isArray(value)) return value.map(one => maskDeep(one, names))
79  if (value !== null && typeof value === 'object') {
80    return Object.fromEntries(
81      Object.entries(value).map(([key, one]) => [key, maskDeep(one, names)]),
82    )
83  }
84
85  return value
86}
87

1declare module 'claude-code' {
2  interface PluginState {
3    'screen-share': { isOn: boolean }
4    // Read-only view of Switchboard's value (it alone writes it).
5    switchboard: {
6      flags: Record<string, { isOn: boolean; isHidden: boolean }>
7      toggles: Record<string, number>
8    }
9  }
10}
11