AssetBase/Assets/FirstVillain/InfinityScrollView/Scripts/VerticalInfiniteScrollView.cs

using System.Collections;
using UnityEngine;

namespace FirstVillain.ScrollView
{
    public class VerticalInfiniteScrollView : InfiniteScrollView
    {
        [SerializeField] private bool _isAtTop = true;
        [SerializeField] private bool _isAtBottom = true;

        public override void Initialize()
        {
            base.Initialize();
            _isAtTop = true;
            _isAtBottom = true;
        }

        protected override void OnValueChanged(Vector2 normalizedPosition)
        {
            if (_dataList.Count == 0)
                return;
            float viewportInterval = _scrollRect.viewport.rect.height;
            float minViewport = _scrollRect.content.anchoredPosition.y;
            Vector2 viewportRange = new Vector2(minViewport - _extendVisibleRange, minViewport + viewportInterval + _extendVisibleRange);
            float contentHeight = _padding.x;
            for (int i = 0; i < _dataList.Count; i++)
            {
                var visibleRange = new Vector2(contentHeight, contentHeight + _dataList[i].CellSize.y);
                if (visibleRange.y < viewportRange.x || visibleRange.x > viewportRange.y)
                {
                    RecycleCell(i);
                }
                contentHeight += _dataList[i].CellSize.y + _spacing;
            }
            contentHeight = _padding.x;
            for (int i = 0; i < _dataList.Count; i++)
            {
                var visibleRange = new Vector2(contentHeight, contentHeight + _dataList[i].CellSize.y);
                if (visibleRange.y >= viewportRange.x && visibleRange.x <= viewportRange.y)
                {
                    SetupCell(i, new Vector2(0, -contentHeight));
                    if (visibleRange.y >= viewportRange.x)
                        _cellList[i].transform.SetAsLastSibling();
                    else
                        _cellList[i].transform.SetAsFirstSibling();
                }
                contentHeight += _dataList[i].CellSize.y + _spacing;
            }
            if (_scrollRect.content.sizeDelta.y > viewportInterval)
            {
                _isAtTop = viewportRange.x + _extendVisibleRange <= 0.001f;
                _isAtBottom = _scrollRect.content.sizeDelta.y - viewportRange.y + _extendVisibleRange <= 0.001f;
            }
            else
            {
                _isAtTop = true;
                _isAtBottom = true;
            }
        }

        public sealed override void Refresh()
        {
            if (!IsInitialized)
            {
                Initialize();
            }
            if (_scrollRect.viewport.rect.height == 0)
                StartCoroutine(DelayToRefresh());
            else
                DoRefresh();
        }

        private void DoRefresh()
        {
            float height = _padding.x;
            for (int i = 0; i < _dataList.Count; i++)
            {
                height += _dataList[i].CellSize.y + _spacing;
            }
            for (int i = 0; i < _cellList.Count; i++)
            {
                RecycleCell(i);
            }
            height += _padding.y;
            _scrollRect.content.sizeDelta = new Vector2(_scrollRect.content.sizeDelta.x, height);
            OnValueChanged(_scrollRect.normalizedPosition);
            OnRefresh?.Invoke();
        }

        private IEnumerator DelayToRefresh()
        {
            yield return _waitEndOfFrame;
            DoRefresh();
        }

        public override void Snap(int index, float duration)
        {
            if (!IsInitialized)
                return;
            if (index >= _dataList.Count)
                return;
            if (_scrollRect.content.rect.height < _scrollRect.viewport.rect.height)
                return;
            float height = _padding.x;
            for (int i = 0; i < index; i++)
            {
                height += _dataList[i].CellSize.y + _spacing;
            }
            height = Mathf.Min(_scrollRect.content.rect.height - _scrollRect.viewport.rect.height, height);
            if (_scrollRect.content.anchoredPosition.y != height)
            {
                DoSnapping(new Vector2(0, height), duration);
            }
        }

        public override void Remove(int index)
        {
            var removeCell = _dataList[index];
            base.Remove(index);
            _scrollRect.content.anchoredPosition -= new Vector2(0, removeCell.CellSize.y + _spacing);
        }
    }
}