throughline 0.3.24 → 0.4.0

package/src/state-file.mjs

@@ -2,12 +2,12 @@
  * state-file.mjs — セッション単位の状態ファイル管理（共有モジュール）
  *
  * パス: ~/.throughline/state/<session_id>.json
- * 書き手: turn-processor (Stop)
+ * 書き手: turn-processor (Claude Stop), codex-hook (Codex Stop)
  * 読み手: token-monitor
  *
  * 設計判断 (docs/PUBLIC_RELEASE_PLAN.md §4.5/4.6):
  *   - ファイル単位分割で last-writer-wins 問題を解消
- *   - PID 生存チェックで stale 削除（時間窓は使わない）
+ *   - updatedAt ベースで stale 判定（短命 hook process の PID には依存しない）
  *   - projectPath は path.resolve → / → 末尾 / 除去 → Windows lowercase で正規化
  */
 
@@ -37,21 +37,43 @@ export function normalizeProjectPath(p) {
 
 /**
  * セッション状態ファイルを書く
- * @param {{sessionId: string, projectPath: string, transcriptPath: string, pid: number, usage?: object|null}} data
+ * @param {{
+ *   sessionId: string,
+ *   projectPath: string,
+ *   transcriptPath?: string|null,
+ *   rolloutPath?: string|null,
+ *   pid?: number,
+ *   usage?: object|null,
+ *   host?: 'claude'|'codex',
+ * }} data
  *
  * usage: monitor が表示する tokens/model/contextWindowSize をここに固定保存する。
  * Stop hook が readLatestUsage の結果を載せることで、monitor 側が毎フレーム JSONL を
  * 再スキャンする必要がなくなる。旧バージョン互換のため optional (無ければ monitor が
  * transcriptPath を読んでフォールバック)。
  */
-export function writeSessionState({ sessionId, projectPath, transcriptPath, pid, usage }) {
+export function writeSessionState({
+  sessionId,
+  projectPath,
+  transcriptPath,
+  rolloutPath,
+  pid,
+  usage,
+  host,
+}) {
   if (!sessionId) throw new Error('writeSessionState: sessionId is required');
+  const normalizedHost = normalizeHost(host);
+  if (host && normalizedHost === 'unknown') {
+    throw new Error(`writeSessionState: unsupported host ${host}`);
+  }
   if (!existsSync(STATE_DIR)) mkdirSync(STATE_DIR, { recursive: true });
-  const file = join(STATE_DIR, `${sessionId}.json`);
+  const file = join(STATE_DIR, stateFilename(sessionId));
   const payload = {
     sessionId,
+    host: normalizedHost === 'unknown' ? 'claude' : normalizedHost,
     projectPath: normalizeProjectPath(projectPath),
     transcriptPath: transcriptPath ?? null,
+    rolloutPath: rolloutPath ?? null,
     pid: pid ?? process.pid,
     updatedAt: Date.now(),
   };
@@ -69,7 +91,7 @@ export const STALE_DELETE_MS = 24 * 60 * 60 * 1000; // 24 時間: ファイル
 /**
  * 全セッション状態を読む。24 時間超のファイルは削除、壊れたファイルも削除する。
  * 15 分超のファイルは「stale」フラグを付けて返す（monitor 側で隠す判断をする）。
- * @returns {Array<{sessionId: string, projectPath: string, transcriptPath: string|null, updatedAt: number, stale: boolean}>}
+ * @returns {Array<{sessionId: string, host: string, projectPath: string, transcriptPath: string|null, rolloutPath: string|null, updatedAt: number, stale: boolean}>}
  */
 export function readAllSessionStates() {
   if (!existsSync(STATE_DIR)) return [];
@@ -112,6 +134,7 @@ export function readAllSessionStates() {
       }
       continue;
     }
+    parsed = normalizeState(parsed);
     const age = now - (parsed.updatedAt ?? 0);
     if (age > STALE_DELETE_MS) {
       // 24h 超: ハード削除（無制限蓄積防止）
@@ -128,6 +151,27 @@ export function readAllSessionStates() {
   return results;
 }
 
+function stateFilename(sessionId) {
+  return `${encodeURIComponent(sessionId)}.json`;
+}
+
+function normalizeHost(host) {
+  if (host === undefined || host === null || host === '') return 'claude';
+  if (host === 'claude' || host === 'codex') return host;
+  return 'unknown';
+}
+
+function normalizeState(parsed) {
+  const host = normalizeHost(parsed?.host);
+  return {
+    ...parsed,
+    host,
+    projectPath: normalizeProjectPath(parsed?.projectPath ?? ''),
+    transcriptPath: parsed?.transcriptPath ?? null,
+    rolloutPath: parsed?.rolloutPath ?? null,
+  };
+}
+
 /**
  * ファイル単位の mtime スナップショットを取る（差分検知用）
  * @returns {Map<string, number>}

package/src/state-file.test.mjs

@@ -146,11 +146,58 @@ test('writeSessionState: usage 付きで書くと JSON に含まれる', async (
     const results = mod.readAllSessionStates();
     assert.equal(results.length, 1);
     assert.ok(results[0].usage);
+    assert.equal(results[0].host, 'claude');
+    assert.equal(results[0].rolloutPath, null);
     assert.equal(results[0].usage.tokens, 123);
     assert.equal(results[0].usage.model, 'claude-opus-4-6');
   });
 });
 
+test('writeSessionState: Codex state は host と rolloutPath を保持しファイル名を encode する', async () => {
+  await withIsolatedStateDir(async ({ stateDir, mod }) => {
+    mod.writeSessionState({
+      sessionId: 'codex:019dfaba-thread',
+      host: 'codex',
+      projectPath: '/tmp/x',
+      transcriptPath: null,
+      rolloutPath: '/tmp/codex/rollout.jsonl',
+      pid: 1,
+      usage: {
+        tokens: 123,
+        model: 'codex',
+        contextWindowSize: 258400,
+        contextWindowEstimated: false,
+        outputTokens: 10,
+        estimated: false,
+        source: 'codex-rollout-token-count',
+      },
+    });
+
+    assert.deepEqual(readdirSync(stateDir), ['codex%3A019dfaba-thread.json']);
+    const results = mod.readAllSessionStates();
+    assert.equal(results.length, 1);
+    assert.equal(results[0].sessionId, 'codex:019dfaba-thread');
+    assert.equal(results[0].host, 'codex');
+    assert.equal(results[0].transcriptPath, null);
+    assert.equal(results[0].rolloutPath, '/tmp/codex/rollout.jsonl');
+    assert.equal(results[0].usage.source, 'codex-rollout-token-count');
+  });
+});
+
+test('writeSessionState: unsupported host は throw する', async () => {
+  await withIsolatedStateDir(async ({ mod }) => {
+    assert.throws(
+      () =>
+        mod.writeSessionState({
+          sessionId: 'sess-bad-host',
+          host: 'unknown-host',
+          projectPath: '/tmp/x',
+        }),
+      /unsupported host/,
+    );
+  });
+});
+
 test('writeSessionState: usage 無しで書いたらフィールド自体が無い (旧フォーマット互換)', async () => {
   await withIsolatedStateDir(async ({ stateDir, mod }) => {
     mod.writeSessionState({
@@ -162,6 +209,7 @@ test('writeSessionState: usage 無しで書いたらフィールド自体が無
     const results = mod.readAllSessionStates();
     assert.equal(results.length, 1);
     assert.equal(results[0].usage, undefined);
+    assert.equal(results[0].host, 'claude');
   });
 });
 
@@ -178,6 +226,8 @@ test('readAllSessionStates: 旧バージョンが書いた usage 無しの state
     }));
     const results = mod.readAllSessionStates();
     assert.equal(results.length, 1);
+    assert.equal(results[0].host, 'claude');
+    assert.equal(results[0].rolloutPath, null);
     assert.equal(results[0].usage, undefined);
     // usage 無しで読めること自体が互換性の証明
   });

package/src/token-monitor.mjs

@@ -14,7 +14,8 @@
  *   - setInterval (1s) + mtime 差分検知で更新を捕捉
  *   - updatedAt 降順ソート、先頭行を ▶ でハイライト
  *   - stale は PID 生存チェックで判定
- *   - トークン数は transcript JSONL の最新 assistant usage を直読
+ *   - Claude は transcript JSONL の最新 assistant usage を直読
+ *   - Codex は Stop hook が state.usage に固定した rollout usage / estimate を表示
  */
 
 import { basename, dirname, join } from 'node:path';
@@ -191,7 +192,7 @@ function parseArgs(argv) {
 }
 
 // --- 表示 ---
-function renderBar(ratio, width = 20) {
+function renderBar(ratio, width = 10) {
   // NaN は 0、+Infinity は 1（オーバーフロー = 満タン表示）、負値 / -Infinity は 0 にクランプ
   let safe;
   if (Number.isNaN(ratio)) safe = 0;
@@ -288,11 +289,10 @@ export function resolveColumns() {
 function formatLine({ state, usage, isActive, now = Date.now() }) {
   const project = basename(state.projectPath || '?');
   const shortId = state.sessionId.slice(0, 8);
+  const host = state.host === 'codex' ? 'Codex' : state.host === 'unknown' ? 'Unknown' : 'Claude';
   const tokens = usage?.tokens ?? 0;
   const max = usage?.contextWindowSize ?? 200_000;
   const ratio = max > 0 ? tokens / max : 0;
-  const pct = Math.round(ratio * 100);
-  const remaining = Math.max(0, max - tokens);
 
   const bar = renderBar(ratio);
   const barColor =
@@ -310,11 +310,15 @@ function formatLine({ state, usage, isActive, now = Date.now() }) {
 
   const marker = isActive ? color(ANSI.bold + ANSI.cyan, '▶') : ' ';
   const projectCol = padCellsEnd(project, 18);
+  const hostCol = color(ANSI.dim, padCellsEnd(host, 6));
   const idCol = color(ANSI.dim, shortId);
   const barCol = color(barColor, bar);
-  const tokCol = `${formatNumber(tokens).padStart(6)} / ${pct.toString().padStart(3)}%`;
-  const remCol = color(ANSI.dim, `残 ${formatNumber(remaining)}`);
-  const modelCol = usage?.model ? color(ANSI.dim, usage.model) : color(ANSI.dim, '(未取得)');
+  const tokCol = `${formatNumber(tokens).padStart(6)} / ${formatNumber(max).padStart(6)}`;
+  const estimateMark = usage?.estimated ? ' est' : '';
+  const windowMark = usage?.contextWindowEstimated ? ' win?' : '';
+  const modelCol = usage?.model
+    ? color(ANSI.dim, `${usage.model}${estimateMark}${windowMark}`)
+    : color(ANSI.dim, '(未取得)');
   // 最終更新からの経過: 表示が「止まって見える」とき、それが idle なのか障害なのかを
   // 即座に判別できるようにする。updatedAt は state.writeSessionState 時の Date.now()。
   // 位置は session id の直後（左寄せ固定幅）。狭いターミナルでもモデル名より先に
@@ -325,7 +329,7 @@ function formatLine({ state, usage, isActive, now = Date.now() }) {
   // 8 セル固定: "just now" が最長 (8 セル)、"99d ago" は 7 セル。括弧なしで OK
   const agoCol = color(ANSI.dim, padCellsEnd(agoText, 8));
 
-  return `${marker} ${projectCol} ${idCol} ${agoCol} ${barCol} ${tokCol}  ${remCol}  ${modelCol}${warn}`;
+  return `${marker} ${projectCol} ${hostCol} ${idCol} ${agoCol} ${barCol} ${tokCol}  ${modelCol}${warn}`;
 }
 
 // --- フィルタ ---
@@ -428,8 +432,8 @@ function renderFrame(args) {
     for (let i = 0; i < filtered.length; i++) {
       const state = filtered[i];
       // Stop hook が state.usage に固定値を入れていればそれを使う（JSONL 再スキャン不要）。
-      // 旧バージョンが書いた state や usage スナップショットが取れなかったターンでは
-      // transcriptPath を直読してフォールバック。state 側の情報が 1 本化されると
+      // 旧バージョンが書いた Claude state や usage スナップショットが取れなかったターンでは
+      // transcriptPath を直読。state 側の情報が 1 本化されると
       // 「state が古い JSONL を指している」時の表示ブレが減る。
       const usage = state.usage
         ?? (state.transcriptPath ? readLatestUsage(state.transcriptPath) : null);

package/src/token-monitor.test.mjs

@@ -254,6 +254,10 @@ test('renderBar: ratio=0 は全部 ░', () => {
   assert.equal(renderBar(0, 5), '░░░░░');
 });
 
+test('renderBar: default width は 10 セル', () => {
+  assert.equal(renderBar(0.5), '█████░░░░░');
+});
+
 test('renderBar: ratio=1 は全部 █', () => {
   assert.equal(renderBar(1, 5), '█████');
 });
@@ -287,6 +291,7 @@ function makeLineArgs(ratio) {
   return {
     state: {
       sessionId: 'abc12345-xxxx',
+      host: 'claude',
       projectPath: '/tmp/foo',
       transcriptPath: null,
       updatedAt: Date.now(),
@@ -303,11 +308,33 @@ function makeLineArgs(ratio) {
 
 test('formatLine: 70% 未満は警告テキストなし', () => {
   const out = stripColors(formatLine(makeLineArgs(0.5)));
+  assert.ok(out.includes('Claude'));
+  assert.ok(out.includes('100.0k / 200.0k'));
+  assert.ok(!out.includes('残'));
+  assert.ok(!out.includes('/  50%'));
   assert.ok(!out.includes('!!'));
   assert.ok(!out.includes('!  '));
   assert.ok(!out.includes('/tl'));
 });
 
+test('formatLine: Codex estimated usage は host と est marker を表示する', () => {
+  const args = makeLineArgs(0.5);
+  args.state.host = 'codex';
+  args.usage = {
+    tokens: 100_000,
+    model: 'codex',
+    contextWindowSize: 200_000,
+    contextWindowEstimated: true,
+    outputTokens: 0,
+    estimated: true,
+    source: 'codex-rollout-chars-div-4',
+  };
+  const out = stripColors(formatLine(args));
+  assert.ok(out.includes('Codex'));
+  assert.ok(out.includes('100.0k / 200.0k'));
+  assert.ok(out.includes('codex est win?'));
+});
+
 test('formatLine: 70% 以上で "!" マーカーと弱めの文言', () => {
   const out = stripColors(formatLine(makeLineArgs(0.75)));
   assert.ok(out.includes('!'), 'should include ! marker');