feat(codex): 为会话管理器添加 Token 使用统计

- 更新 CodexSessionRecord 以包含 inputTokens、outputTokens 和 totalTokens。
- 实现 formatLargeNumber 和 formatTokenStats 函数来显示 Token 使用情况。
- 增强 CodexSessionManager 以在 UI 中显示 Token 使用情况。
- 添加 Token 使用统计对应的英文和中文翻译。
- 更新 Token 使用显示的样式。

Author: zz314657917

package-lock.json

@@ -34,6 +34,15 @@ pub struct CodexSessionRecord {
     pub updated_at: Option<i64>,
     pub location_count: usize,
     pub locations: Vec<CodexSessionLocation>,
+    /// 输入token数量（来自rollout文件的token_count记录）
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub input_tokens: Option<u64>,
+    /// 输出token数量（来自rollout文件的token_count记录）
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub output_tokens: Option<u64>,
+    /// 总token数量（来自rollout文件的token_count记录）
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub total_tokens: Option<u64>,
 }
 
 #[derive(Debug, Clone, Serialize)]
@@ -148,6 +157,54 @@ struct TrashedSessionEntry {
     trashed_rollout_path: PathBuf,
 }
 
+/// 从 rollout JSONL 文件中读取 token 统计信息
+/// 返回 (input_tokens, output_tokens, total_tokens)
+fn read_token_stats_from_rollout(rollout_path: &Path) -> Option<(u64, u64, u64)> {
+    let content = fs::read_to_string(rollout_path).ok()?;
+
+    // 从文件末尾向前读取，找到第一条 token_count 记录
+    // token_count 记录通常在文件末尾
+    for line in content.lines().rev() {
+        let trimmed = line.trim();
+        if trimmed.is_empty() {
+            continue;
+        }
+
+        if let Ok(parsed) = serde_json::from_str::<JsonValue>(trimmed) {
+            // 检查 type == "event_msg"
+            if parsed.get("type").and_then(|v| v.as_str()) != Some("event_msg") {
+                continue;
+            }
+
+            if let Some(payload) = parsed.get("payload") {
+                // 检查 payload.type == "token_count"
+                if payload.get("type").and_then(|v| v.as_str()) != Some("token_count") {
+                    continue;
+                }
+
+                if let Some(info) = payload.get("info") {
+                    if let Some(usage) = info.get("total_token_usage") {
+                        let input = usage
+                            .get("input_tokens")
+                            .and_then(|v| v.as_u64())
+                            .unwrap_or(0);
+                        let output = usage
+                            .get("output_tokens")
+                            .and_then(|v| v.as_u64())
+                            .unwrap_or(0);
+                        let total = usage
+                            .get("total_tokens")
+                            .and_then(|v| v.as_u64())
+                            .unwrap_or(0);
+                        return Some((input, output, total));
+                    }
+                }
+            }
+        }
+    }
+    None
+}
+
 pub fn list_sessions_across_instances() -> Result<Vec<CodexSessionRecord>, String> {
     let instances = collect_instances()?;
     let process_entries = modules::process::collect_codex_process_entries();
@@ -156,6 +213,9 @@ pub fn list_sessions_across_instances() -> Result<Vec<CodexSessionRecord>, Strin
     for instance in &instances {
         let running = is_instance_running(instance, &process_entries);
         for snapshot in load_thread_snapshots(instance)? {
+            // 读取 token 统计
+            let token_stats = read_token_stats_from_rollout(&snapshot.rollout_path);
+
             let entry =
                 session_map
                     .entry(snapshot.id.clone())
@@ -166,6 +226,9 @@ pub fn list_sessions_across_instances() -> Result<Vec<CodexSessionRecord>, Strin
                         updated_at: snapshot.updated_at,
                         location_count: 0,
                         locations: Vec::new(),
+                        input_tokens: token_stats.as_ref().map(|(i, _, _)| *i),
+                        output_tokens: token_stats.as_ref().map(|(_, o, _)| *o),
+                        total_tokens: token_stats.as_ref().map(|(_, _, t)| *t),
                     });
 
             if entry.updated_at.is_none() {

src-tauri/src/modules/codex_session_manager.rs

@@ -75,6 +75,26 @@ function formatSessionId(sessionId: string): string {
   return `${sessionId.slice(0, 8)}...${sessionId.slice(-6)}`;
 }
 
+function formatLargeNumber(value: number): string {
+  if (value >= 1_000_000) {
+    return `${(value / 1_000_000).toFixed(1)}M`;
+  }
+  if (value >= 1_000) {
+    return `${(value / 1_000).toFixed(1)}K`;
+  }
+  return value.toLocaleString();
+}
+
+function formatTokenStats(session: CodexSessionRecord): string {
+  const { inputTokens, outputTokens } = session;
+
+  if (inputTokens !== undefined && outputTokens !== undefined) {
+    return `${formatLargeNumber(inputTokens)} / ${formatLargeNumber(outputTokens)} tokens`;
+  }
+
+  return '';
+}
+
 export function CodexSessionManager() {
   const { t, i18n } = useTranslation();
   const instances = useCodexInstanceStore((state) => state.instances);
@@ -547,6 +567,11 @@ export function CodexSessionManager() {
                             >
                               {copiedSessionId === session.sessionId ? <Check size={14} /> : <Copy size={14} />}
                             </button>
+                            {formatTokenStats(session) && (
+                              <span className="codex-session-row__tokens" title={t('codex.sessionManager.labels.tokenUsage', 'Token使用')}>
+                                {formatTokenStats(session)}
+                              </span>
+                            )}
                             <span className="codex-session-row__time">
                               {formatRelativeTime(session.updatedAt, isZh)}
                             </span>

src/components/codex/CodexSessionManager.tsx

@@ -2191,11 +2191,12 @@
     "groupCount": "{{count}} sessions",
     "updatedAt": "Updated",
     "labels": {
-      "sessionId": "Session ID"
+      "sessionId": "Session ID",
+      "tokenUsage": "Token Usage"
     },
     "locationRunning": " (running)",
     "untitled": "Untitled session",
-      "restoreModal": {
+    "restoreModal": {
         "title": "Restore Sessions",
         "hint": "Restoring puts the rollout file, SQLite thread row, and session_index entry back into the original instance together.",
         "emptyTitle": "Trash is empty",

src/locales/en.json

@@ -2191,11 +2191,12 @@
     "groupCount": "{{count}} 条会话",
     "updatedAt": "最近更新",
     "labels": {
-      "sessionId": "会话 ID"
+      "sessionId": "会话 ID",
+      "tokenUsage": "Token使用"
     },
     "locationRunning": "（运行中）",
     "untitled": "未命名会话",
-      "restoreModal": {
+    "restoreModal": {
         "title": "恢复会话",
         "hint": "恢复会把 rollout 文件、SQLite 线程记录和 session_index 条目一起放回原实例。",
         "emptyTitle": "废纸篓里还没有会话",

src/locales/zh-CN.json

@@ -5421,6 +5421,14 @@
   outline-offset: 2px;
 }
 
+.codex-session-row__tokens {
+  font-size: 11px;
+  color: var(--text-muted);
+  margin-right: 8px;
+  white-space: nowrap;
+  opacity: 0.85;
+}
+
 .codex-session-restore-modal {
   width: min(760px, calc(100vw - 32px));
 }

src/styles/pages/codex.css

@@ -137,6 +137,12 @@ export interface CodexSessionRecord {
   updatedAt?: number | null;
   locationCount: number;
   locations: CodexSessionLocation[];
+  /** 输入token数量（来自rollout文件的token_count记录） */
+  inputTokens?: number;
+  /** 输出token数量（来自rollout文件的token_count记录） */
+  outputTokens?: number;
+  /** 总token数量（来自rollout文件的token_count记录） */
+  totalTokens?: number;
 }
 
 export interface CodexSessionTrashSummary {