namespace UIWidgets.Issues
{
    using UIWidgets;
    using UnityEngine;
    using UnityEngine.EventSystems;

    public class TreeViewMoveDown : MonoBehaviour
    {
        public TreeView TreeView;

        protected virtual void Start()
        {
            TreeView.ItemsEvents.Move.AddListener(ProcessMove);
        }

        void ProcessMove(int nodeIndex, ListViewItem instance, AxisEventData eventData)
        {
            if (eventData.moveDir == MoveDirection.Down)
            {
                var node = TreeView.DataSource[nodeIndex].Node;
                if (!node.IsExpanded)
                {
                    node.IsExpanded = true;
                    var next_instance = TreeView.GetItemComponent(nodeIndex + 1);
                    if (next_instance != null)
                    {
                        EventSystem.current.SetSelectedGameObject(next_instance.gameObject);
                    }
                }
            }
        }
    }
}