import { useEffect, useRef, useState } from "react";
import { Empty } from "@arco-design/web-react";

const InfiniteScroll = ({
  loadMore,
  hasMore,
  children,
  threshold = 50,
  className = "",
}) => {
  const containerRef = useRef(null);
  const [loading, setLoading] = useState(false);

  const handleScroll = () => {
    if (loading) return;
    setLoading(true);
    if (!containerRef.current || !hasMore) return;

    const { scrollTop, scrollHeight, clientHeight } = containerRef.current;

    if (scrollTop + clientHeight >= scrollHeight - threshold) {
      loadMore().finally(() => {
        setLoading(false);
      });
    }
  };

  useEffect(() => {
    const container = containerRef.current;
    if (container) {
      container.addEventListener("scroll", handleScroll);
      // 初始加载时检查是否需要加载更多
      handleScroll();
    }

    return () => {
      if (container) {
        container.removeEventListener("scroll", handleScroll);
      }
    };
  }, [loadMore, hasMore, threshold]);

  return (
    <div
      ref={containerRef}
      className={`infinite-scroll-container ${className}`}
    >
      {children}
      {!hasMore && !loading && <Empty description="没有更多了" />}
    </div>
  );
};

export default InfiniteScroll;
