model-by-title

1import type { EngineInterface, Register } from 'claude-code'
2import { expectedFamily, lastModel, planFor, transcriptDir } from './rules'
3
4// 親が子からのメッセージ (起動報告など) を受けたら、自分が起動した子を見直し、
5// タイトル規約と違うモデルの子を set_session_model で切り替える。
6// - きっかけ: デスクトップでは子の send_message は session.receive も prompt.submit も通らず、
7//   classic.UserPromptSubmit の e.prompt に <cross-session-message …> を含む形で届く。
8//   それを含むときだけ見直す (ユーザーの入力では動かさない)
9// - set_session_model は自分自身には効かないので、起動元の親から直す。
10//   mod からの呼び出しも auto mode の判定を通るので、permissions.allow に
11//   mcp__ccd_session_mgmt__set_session_model が要る (README)
12// - 今のモデルは transcript の最後の応答から読む (get_session の model は claude.ai 側の切り替えを拾わない)
13// - 1 つの子につき判定は 1 回 (期待どおり / 切り替えた / 失敗)。以後は手で変えたものを尊重する
14// - $ を渡す先は最上位の function (claude plugin validate の要件)。状態はモジュール変数に置く
15const SERVER = 'ccd_session_mgmt'
16const MARKER = '<' + 'cross-session-message'
17
18type Row = { sessionId: string; title?: string; cwd?: string; isArchived?: boolean }
19type Session = { sessionId?: string; parentSessionId?: string }
20
21const textOf = (r: { content: { type: string; text?: string }[] }) => r.content.find(c => c.type === 'text')?.text ?? ''
22
23// MCP 結果のテキストは JSON の後ろに注記が付くことがある。最初の [ か { から最後の ] か } までを parse する
24const parseJson = <T>(text: string): T => {
25  const start = text.search(/[[{]/)
26  const end = Math.max(text.lastIndexOf(']'), text.lastIndexOf('}'))
27  return JSON.parse(text.slice(start, end + 1)) as T
28}
29
30let isBusy = false
31const othersChildren = new Set<string>() // 他の親の子。get_session を毎回打たない
32
33export const register: Register = on => {
34  // 読み込まれた印 (子からメッセージが来るまで何もしないので、これが無いと入ったか分からない)
35  on('session.start', async ($, e, next) => {
36    const result = await next(e)
37    $.ui.status('model-by-title')
38    return result
39  })
40
41  on('classic.UserPromptSubmit', async ($, e, next) => {
42    if (typeof e.prompt === 'string' && e.prompt.includes(MARKER)) await review($)
43    return next(e)
44  })
45}
46
47async function review($: EngineInterface) {
48  if (isBusy) return
49  isBusy = true
50  try {
51    const home = (await $.env.get('HOME')) ?? ''
52    const me = parseJson<Session>(textOf(await $.mcp.call(SERVER, 'get_session', { session_id: 'self' }))).sessionId
53    // linked: true は start_session の家族だけで、spawn_task のチップから起動した子は入らない。
54    // 全件から題で絞り、get_session の parentSessionId で自分の子か確かめる
55    const rows = parseJson<Row[]>(textOf(await $.mcp.call(SERVER, 'list_sessions', { limit: 50 })))
56    for (const row of rows) {
57      if (row.isArchived || !row.cwd || expectedFamily(row.title) === null) continue
58      const key = `settled:${row.sessionId}`
59      if (othersChildren.has(row.sessionId) || (await $.store.get(key))) continue
60      const info = parseJson<Session>(textOf(await $.mcp.call(SERVER, 'get_session', { session_id: row.sessionId })))
61      if (!me) continue
62      if (info.parentSessionId !== me) {
63        othersChildren.add(row.sessionId)
64        continue
65      }
66
67      const model = await currentModel($, transcriptDir(home, row.cwd))
68      if (model === undefined) continue
69      const to = planFor(row.title, model)
70      if (to === null) {
71        await $.store.set(key, 'ok')
72        await log($, `ok ${model} ${row.title}`)
73        continue
74      }
75
76      const r = await $.mcp.call(SERVER, 'set_session_model', { session_id: row.sessionId, model: to })
77      await $.store.set(key, r.isError ? 'failed' : 'switched')
78      await log($, `${r.isError ? 'failed' : 'switched'} ${model} → ${to} ${row.title}${r.isError ? ` :: ${textOf(r).slice(0, 200)}` : ''}`)
79      $.ui.toast(r.isError ? `model-by-title: 切替失敗 ${row.title}` : `model-by-title: ${row.title} → ${to}`)
80    }
81  } catch (err) {
82    await log($, `error ${String(err).slice(0, 200)}`)
83  } finally {
84    isBusy = false
85  }
86}
87
88// 応答がまだ無い (起動直後) なら undefined。次のメッセージでもう一度見る
89async function currentModel($: EngineInterface, dir: string) {
90  if (!(await $.fs.exists(dir))) return undefined
91  const newest = (await $.fs.list(dir))
92    .filter(f => f.kind === 'file' && f.name.endsWith('.jsonl'))
93    .sort((a, b) => b.mtimeMs - a.mtimeMs)[0]
94  return newest ? lastModel(await $.fs.read(`${dir}/${newest.name}`)) : undefined
95}
96
97async function log($: EngineInterface, line: string) {
98  const prev = ((await $.store.get('log')) as string[] | undefined) ?? []
99  await $.store.set('log', [...prev, `${new Date(await $.clock.now()).toISOString()} ${line}`].slice(-100))
100}
101

1// タイトル規約 (ippoan/claude-skills task-split §1)。claude-code-web-split-view の content.js と同じ判定。
2//   親:  #p<issue> <題>                                → Opus (Fable でも可)
3//   子:  [S]/[O] #p<issue>-c<番号>(-<n>) <題>   (c の後は子 issue 番号でも分岐番号でもよい。-<n> は再起票の連番)
4//        [S] = Sonnet、それ以外 (= 既定) は Opus
5//   [旧] #p… は旧親 (何もしない)。規約外も触らない。旧形 #c<issue>-<n> は廃止 (#19) で規約外
6export type Family = 'sonnet' | 'opus'
7
8export const cleanTitle = (t: string | undefined): string =>
9  String(t ?? '').replace(/[\uE000-\uF8FF\u200B-\u200D\u2060\uFEFF]/g, '').replace(/\s+/g, ' ').trim()
10
11export const expectedFamily = (title: string | undefined): Family | null => {
12  const s = cleanTitle(title)
13  const m0 = s.match(/^\[(S|O|旧)\]\s*/i)
14  const tag = m0?.[1]?.toUpperCase() ?? null
15  const rest = m0 ? s.slice(m0[0].length) : s
16  if (tag === '旧') return null
17  const tagged: Family = tag === 'S' ? 'sonnet' : 'opus'
18  if (/^#p\d+-c\d+(?:-\d+)?(\s|$)/.test(rest)) return tagged
19  if (/^#p\d+(\s|$)/.test(rest)) return 'opus'
20  if (/^#c\d/.test(rest)) return null // 廃止した旧形 (#19)。タグがあっても触らない
21  return tag ? tagged : null
22}
23
24// 期待どおりなら null、違えば差し替え先のモデル id
25const OK: Record<Family, RegExp> = { sonnet: /sonnet/i, opus: /opus|fable/i }
26const PICK: Record<Family, string> = { sonnet: 'claude-sonnet-5-5', opus: 'claude-opus-5-5' }
27
28export const rewriteTo = (family: Family, model: string): string | null =>
29  OK[family].test(model) ? null : PICK[family]
30
31// 題とモデルから、切り替え先 (不要なら null) を出す
32export const planFor = (title: string | undefined, model: string): string | null => {
33  const family = expectedFamily(title)
34  return family === null ? null : rewriteTo(family, model)
35}
36
37// transcript (jsonl) の置き場: ~/.claude/projects/<cwd の / と . を - にしたもの>
38export const transcriptDir = (home: string, cwd: string): string =>
39  `${home}/.claude/projects/${cwd.replace(/[/.]/g, '-')}`
40
41// transcript の最後の assistant 応答のモデル。get_session の model は Remote Control 側の切り替えを拾わないので、こちらを正とする
42export const lastModel = (jsonl: string): string | undefined => {
43  const lines = jsonl.split('\n')
44  for (let i = lines.length - 1; i >= 0; i--) {
45    const line = lines[i]
46    if (!line?.includes('"assistant"')) continue
47    try {
48      const row = JSON.parse(line) as { type?: string; message?: { model?: string } }
49      if (row.type === 'assistant' && row.message?.model) return row.message.model
50    } catch {
51      // 書きかけの行は飛ばす
52    }
53  }
54  return undefined
55}
56