mods-test-lab

1import type { On } from 'claude-code'
2
3const paneId = 'mods-test-lab'
4const contextMarker = '[mods-test-lab: one-shot context]'
5const usage = '/mods-test [open|background|status|reset|close|context|register|toolsearch NONCE|suggest TEXT]'
6const generation = 'base'
7const emptyCounts = () => ({ command: 0, tool: 0, prompt: 0, turn: 0 })
8type Event = 'activation' | 'open' | 'status' | 'reset' | 'close' | 'context' |
9  'tool' | 'prompt' | 'prompt.context' | 'turn' | 'button' | 'input' | 'submit' | 'select'
10
11let counts = emptyCounts()
12let persistent = { activations: 0, ...emptyCounts() }
13let controls = { button: 0, input: 0, submit: 0, select: 0 }
14let events: Event[] = []
15let input = ''
16let selection = 'ascii'
17let pendingContext = false
18let pendingToolSearch: string | undefined
19let lastTurn: { reason: string; aborted: boolean } | undefined
20
21function record(event: Event) {
22  events.push(event)
23  if (events.length > 20) events.shift()
24}
25
26function summary() {
27  return [
28    `Mods test lab | activation=${persistent.activations}`,
29    `Activation counters (since load/reset): ${JSON.stringify(counts)}`,
30    `Persistent counters (since reset; activations retained): ${JSON.stringify(persistent)}`,
31    `Controls: ${JSON.stringify(controls)}`,
32    `Input length=${input.length}/256 | selection=${selection} | context=${pendingContext ? 'pending' : 'idle'}`,
33    `Last turn: ${lastTurn ? JSON.stringify(lastTurn) : 'none'}`,
34    `Events (${events.length}/20): ${events.join(', ')}`,
35  ].join('\n')
36}
37
38export function register(on: On) {
39  on('session.start', async ($, e, next) => {
40    const saved = await $.store.get('counters')
41    if (saved && typeof saved === 'object' && !Array.isArray(saved)) {
42      for (const key of Object.keys(persistent) as (keyof typeof persistent)[]) {
43        const value = (saved as Record<string, unknown>)[key]
44        if (typeof value === 'number' && Number.isSafeInteger(value) && value >= 0) persistent[key] = value
45      }
46    }
47    persistent.activations++
48    await $.store.set('counters', persistent)
49    await $.command.register({
50      name: 'mods-test',
51      description: 'Open the local Mods test lab (no model request)',
52      argumentHint: '[open|background|status|reset|close|context|register|toolsearch NONCE|suggest TEXT]',
53      immediate: true,
54    })
55    record('activation')
56    return next(e)
57  })
58
59  on('command.run', { command: 'mods-test' }, async ($, e) => {
60    const [action = 'open', argument] = e.args.trim() ? e.args.trim().split(/\s+/, 2) : []
61    if (action === 'toolsearch') {
62      if (!argument || !/^[A-Za-z0-9_-]{1,32}$/.test(argument)) return { text: usage }
63      pendingToolSearch = argument
64      return { text: `MODS_TOOLSEARCH_ARMED:${argument}` }
65    }
66    if (action === 'suggest') {
67      if (!argument || !/^[A-Za-z0-9_-]{1,32}$/.test(argument)) return { text: usage }
68      const result = await $.prompt.suggest({ text: `MODS_SUGGESTION:${generation}:${argument}` })
69      return { text: `suggestion=${result.isShown ? 'shown' : 'hidden'}:${generation}:${argument}` }
70    }
71    if (!['open', 'background', 'status', 'reset', 'close', 'context', 'register'].includes(action)) return { text: usage }
72    if (action === 'register') {
73      try {
74        const registered = await $.tool.register({
75          name: 'acceptance-probe',
76          description: 'Compiled native sec-default policy acceptance probe',
77        })
78        return { text: `registration=${registered.tool}` }
79      } catch (error) {
80        return { text: `registration=${error instanceof Error ? error.message : String(error)}` }
81      }
82    }
83    if (action === 'reset') {
84      counts = emptyCounts()
85      persistent = { activations: persistent.activations, ...emptyCounts() }
86      controls = { button: 0, input: 0, submit: 0, select: 0 }
87      events = []
88      input = ''
89      selection = 'ascii'
90      pendingContext = false
91      lastTurn = undefined
92    } else {
93      counts.command++
94      persistent.command++
95      record(action as Event)
96    }
97    if (action === 'context') pendingContext = true
98    await $.store.set('counters', persistent)
99    if (action === 'open') await $.ui.open({ id: paneId, title: 'Mods test lab', focus: true, closeOnEscape: true })
100    if (action === 'background') await $.ui.open({ id: paneId, title: 'Mods test lab', closeOnEscape: true })
101    if (action === 'close') await $.ui.close({ id: paneId })
102    $.ui.invalidate('ui.render')
103    if (action === 'status' || action === 'reset') return { text: summary() }
104    if (action === 'context') return { text: 'Mods test lab: fixed context armed for the next prompt only.' }
105    return {}
106  })
107
108  on('tool.describe', async ($, e, next) => {
109    const result = await next(e)
110    return /^mcp__mods-test-lab__dynamic_[A-Za-z0-9_-]{1,32}$/.test(e.tool)
111      ? { ...result, isDeferred: true }
112      : result
113  })
114
115  on('tool.call', async ($, e, next) => {
116    counts.tool++
117    persistent.tool++
118    record('tool')
119    await $.store.set('counters', persistent)
120    $.ui.invalidate('ui.render')
121    if (e.tool === 'ToolSearch' && pendingToolSearch) {
122      const nonce = pendingToolSearch
123      const expected = `select:mcp__mods-test-lab__dynamic_${nonce}`
124      if (e.query === expected) {
125        pendingToolSearch = undefined
126        await $.tool.register({
127          name: `dynamic_${nonce}`,
128          description: `Mods test lab dynamic tool ${nonce}`,
129          inputSchema: {
130            type: 'object',
131            properties: { value: { type: 'string' } },
132            required: ['value'],
133          },
134        })
135      }
136    }
137    const dynamic = /^mcp__mods-test-lab__dynamic_([A-Za-z0-9_-]{1,32})$/.exec(e.tool)
138    if (dynamic && e.value === dynamic[1]) return { result: `MODS_DYNAMIC_RESULT:${dynamic[1]}` }
139    return next(e)
140  })
141
142  on('prompt.submit', async ($, e, next) => {
143    const attach = pendingContext
144    pendingContext = false
145    counts.prompt++
146    persistent.prompt++
147    record(attach ? 'prompt.context' : 'prompt')
148    await $.store.set('counters', persistent)
149    $.ui.invalidate('ui.render')
150    return next(attach ? { ...e, context: [...(e.context ?? []), contextMarker] } : e)
151  })
152
153  on('turn.complete', async ($, e, next) => {
154    counts.turn++
155    persistent.turn++
156    lastTurn = {
157      reason: ['answer', 'aborted', 'refusal', 'error'].includes(e.reason) ? e.reason : 'unknown',
158      aborted: e.isAborted === true,
159    }
160    record('turn')
161    await $.store.set('counters', persistent)
162    $.ui.invalidate('ui.render')
163    return next(e)
164  })
165
166  on('ui.render', { component: 'Pane', requestId: paneId }, ($, e, next) => {
167    if (e.surface !== 'terminal') return next(e)
168    const { Box, Text, Button, Input, Select } = $.ui.resolve(e)
169    return Box({ flexDirection: 'column', children: [
170      Text({ children: summary() }),
171      Text({ dimColor: true, children: 'Tab/Shift+Tab: focus | Enter: activate/submit | arrows/wheel: scroll | Esc: close' }),
172      Button({ key: 'count', autoFocus: true, label: `Count: ${controls.button}`, onPress: async () => {
173        controls.button++
174        record('button')
175        $.ui.invalidate('ui.render')
176      } }),
177      Input({ key: 'input', label: 'Local input', placeholder: 'Not logged or persisted (max 256)', value: input,
178        onInput: async value => {
179          input = value.slice(0, 256)
180          controls.input++
181          record('input')
182          $.ui.invalidate('ui.render')
183        },
184        onSubmit: async value => {
185          input = value.slice(0, 256)
186          controls.submit++
187          record('submit')
188          $.ui.invalidate('ui.render')
189        },
190      }),
191      Select({ key: 'selection', label: 'Diff labels', value: selection,
192        options: [{ value: 'ascii', label: 'ASCII' }, { value: 'cjk', label: '中文 / CJK' }],
193        onSelect: async value => {
194          if (value !== 'ascii' && value !== 'cjk') return
195          selection = value
196          controls.select++
197          record('select')
198          $.ui.invalidate('ui.render')
199        },
200      }),
201      Button({ key: 'close', label: 'Close panel', onPress: async () => {
202        record('close')
203        await $.ui.close({ id: paneId })
204      } }),
205      Text({ bold: true, children: selection === 'cjk' ? '固定示例差异 / CJK diff' : 'Fixed ASCII / CJK diff' }),
206      Text({ children: '--- before.txt\n+++ after.txt\n@@ -1,2 +1,2 @@' }),
207      Text({ color: 'red', children: '- hello world\n- 旧行：你好世界' }),
208      Text({ color: 'green', children: '+ hello mods\n+ 新行：宽度测试' }),
209      ...Array.from({ length: 60 }, (_, index) => Text({ children: `Row ${String(index + 1).padStart(2, '0')} | fixed scroll fixture | 中文宽度测试` })),
210    ] })
211  })
212}
213