reflexo

1import { atom, read, update } from 'claude-code'
2import type { Register, EngineInterface } from 'claude-code'
3import { looksLikeRule } from './rule'
4
5const isEnabledRef = atom({ plugin: 'reflexo', key: 'isEnabled' } as const, true)
6
7const PROJECT = 'Projeto (CLAUDE.md)'
8const GLOBAL = 'Global (~/.claude/CLAUDE.md)'
9const IGNORE = 'Ignorar'
10const HEADING = '## Regras capturadas'
11// Only what a person typed is a correction; task notifications and other agents' prompts are not.
12const HUMAN = new Set(['composer', 'bridge'])
13
14export const register: Register = on => {
15  // The rule waits from prompt.submit to the end of that turn; one session runs one turn at a time, so one slot is enough.
16  let queuedRule: string | null = null
17
18  on('session.start', async ($, e, next) => {
19    const saved = await $.store.get('isEnabled').catch(() => undefined)
20    await update($, isEnabledRef, () => (typeof saved === 'boolean' ? saved : isEnabledRef.initial))
21
22    await $.command.register({
23      name: 'reflexo',
24      description: 'Liga/desliga o reflexo (salvar correções como regras).',
25      argumentHint: 'on|off',
26    })
27    return next(e)
28  })
29
30  on('command.run', { command: 'reflexo' }, async ($, e) => {
31    const arg = e.args.trim().toLowerCase()
32    if (arg !== 'on' && arg !== 'off') {
33      const isEnabled = await read($, isEnabledRef)
34      return { text: `reflexo esta ${isEnabled ? 'ativado' : 'desativado'}. Uso: /reflexo on · /reflexo off` }
35    }
36    const isEnabled = arg === 'on'
37    await update($, isEnabledRef, () => isEnabled)
38    await $.store.set('isEnabled', isEnabled)
39    return { text: `reflexo ${isEnabled ? 'ativado' : 'desativado'}.` }
40  })
41
42  on('prompt.submit', async ($, e, next) => {
43    if (HUMAN.has(e.origin.kind) && (await read($, isEnabledRef)) && looksLikeRule(e.text)) queuedRule = e.text
44    return next(e) // never blocks or drops the prompt
45  }).catch(($, e, next) => next(e))
46
47  on('turn.complete', async ($, e, next) => {
48    if (e.agentId) return next(e)
49    const rule = queuedRule
50    queuedRule = null
51    if (rule === null) return next(e)
52    const answer = await $.ui.ask(`Salvar como regra: "${rule}"?`, { header: 'Reflexo', options: [PROJECT, GLOBAL, IGNORE] })
53    if (answer === PROJECT) await appendRule($, 'CLAUDE.md', rule) // relative = session cwd
54    else if (answer === GLOBAL) {
55      const home = (await $.env.get('USERPROFILE')) ?? (await $.env.get('HOME'))
56      if (home) await appendRule($, `${home}/.claude/CLAUDE.md`, rule)
57    }
58    return next(e) // IGNORE or free text: nothing written
59  }).catch(($, e, next) => next(e)) // fail open: a dismissed ask (rejects) or a write error never blocks the turn
60}
61
62/** Appends one line under the heading; never rewrites what is already in the file. */
63async function appendRule($: EngineInterface, path: string, rule: string): Promise<void> {
64  // Only a missing file counts as empty; any other read error rethrows so nothing is written (fail open upstream).
65  const current = await $.fs.read(path).then(r => (typeof r === 'string' ? r : ''), (err: unknown) => {
66    const e = err as { code?: string; message?: string }
67    if (e?.code === 'ENOENT' || /ENOENT/.test(e?.message ?? '')) return ''
68    throw err
69  })
70  const nl = current.includes('\r\n') ? '\r\n' : '\n'
71  const lines = current.split('\n')
72  const at = lines.findIndex(l => l.trim() === HEADING)
73  let end = at === -1 ? -1 : lines.findIndex((l, i) => i > at && /^#{1,6}\s/.test(l))
74  if (end !== -1) {
75    // Another section follows: insert after the section's last non-blank line, keep every other line as is.
76    while (end > at + 1 && lines[end - 1]!.trim() === '') end--
77    lines.splice(end, 0, `- ${rule}${nl === '\r\n' ? '\r' : ''}`)
78    return $.fs.write(path, lines.join('\n'))
79  }
80  const eol = current === '' || current.endsWith('\n') ? '' : nl
81  const head = at !== -1 ? '' : `${current === '' ? '' : nl}${HEADING}${nl}`
82  await $.fs.write(path, `${current}${eol}${head}- ${rule}${nl}`)
83}
84

1export function looksLikeRule(text: string): boolean {
2  // One line only: a newline would land in CLAUDE.md as a line of its own, maybe a fake "## " heading.
3  if (!text || text.length > 200 || /[\r\n]/.test(text)) return false
4  // "não sei", "não entendi", "não funcionou" and similar are answers, not rules
5  if (/^(n[ãa]o (sei|entendi|funcion|consegui|deu|rodou)|i don't know)/i.test(text)) return false
6  // 'no,' needs the comma: bare 'no' is Portuguese "in the" ("No arquivo X...")
7  return /^(?:(?:não|nao|nunca|sempre|pare de|prefiro|don't|always|never|stop)\b|no,)/i.test(text)
8}
9

1declare module 'claude-code' {
2  interface PluginState {
3    reflexo: {
4      isEnabled: boolean
5    }
6  }
7}
8