fix: scroll issue playground
This commit is contained in:
35
src/hooks/useSmartScroll.tsx
Normal file
35
src/hooks/useSmartScroll.tsx
Normal file
@@ -0,0 +1,35 @@
|
||||
import { useRef, useEffect, useState } from 'react';
|
||||
|
||||
export const useSmartScroll = (messages: any[], streaming: boolean) => {
|
||||
const containerRef = useRef<HTMLDivElement>(null);
|
||||
const [isAtBottom, setIsAtBottom] = useState(true);
|
||||
|
||||
useEffect(() => {
|
||||
const container = containerRef.current;
|
||||
if (!container) return;
|
||||
|
||||
const handleScroll = () => {
|
||||
const { scrollTop, scrollHeight, clientHeight } = container;
|
||||
setIsAtBottom(scrollHeight - scrollTop - clientHeight < 50);
|
||||
};
|
||||
|
||||
container.addEventListener('scroll', handleScroll);
|
||||
return () => container.removeEventListener('scroll', handleScroll);
|
||||
}, []);
|
||||
|
||||
useEffect(() => {
|
||||
if (isAtBottom && containerRef.current) {
|
||||
const scrollOptions: ScrollIntoViewOptions = streaming
|
||||
? { behavior: 'smooth', block: 'end' }
|
||||
: { behavior: 'auto', block: 'end' };
|
||||
containerRef.current.lastElementChild?.scrollIntoView(scrollOptions);
|
||||
}
|
||||
}, [messages, streaming, isAtBottom]);
|
||||
|
||||
const scrollToBottom = () => {
|
||||
containerRef.current?.lastElementChild?.scrollIntoView({ behavior: 'smooth', block: 'end' });
|
||||
};
|
||||
|
||||
|
||||
return { containerRef, isAtBottom, scrollToBottom };
|
||||
};
|
||||
Reference in New Issue
Block a user