refactor: streamline chat scroll manager

- Removed unused activeTurnAnchorId and activeTurnSpacerHeight from MemoryDebugPanel.
- Simplified useChatScrollManager by eliminating unnecessary state and functions related to active turn anchoring.
- Updated loadMessages function signatures to accept an optional limit parameter across message and session stores.
- Enhanced scroll behavior in useScrollEngine to support dynamic bottom tracking during animations.
- Improved event stream handling to allow for resyncing messages with a specified limit.
- Cleaned up session memory state to remove active turn properties, focusing on viewport anchoring.
This commit is contained in:
Bohdan Triapitsyn
2026-01-20 03:23:39 +02:00
parent 8fa16baf3b
commit 6544b12e78
15 changed files with 318 additions and 937 deletions
@@ -25,8 +25,6 @@ export const ChatContainer: React.FC = () => {
loadMessages,
loadMoreMessages,
updateViewportAnchor,
updateActiveTurnAnchor,
getActiveTurnAnchor,
sessionMemoryState,
openNewSessionDraft,
isSyncing,
@@ -75,26 +73,22 @@ export const ChatContainer: React.FC = () => {
const {
scrollRef,
handleMessageContentChange,
getAnimationHandlers,
showScrollButton,
scrollToBottom,
spacerHeight,
pendingAnchorId,
hasActiveAnchor,
getAnimationHandlers,
showScrollButton,
scrollToBottom,
scrollToPosition,
isPinned,
} = useChatScrollManager({
currentSessionId,
sessionMessages,
streamingMessageId,
sessionMemoryState,
updateViewportAnchor,
updateActiveTurnAnchor,
getActiveTurnAnchor,
isSyncing,
isMobile,
messageStreamStates,
sessionPermissions: sessionBlockingCards,
trimToViewportWindow,
sessionActivityPhase,
});
const memoryState = React.useMemo(() => {
@@ -123,12 +117,12 @@ export const ChatContainer: React.FC = () => {
await loadMoreMessages(currentSessionId, 'up');
if (container && prevHeight !== null && prevTop !== null) {
const heightDiff = container.scrollHeight - prevHeight;
container.scrollTop = prevTop + heightDiff;
scrollToPosition(prevTop + heightDiff, { instant: true });
}
} finally {
setIsLoadingOlder(false);
}
}, [currentSessionId, isLoadingOlder, loadMoreMessages, scrollRef]);
}, [currentSessionId, isLoadingOlder, loadMoreMessages, scrollRef, scrollToPosition]);
// Scroll to a specific message by ID (for timeline dialog)
const scrollToMessage = React.useCallback((messageId: string) => {
@@ -169,7 +163,8 @@ export const ChatContainer: React.FC = () => {
} finally {
const currentPhase = sessionActivityPhase?.get(currentSessionId) ?? 'idle';
const isActivePhase = currentPhase === 'busy' || currentPhase === 'cooldown';
const shouldSkipScroll = isActivePhase && hasActiveAnchor;
// When pinned and active, scroll is already maintained automatically
const shouldSkipScroll = isActivePhase && isPinned;
if (!shouldSkipScroll) {
if (typeof window === 'undefined') {
@@ -184,7 +179,7 @@ export const ChatContainer: React.FC = () => {
};
void load();
}, [currentSessionId, hasActiveAnchor, loadMessages, messages, scrollToBottom, sessionActivityPhase]);
}, [currentSessionId, isPinned, loadMessages, messages, scrollToBottom, sessionActivityPhase]);
if (!currentSessionId && !draftOpen) {
return (
@@ -277,7 +272,6 @@ export const ChatContainer: React.FC = () => {
}}
data-scroll-shadow="true"
data-scrollbar="chat"
hideBottomShadow={!!pendingAnchorId}
>
<div className="relative z-0 min-h-full">
<MessageList
@@ -290,16 +284,7 @@ export const ChatContainer: React.FC = () => {
isLoadingOlder={isLoadingOlder}
onLoadOlder={handleLoadOlder}
scrollToBottom={scrollToBottom}
pendingAnchorId={pendingAnchorId}
/>
{}
{spacerHeight > 0 && hasActiveAnchor && (
<div
data-role="active-turn-spacer"
style={{ height: spacerHeight }}
aria-hidden="true"
/>
)}
</div>
</ScrollShadow>
<OverlayScrollbar containerRef={scrollRef} />
+18 -14
View File
@@ -360,6 +360,7 @@ export const ChatInput: React.FC<ChatInputProps> = ({ onOpenSettings, scrollToBo
if (!canSend || (!currentSessionId && !newSessionDraftOpen)) return;
// Re-pin and scroll to bottom when sending
scrollToBottom?.({ instant: true, force: true });
if (!currentProviderId || !currentModelId) {
@@ -1356,25 +1357,28 @@ export const ChatInput: React.FC<ChatInputProps> = ({ onOpenSettings, scrollToBo
<form
onSubmit={handleSubmit}
className={cn(
"pt-0 pb-2 md:pb-4",
"relative pt-0 pb-2 md:pb-4",
isMobile && isKeyboardOpen ? "ios-keyboard-safe-area" : "bottom-safe-area"
)}
data-keyboard-avoid="true"
style={isMobile && inputBarOffset > 0 && !isKeyboardOpen ? { marginBottom: `${inputBarOffset}px` } : undefined}
>
<StatusRow
isWorking={working.isWorking}
statusText={workingStatusText}
isGenericStatus={working.isGenericStatus}
isWaitingForPermission={working.isWaitingForPermission}
wasAborted={working.wasAborted}
abortActive={working.abortActive}
completionId={working.lastCompletionId}
isComplete={working.isComplete}
showAbort={showAbortInStatusRow}
onAbort={handleAbort}
showAbortStatus={showAbortStatus}
/>
{/* Absolute positioned above input - no layout shift */}
<div className="absolute bottom-full left-0 right-0">
<StatusRow
isWorking={working.isWorking}
statusText={workingStatusText}
isGenericStatus={working.isGenericStatus}
isWaitingForPermission={working.isWaitingForPermission}
wasAborted={working.wasAborted}
abortActive={working.abortActive}
completionId={working.lastCompletionId}
isComplete={working.isComplete}
showAbort={showAbortInStatusRow}
onAbort={handleAbort}
showAbortStatus={showAbortStatus}
/>
</div>
<div
ref={dropZoneRef}
className={cn(
@@ -63,8 +63,7 @@ interface ChatMessageProps {
};
onContentChange?: (reason?: ContentChangeReason) => void;
animationHandlers?: AnimationHandlers;
scrollToBottom?: (options?: { instant?: boolean; force?: boolean; clearAnchor?: boolean }) => void;
isPendingAnchor?: boolean;
scrollToBottom?: (options?: { instant?: boolean; force?: boolean }) => void;
turnGroupingContext?: TurnGroupingContext;
}
@@ -74,7 +73,6 @@ const ChatMessage: React.FC<ChatMessageProps> = ({
nextMessage,
onContentChange,
animationHandlers,
isPendingAnchor = false,
turnGroupingContext,
}) => {
const { isMobile, hasTouchInput } = useDeviceInfo();
@@ -834,7 +832,6 @@ const ChatMessage: React.FC<ChatMessageProps> = ({
)}
data-message-id={message.info.id}
ref={messageContainerRef}
style={isPendingAnchor ? { visibility: 'hidden' } : undefined}
>
<div className="chat-column">
{isUser ? (
@@ -19,8 +19,7 @@ interface MessageListProps {
hasMoreAbove: boolean;
isLoadingOlder: boolean;
onLoadOlder: () => void;
scrollToBottom?: (options?: { instant?: boolean; force?: boolean; clearAnchor?: boolean }) => void;
pendingAnchorId?: string | null;
scrollToBottom?: (options?: { instant?: boolean; force?: boolean }) => void;
}
const MessageList: React.FC<MessageListProps> = ({
@@ -33,7 +32,6 @@ const MessageList: React.FC<MessageListProps> = ({
isLoadingOlder,
onLoadOlder,
scrollToBottom,
pendingAnchorId,
}) => {
React.useEffect(() => {
if (permissions.length === 0 && questions.length === 0) {
@@ -101,7 +99,6 @@ const MessageList: React.FC<MessageListProps> = ({
onContentChange={onMessageContentChange}
animationHandlers={getAnimationHandlers(message.info.id)}
scrollToBottom={scrollToBottom}
isPendingAnchor={pendingAnchorId === message.info.id}
turnGroupingContext={getContextForMessage(message.info.id)}
/>
))}
@@ -118,6 +115,9 @@ const MessageList: React.FC<MessageListProps> = ({
))}
</div>
)}
{/* Bottom spacer - always 10% of viewport height */}
<div className="flex-shrink-0" style={{ height: '10vh' }} aria-hidden="true" />
</div>
);
};
@@ -56,15 +56,12 @@ export interface TurnGroupingContext {
isWorking: boolean;
isGroupExpanded: boolean;
previewedPartIds: Set<string>;
toggleGroup: () => void;
markPartsPreviewed: (partIds: string[]) => void;
}
interface TurnUiState {
isExpanded: boolean;
previewedPartIds: Set<string>;
}
interface TurnActivityInfo {
@@ -424,7 +421,7 @@ export const useTurnGrouping = (messages: ChatMessageEntry[]): UseTurnGroupingRe
(turnId: string): TurnUiState => {
const existing = turnUiStates.get(turnId);
if (existing) return existing;
return { isExpanded: defaultActivityExpanded, previewedPartIds: new Set<string>() };
return { isExpanded: defaultActivityExpanded };
},
[turnUiStates, defaultActivityExpanded]
);
@@ -432,25 +429,8 @@ export const useTurnGrouping = (messages: ChatMessageEntry[]): UseTurnGroupingRe
const toggleGroup = React.useCallback((turnId: string) => {
setTurnUiStates((prev) => {
const next = new Map(prev);
const current = next.get(turnId) ?? { isExpanded: defaultActivityExpanded, previewedPartIds: new Set<string>() };
next.set(turnId, { ...current, isExpanded: !current.isExpanded });
return next;
});
}, [defaultActivityExpanded]);
const markPartsPreviewedInternal = React.useCallback((turnId: string, partIds: string[]) => {
if (partIds.length === 0) return;
setTurnUiStates((prev) => {
const next = new Map(prev);
const state = next.get(turnId) ?? { isExpanded: defaultActivityExpanded, previewedPartIds: new Set<string>() };
const newPreviewed = new Set(state.previewedPartIds);
partIds.forEach((id) => {
if (id && id.trim().length > 0) {
newPreviewed.add(id);
}
});
next.set(turnId, { ...state, previewedPartIds: newPreviewed });
const current = next.get(turnId) ?? { isExpanded: defaultActivityExpanded };
next.set(turnId, { isExpanded: !current.isExpanded });
return next;
});
}, [defaultActivityExpanded]);
@@ -506,12 +486,10 @@ export const useTurnGrouping = (messages: ChatMessageEntry[]): UseTurnGroupingRe
userMessageCreatedAt,
isWorking: isTurnWorking,
isGroupExpanded: uiState.isExpanded,
previewedPartIds: uiState.previewedPartIds,
toggleGroup: () => toggleGroup(turn.turnId),
markPartsPreviewed: (partIds: string[]) => markPartsPreviewedInternal(turn.turnId, partIds),
} satisfies TurnGroupingContext;
},
[getOrCreateTurnState, lastTurnId, markPartsPreviewedInternal, messageToTurn, sessionIsWorking, toggleGroup, turnActivityInfo]
[getOrCreateTurnState, lastTurnId, messageToTurn, sessionIsWorking, toggleGroup, turnActivityInfo]
);
@@ -4,15 +4,16 @@ import { cn } from '@/lib/utils';
interface FadeInOnRevealProps {
children: React.ReactNode;
className?: string;
skipAnimation?: boolean;
}
const FADE_ANIMATION_ENABLED = true;
export const FadeInOnReveal: React.FC<FadeInOnRevealProps> = ({ children, className }) => {
const [visible, setVisible] = React.useState(false);
export const FadeInOnReveal: React.FC<FadeInOnRevealProps> = ({ children, className, skipAnimation }) => {
const [visible, setVisible] = React.useState(skipAnimation ?? false);
React.useEffect(() => {
if (!FADE_ANIMATION_ENABLED) {
if (!FADE_ANIMATION_ENABLED || skipAnimation) {
return;
}
@@ -35,9 +36,9 @@ export const FadeInOnReveal: React.FC<FadeInOnRevealProps> = ({ children, classN
window.cancelAnimationFrame(frame);
}
};
}, []);
}, [skipAnimation]);
if (!FADE_ANIMATION_ENABLED) {
if (!FADE_ANIMATION_ENABLED || skipAnimation) {
return <>{children}</>;
}
@@ -1,12 +1,9 @@
import React from 'react';
import type { Part } from '@opencode-ai/sdk/v2';
import AssistantTextPart from './parts/AssistantTextPart';
import UserTextPart from './parts/UserTextPart';
import ReasoningPart from './parts/ReasoningPart';
import ToolPart from './parts/ToolPart';
import ProgressiveGroup from './parts/ProgressiveGroup';
import MigratingPart from './parts/MigratingPart';
import { MessageFilesDisplay } from '../FileAttachment';
import type { ToolPart as ToolPartType } from '@opencode-ai/sdk/v2';
import type { StreamPhase, ToolPopupContent, AgentMentionInfo } from './types';
@@ -37,77 +34,6 @@ const formatTurnDuration = (durationMs: number): string => {
return `${minutes}m ${seconds}s`;
};
const useMigrationTimer = (
turnGroupingContext: TurnGroupingContext | undefined,
previewablePartIds: Set<string>
): { isAnimating: boolean } => {
const timerRef = React.useRef<number | null>(null);
const animationTimerRef = React.useRef<number | null>(null);
const [isAnimating, setIsAnimating] = React.useState(false);
const contextRef = React.useRef(turnGroupingContext);
contextRef.current = turnGroupingContext;
const partIdsRef = React.useRef(previewablePartIds);
partIdsRef.current = previewablePartIds;
const timerStartedRef = React.useRef(false);
const hasPreviewableParts = previewablePartIds.size > 0;
React.useEffect(() => {
if (!turnGroupingContext) return;
if (!turnGroupingContext.isWorking) return;
if (!hasPreviewableParts) return;
if (timerStartedRef.current) return;
timerStartedRef.current = true;
timerRef.current = window.setTimeout(() => {
timerRef.current = null;
setIsAnimating(true);
animationTimerRef.current = window.setTimeout(() => {
animationTimerRef.current = null;
setIsAnimating(false);
const context = contextRef.current;
if (!context) {
return;
}
const idsToPreview = Array.from(partIdsRef.current);
if (idsToPreview.length > 0) {
context.markPartsPreviewed(idsToPreview);
}
}, 300);
}, 1000);
}, [hasPreviewableParts, turnGroupingContext]);
React.useEffect(() => {
if (!turnGroupingContext) return;
if (!turnGroupingContext.isWorking || !hasPreviewableParts) {
if (timerRef.current) {
window.clearTimeout(timerRef.current);
timerRef.current = null;
}
if (animationTimerRef.current) {
window.clearTimeout(animationTimerRef.current);
animationTimerRef.current = null;
}
setIsAnimating(false);
timerStartedRef.current = false;
}
}, [hasPreviewableParts, turnGroupingContext]);
React.useEffect(() => {
return () => {
if (timerRef.current) window.clearTimeout(timerRef.current);
if (animationTimerRef.current) window.clearTimeout(animationTimerRef.current);
};
}, []);
return { isAnimating };
};
const ACTIVITY_STANDALONE_TOOL_NAMES = new Set<string>(['task']);
const isActivityStandaloneTool = (toolName: unknown): boolean => {
@@ -429,15 +355,6 @@ const AssistantMessageBody: React.FC<Omit<MessageBodyProps, 'isUser'>> = ({
return toolParts.every((toolPart) => isToolFinalized(toolPart));
}, [toolParts, hasPendingTools, isToolFinalized]);
const assistantTextReady = React.useMemo(() => {
if (assistantTextParts.length === 0) {
return true;
}
return assistantTextParts.every((part) => {
const time = (part as Record<string, unknown>).time as Record<string, unknown> | undefined;
return typeof time?.end === 'number';
});
}, [assistantTextParts]);
const reasoningParts = React.useMemo(() => {
return visibleParts.filter((part) => part.type === 'reasoning');
@@ -461,19 +378,7 @@ const AssistantMessageBody: React.FC<Omit<MessageBodyProps, 'isUser'>> = ({
hasTools &&
(hasPendingTools || hasOpenStep || !allToolsFinalized);
const shouldCoordinateRendering = React.useMemo(() => {
if (!hasTools) {
return assistantTextParts.length > 0 ? shouldHoldForReasoning : false;
}
if (assistantTextParts.length === 0) {
return hasOpenStep || hasPendingTools || !allToolsFinalized;
}
return true;
}, [assistantTextParts.length, hasOpenStep, hasPendingTools, hasTools, shouldHoldForReasoning, allToolsFinalized]);
const shouldHoldAssistantText = awaitingMessageCompletion
|| (shouldCoordinateRendering && (!assistantTextReady || !allToolsFinalized || hasPendingTools || hasOpenStep))
|| shouldHoldForReasoning;
const shouldHoldTools = awaitingMessageCompletion
|| (hasTools && (hasPendingTools || hasOpenStep || !allToolsFinalized));
const shouldHoldReasoning = awaitingMessageCompletion || shouldHoldForReasoning;
@@ -682,70 +587,6 @@ const AssistantMessageBody: React.FC<Omit<MessageBodyProps, 'isUser'>> = ({
const shouldShowActivityGroup = Boolean(turnGroupingContext && hasEverHadMultipleVisibleActivities);
const previewableActivityPartsForMessage = React.useMemo(() => {
if (!turnGroupingContext) return [];
if (!shouldShowActivityGroup) return [];
if (!turnGroupingContext.isWorking) {
return [];
}
const previewable: (typeof activityPartsForMessage) = [];
activityPartsForMessage.forEach((activity) => {
if (turnGroupingContext.previewedPartIds.has(activity.id)) {
return;
}
if (!showReasoningTraces && activity.kind !== 'tool') {
return;
}
const part = activity.part;
if (activity.kind === 'tool') {
const toolPart = part as ToolPartType;
if (isActivityStandaloneTool(toolPart.tool)) {
return;
}
if (shouldHoldTools) return;
if (!isToolFinalized(toolPart)) return;
} else if (activity.kind === 'reasoning') {
if (!showReasoningTraces) return;
if (shouldHoldReasoning) return;
const time = (part as { time?: { end?: number | null | undefined } | null | undefined }).time;
if (typeof time?.end !== 'number') return;
} else if (activity.kind === 'justification') {
if (!showReasoningTraces) return;
if (shouldHoldAssistantText) return;
const time = (part as { time?: { end?: number | null | undefined } | null | undefined }).time;
if (typeof time?.end !== 'number') return;
}
previewable.push(activity);
});
return previewable;
}, [
activityPartsForMessage,
isToolFinalized,
shouldHoldAssistantText,
shouldHoldReasoning,
shouldHoldTools,
showReasoningTraces,
shouldShowActivityGroup,
turnGroupingContext,
]);
const previewableActivityPartIds = React.useMemo(() => {
const ids = new Set<string>();
previewableActivityPartsForMessage.forEach((activity) => {
ids.add(activity.id);
});
return ids;
}, [previewableActivityPartsForMessage]);
const { isAnimating: isMessageAnimating } = useMigrationTimer(turnGroupingContext, previewableActivityPartIds);
const shouldRenderActivityGroup = Boolean(
turnGroupingContext &&
shouldShowActivityGroup &&
@@ -789,8 +630,6 @@ const AssistantMessageBody: React.FC<Omit<MessageBodyProps, 'isUser'>> = ({
onToggleTool={onToggleTool}
onShowPopup={onShowPopup}
onContentChange={onContentChange}
isWorking={turnGroupingContext.isWorking}
previewedPartIds={turnGroupingContext.previewedPartIds}
diffStats={turnGroupingContext.diffStats}
/>
);
@@ -885,146 +724,6 @@ const AssistantMessageBody: React.FC<Omit<MessageBodyProps, 'isUser'>> = ({
element,
});
}
return;
}
if (!turnGroupingContext.isWorking || turnGroupingContext.isGroupExpanded) {
return;
}
if (turnGroupingContext.previewedPartIds.has(activity.id)) {
return;
}
if (!showReasoningTraces && activity.kind !== 'tool') {
return;
}
const wrapForMigration = previewableActivityPartIds.has(activity.id);
switch (activity.kind) {
case 'tool': {
const toolPart = part as ToolPartType;
if (isActivityStandaloneTool(toolPart.tool)) {
break;
}
const toolState = (toolPart as { state?: { time?: { end?: number | null | undefined } | null | undefined } | null | undefined }).state;
const time = toolState?.time;
const isFinalized = isToolFinalized(toolPart);
const shouldShowTool = !shouldHoldTools && isFinalized;
if (!shouldShowTool) {
break;
}
const connection = toolConnections[toolPart.id];
const toolElement = (
<FadeInOnReveal key={`tool-${toolPart.id}`}>
<ToolPart
part={toolPart}
isExpanded={expandedTools.has(toolPart.id)}
onToggle={onToggleTool}
syntaxTheme={syntaxTheme}
isMobile={isMobile}
onContentChange={onContentChange}
hasPrevTool={connection?.hasPrev ?? false}
hasNextTool={connection?.hasNext ?? false}
/>
</FadeInOnReveal>
);
element = wrapForMigration ? (
<MigratingPart key={`migrating-tool-${toolPart.id}`} isMigrating={isMessageAnimating}>
{toolElement}
</MigratingPart>
) : toolElement;
endTime = isFinalized && typeof time?.end === 'number' ? time.end : null;
break;
}
case 'reasoning': {
if (!showReasoningTraces) {
break;
}
const reasoningTime = (part as { time?: { end?: number | null | undefined } | null | undefined }).time;
const hasEndTime = typeof reasoningTime?.end === 'number';
const shouldShowReasoning = hasEndTime && !shouldHoldReasoning;
if (!shouldShowReasoning) {
break;
}
const reasoningElement = (
<FadeInOnReveal key={`reasoning-${index}`}>
<ReasoningPart
part={part}
messageId={messageId}
onContentChange={onContentChange}
/>
</FadeInOnReveal>
);
element = wrapForMigration ? (
<MigratingPart key={`migrating-reasoning-${index}`} isMigrating={isMessageAnimating}>
{reasoningElement}
</MigratingPart>
) : reasoningElement;
endTime = hasEndTime ? reasoningTime?.end ?? null : null;
break;
}
case 'justification': {
if (!showReasoningTraces) {
break;
}
const time = (part as { time?: { end?: number | null | undefined } | null | undefined }).time;
const hasEndTime = typeof time?.end === 'number';
const shouldShowJustification = hasEndTime && !shouldHoldAssistantText;
if (!shouldShowJustification) {
break;
}
const textElement = (
<FadeInOnReveal key={`assistant-text-${index}`}>
<AssistantTextPart
part={part}
messageId={messageId}
streamPhase="completed"
allowAnimation={false}
onContentChange={onContentChange}
renderAsReasoning
/>
</FadeInOnReveal>
);
element = wrapForMigration ? (
<MigratingPart key={`migrating-text-${index}`} isMigrating={isMessageAnimating}>
{textElement}
</MigratingPart>
) : textElement;
endTime = hasEndTime ? time?.end ?? null : null;
break;
}
default:
break;
}
if (element) {
partsWithTime.push({
part,
index,
endTime,
element,
});
}
});
@@ -1050,16 +749,11 @@ const AssistantMessageBody: React.FC<Omit<MessageBodyProps, 'isUser'>> = ({
activityPartsByPart,
activityGroupSegmentsForMessage,
expandedTools,
isMessageAnimating,
isMobile,
isToolFinalized,
messageId,
onContentChange,
onShowPopup,
onToggleTool,
previewableActivityPartIds,
shouldHoldAssistantText,
shouldHoldReasoning,
shouldHoldTools,
shouldShowActivityGroup,
showReasoningTraces,
@@ -10,6 +10,8 @@ import ReasoningPart from './ReasoningPart';
import JustificationBlock from './JustificationBlock';
import { FadeInOnReveal } from '../FadeInOnReveal';
const MAX_VISIBLE_COLLAPSED = 6;
interface DiffStats {
additions: number;
deletions: number;
@@ -26,32 +28,9 @@ interface ProgressiveGroupProps {
onToggleTool: (toolId: string) => void;
onShowPopup: (content: ToolPopupContent) => void;
onContentChange?: (reason?: ContentChangeReason) => void;
isWorking: boolean;
previewedPartIds: Set<string>;
diffStats?: DiffStats;
}
const getGroupSummary = (parts: TurnActivityPart[]): string => {
const counts = {
tools: parts.filter((p) => p.kind === 'tool').length,
reasoning: parts.filter((p) => p.kind === 'reasoning').length,
justifications: parts.filter((p) => p.kind === 'justification').length,
};
const segments: string[] = [];
if (counts.tools > 0) {
segments.push(`${counts.tools} tool${counts.tools > 1 ? 's' : ''}`);
}
if (counts.reasoning > 0) {
segments.push(`${counts.reasoning} reasoning`);
}
if (counts.justifications > 0) {
segments.push(`${counts.justifications} justification${counts.justifications > 1 ? 's' : ''}`);
}
return segments.join(', ');
};
const sortPartsByTime = (parts: TurnActivityPart[]): TurnActivityPart[] => {
return [...parts].sort((a, b) => {
const aTime = typeof a.endedAt === 'number' ? a.endedAt : undefined;
@@ -93,13 +72,12 @@ const ProgressiveGroup: React.FC<ProgressiveGroupProps> = ({
expandedTools,
onToggleTool,
onContentChange,
isWorking,
previewedPartIds,
diffStats,
}) => {
const previousExpandedRef = React.useRef<boolean | undefined>(isExpanded);
// Track if we just expanded from collapsed state
const [justExpandedFromCollapsed, setJustExpandedFromCollapsed] = React.useState(false);
// Track expansion count to force re-mount of items when group expands from collapsed
const [expansionKey, setExpansionKey] = React.useState(0);
React.useEffect(() => {
@@ -108,85 +86,101 @@ const ProgressiveGroup: React.FC<ProgressiveGroupProps> = ({
previousExpandedRef.current = isExpanded;
onContentChange?.('structural');
// Increment key when expanding to trigger fresh animations
if (isExpanded && wasCollapsed) {
setExpansionKey((k) => k + 1);
setJustExpandedFromCollapsed(true);
// Reset after a short delay (after animations would have started)
const timer = setTimeout(() => setJustExpandedFromCollapsed(false), 50);
return () => clearTimeout(timer);
} else {
setJustExpandedFromCollapsed(false);
}
}, [isExpanded, onContentChange]);
const displayParts = React.useMemo(() => {
if (!isWorking) {
return sortPartsByTime(parts);
}
return sortPartsByTime(parts);
}, [parts]);
// While turn is working, only show parts that have been "previewed".
// Collapsed mode previews them in-chat first, then migrates into Activity.
// Summary/Detailed modes skip in-chat preview, but still use the same migration gate.
return sortPartsByTime(
parts.filter((activity) => {
const partId = activity.part.id;
return partId && previewedPartIds.has(activity.id);
})
);
}, [parts, isWorking, previewedPartIds]);
const summary = getGroupSummary(displayParts);
const toolConnections = getToolConnections(displayParts);
// For collapsed state: show last N items
const visibleCollapsedParts = React.useMemo(() => {
return displayParts.slice(-MAX_VISIBLE_COLLAPSED);
}, [displayParts]);
// Set of part IDs that were visible in collapsed state
const visibleInCollapsedIds = React.useMemo(() => {
const ids = new Set<string>();
visibleCollapsedParts.forEach((p) => {
if (p.part.id) ids.add(p.part.id);
});
return ids;
}, [visibleCollapsedParts]);
// Connections for collapsed view (based on visible parts only)
const collapsedToolConnections = React.useMemo(() => {
return getToolConnections(visibleCollapsedParts);
}, [visibleCollapsedParts]);
const hiddenCount = Math.max(0, displayParts.length - MAX_VISIBLE_COLLAPSED);
if (displayParts.length === 0) {
return null;
}
const partsToRender = isExpanded ? displayParts : visibleCollapsedParts;
const connectionsToUse = isExpanded ? toolConnections : collapsedToolConnections;
// If there are no hidden items, header is not interactive
const isHeaderInteractive = hiddenCount > 0;
return (
<FadeInOnReveal>
<div className="my-1">
{}
<div
className={cn(
'group/tool flex items-center gap-2 pr-2 pl-px pt-0 pb-1.5 rounded-xl cursor-pointer'
'group/tool flex items-center gap-2 pr-2 pl-px pt-0 pb-1.5 rounded-xl',
isHeaderInteractive && 'cursor-pointer'
)}
onClick={onToggle}
onClick={isHeaderInteractive ? onToggle : undefined}
>
<div className="flex items-center gap-2 flex-shrink-0">
{}
<div className="relative h-3.5 w-3.5 flex-shrink-0">
{}
<div
className={cn(
'absolute inset-0 transition-opacity',
isExpanded && 'opacity-0',
!isExpanded && !isMobile && 'group-hover/tool:opacity-0'
)}
>
<RiStackLine className="h-3.5 w-3.5" />
</div>
{}
<div
className={cn(
'absolute inset-0 transition-opacity flex items-center justify-center',
isExpanded && 'opacity-100',
!isExpanded && isMobile && 'opacity-0',
!isExpanded && !isMobile && 'opacity-0 group-hover/tool:opacity-100'
)}
>
{isExpanded ? (
<RiArrowDownSLine className="h-3.5 w-3.5" />
<div className="flex items-center gap-2 flex-shrink-0">
<div className="relative h-3.5 w-3.5 flex-shrink-0">
{isHeaderInteractive ? (
<>
<div
className={cn(
'absolute inset-0 transition-opacity',
isExpanded && 'opacity-0',
!isExpanded && !isMobile && 'group-hover/tool:opacity-0'
)}
>
<RiStackLine className="h-3.5 w-3.5" />
</div>
<div
className={cn(
'absolute inset-0 transition-opacity flex items-center justify-center',
isExpanded && 'opacity-100',
!isExpanded && isMobile && 'opacity-0',
!isExpanded && !isMobile && 'opacity-0 group-hover/tool:opacity-100'
)}
>
{isExpanded ? (
<RiArrowDownSLine className="h-3.5 w-3.5" />
) : (
<RiArrowRightSLine className="h-3.5 w-3.5" />
)}
</div>
</>
) : (
<RiArrowRightSLine className="h-3.5 w-3.5" />
<RiStackLine className="h-3.5 w-3.5" />
)}
</div>
<span className="typography-meta font-medium">Activity</span>
</div>
<span className="typography-meta font-medium">Activity</span>
</div>
{(summary || diffStats) && (
<div className="flex-1 min-w-0 typography-meta text-muted-foreground/70 flex items-center gap-2">
{summary && (
<span className="truncate block">{summary}</span>
)}
{diffStats && (diffStats.additions > 0 || diffStats.deletions > 0) && (
{diffStats && (diffStats.additions > 0 || diffStats.deletions > 0) && (
<div className="flex-1 min-w-0 typography-meta text-muted-foreground/70 flex items-center gap-2">
<span className="flex-shrink-0 leading-none">
<span className="text-[color:var(--status-success)]">
+{Math.max(0, diffStats.additions)}
@@ -196,13 +190,10 @@ const ProgressiveGroup: React.FC<ProgressiveGroupProps> = ({
-{Math.max(0, diffStats.deletions)}
</span>
</span>
)}
</div>
)}
</div>
</div>
)}
</div>
{}
{isExpanded && (
<div
className={cn(
'relative pr-2 pb-1 pt-1 pl-[1.4375rem]',
@@ -210,16 +201,31 @@ const ProgressiveGroup: React.FC<ProgressiveGroupProps> = ({
'before:top-[-0.25rem] before:bottom-0'
)}
>
{displayParts.map((activity, index) => {
{!isExpanded && hiddenCount > 0 && (
<div
className="typography-micro text-muted-foreground/70 mb-1 cursor-pointer hover:text-muted-foreground"
onClick={onToggle}
>
+{hiddenCount} more...
</div>
)}
{partsToRender.map((activity, index) => {
const partId = activity.part.id || `group-part-${index}`;
const connection = toolConnections[partId];
const connection = connectionsToUse[partId];
const animationKey = `${partId}-exp${expansionKey}`;
// Skip animation if:
// - We just expanded from collapsed AND
// - This part was already visible in collapsed state
const wasVisibleInCollapsed = activity.part.id ? visibleInCollapsedIds.has(activity.part.id) : false;
const skipAnimation = justExpandedFromCollapsed && wasVisibleInCollapsed;
switch (activity.kind) {
case 'tool':
return (
<FadeInOnReveal key={animationKey}>
<FadeInOnReveal key={animationKey} skipAnimation={skipAnimation}>
<ToolPart
part={activity.part as ToolPartType}
isExpanded={expandedTools.has(partId)}
@@ -235,7 +241,7 @@ const ProgressiveGroup: React.FC<ProgressiveGroupProps> = ({
case 'reasoning':
return (
<FadeInOnReveal key={animationKey}>
<FadeInOnReveal key={animationKey} skipAnimation={skipAnimation}>
<ReasoningPart
part={activity.part}
messageId={activity.messageId}
@@ -246,7 +252,7 @@ const ProgressiveGroup: React.FC<ProgressiveGroupProps> = ({
case 'justification':
return (
<FadeInOnReveal key={animationKey}>
<FadeInOnReveal key={animationKey} skipAnimation={skipAnimation}>
<JustificationBlock
part={activity.part}
messageId={activity.messageId}
@@ -260,7 +266,6 @@ const ProgressiveGroup: React.FC<ProgressiveGroupProps> = ({
}
})}
</div>
)}
</div>
</FadeInOnReveal>
);
@@ -42,8 +42,6 @@ export const MemoryDebugPanel: React.FC<MemoryDebugPanelProps> = ({ onClose }) =
isZombie: memoryState?.isZombie || false,
backgroundCount: memoryState?.backgroundMessageCount || 0,
lastAccessed: memoryState?.lastAccessedAt || 0,
activeTurnAnchorId: memoryState?.activeTurnAnchorId ?? null,
activeTurnSpacerHeight: memoryState?.activeTurnSpacerHeight ?? 0,
isCurrent: session.id === currentSessionId
};
}).sort((a, b) => b.lastAccessed - a.lastAccessed);
@@ -145,11 +143,6 @@ export const MemoryDebugPanel: React.FC<MemoryDebugPanelProps> = ({ onClose }) =
}`}>
{stat.messageCount} msgs
</span>
{stat.activeTurnAnchorId && stat.activeTurnSpacerHeight > 0 && (
<span className="font-mono text-xs text-primary">
anchor+{Math.round(stat.activeTurnSpacerHeight)}px
</span>
)}
{stat.backgroundCount > 0 && (
<span className="text-primary">+{stat.backgroundCount}</span>
)}
+148 -402
View File
@@ -1,5 +1,4 @@
import React from 'react';
import { flushSync } from 'react-dom';
import type { Part } from '@opencode-ai/sdk/v2';
import { MessageFreshnessDetector } from '@/lib/messageFreshness';
@@ -26,7 +25,6 @@ interface SessionMemoryState {
isZombie?: boolean;
}
type SessionActivityPhase = 'idle' | 'busy' | 'cooldown';
interface UseChatScrollManagerOptions {
currentSessionId: string | null;
@@ -35,13 +33,10 @@ interface UseChatScrollManagerOptions {
streamingMessageId: string | null;
sessionMemoryState: Map<string, SessionMemoryState>;
updateViewportAnchor: (sessionId: string, anchor: number) => void;
updateActiveTurnAnchor: (sessionId: string, anchorId: string | null, spacerHeight: number) => void;
getActiveTurnAnchor: (sessionId: string) => { anchorId: string | null; spacerHeight: number } | null;
isSyncing: boolean;
isMobile: boolean;
messageStreamStates: Map<string, unknown>;
trimToViewportWindow: (sessionId: string, targetSize?: number) => void;
sessionActivityPhase?: Map<string, SessionActivityPhase>;
}
export interface AnimationHandlers {
@@ -59,128 +54,76 @@ interface UseChatScrollManagerResult {
handleMessageContentChange: (reason?: ContentChangeReason) => void;
getAnimationHandlers: (messageId: string) => AnimationHandlers;
showScrollButton: boolean;
scrollToBottom: (options?: { instant?: boolean; force?: boolean; clearAnchor?: boolean }) => void;
spacerHeight: number;
pendingAnchorId: string | null;
hasActiveAnchor: boolean;
scrollToBottom: (options?: { instant?: boolean; force?: boolean }) => void;
scrollToPosition: (position: number, options?: { instant?: boolean }) => void;
isPinned: boolean;
}
const ANCHOR_TARGET_OFFSET = 8;
const DEFAULT_SCROLL_BUTTON_THRESHOLD = 40;
const NEW_USER_ANCHOR_WINDOW_MS = 20_000;
const PROGRAMMATIC_SCROLL_SUPPRESS_MS = 200;
// After we set an anchor/spacer, ignore incidental scroll events for a bit.
const ANCHOR_CLEAR_GRACE_MS = 1200;
// Require recent direct user input (wheel/touch) to treat scroll as intentional.
const DIRECT_SCROLL_INTENT_WINDOW_MS = 250;
const ANCHOR_CLEAR_TOLERANCE_PX = 24;
const getMessageId = (message: ChatMessageRecord): string | null => {
const info = message.info;
if (typeof info?.id === 'string') {
return info.id;
}
return null;
};
const isUserMessage = (message: ChatMessageRecord): boolean => {
const info = message.info;
if (info?.userMessageMarker === true) {
return true;
}
const clientRole = info?.clientRole;
const serverRole = info?.role;
return clientRole === 'user' || serverRole === 'user';
};
const getMessageCreatedAt = (message: ChatMessageRecord): number => {
const info = message.info as { time?: { created?: unknown } };
const created = info?.time?.created;
return typeof created === 'number' ? created : 0;
};
// Threshold for re-pinning: 10% of container height (matches bottom spacer)
const PIN_THRESHOLD_RATIO = 0.10;
export const useChatScrollManager = ({
currentSessionId,
sessionMessages,
streamingMessageId,
updateViewportAnchor,
updateActiveTurnAnchor,
getActiveTurnAnchor,
isSyncing,
isMobile,
sessionActivityPhase,
}: UseChatScrollManagerOptions): UseChatScrollManagerResult => {
const scrollRef = React.useRef<HTMLDivElement | null>(null);
const scrollEngine = useScrollEngine({ containerRef: scrollRef, isMobile });
const [anchorId, setAnchorId] = React.useState<string | null>(null);
const [spacerHeight, setSpacerHeight] = React.useState(0);
const [showScrollButton, setShowScrollButton] = React.useState(false);
const [pendingAnchorId, setPendingAnchorId] = React.useState<string | null>(null);
const getPinThreshold = React.useCallback(() => {
const container = scrollRef.current;
if (!container || container.clientHeight <= 0) {
return 0;
}
const raw = container.clientHeight * PIN_THRESHOLD_RATIO;
return Math.max(24, Math.min(200, raw));
}, []);
const [showScrollButton, setShowScrollButton] = React.useState(false);
const [isPinned, setIsPinned] = React.useState(true);
const lastScrolledAnchorIdRef = React.useRef<string | null>(null);
const lastSessionIdRef = React.useRef<string | null>(null);
const currentSessionIdRef = React.useRef<string | null>(currentSessionId ?? null);
const suppressUserScrollUntilRef = React.useRef<number>(0);
const anchorClearIgnoreUntilRef = React.useRef<number>(0);
const lastDirectScrollIntentAtRef = React.useRef<number>(0);
const previousMessageIdsRef = React.useRef<Set<string>>(new Set());
const lastMessageCountRef = React.useRef<number>(sessionMessages.length);
const spacerHeightRef = React.useRef(0);
const isPinnedRef = React.useRef(true);
const lastScrollTopRef = React.useRef<number>(0);
const anchorIdRef = React.useRef<string | null>(null);
const pendingRestoreAnchorRef = React.useRef<{ sessionId: string; anchorId: string; startedAt: number } | null>(null);
const userScrollOverrideRef = React.useRef<boolean>(false);
const currentPhase = currentSessionId
? sessionActivityPhase?.get(currentSessionId) ?? 'idle'
: 'idle';
const isActivePhase = currentPhase === 'busy' || currentPhase === 'cooldown';
React.useEffect(() => {
currentSessionIdRef.current = currentSessionId ?? null;
}, [currentSessionId]);
const updateSpacerHeight = React.useCallback((height: number) => {
const newHeight = Math.max(0, height);
if (spacerHeightRef.current !== newHeight) {
spacerHeightRef.current = newHeight;
setSpacerHeight(newHeight);
}
}, []);
const calculateAnchorPosition = React.useCallback((anchorElement: HTMLElement): number => {
const messageTop = anchorElement.offsetTop;
return messageTop - ANCHOR_TARGET_OFFSET;
}, []);
const isAnchorStillPinned = React.useCallback((): boolean => {
const container = scrollRef.current;
const anchorId = anchorIdRef.current;
if (!container || !anchorId) return false;
const anchorElement = container.querySelector(`[data-message-id="${anchorId}"]`) as HTMLElement | null;
if (!anchorElement) return false;
const expectedTop = calculateAnchorPosition(anchorElement);
const distance = Math.abs(container.scrollTop - expectedTop);
return distance <= ANCHOR_CLEAR_TOLERANCE_PX;
}, [calculateAnchorPosition]);
const clearActiveTurnAnchor = React.useCallback((sessionId: string) => {
anchorIdRef.current = null;
lastScrolledAnchorIdRef.current = null;
pendingRestoreAnchorRef.current = null;
setAnchorId(null);
updateSpacerHeight(0);
updateActiveTurnAnchor(sessionId, null, 0);
}, [updateActiveTurnAnchor, updateSpacerHeight]);
const markProgrammaticScroll = React.useCallback(() => {
suppressUserScrollUntilRef.current = Date.now() + PROGRAMMATIC_SCROLL_SUPPRESS_MS;
}, []);
const getDistanceFromBottom = React.useCallback(() => {
const container = scrollRef.current;
if (!container) return 0;
return container.scrollHeight - container.scrollTop - container.clientHeight;
}, []);
const updatePinnedState = React.useCallback((newPinned: boolean) => {
if (isPinnedRef.current !== newPinned) {
isPinnedRef.current = newPinned;
setIsPinned(newPinned);
}
}, []);
const scrollToBottomInternal = React.useCallback((options?: { instant?: boolean; followBottom?: boolean }) => {
const container = scrollRef.current;
if (!container) return;
const bottom = container.scrollHeight - container.clientHeight;
markProgrammaticScroll();
scrollEngine.scrollToPosition(Math.max(0, bottom), options);
}, [markProgrammaticScroll, scrollEngine]);
const updateScrollButtonVisibility = React.useCallback(() => {
const container = scrollRef.current;
if (!container) {
@@ -188,126 +131,35 @@ export const useChatScrollManager = ({
return;
}
if (pendingAnchorId) {
setShowScrollButton(false);
return;
}
const hasScrollableContent = container.scrollHeight > container.clientHeight;
if (!hasScrollableContent) {
setShowScrollButton(false);
return;
}
const distanceFromBottom = container.scrollHeight - container.scrollTop - container.clientHeight;
const currentSpacerHeight = spacerHeightRef.current;
// Show scroll button when scrolled above the 10vh threshold
const distanceFromBottom = getDistanceFromBottom();
setShowScrollButton(distanceFromBottom > getPinThreshold());
}, [getDistanceFromBottom, getPinThreshold]);
if (currentSpacerHeight > 0) {
const spacerStartPosition = container.scrollHeight - currentSpacerHeight;
const viewportBottom = container.scrollTop + container.clientHeight;
setShowScrollButton(viewportBottom < spacerStartPosition);
} else {
setShowScrollButton(distanceFromBottom > DEFAULT_SCROLL_BUTTON_THRESHOLD);
}
}, [pendingAnchorId]);
const scrollToBottom = React.useCallback((options?: { instant?: boolean; force?: boolean; clearAnchor?: boolean }) => {
const scrollToPosition = React.useCallback((position: number, options?: { instant?: boolean }) => {
const container = scrollRef.current;
if (!container) return;
const distanceFromBottom = container.scrollHeight - container.scrollTop - container.clientHeight;
const shouldRespectUserScroll =
userScrollOverrideRef.current &&
currentPhase === 'idle' &&
!isSyncing &&
!options?.force &&
distanceFromBottom > DEFAULT_SCROLL_BUTTON_THRESHOLD;
if (shouldRespectUserScroll) {
return;
}
if (options?.force) {
userScrollOverrideRef.current = false;
}
if (options?.clearAnchor && currentSessionId && anchorIdRef.current) {
clearActiveTurnAnchor(currentSessionId);
}
const bottom = container.scrollHeight - container.clientHeight;
markProgrammaticScroll();
scrollEngine.scrollToPosition(Math.max(0, bottom), options);
}, [clearActiveTurnAnchor, currentPhase, currentSessionId, isSyncing, markProgrammaticScroll, scrollEngine]);
scrollEngine.scrollToPosition(Math.max(0, position), options);
}, [markProgrammaticScroll, scrollEngine]);
const scrollToNewAnchor = React.useCallback((messageId: string) => {
if (lastScrolledAnchorIdRef.current === messageId) {
return;
}
lastScrolledAnchorIdRef.current = messageId;
const scrollToBottom = React.useCallback((options?: { instant?: boolean; force?: boolean }) => {
const container = scrollRef.current;
if (!container) return;
// Give the UI a grace window so incidental scroll/layout events don't clear the anchor.
anchorClearIgnoreUntilRef.current = Date.now() + ANCHOR_CLEAR_GRACE_MS;
// Re-pin when explicitly scrolling to bottom
updatePinnedState(true);
setPendingAnchorId(messageId);
const expectedSessionId = currentSessionIdRef.current;
window.requestAnimationFrame(() => {
if (expectedSessionId !== currentSessionIdRef.current) {
return;
}
const container = scrollRef.current;
if (!container) {
setPendingAnchorId(null);
return;
}
const anchorElement = container.querySelector(`[data-message-id="${messageId}"]`) as HTMLElement | null;
if (!anchorElement) {
setPendingAnchorId(null);
return;
}
const containerHeight = container.clientHeight;
const targetScrollTop = calculateAnchorPosition(anchorElement);
const contentHeight = container.scrollHeight;
const currentSpacer = spacerHeightRef.current;
const contentWithoutSpacer = contentHeight - currentSpacer;
const requiredHeight = targetScrollTop + containerHeight;
let newSpacerHeight = 0;
if (contentWithoutSpacer < requiredHeight) {
newSpacerHeight = requiredHeight - contentWithoutSpacer;
}
if (newSpacerHeight !== currentSpacer) {
updateSpacerHeight(newSpacerHeight);
}
if (currentSessionIdRef.current) {
updateActiveTurnAnchor(currentSessionIdRef.current, messageId, newSpacerHeight);
}
window.requestAnimationFrame(() => {
if (expectedSessionId !== currentSessionIdRef.current) {
return;
}
markProgrammaticScroll();
scrollEngine.scrollToPosition(Math.max(0, targetScrollTop), { instant: true });
window.requestAnimationFrame(() => {
if (expectedSessionId !== currentSessionIdRef.current) {
return;
}
setPendingAnchorId(null);
});
});
});
}, [calculateAnchorPosition, markProgrammaticScroll, scrollEngine, updateActiveTurnAnchor, updateSpacerHeight]);
scrollToBottomInternal(options);
setShowScrollButton(false);
}, [scrollToBottomInternal, updatePinnedState]);
const handleScrollEvent = React.useCallback((event?: Event) => {
const container = scrollRef.current;
@@ -316,46 +168,44 @@ export const useChatScrollManager = ({
}
const now = Date.now();
const isProgrammatic = now < suppressUserScrollUntilRef.current || pendingAnchorId !== null;
const isProgrammatic = now < suppressUserScrollUntilRef.current;
const hasDirectIntent = now - lastDirectScrollIntentAtRef.current <= DIRECT_SCROLL_INTENT_WINDOW_MS;
if (event?.isTrusted && !isProgrammatic && hasDirectIntent) {
userScrollOverrideRef.current = true;
}
scrollEngine.handleScroll();
updateScrollButtonVisibility();
const shouldIgnoreAnchorClear = now < anchorClearIgnoreUntilRef.current;
// Handle pin/unpin logic
const currentScrollTop = container.scrollTop;
if (
event?.isTrusted &&
!isProgrammatic &&
!shouldIgnoreAnchorClear &&
hasDirectIntent &&
currentPhase === 'idle' &&
anchorIdRef.current !== null &&
spacerHeightRef.current > 0 &&
// Only clear when the user actually scrolls away from the pinned anchor.
// (Spacer being out of viewport is expected while anchored.)
!isAnchorStillPinned()
) {
clearActiveTurnAnchor(currentSessionId);
// Unpin requires strict user intent check
if (event?.isTrusted && !isProgrammatic && hasDirectIntent) {
const scrollingUp = currentScrollTop < lastScrollTopRef.current;
if (scrollingUp && isPinnedRef.current) {
updatePinnedState(false);
}
}
// Re-pin at bottom should always work (even momentum scroll)
if (!isPinnedRef.current) {
const distanceFromBottom = getDistanceFromBottom();
if (distanceFromBottom <= getPinThreshold()) {
updatePinnedState(true);
}
}
lastScrollTopRef.current = currentScrollTop;
const { scrollTop, scrollHeight, clientHeight } = container;
const position = (scrollTop + clientHeight / 2) / Math.max(scrollHeight, 1);
const estimatedIndex = Math.floor(position * sessionMessages.length);
updateViewportAnchor(currentSessionId, estimatedIndex);
}, [
clearActiveTurnAnchor,
currentPhase,
currentSessionId,
isAnchorStillPinned,
pendingAnchorId,
getDistanceFromBottom,
getPinThreshold,
scrollEngine,
sessionMessages.length,
updatePinnedState,
updateScrollButtonVisibility,
updateViewportAnchor,
]);
@@ -379,6 +229,7 @@ export const useChatScrollManager = ({
};
}, [handleScrollEvent]);
// Session switch - always start pinned at bottom
useIsomorphicLayoutEffect(() => {
if (!currentSessionId || currentSessionId === lastSessionIdRef.current) {
return;
@@ -387,175 +238,69 @@ export const useChatScrollManager = ({
lastSessionIdRef.current = currentSessionId;
MessageFreshnessDetector.getInstance().recordSessionStart(currentSessionId);
previousMessageIdsRef.current = new Set(
sessionMessages.map(getMessageId).filter((id): id is string => Boolean(id))
);
lastMessageCountRef.current = sessionMessages.length;
if (isActivePhase) {
const persistedAnchor = getActiveTurnAnchor(currentSessionId);
if (persistedAnchor && persistedAnchor.anchorId) {
anchorIdRef.current = persistedAnchor.anchorId;
lastScrolledAnchorIdRef.current = persistedAnchor.anchorId;
const container = scrollRef.current;
const anchorElement = container
? (container.querySelector(`[data-message-id="${persistedAnchor.anchorId}"]`) as HTMLElement | null)
: null;
const messageHeight = anchorElement?.offsetHeight ?? 0;
const restoredSpacerHeight = Math.max(0, persistedAnchor.spacerHeight - (messageHeight - 50));
flushSync(() => {
setAnchorId(persistedAnchor.anchorId);
updateSpacerHeight(restoredSpacerHeight);
});
pendingRestoreAnchorRef.current = { sessionId: currentSessionId, anchorId: persistedAnchor.anchorId, startedAt: Date.now() };
} else {
lastScrolledAnchorIdRef.current = null;
anchorIdRef.current = null;
setAnchorId(null);
updateSpacerHeight(0);
pendingRestoreAnchorRef.current = null;
const container = scrollRef.current;
if (container) {
const bottom = container.scrollHeight - container.clientHeight;
markProgrammaticScroll();
scrollEngine.scrollToPosition(Math.max(0, bottom), { instant: true });
}
}
} else {
lastScrolledAnchorIdRef.current = null;
anchorIdRef.current = null;
setAnchorId(null);
updateSpacerHeight(0);
pendingRestoreAnchorRef.current = null;
updateActiveTurnAnchor(currentSessionId, null, 0);
const container = scrollRef.current;
if (container) {
const bottom = container.scrollHeight - container.clientHeight;
markProgrammaticScroll();
scrollEngine.scrollToPosition(Math.max(0, bottom), { instant: true });
}
}
setPendingAnchorId(null);
// Always start pinned at bottom on session switch
updatePinnedState(true);
setShowScrollButton(false);
userScrollOverrideRef.current = false;
}, [
currentSessionId,
getActiveTurnAnchor,
isActivePhase,
markProgrammaticScroll,
scrollEngine,
updateActiveTurnAnchor,
updateSpacerHeight,
sessionMessages,
]);
useIsomorphicLayoutEffect(() => {
if (typeof window === 'undefined') return;
if (!currentSessionId) return;
const container = scrollRef.current;
if (container) {
markProgrammaticScroll();
scrollToBottomInternal({ instant: true });
}
}, [currentSessionId, markProgrammaticScroll, scrollToBottomInternal, updatePinnedState]);
const pending = pendingRestoreAnchorRef.current;
if (!pending || pending.sessionId !== currentSessionId) return;
// Maintain pin-to-bottom when content changes
React.useEffect(() => {
if (!isPinnedRef.current) return;
if (isSyncing) return;
const container = scrollRef.current;
if (!container) return;
const anchorElement = container.querySelector(`[data-message-id="${pending.anchorId}"]`) as HTMLElement | null;
if (!anchorElement) {
// When the anchor is created from a just-sent user message, the persisted anchor can
// show up before the message is in the rendered list. Give it a short window.
if (Date.now() - pending.startedAt < 1200) {
return;
}
clearActiveTurnAnchor(currentSessionId);
return;
// When pinned and content grows, scroll to bottom instantly
const distanceFromBottom = getDistanceFromBottom();
if (distanceFromBottom > getPinThreshold()) {
markProgrammaticScroll();
scrollToBottomInternal({ instant: true });
}
}, [getDistanceFromBottom, getPinThreshold, isSyncing, markProgrammaticScroll, scrollToBottomInternal, sessionMessages]);
const targetScrollTop = calculateAnchorPosition(anchorElement);
markProgrammaticScroll();
scrollEngine.scrollToPosition(targetScrollTop, { instant: true });
pendingRestoreAnchorRef.current = null;
}, [calculateAnchorPosition, clearActiveTurnAnchor, currentSessionId, markProgrammaticScroll, scrollEngine, sessionMessages]);
useIsomorphicLayoutEffect(() => {
if (isSyncing) {
return;
}
if (lastSessionIdRef.current !== currentSessionId) {
return;
}
const previousIds = previousMessageIdsRef.current;
const nextIds = new Set(sessionMessages.map(getMessageId).filter((id): id is string => Boolean(id)));
const nextCount = sessionMessages.length;
if (nextCount > lastMessageCountRef.current) {
const addedIds: string[] = [];
nextIds.forEach((id) => {
if (!previousIds.has(id)) {
addedIds.push(id);
}
});
if (addedIds.length > 0) {
const now = Date.now();
let latestNewUserMessageId: string | null = null;
let latestNewUserCreatedAt = 0;
for (let i = 0; i < sessionMessages.length; i++) {
const message = sessionMessages[i];
const id = getMessageId(message);
if (!id || !addedIds.includes(id)) continue;
if (!isUserMessage(message)) continue;
let createdAt = getMessageCreatedAt(message);
if (createdAt <= 0 && (Boolean(streamingMessageId) || isActivePhase)) {
createdAt = now;
}
if (createdAt >= latestNewUserCreatedAt) {
latestNewUserCreatedAt = createdAt;
latestNewUserMessageId = id;
}
}
const shouldAnchorNewUser =
latestNewUserMessageId !== null &&
(Boolean(streamingMessageId) ||
isActivePhase ||
now - latestNewUserCreatedAt <= NEW_USER_ANCHOR_WINDOW_MS);
if (shouldAnchorNewUser && latestNewUserMessageId) {
anchorIdRef.current = latestNewUserMessageId;
setAnchorId(latestNewUserMessageId);
scrollToNewAnchor(latestNewUserMessageId);
}
}
}
lastMessageCountRef.current = nextCount;
previousMessageIdsRef.current = nextIds;
}, [currentSessionId, isActivePhase, isSyncing, scrollToNewAnchor, sessionMessages, streamingMessageId]);
// Use ResizeObserver to detect content changes and maintain pin
React.useEffect(() => {
const container = scrollRef.current;
if (!container || typeof ResizeObserver === 'undefined') return;
const observer = new ResizeObserver(() => {
updateScrollButtonVisibility();
// Maintain pin when content grows - always instant for smooth experience
if (isPinnedRef.current) {
const distanceFromBottom = getDistanceFromBottom();
if (distanceFromBottom > getPinThreshold()) {
scrollToBottomInternal({ instant: true });
}
}
});
observer.observe(container);
// Also observe children for content changes
const childObserver = new MutationObserver(() => {
if (isPinnedRef.current) {
const distanceFromBottom = getDistanceFromBottom();
if (distanceFromBottom > getPinThreshold()) {
scrollToBottomInternal({ instant: true });
}
}
});
childObserver.observe(container, { childList: true, subtree: true });
return () => {
observer.disconnect();
childObserver.disconnect();
};
}, [updateScrollButtonVisibility]);
}, [getDistanceFromBottom, getPinThreshold, scrollToBottomInternal, updateScrollButtonVisibility]);
React.useEffect(() => {
if (typeof window === 'undefined') {
@@ -572,23 +317,21 @@ export const useChatScrollManager = ({
};
}, [currentSessionId, sessionMessages.length, updateScrollButtonVisibility]);
React.useEffect(() => {
if (anchorId) {
updateScrollButtonVisibility();
}
}, [anchorId, updateScrollButtonVisibility]);
React.useEffect(() => {
updateScrollButtonVisibility();
}, [spacerHeight, updateScrollButtonVisibility]);
const animationHandlersRef = React.useRef<Map<string, AnimationHandlers>>(new Map());
const handleMessageContentChange = React.useCallback(() => {
updateScrollButtonVisibility();
}, [updateScrollButtonVisibility]);
const getAnimationHandlers = React.useCallback((messageId: string): AnimationHandlers => {
// Maintain pin when content changes - always instant
if (isPinnedRef.current) {
const distanceFromBottom = getDistanceFromBottom();
if (distanceFromBottom > getPinThreshold()) {
scrollToBottomInternal({ instant: true });
}
}
}, [getDistanceFromBottom, getPinThreshold, scrollToBottomInternal, updateScrollButtonVisibility]);
const getAnimationHandlers = React.useCallback((messageId: string): AnimationHandlers => {
const existing = animationHandlersRef.current.get(messageId);
if (existing) {
return existing;
@@ -597,30 +340,34 @@ export const useChatScrollManager = ({
const handlers: AnimationHandlers = {
onChunk: () => {
updateScrollButtonVisibility();
if (isPinnedRef.current) {
const distanceFromBottom = getDistanceFromBottom();
if (distanceFromBottom > getPinThreshold()) {
scrollToBottomInternal({ instant: true });
}
}
},
onComplete: () => {
updateScrollButtonVisibility();
},
onStreamingCandidate: () => {
},
onAnimationStart: () => {
},
onStreamingCandidate: () => {},
onAnimationStart: () => {},
onAnimatedHeightChange: () => {
updateScrollButtonVisibility();
if (isPinnedRef.current) {
const distanceFromBottom = getDistanceFromBottom();
if (distanceFromBottom > getPinThreshold()) {
scrollToBottomInternal({ instant: true });
}
}
},
onReservationCancelled: () => {
},
onReasoningBlock: () => {
},
onReservationCancelled: () => {},
onReasoningBlock: () => {},
};
animationHandlersRef.current.set(messageId, handlers);
return handlers;
}, [updateScrollButtonVisibility]);
}, [getDistanceFromBottom, getPinThreshold, scrollToBottomInternal, updateScrollButtonVisibility]);
return {
scrollRef,
@@ -628,8 +375,7 @@ export const useChatScrollManager = ({
getAnimationHandlers,
showScrollButton,
scrollToBottom,
spacerHeight,
pendingAnchorId,
hasActiveAnchor: anchorId !== null,
scrollToPosition,
isPinned,
};
};
+7 -6
View File
@@ -2,6 +2,7 @@ import React from 'react';
import { opencodeClient, type RoutedOpencodeEvent } from '@/lib/opencode/client';
import { saveSessionCursor } from '@/lib/messageCursorPersistence';
import { useSessionStore } from '@/stores/useSessionStore';
import { getActiveSessionWindow } from '@/stores/types/sessionTypes';
import { useConfigStore } from '@/stores/useConfigStore';
import { useUIStore, type EventStreamStatus } from '@/stores/useUIStore';
import { useDirectoryStore } from '@/stores/useDirectoryStore';
@@ -276,7 +277,7 @@ export const useEventStream = () => {
);
const resyncMessages = React.useCallback(
(sessionId: string, reason: string) => {
(sessionId: string, reason: string, limit?: number) => {
if (!sessionId) {
return Promise.resolve();
}
@@ -287,7 +288,7 @@ export const useEventStream = () => {
if (now - lastResyncAtRef.current < RESYNC_DEBOUNCE_MS) {
return Promise.resolve();
}
const task = loadMessages(sessionId)
const task = loadMessages(sessionId, limit)
.catch((error) => {
console.warn(`[useEventStream] Failed to resync messages (${reason}):`, error);
})
@@ -309,7 +310,7 @@ export const useEventStream = () => {
try {
await Promise.all([
loadSessions(),
currentSessionId ? resyncMessages(currentSessionId, reason) : Promise.resolve(),
currentSessionId ? resyncMessages(currentSessionId, reason, Infinity) : Promise.resolve(),
]);
} catch (error) {
console.warn('[useEventStream] Bootstrap failed:', reason, error);
@@ -1451,7 +1452,7 @@ export const useEventStream = () => {
const sessionId = currentSessionIdRef.current;
if (sessionId) {
setTimeout(() => {
resyncMessages(sessionId, 'sse_reconnected')
resyncMessages(sessionId, 'sse_reconnected', Infinity)
.then(() => requestSessionMetadataRefresh(sessionId))
.catch((error) => {
console.warn('[useEventStream] Failed to resync messages after reconnect:', error);
@@ -1618,7 +1619,7 @@ export const useEventStream = () => {
console.info('[useEventStream] Visibility restored, triggering soft refresh...');
const sessionId = currentSessionIdRef.current;
if (sessionId) {
resyncMessages(sessionId, 'visibility_restore').catch(() => {});
resyncMessages(sessionId, 'visibility_restore', getActiveSessionWindow()).catch(() => {});
requestSessionMetadataRefresh(sessionId);
}
@@ -1644,7 +1645,7 @@ export const useEventStream = () => {
const sessionId = currentSessionIdRef.current;
if (sessionId) {
requestSessionMetadataRefresh(sessionId);
resyncMessages(sessionId, 'window_focus')
resyncMessages(sessionId, 'window_focus', getActiveSessionWindow())
.then(() => console.info('[useEventStream] Messages refreshed on focus'))
.catch((err) => console.warn('[useEventStream] Failed to refresh messages:', err));
}
+15
View File
@@ -7,6 +7,7 @@ type ScrollEngineOptions = {
type ScrollOptions = {
instant?: boolean;
followBottom?: boolean; // Dynamically track bottom during animation
};
type ScrollEngineResult = {
@@ -33,6 +34,7 @@ export const useScrollEngine = ({
const animationStartRef = React.useRef<number | null>(null);
const animationFromRef = React.useRef(0);
const animationTargetRef = React.useRef(0);
const followBottomRef = React.useRef(false);
const cancelAnimation = React.useCallback(() => {
if (animationFrameRef.current !== null && typeof window !== 'undefined') {
@@ -41,6 +43,7 @@ export const useScrollEngine = ({
animationFrameRef.current = null;
animationStartRef.current = null;
followBottomRef.current = false;
}, []);
const runAnimationFrame = React.useCallback(
@@ -55,6 +58,11 @@ export const useScrollEngine = ({
animationStartRef.current = timestamp;
}
// If followBottom mode, dynamically update target to current bottom
if (followBottomRef.current) {
animationTargetRef.current = container.scrollHeight - container.clientHeight;
}
const progress = Math.min(1, (timestamp - animationStartRef.current) / ANIMATION_DURATION_MS);
const easedProgress = 1 - Math.pow(1 - progress, 3);
const from = animationFromRef.current;
@@ -86,6 +94,7 @@ export const useScrollEngine = ({
const target = Math.max(0, position);
const preferInstant = options?.instant ?? false;
const followBottom = options?.followBottom ?? false;
manualOverrideRef.current = false;
@@ -102,6 +111,11 @@ export const useScrollEngine = ({
return;
}
// If followBottom animation is already running, don't restart - let it continue
if (followBottom && followBottomRef.current && animationFrameRef.current !== null) {
return;
}
cancelAnimation();
const distance = Math.abs(target - container.scrollTop);
@@ -120,6 +134,7 @@ export const useScrollEngine = ({
animationFromRef.current = container.scrollTop;
animationTargetRef.current = target;
animationStartRef.current = null;
followBottomRef.current = followBottom;
animationFrameRef.current = window.requestAnimationFrame(runAnimationFrame);
},
[cancelAnimation, containerRef, runAnimationFrame, setIsAtTop]
+5 -35
View File
@@ -343,7 +343,7 @@ interface MessageState {
}
interface MessageActions {
loadMessages: (sessionId: string) => Promise<void>;
loadMessages: (sessionId: string, limit?: number) => Promise<void>;
sendMessage: (content: string, providerID: string, modelID: string, agent?: string, currentSessionId?: string, attachments?: AttachedFile[], agentMentionName?: string | null, additionalParts?: Array<{ text: string; attachments?: AttachedFile[] }>, variant?: string) => Promise<void>;
abortCurrentOperation: (currentSessionId?: string) => Promise<void>;
_addStreamingPartImmediate: (sessionId: string, messageId: string, part: Part, role?: string, currentSessionId?: string) => void;
@@ -354,8 +354,6 @@ interface MessageActions {
updateMessageInfo: (sessionId: string, messageId: string, messageInfo: any) => void;
syncMessages: (sessionId: string, messages: { info: Message; parts: Part[] }[]) => void;
updateViewportAnchor: (sessionId: string, anchor: number) => void;
updateActiveTurnAnchor: (sessionId: string, anchorId: string | null, spacerHeight: number) => void;
getActiveTurnAnchor: (sessionId: string) => { anchorId: string | null; spacerHeight: number } | null;
trimToViewportWindow: (sessionId: string, targetSize?: number, currentSessionId?: string) => void;
evictLeastRecentlyUsed: (currentSessionId?: string) => void;
loadMoreMessages: (sessionId: string, direction: "up" | "down") => Promise<void>;
@@ -387,10 +385,12 @@ export const useMessageStore = create<MessageStore>()(
loadMessages: async (sessionId: string, limit?: number) => {
const memLimits = getMemoryLimits();
const effectiveLimit = limit ?? memLimits.HISTORICAL_MESSAGES;
const noLimit = limit === Infinity;
const effectiveLimit = noLimit ? Infinity : (limit ?? memLimits.HISTORICAL_MESSAGES);
const isStreaming = get().sessionMemoryState.get(sessionId)?.isStreaming;
const targetLimit = isStreaming ? memLimits.VIEWPORT_MESSAGES : effectiveLimit;
const fetchLimit = isStreaming ? undefined : targetLimit + memLimits.FETCH_BUFFER;
// Don't pass Infinity to API - use undefined for "fetch all"
const fetchLimit = isStreaming || noLimit ? undefined : targetLimit + memLimits.FETCH_BUFFER;
const allMessages = await executeWithSessionDirectory(sessionId, () => opencodeClient.getSessionMessages(sessionId, fetchLimit));
// Filter out reverted messages first
@@ -2201,34 +2201,6 @@ export const useMessageStore = create<MessageStore>()(
});
},
updateActiveTurnAnchor: (sessionId: string, anchorId: string | null, spacerHeight: number) => {
set((state) => {
const memoryState = state.sessionMemoryState.get(sessionId) || {
viewportAnchor: 0,
isStreaming: false,
lastAccessedAt: Date.now(),
backgroundMessageCount: 0,
};
const newMemoryState = new Map(state.sessionMemoryState);
newMemoryState.set(sessionId, {
...memoryState,
activeTurnAnchorId: anchorId ?? undefined,
activeTurnSpacerHeight: spacerHeight,
});
return { sessionMemoryState: newMemoryState };
});
},
getActiveTurnAnchor: (sessionId: string) => {
const memoryState = get().sessionMemoryState.get(sessionId);
if (!memoryState) return null;
return {
anchorId: memoryState.activeTurnAnchorId ?? null,
spacerHeight: memoryState.activeTurnSpacerHeight ?? 0,
};
},
trimToViewportWindow: (sessionId: string, targetSize?: number, currentSessionId?: string) => {
const effectiveTargetSize = targetSize ?? getMemoryLimits().VIEWPORT_MESSAGES;
const state = get();
@@ -2502,8 +2474,6 @@ export const useMessageStore = create<MessageStore>()(
totalAvailableMessages: memory.totalAvailableMessages,
hasMoreAbove: memory.hasMoreAbove,
trimmedHeadMaxId: memory.trimmedHeadMaxId,
activeTurnAnchorId: memory.activeTurnAnchorId,
activeTurnSpacerHeight: memory.activeTurnSpacerHeight,
},
]),
sessionAbortFlags: Array.from(state.sessionAbortFlags.entries()).map(([sessionId, record]) => [
+1 -7
View File
@@ -35,10 +35,6 @@ export interface SessionMemoryState {
hasMoreAbove?: boolean;
trimmedHeadMaxId?: string;
streamingCooldownUntil?: number;
/** Message ID of the user's active turn anchor (for scroll position preservation) */
activeTurnAnchorId?: string;
/** Height of the spacer below messages for active turn positioning */
activeTurnSpacerHeight?: number;
}
export interface SessionContextUsage {
@@ -163,7 +159,7 @@ export interface SessionStore {
shareSession: (id: string) => Promise<Session | null>;
unshareSession: (id: string) => Promise<Session | null>;
setCurrentSession: (id: string | null) => void;
loadMessages: (sessionId: string) => Promise<void>;
loadMessages: (sessionId: string, limit?: number) => Promise<void>;
sendMessage: (content: string, providerID: string, modelID: string, agent?: string, attachments?: AttachedFile[], agentMentionName?: string, additionalParts?: Array<{ text: string; attachments?: AttachedFile[] }>, variant?: string) => Promise<void>;
abortCurrentOperation: () => Promise<void>;
acknowledgeSessionAbort: (sessionId: string) => void;
@@ -197,8 +193,6 @@ export interface SessionStore {
clearAttachedFiles: () => void;
updateViewportAnchor: (sessionId: string, anchor: number) => void;
updateActiveTurnAnchor: (sessionId: string, anchorId: string | null, spacerHeight: number) => void;
getActiveTurnAnchor: (sessionId: string) => { anchorId: string | null; spacerHeight: number } | null;
trimToViewportWindow: (sessionId: string, targetSize?: number) => void;
evictLeastRecentlyUsed: () => void;
loadMoreMessages: (sessionId: string, direction: "up" | "down") => Promise<void>;
+2 -4
View File
@@ -310,7 +310,7 @@ export const useSessionStore = create<SessionStore>()(
get().evictLeastRecentlyUsed();
},
loadMessages: (sessionId: string) => useMessageStore.getState().loadMessages(sessionId),
loadMessages: (sessionId: string, limit?: number) => useMessageStore.getState().loadMessages(sessionId, limit),
sendMessage: async (content: string, providerID: string, modelID: string, agent?: string, attachments?: AttachedFile[], agentMentionName?: string, additionalParts?: Array<{ text: string; attachments?: AttachedFile[] }>, variant?: string) => {
const draft = get().newSessionDraft;
const trimmedAgent = typeof agent === 'string' && agent.trim().length > 0 ? agent.trim() : undefined;
@@ -502,11 +502,9 @@ export const useSessionStore = create<SessionStore>()(
clearAttachedFiles: () => useFileStore.getState().clearAttachedFiles(),
updateViewportAnchor: (sessionId: string, anchor: number) => useMessageStore.getState().updateViewportAnchor(sessionId, anchor),
updateActiveTurnAnchor: (sessionId: string, anchorId: string | null, spacerHeight: number) => useMessageStore.getState().updateActiveTurnAnchor(sessionId, anchorId, spacerHeight),
getActiveTurnAnchor: (sessionId: string) => useMessageStore.getState().getActiveTurnAnchor(sessionId),
trimToViewportWindow: (sessionId: string, targetSize?: number) => {
const currentSessionId = useSessionManagementStore.getState().currentSessionId;
// Skip trimming for sessions in active phase (busy/cooldown) to preserve anchor/spacer
// Skip trimming for sessions in active phase (busy/cooldown)
const phase = get().sessionActivityPhase?.get(sessionId);
if (phase === 'busy' || phase === 'cooldown') {
return;