cc-haha/desktop/src/pages/ActiveSession.tsx
程序员阿江(Relakkes) a94e5b641a fix: unify token usage display across desktop and CLI (#757)
Token counts were rendered with five inconsistent formats across the
chat UI (header "181,117 t", in-progress bare "↓ 2514", background
agents "12,345 tokens", compact summary "1.5k", trace "1.2k"), and the
in-progress count was actually stale: the server never set the status
event's tokens field, so the indicator showed the previous turn's value
(hence "first message shows nothing", "sometimes appears, never moves").

- Add shared desktop lib/formatTokenCount; route header, streaming
  indicator, background agents, compact summary, and trace through it.
- Header shows compact "124.3k tokens" with the exact count on hover.
- Add common.tokens i18n key (en/zh/zh-TW/jp/kr).
- Estimate the in-progress count from streamed chars (÷4, mirroring the
  CLI spinner) via a new streamingResponseChars per-session field, reset
  on each send; drop the dead status.tokens/elapsed fields.
- CLI spinner rows use formatTokens to drop the "1.0k" artifact.
2026-06-12 15:25:56 +08:00

593 lines
25 KiB
TypeScript

import { useEffect, useMemo, useRef, useState } from 'react'
import { Target } from 'lucide-react'
import {
SCHEDULED_TAB_ID,
SETTINGS_TAB_ID,
TERMINAL_TAB_PREFIX,
TRACE_TAB_PREFIX,
useTabStore,
type TabType,
} from '../stores/tabStore'
import { useSessionStore } from '../stores/sessionStore'
import { useChatStore } from '../stores/chatStore'
import { useCLITaskStore } from '../stores/cliTaskStore'
import { useTeamStore } from '../stores/teamStore'
import { useWorkspacePanelStore } from '../stores/workspacePanelStore'
import {
TERMINAL_PANEL_DEFAULT_HEIGHT,
TERMINAL_PANEL_MAX_HEIGHT,
TERMINAL_PANEL_MIN_HEIGHT,
useTerminalPanelStore,
} from '../stores/terminalPanelStore'
import { useTranslation } from '../i18n'
import { MessageList } from '../components/chat/MessageList'
import { ChatInput } from '../components/chat/ChatInput'
import { ComputerUsePermissionModal } from '../components/chat/ComputerUsePermissionModal'
import { SessionTaskBar } from '../components/chat/SessionTaskBar'
import { WorkbenchPanel } from '../components/workbench/WorkbenchPanel'
import { TeamStatusBar } from '../components/teams/TeamStatusBar'
import { TerminalSettings } from './TerminalSettings'
import type { SessionListItem } from '../types/session'
import type { ActiveGoalState } from '../types/chat'
import { useMobileViewport } from '../hooks/useMobileViewport'
import { isDesktopRuntime } from '../lib/desktopRuntime'
import { formatTokenCount } from '../lib/formatTokenCount'
import { publicAssetPath } from '../lib/publicAsset'
const TASK_POLL_INTERVAL_MS = 1000
const WORKSPACE_RESIZE_STEP = 32
const TERMINAL_RESIZE_STEP = 24
const CHAT_COLUMN_WITH_WORKSPACE_CLASS =
'min-w-[320px] flex-1 border-r border-[var(--color-border)] bg-[var(--color-surface)]'
function isSessionTabState(activeTabId: string | null, activeTabType: TabType | null | undefined) {
if (!activeTabId) return false
if (activeTabType === 'session') return true
if (activeTabType) return false
return activeTabId !== SETTINGS_TAB_ID &&
activeTabId !== SCHEDULED_TAB_ID &&
!activeTabId.startsWith(TERMINAL_TAB_PREFIX) &&
!activeTabId.startsWith(TRACE_TAB_PREFIX)
}
function getSessionTerminalCwd(session: SessionListItem | undefined) {
if (!session) return undefined
if (session.workDir && session.workDirExists !== false) return session.workDir
return session.projectPath || undefined
}
function ActiveGoalStrip({
goal,
isRunning,
compact,
}: {
goal: ActiveGoalState | null | undefined
isRunning: boolean
compact: boolean
}) {
const t = useTranslation()
if (!goal || goal.action === 'completed') return null
const objective = goal.objective ?? goal.message
if (!objective) return null
const statusLabel = isRunning
? t('chat.activeGoal.running')
: goal.status === 'paused'
? t('chat.activeGoal.paused')
: t('chat.activeGoal.active')
const meta = [
goal.budget ? t('chat.activeGoal.budget', { value: goal.budget }) : null,
goal.elapsed ? t('chat.activeGoal.elapsed', { value: goal.elapsed }) : null,
goal.continuations ? t('chat.activeGoal.continuations', { value: goal.continuations }) : null,
].filter((value): value is string => value !== null)
return (
<div
data-testid="active-goal-strip"
className={[
'mt-2 flex max-w-full items-center gap-2 rounded-[8px] border border-[var(--color-memory-border)] bg-[var(--color-memory-surface)] px-2.5 py-1.5',
compact ? 'text-[11px]' : 'text-[12px]',
].join(' ')}
>
<Target size={compact ? 13 : 14} className="shrink-0 text-[var(--color-memory-accent)]" strokeWidth={2.25} aria-hidden="true" />
<span className="shrink-0 font-semibold text-[var(--color-text-primary)]">
{t('chat.activeGoal.title')}
</span>
<span className="h-1.5 w-1.5 shrink-0 rounded-full bg-[var(--color-memory-accent)]" aria-hidden="true" />
<span className="shrink-0 text-[var(--color-text-tertiary)]">{statusLabel}</span>
<span className="min-w-0 flex-1 truncate font-medium text-[var(--color-text-primary)]" title={objective}>
{objective}
</span>
{meta.length > 0 ? (
<span className="hidden shrink-0 items-center gap-1.5 text-[11px] text-[var(--color-text-tertiary)] lg:flex">
{meta.map((item) => (
<span key={item} className="max-w-[140px] truncate">{item}</span>
))}
</span>
) : null}
</div>
)
}
function WorkspaceResizeHandle() {
const t = useTranslation()
const width = useWorkspacePanelStore((state) => state.width)
const setWidth = useWorkspacePanelStore((state) => state.setWidth)
const [dragState, setDragState] = useState<{ startX: number; startWidth: number } | null>(null)
const dragStateRef = useRef(dragState)
useEffect(() => {
dragStateRef.current = dragState
}, [dragState])
useEffect(() => {
if (!dragState) return
const handlePointerMove = (event: PointerEvent) => {
const current = dragStateRef.current
if (!current) return
setWidth(current.startWidth + current.startX - event.clientX)
}
const handlePointerUp = () => {
setDragState(null)
}
document.body.style.cursor = 'col-resize'
document.body.style.userSelect = 'none'
window.addEventListener('pointermove', handlePointerMove)
window.addEventListener('pointerup', handlePointerUp)
window.addEventListener('pointercancel', handlePointerUp)
return () => {
document.body.style.cursor = ''
document.body.style.userSelect = ''
window.removeEventListener('pointermove', handlePointerMove)
window.removeEventListener('pointerup', handlePointerUp)
window.removeEventListener('pointercancel', handlePointerUp)
}
}, [dragState, setWidth])
return (
<div
role="separator"
aria-label={t('workspace.resizePanel')}
aria-orientation="vertical"
aria-valuenow={width}
tabIndex={0}
data-testid="workspace-resize-handle"
onPointerDown={(event) => {
if (event.button !== 0) return
event.preventDefault()
setDragState({ startX: event.clientX, startWidth: width })
}}
onKeyDown={(event) => {
if (event.key === 'ArrowLeft') {
event.preventDefault()
setWidth(width + WORKSPACE_RESIZE_STEP)
}
if (event.key === 'ArrowRight') {
event.preventDefault()
setWidth(width - WORKSPACE_RESIZE_STEP)
}
}}
className="group relative z-10 flex w-2 shrink-0 cursor-col-resize items-stretch justify-center bg-[var(--color-surface)] outline-none focus-visible:bg-[var(--color-surface-container)]"
>
<div className="my-3 w-px rounded-full bg-[var(--color-border)] transition-colors group-hover:bg-[var(--color-border-focus)] group-focus-visible:bg-[var(--color-border-focus)]" />
</div>
)
}
function TerminalResizeHandle() {
const t = useTranslation()
const height = useTerminalPanelStore((state) => state.height)
const setHeight = useTerminalPanelStore((state) => state.setHeight)
const [dragState, setDragState] = useState<{ startY: number; startHeight: number } | null>(null)
const dragStateRef = useRef(dragState)
useEffect(() => {
dragStateRef.current = dragState
}, [dragState])
useEffect(() => {
if (!dragState) return
const handlePointerMove = (event: PointerEvent) => {
const current = dragStateRef.current
if (!current) return
setHeight(current.startHeight + current.startY - event.clientY)
}
const handlePointerUp = () => {
setDragState(null)
}
document.body.style.cursor = 'row-resize'
document.body.style.userSelect = 'none'
window.addEventListener('pointermove', handlePointerMove)
window.addEventListener('pointerup', handlePointerUp)
window.addEventListener('pointercancel', handlePointerUp)
return () => {
document.body.style.cursor = ''
document.body.style.userSelect = ''
window.removeEventListener('pointermove', handlePointerMove)
window.removeEventListener('pointerup', handlePointerUp)
window.removeEventListener('pointercancel', handlePointerUp)
}
}, [dragState, setHeight])
return (
<div
role="separator"
aria-label={t('terminal.resizePanel')}
aria-orientation="horizontal"
aria-valuemin={TERMINAL_PANEL_MIN_HEIGHT}
aria-valuemax={TERMINAL_PANEL_MAX_HEIGHT}
aria-valuenow={height}
tabIndex={0}
data-testid="terminal-resize-handle"
onPointerDown={(event) => {
if (event.button !== 0) return
event.preventDefault()
setDragState({ startY: event.clientY, startHeight: height })
}}
onKeyDown={(event) => {
if (event.key === 'ArrowUp') {
event.preventDefault()
setHeight(height + TERMINAL_RESIZE_STEP)
}
if (event.key === 'ArrowDown') {
event.preventDefault()
setHeight(height - TERMINAL_RESIZE_STEP)
}
if (event.key === 'Home') {
event.preventDefault()
setHeight(TERMINAL_PANEL_MIN_HEIGHT)
}
if (event.key === 'End') {
event.preventDefault()
setHeight(TERMINAL_PANEL_MAX_HEIGHT)
}
}}
onDoubleClick={() => setHeight(TERMINAL_PANEL_DEFAULT_HEIGHT)}
className="group flex h-2.5 shrink-0 cursor-row-resize items-center bg-[var(--color-surface)] outline-none focus-visible:bg-[var(--color-surface-container)]"
>
<div className="mx-3 h-px flex-1 rounded-full bg-[var(--color-border)] transition-colors group-hover:bg-[var(--color-border-focus)] group-focus-visible:bg-[var(--color-border-focus)]" />
</div>
)
}
export function ActiveSession() {
const isMobileLayout = useMobileViewport() && !isDesktopRuntime()
const activeTabId = useTabStore((s) => s.activeTabId)
const activeTabType = useTabStore((s) => s.tabs.find((tab) => tab.sessionId === s.activeTabId)?.type ?? null)
const sessions = useSessionStore((s) => s.sessions)
const connectToSession = useChatStore((s) => s.connectToSession)
const sessionState = useChatStore((s) => activeTabId ? s.sessions[activeTabId] : undefined)
const pendingComputerUsePermission = sessionState?.pendingComputerUsePermission ?? null
const fetchSessionTasks = useCLITaskStore((s) => s.fetchSessionTasks)
const trackedTaskSessionId = useCLITaskStore((s) => s.sessionId)
const hasIncompleteTasks = useCLITaskStore((s) => s.tasks.some((task) => task.status !== 'completed'))
const hasRunningTasks = useCLITaskStore((s) => s.tasks.some((task) => task.status === 'in_progress'))
const chatState = sessionState?.chatState ?? 'idle'
const tokenUsage = sessionState?.tokenUsage ?? { input_tokens: 0, output_tokens: 0 }
const hasRunningBackgroundTasks = Object.values(sessionState?.backgroundAgentTasks ?? {})
.some((task) => task.status === 'running')
const session = sessions.find((s) => s.id === activeTabId)
const memberInfo = useTeamStore((s) => activeTabId ? s.getMemberBySessionId(activeTabId) : null)
const activeTeam = useTeamStore((s) => s.activeTeam)
const isMemberSession = !!memberInfo
const showWorkbench = useWorkspacePanelStore((state) =>
activeTabId && isSessionTabState(activeTabId, activeTabType) && !isMemberSession && !isMobileLayout
? state.isPanelOpen(activeTabId)
: false,
)
const showRightPanel = showWorkbench
const rightPanelWidth = useWorkspacePanelStore((state) => state.width)
const showTerminalPanel = useTerminalPanelStore((state) =>
activeTabId && isSessionTabState(activeTabId, activeTabType) && !isMemberSession && !isMobileLayout
? state.isPanelOpen(activeTabId)
: false,
)
const terminalPanelRuntimeId = useTerminalPanelStore((state) =>
activeTabId && isSessionTabState(activeTabId, activeTabType) && !isMemberSession && !isMobileLayout
? state.panelBySession[activeTabId]?.runtimeId
: undefined,
)
const terminalPanelHeight = useTerminalPanelStore((state) => state.height)
useEffect(() => {
if (activeTabId && !isMemberSession) {
connectToSession(activeTabId)
}
}, [activeTabId, isMemberSession, connectToSession])
useEffect(() => {
if (!activeTabId || isMemberSession) return
const shouldPollTasks =
chatState !== 'idle' ||
(trackedTaskSessionId === activeTabId && hasIncompleteTasks)
if (!shouldPollTasks) return
void fetchSessionTasks(activeTabId)
const timer = setInterval(() => {
void fetchSessionTasks(activeTabId)
}, TASK_POLL_INTERVAL_MS)
return () => clearInterval(timer)
}, [
activeTabId,
isMemberSession,
chatState,
trackedTaskSessionId,
hasIncompleteTasks,
fetchSessionTasks,
])
const t = useTranslation()
const messages = sessionState?.messages ?? []
const streamingText = sessionState?.streamingText ?? ''
const activeGoal = sessionState?.activeGoal ?? null
const isEmpty = messages.length === 0 && !streamingText && (session?.messageCount ?? 0) === 0
const isHistoryLoading =
!isMemberSession &&
(session?.messageCount ?? 0) > 0 &&
messages.length === 0 &&
sessionState?.historyStatus === 'loading'
const historyError =
!isMemberSession &&
(session?.messageCount ?? 0) > 0 &&
messages.length === 0 &&
sessionState?.historyStatus === 'error'
? sessionState.historyError || t('session.historyLoadFailed')
: null
const visibleMessageCount = messages.length > 0 ? messages.length : session?.messageCount ?? 0
const isActive = chatState !== 'idle' ||
(trackedTaskSessionId === activeTabId && hasRunningTasks) ||
hasRunningBackgroundTasks
const totalTokens = tokenUsage.input_tokens + tokenUsage.output_tokens
const lastUpdated = useMemo(() => {
if (!session?.modifiedAt) return ''
const diff = Date.now() - new Date(session.modifiedAt).getTime()
if (diff < 60000) return t('session.timeJustNow')
if (diff < 3600000) return t('session.timeMinutes', { n: Math.floor(diff / 60000) })
if (diff < 86400000) return t('session.timeHours', { n: Math.floor(diff / 3600000) })
return t('session.timeDays', { n: Math.floor(diff / 86400000) })
}, [session?.modifiedAt, t])
if (!activeTabId) return null
return (
<div className="flex-1 flex relative overflow-hidden bg-background text-on-surface">
<div data-testid="active-session-content-row" className="flex min-h-0 min-w-0 flex-1">
<div
data-testid="active-session-chat-column"
className={`flex flex-col ${showRightPanel ? CHAT_COLUMN_WITH_WORKSPACE_CLASS : isMobileLayout ? 'min-w-0 flex-1' : 'min-w-[360px] flex-1'}`}
>
{isMemberSession && (
<div className="shrink-0 border-b border-[var(--color-border)] bg-[var(--color-surface-container)]">
<div className="mx-auto max-w-[860px] flex items-center justify-between gap-4 px-8 py-2">
<div className="min-w-0">
<div className="flex items-center gap-3">
{memberInfo?.status === 'running' && (
<span className="flex h-2 w-2 rounded-full bg-[var(--color-warning)] animate-pulse-dot" />
)}
{memberInfo?.status === 'completed' && (
<span className="material-symbols-outlined text-[14px] text-[var(--color-success)]" style={{ fontVariationSettings: "'FILL' 1" }}>check_circle</span>
)}
<span className="material-symbols-outlined text-[14px] text-[var(--color-text-tertiary)]">smart_toy</span>
<span className="text-sm font-semibold text-[var(--color-text-primary)]">
{memberInfo?.role}
</span>
{activeTeam && (
<span className="text-[10px] text-[var(--color-text-tertiary)]">
@ {activeTeam.name}
</span>
)}
</div>
<p className="mt-1 text-[11px] text-[var(--color-text-tertiary)]">
{t('teams.memberSessionHint')}
</p>
</div>
<button
onClick={() => {
if (activeTeam?.leadSessionId) {
useTabStore.getState().openTab(
activeTeam.leadSessionId,
t('teams.leader'),
'session',
)
}
}}
disabled={!activeTeam?.leadSessionId}
className="flex shrink-0 items-center gap-1 text-xs font-medium text-[var(--color-text-secondary)] hover:text-[var(--color-text-primary)] transition-colors disabled:opacity-50 disabled:hover:text-[var(--color-text-secondary)]"
>
<span className="material-symbols-outlined text-[14px]">arrow_back</span>
{t('teams.backToLeader')}
</button>
</div>
</div>
)}
{isEmpty ? (
<div className="flex flex-1 flex-col items-center justify-center p-8 pb-32">
<div className="flex max-w-md flex-col items-center text-center">
{isMemberSession ? (
<>
<span className="material-symbols-outlined text-[48px] mb-4 text-[var(--color-text-tertiary)]">smart_toy</span>
<p className="text-[var(--color-text-secondary)]">
{memberInfo?.status === 'running'
? `${memberInfo.role} ${t('teams.working')}`
: t('teams.noMessages')}
</p>
</>
) : (
<>
<img src={publicAssetPath('app-icon.png')} alt="Claude Code Haha" className="mb-6 h-24 w-24" />
<h1 className="mb-2 text-3xl font-extrabold tracking-tight text-[var(--color-text-primary)]" style={{ fontFamily: 'var(--font-headline)' }}>
{t('empty.title')}
</h1>
<p className="mx-auto max-w-xs text-[var(--color-text-secondary)]" style={{ fontFamily: 'var(--font-body)' }}>
{t('empty.subtitle')}
</p>
</>
)}
</div>
</div>
) : (
<>
{!isMemberSession && !isMobileLayout && (
<div
className={
showRightPanel
? 'flex w-full items-center border-b border-[var(--color-border)]/70 px-4 py-3'
: 'w-full border-b border-outline-variant/10 px-4 py-3'
}
>
<div className={showRightPanel ? 'min-w-0 flex-1' : 'mx-auto w-full max-w-[860px] min-w-0'}>
<div className="flex min-w-0 items-center gap-3">
<h1
className={
showRightPanel
? 'min-w-0 flex-1 truncate text-[15px] font-bold font-headline leading-tight text-on-surface'
: 'min-w-0 flex-1 text-lg font-bold font-headline text-on-surface leading-tight'
}
>
{session?.title || t('session.untitled')}
</h1>
</div>
<div
className={
showRightPanel
? 'mt-1 flex min-w-0 items-center gap-1.5 overflow-hidden whitespace-nowrap text-[10px] font-medium text-outline'
: 'flex items-center gap-2 text-[10px] text-outline font-medium mt-1'
}
>
{isActive && (
<span className="flex shrink-0 items-center gap-1">
<span className="w-1.5 h-1.5 rounded-full bg-[var(--color-success)] animate-pulse-dot" />
{t('session.active')}
</span>
)}
{totalTokens > 0 && (
<>
<span className="text-[var(--color-outline)]">·</span>
<span title={t('common.tokens', { count: totalTokens.toLocaleString() })}>
{t('common.tokens', { count: formatTokenCount(totalTokens) })}
</span>
</>
)}
{lastUpdated && (
<>
<span className="shrink-0 text-[var(--color-outline)]">·</span>
<span className="truncate">{t('session.lastUpdated', { time: lastUpdated })}</span>
</>
)}
{!showRightPanel && visibleMessageCount > 0 && (
<>
<span className="text-[var(--color-outline)]">·</span>
<span>{t('session.messages', { count: visibleMessageCount })}</span>
</>
)}
</div>
{session?.workDirExists === false && (
<div className="mt-2 inline-flex max-w-full items-center gap-2 rounded-lg border border-[var(--color-error)]/20 bg-[var(--color-error)]/8 px-3 py-1.5 text-[11px] text-[var(--color-error)]">
<span className="material-symbols-outlined text-[14px]">warning</span>
<span className="truncate">
{t('session.workspaceUnavailable', { dir: session.workDir || 'directory no longer exists' })}
</span>
</div>
)}
<ActiveGoalStrip
goal={activeGoal}
isRunning={isActive}
compact={showRightPanel}
/>
</div>
</div>
)}
{isHistoryLoading ? (
<div role="status" className="flex flex-1 items-center justify-center p-8 text-sm text-[var(--color-text-secondary)]">
<span className="material-symbols-outlined mr-2 animate-spin text-[18px]">progress_activity</span>
{t('common.loading')}
</div>
) : historyError ? (
<div role="alert" className="flex flex-1 items-center justify-center p-8 text-sm text-[var(--color-error)]">
{historyError}
</div>
) : (
<MessageList compact={showRightPanel} />
)}
</>
)}
{!isMemberSession && <SessionTaskBar />}
<TeamStatusBar />
<ChatInput
variant={isEmpty && !isMemberSession && !showRightPanel ? 'hero' : 'default'}
compact={showRightPanel}
/>
{terminalPanelRuntimeId && activeTabId ? (
<div
data-testid="session-terminal-panel"
className={[
'flex shrink-0 flex-col border-t border-[var(--color-border)] bg-[var(--color-surface-container-lowest)]',
showTerminalPanel ? '' : 'hidden',
].join(' ')}
style={{ height: showTerminalPanel ? terminalPanelHeight : 0 }}
>
{showTerminalPanel && <TerminalResizeHandle />}
<TerminalSettings
active={showTerminalPanel}
docked
cwd={getSessionTerminalCwd(session)}
runtimeId={terminalPanelRuntimeId}
preserveOnUnmount
testId={`session-terminal-host-${activeTabId}`}
onOpenInTab={() => {
useTerminalPanelStore.getState().closePanel(activeTabId)
useTabStore.getState().openTerminalTab(getSessionTerminalCwd(session), terminalPanelRuntimeId)
useTerminalPanelStore.getState().detachRuntime(activeTabId)
}}
onClose={() => useTerminalPanelStore.getState().closePanel(activeTabId)}
/>
</div>
) : null}
</div>
{showWorkbench ? (
<>
<WorkspaceResizeHandle />
<aside
data-testid="workbench-panel"
className="flex h-full shrink-0 flex-col border-l border-[var(--color-border)] bg-[var(--color-surface)]"
style={{ width: rightPanelWidth, maxWidth: '62%', minWidth: 'min(420px, 54%)' }}
>
<WorkbenchPanel sessionId={activeTabId} />
</aside>
</>
) : null}
</div>
{!isMemberSession && activeTabId ? (
<ComputerUsePermissionModal
sessionId={activeTabId}
request={pendingComputerUsePermission?.request ?? null}
/>
) : null}
</div>
)
}