using System.Collections.Generic;
using UnityEngine;
using UnityEngine.UI;

namespace SpellBoundAR.Items.Selections.UI
{
    [RequireComponent(typeof(Button))]
    public class NextItemTypeButton : MonoBehaviour
    {
        [SerializeField] private ItemTypeSelectionMenu itemTypeSelectionMenu;
        
        [Header("Cache")]
        private Button _button;

        public ItemTypeSelectionMenu ItemTypeSelectionMenu
        {
            get => itemTypeSelectionMenu;
            set => itemTypeSelectionMenu = value;
        }

        private void Awake()
        {
            _button = GetComponent<Button>();
        }

        private void OnEnable()
        {
            if (_button) _button.onClick.AddListener(OnClick);
        }

        private void OnDisable()
        {
            if (_button) _button.onClick.RemoveListener(OnClick);
        }

        private void OnClick()
        {
            if (!itemTypeSelectionMenu) return;
            List<ScriptedItemTypeData> items = itemTypeSelectionMenu.Items.ItemTypes;
            IItemTypeData currentItem = itemTypeSelectionMenu.CurrentSelection;
            int currentIndex = currentItem != null
               ? items.IndexOf(currentItem as ScriptedItemTypeData)
               : -1;
            int nextIndex = currentIndex + 1;
            if (nextIndex >= items.Count) nextIndex = 0;
            itemTypeSelectionMenu.CurrentSelection = nextIndex < items.Count ? items[nextIndex] : null;
        }
    }
}