claude-code-best/src/components/messages/teamMemCollapsed.tsx
2026-04-06 23:56:45 +08:00

99 lines
2.5 KiB
TypeScript

import React from 'react'
import { Text } from '@anthropic/ink'
import type { CollapsedReadSearchGroup } from '../../types/message.js'
/**
* Plain function (not a React component) so the React Compiler won't
* hoist the teamMemory* property accesses for memoization. This module
* is only loaded when feature('TEAMMEM') is true.
*/
export function checkHasTeamMemOps(message: CollapsedReadSearchGroup): boolean {
return (
(message.teamMemorySearchCount ?? 0) > 0 ||
(message.teamMemoryReadCount ?? 0) > 0 ||
(message.teamMemoryWriteCount ?? 0) > 0
)
}
/**
* Renders team memory count parts for the collapsed read/search UI.
* This module is only loaded when feature('TEAMMEM') is true,
* so DCE removes it entirely from external builds.
*/
export function TeamMemCountParts({
message,
isActiveGroup,
hasPrecedingParts,
}: {
message: CollapsedReadSearchGroup
isActiveGroup: boolean | undefined
hasPrecedingParts: boolean
}): React.ReactNode {
const tmReadCount = message.teamMemoryReadCount ?? 0
const tmSearchCount = message.teamMemorySearchCount ?? 0
const tmWriteCount = message.teamMemoryWriteCount ?? 0
if (tmReadCount === 0 && tmSearchCount === 0 && tmWriteCount === 0) {
return null
}
const nodes: React.ReactNode[] = []
let count = hasPrecedingParts ? 1 : 0
if (tmReadCount > 0) {
const verb = isActiveGroup
? count === 0
? 'Recalling'
: 'recalling'
: count === 0
? 'Recalled'
: 'recalled'
if (count > 0) {
nodes.push(<Text key="comma-tmr">, </Text>)
}
nodes.push(
<Text key="team-mem-read">
{verb} <Text bold>{tmReadCount}</Text> team{' '}
{tmReadCount === 1 ? 'memory' : 'memories'}
</Text>,
)
count++
}
if (tmSearchCount > 0) {
const verb = isActiveGroup
? count === 0
? 'Searching'
: 'searching'
: count === 0
? 'Searched'
: 'searched'
if (count > 0) {
nodes.push(<Text key="comma-tms">, </Text>)
}
nodes.push(<Text key="team-mem-search">{`${verb} team memories`}</Text>)
count++
}
if (tmWriteCount > 0) {
const verb = isActiveGroup
? count === 0
? 'Writing'
: 'writing'
: count === 0
? 'Wrote'
: 'wrote'
if (count > 0) {
nodes.push(<Text key="comma-tmw">, </Text>)
}
nodes.push(
<Text key="team-mem-write">
{verb} <Text bold>{tmWriteCount}</Text> team{' '}
{tmWriteCount === 1 ? 'memory' : 'memories'}
</Text>,
)
}
return <>{nodes}</>
}