pocket-pet

1// pocket-pet: 바탕화면에 사는 픽셀 펫 "클로"(주황 몸에 민트 스카프)의 Claude Code 쪽 절반.
2//
3// 이 mod는 Claude Code 화면에 아무것도 그리지 않는다. Claude가 하는 일(생각·도구·허락 대기·
4// 완료·멈춤)을 세션마다 상태 파일(~/.claude/clo/sessions/<세션 ID>.json)에 적고, 바탕화면 클로
5// 프로그램(desktop/clo_pet.pyw)을 띄운다. 프로그램은 살아 있는 세션들의 파일을 읽어 반응한다
6// (허락 대기가 먼저, 그다음 가장 최근에 바뀐 세션). 창은 마우스로 끌어 옮길 수 있다.
7//
8// 바탕화면 클로의 입력 카드에서 이 세션을 골라 보내면, 클로가 ~/.claude/clo/inbox/<세션 ID>.json 에
9// 메시지를 남기고, 이 mod가 1초마다 확인해 사용자가 입력한 것처럼 이 세션에 넣는다.
10//
11//   /pet          바탕화면 클로 켜기(이미 떠 있으면 화면 안으로 불러오기)
12//   /pet pat      쓰다듬기
13//   /pet 끄기     바탕화면 클로 끄기
14
15import type { EngineInterface, Register } from 'claude-code'
16
17type Mood = 'idle' | 'thinking' | 'working' | 'waiting' | 'done' | 'oops'
18type Command = 'show' | 'pat' | 'quit'
19
20/** 바탕화면 클로가 읽는 상태 파일의 모양 */
21export type PetState = {
22  v: 1
23  name: string
24  mood: Mood
25  /** 말풍선에 쓸 한 줄 */
26  detail: string
27  /** 읽기·찾기 도구 중인지(살펴보는 동작) */
28  isReading: boolean
29  toolCount: number
30  turnStartedAt: number
31  /** mood가 바뀐 시각(ms) */
32  at: number
33  /** 세션이 살아 있다는 표시(1분마다) */
34  alive: number
35  /** 세션의 작업 폴더: 바탕화면 클로에서 새 세션을 열 때 고를 수 있다 */
36  cwd?: string
37  /** 세션 제목: Claude 앱에 보이는 세션 제목(아직 없으면 첫 질문의 앞부분). 바탕화면 클로의 세션 목록에 보인다 */
38  title?: string
39  /** 바탕화면 클로가 남긴 메시지를 받을 수 있다(이 표시가 있는 세션만 '열린 세션' 목록에 나온다) */
40  canReceive?: boolean
41  command?: { id: number; kind: Command }
42}
43
44const HEARTBEAT_MS = 60_000
45const INBOX_MS = 1_000 // 바탕화면 클로가 남긴 메시지를 확인하는 간격
46const INBOX_FRESH_MS = 2 * 60_000 // 이보다 오래된 메시지는 넣지 않는다
47
48/** 바탕화면 클로가 이 세션 앞으로 남기는 메시지 */
49export type InboxMessage = { id: number; text: string; at: number }
50const READING_TOOLS = new Set(['Read', 'Grep', 'Glob', 'WebFetch', 'WebSearch', 'LS', 'NotebookRead'])
51
52// 모듈 상태(다시 불러오면 처음부터)
53let petName = '클로'
54let isAutoStart = true
55let python = ''
56let stateDir = ''
57let state: PetState = { v: 1, name: '클로', mood: 'idle', detail: '', isReading: false, toolCount: 0, turnStartedAt: 0, at: 0, alive: 0 }
58let isTurnRunning = false
59let toolsRunning = 0
60let commandId = 0
61let inboxDir = ''
62let lastInboxId = 0 // 이미 넣은(또는 시작 전에 있던) 메시지
63let isCheckingInbox = false
64
65async function homeOf($: EngineInterface): Promise<string> {
66  return (await $.env.get('USERPROFILE')) ?? (await $.env.get('HOME')) ?? '.'
67}
68
69const isWindowsPath = (p: string) => p.includes('\\') || /^[A-Za-z]:/.test(p)
70const join = (base: string, ...parts: string[]) => [base, ...parts].join(isWindowsPath(base) ? '\\' : '/')
71
72async function save($: EngineInterface, patch: Partial<PetState>) {
73  const now = await $.clock.now()
74  if (patch.mood !== undefined && patch.mood !== state.mood) patch.at = now
75  state = { ...state, ...patch, name: petName, alive: patch.alive ?? now }
76  if (!stateDir) return
77  try {
78    // /clear 뒤에는 세션 ID가 바뀌므로 쓸 때마다 묻는다
79    await $.fs.write(join(stateDir, `${await $.session.id()}.json`), JSON.stringify(state))
80  } catch {
81    // 파일을 못 써도 Claude Code 작업은 계속된다
82  }
83}
84
85/** 바탕화면 클로 프로그램 띄우기. 이미 떠 있으면 프로그램이 알아서 하나만 남긴다 */
86async function launch($: EngineInterface): Promise<boolean> {
87  const script = join($.plugin.root, 'desktop', 'clo_pet.pyw')
88  const isWindows = isWindowsPath(script)
89  const argv = isWindows
90    ? [
91        'powershell.exe',
92        '-NoProfile',
93        '-NonInteractive',
94        '-Command',
95        `Start-Process -FilePath '${(python || 'pythonw').replace(/'/g, "''")}' -ArgumentList @('"${script.replace(/'/g, "''")}"')`,
96      ]
97    : ['sh', '-c', `nohup "${python || 'python3'}" "$1" >/dev/null 2>&1 &`, 'sh', script]
98  try {
99    const r = await $.process.run(argv, { timeoutMs: 15_000 })
100    return r.exitCode === 0
101  } catch {
102    return false
103  }
104}
105
106async function send($: EngineInterface, kind: Command) {
107  commandId += 1
108  await save($, { command: { id: (await $.clock.now()) * 10 + (commandId % 10), kind } })
109}
110
111async function readInbox($: EngineInterface): Promise<InboxMessage | undefined> {
112  if (!inboxDir) return undefined
113  const path = join(inboxDir, `${await $.session.id()}.json`)
114  try {
115    if (!(await $.fs.exists(path))) return undefined
116    const m = JSON.parse(await $.fs.read(path)) as InboxMessage
117    return typeof m.id === 'number' && typeof m.text === 'string' ? m : undefined
118  } catch {
119    return undefined // 쓰는 도중이면 다음에 다시
120  }
121}
122
123/** 바탕화면 클로가 남긴 새 메시지를 이 세션에 사용자 입력처럼 넣는다 */
124async function checkInbox($: EngineInterface) {
125  if (isCheckingInbox) return
126  isCheckingInbox = true
127  try {
128    const m = await readInbox($)
129    if (!m || m.id <= lastInboxId) return
130    lastInboxId = m.id
131    if (!m.text.trim() || (await $.clock.now()) - m.at > INBOX_FRESH_MS) return
132    await $.prompt.submit({ text: m.text, asUser: true })
133  } catch {
134    // 넣지 못해도 다음 메시지는 계속 받는다
135  } finally {
136    isCheckingInbox = false
137  }
138}
139
140/** 도구 호출을 사람이 읽는 한 줄로 */
141export function describeTool(tool: string, args: Record<string, unknown>): string {
142  const str = (k: string) => (typeof args[k] === 'string' ? (args[k] as string) : '')
143  const base = (p: string) => p.split(/[\\/]/).pop() || p
144  const cut = (s: string, n = 40) => (s.length > n ? s.slice(0, n - 1) + '…' : s)
145  switch (tool) {
146    case 'Read':
147      return `읽는 중: ${base(str('file_path'))}`
148    case 'Edit':
149    case 'MultiEdit':
150    case 'NotebookEdit':
151      return `고치는 중: ${base(str('file_path') || str('notebook_path'))}`
152    case 'Write':
153      return `쓰는 중: ${base(str('file_path'))}`
154    case 'Bash':
155    case 'PowerShell':
156      return `실행 중: ${cut(str('command').split('\n')[0] ?? '')}`
157    case 'Grep':
158    case 'Glob':
159      return `찾는 중: ${cut(str('pattern'), 30)}`
160    case 'WebSearch':
161      return `웹 검색: ${cut(str('query'), 30)}`
162    case 'WebFetch':
163      return `웹 페이지 읽는 중`
164    case 'Task':
165    case 'Agent':
166      return `도우미를 보내는 중`
167    case 'TodoWrite':
168      return `할 일 정리 중`
169    default: {
170      const m = /^mcp__(.+?)__(.+)$/.exec(tool)
171      return m ? `${m[1]} 도구: ${m[2]}` : `${tool} 쓰는 중`
172    }
173  }
174}
175
176function durationText(ms: number): string {
177  const s = Math.max(1, Math.round(ms / 1000))
178  return s < 60 ? `${s}초` : `${Math.floor(s / 60)}분 ${s % 60}초`
179}
180
181export const register: Register = (on, options) => {
182  if (typeof options.pet_name === 'string' && options.pet_name.trim()) petName = options.pet_name.trim()
183  isAutoStart = options.auto_start !== 'off'
184  python = typeof options.python === 'string' ? options.python.trim() : ''
185
186  on('session.start', async ($, e, next) => {
187    const result = await next(e)
188    stateDir = join(await homeOf($), '.claude', 'clo', 'sessions')
189    await save($, { mood: 'idle', detail: '', cwd: e.cwd, canReceive: true })
190    // 바탕화면 클로가 남긴 메시지: 시작 전에 있던 것은 넣지 않는다
191    inboxDir = join(await homeOf($), '.claude', 'clo', 'inbox')
192    const old = await readInbox($)
193    if (old) lastInboxId = old.id
194    $.clock.every(INBOX_MS, () => {
195      void checkInbox($)
196    })
197    $.clock.every(HEARTBEAT_MS, () => {
198      void save($, {})
199    })
200    if (isAutoStart) void launch($)
201
202    try {
203      await $.command.register({
204        name: 'pet',
205        description: '바탕화면 클로 켜기 (pat 쓰다듬기 · 끄기)',
206        argumentHint: '[pat|끄기]',
207        immediate: true,
208      })
209    } catch {
210      // /pet 이 이미 있으면 명령 없이 상태만 적는다
211    }
212    return result
213  })
214
215  on('command.run', { command: 'pet' }, async ($, e) => {
216    const arg = e.args.trim().toLowerCase()
217    if (arg === 'pat' || arg === '쓰다듬기') {
218      await send($, 'pat')
219    } else if (arg === '끄기' || arg === 'quit' || arg === 'off' || arg === '숨기기') {
220      await send($, 'quit')
221      $.ui.toast(`${petName}가 쉬러 갔어요 (/pet 으로 부르기)`)
222    } else {
223      await send($, 'show')
224      const ok = await launch($)
225      $.ui.toast(ok ? `${petName}가 바탕화면에 나왔어요` : `${petName}를 띄우지 못했어요: Python(pythonw)이 있는지 확인해 주세요`)
226    }
227    return {}
228  })
229
230  // 세션이 끝나면 살아 있다는 표시를 지워 바탕화면 클로가 이 세션을 잊게 한다
231  on('session.end', async ($, e, next) => {
232    await save($, { mood: 'idle', detail: '', alive: 0 })
233    return next(e)
234  })
235
236  // ── Claude가 하는 일 ──────────────────────────────────────────────
237  // Claude 앱에 보이는 세션 제목: 질문을 받을 때·세션을 열 때 함께 온다(이름을 바꾸면 다음 질문 때 따라간다)
238  on('classic.UserPromptSubmit', async ($, e, next) => {
239    if (e.session_title && e.session_title !== state.title) await save($, { title: e.session_title })
240    return next(e)
241  })
242  on('classic.SessionStart', async ($, e, next) => {
243    const result = await next(e)
244    if (e.session_title && e.session_title !== state.title) await save($, { title: e.session_title })
245    return result
246  })
247
248  on('turn.start', async ($, e, next) => {
249    // 앱의 세션 제목이 아직 없으면 첫 질문을 제목으로(바탕화면 클로의 세션 목록에 보인다)
250    if (!state.title && e.text.trim()) {
251      const first = e.text.trim().split('\n')[0] ?? ''
252      state.title = first.length > 40 ? first.slice(0, 39) + '…' : first
253    }
254    isTurnRunning = true
255    toolsRunning = 0
256    await save($, { mood: 'thinking', detail: '', toolCount: 0, isReading: false, turnStartedAt: await $.clock.now() })
257    return next(e)
258  })
259
260  on('tool.call', async ($, e, next) => {
261    toolsRunning += 1
262    const words = (e.agentId ? '도우미: ' : '') + describeTool(String(e.tool), e as unknown as Record<string, unknown>)
263    const isReading = READING_TOOLS.has(String(e.tool))
264    if (state.mood === 'waiting') await save($, { toolCount: state.toolCount + 1, isReading })
265    else await save($, { mood: 'working', detail: words, toolCount: state.toolCount + 1, isReading })
266    try {
267      return await next(e)
268    } finally {
269      toolsRunning -= 1
270      // 동시에 돌던 도구가 모두 끝났을 때만 생각 중으로 돌아간다
271      if (toolsRunning === 0 && isTurnRunning && (state.mood === 'working' || state.mood === 'waiting')) {
272        await save($, { mood: 'thinking', detail: '', isReading: false })
273      }
274    }
275  })
276
277  on('classic.PermissionRequest', async ($, e, next) => {
278    await save($, { mood: 'waiting', detail: `허락이 필요해요 → ${describeTool(e.tool_name, (e.tool_input ?? {}) as Record<string, unknown>)}` })
279    return next(e)
280  })
281
282  on('turn.complete', async ($, e, next) => {
283    const result = await next(e)
284    if (e.agentId) return result // 도우미의 턴은 건너뛴다
285    isTurnRunning = false
286    const tools = state.toolCount > 0 ? ` · 도구 ${state.toolCount}번` : ''
287    if (e.reason === 'answer') await save($, { mood: 'done', detail: `다 했어요! ${durationText(e.durationMs)}${tools}`, isReading: false })
288    else if (e.reason === 'aborted') await save($, { mood: 'oops', detail: '멈췄어요. 다시 말해 주세요', isReading: false })
289    else await save($, { mood: 'oops', detail: '문제가 생겼어요… 다시 해 볼까요?', isReading: false })
290    return result
291  })
292}
293