feishin/src/renderer/components/item-list/helpers/use-item-list-scroll-persist.ts

import { useCallback, useMemo } from 'react';
import { useLocation, useNavigationType } from 'react-router';

import { useScrollStore } from '/@/renderer/store/scroll.store';

interface UseItemListScrollPersistProps {
    enabled: boolean;
}

export const useItemListScrollPersist = ({ enabled }: UseItemListScrollPersistProps) => {
    const location = useLocation();
    const navigationType = useNavigationType();
    const setOffset = useScrollStore((s) => s.setOffset);
    const getOffset = useScrollStore((s) => s.getOffset);

    const scrollOffset = useMemo(() => {
        if (navigationType !== 'POP') return undefined;
        return getOffset(location.key);
    }, [getOffset, location.key, navigationType]);

    const handleOnScrollEnd = useCallback(
        (offset: number) => {
            if (!enabled) return;
            setOffset(location.key, offset);
        },
        [enabled, location.key, setOffset],
    );

    return { handleOnScrollEnd, scrollOffset };
};