Press n or j to go to the next uncovered block, b, p or k for the previous block.
| 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 | import { AiHydroMessage } from "@shared/ExtensionMessage"
import React, { useCallback } from "react"
import { Virtuoso } from "react-virtuoso"
import { ChatState, MessageHandlers, ScrollBehavior } from "../../types/chatTypes"
import { createMessageRenderer } from "../messages/MessageRenderer"
interface MessagesAreaProps {
task: AiHydroMessage
groupedMessages: (AiHydroMessage | AiHydroMessage[])[]
modifiedMessages: AiHydroMessage[]
scrollBehavior: ScrollBehavior
chatState: ChatState
messageHandlers: MessageHandlers
}
/**
* The scrollable messages area with virtualized list
* Handles rendering of chat rows and browser sessions
*/
export const MessagesArea: React.FC<MessagesAreaProps> = ({
task,
groupedMessages,
modifiedMessages,
scrollBehavior,
chatState,
messageHandlers,
}) => {
const {
virtuosoRef,
scrollContainerRef,
toggleRowExpansion,
handleRowHeightChange,
setIsAtBottom,
setShowScrollToBottom,
disableAutoScrollRef,
} = scrollBehavior
const { expandedRows, inputValue, setActiveQuote } = chatState
const itemContent = useCallback(
createMessageRenderer(
groupedMessages,
modifiedMessages,
expandedRows,
toggleRowExpansion,
handleRowHeightChange,
setActiveQuote,
inputValue,
messageHandlers,
),
[
groupedMessages,
modifiedMessages,
expandedRows,
toggleRowExpansion,
handleRowHeightChange,
setActiveQuote,
inputValue,
messageHandlers,
],
)
return (
<div className="overflow-hidden flex flex-col h-full">
<div className="flex-grow flex" ref={scrollContainerRef}>
<Virtuoso
atBottomStateChange={(isAtBottom) => {
setIsAtBottom(isAtBottom)
if (isAtBottom) {
disableAutoScrollRef.current = false
}
setShowScrollToBottom(disableAutoScrollRef.current && !isAtBottom)
}}
atBottomThreshold={10} // trick to make sure virtuoso re-renders when task changes, and we use initialTopMostItemIndex to start at the bottom
className="scrollable"
components={{
Footer: () => <div style={{ height: 5 }} />, // Add empty padding at the bottom
}}
data={groupedMessages}
// increasing top by 3_000 to prevent jumping around when user collapses a row
increaseViewportBy={{
top: 3_000,
bottom: Number.MAX_SAFE_INTEGER,
}} // hack to make sure the last message is always rendered to get truly perfect scroll to bottom animation when new messages are added (Number.MAX_SAFE_INTEGER is safe for arithmetic operations, which is all virtuoso uses this value for in src/sizeRangeSystem.ts)
initialTopMostItemIndex={groupedMessages.length - 1} // messages is the raw format returned by extension, modifiedMessages is the manipulated structure that combines certain messages of related type, and visibleMessages is the filtered structure that removes messages that should not be rendered
itemContent={itemContent}
key={task.ts}
ref={virtuosoRef} // anything lower causes issues with followOutput
style={{
flexGrow: 1,
overflowY: "scroll", // always show scrollbar
}}
/>
</div>
</div>
)
}
|