using System;
using System.Collections.Generic;
using UnityEngine;
using UnityEngine.Experimental.UIElements;

namespace UnityEditor.PackageManager.UI
{
#if UNITY_2018_3_OR_NEWER
    internal class PopupField<T> : Experimental.UIElements.PopupField<T>
    {
        private Func<T, string> m_Callback;

        public override T value
        {
            get { return base.value; }
            set
            {
                base.value = value;
                if (m_Callback != null)
                    m_TextElement.text = m_Callback(m_Value);
                else
                    m_TextElement.text = m_Value.ToString();
            }
        }

        /// <summary>
        /// Callback that will return the string to be set in the field's label.
        /// </summary>
        /// <param name="callback"></param>
        public void SetLabelCallback(Func<T, string> callback)
        {
            m_Callback = callback;
        }

        public PopupField(List<T> choices, T defaultValue) :
            base(choices, defaultValue)
        {
        }

        public PopupField(List<T> choices, int defaultIndex) :
            base(choices, defaultIndex)
        {
        }
    }
#else
    internal class PopupField<T> : BaseTextElement, INotifyValueChanged<T>
    {
        private readonly List<T> m_PossibleValues;
        private Func<T, string> m_Callback;
        private EventCallback<ChangeEvent<T>> m_valueCallback;
        private T m_Value;
        public T value
        {
            get { return m_Value; }
            set
            {
                if (EqualityComparer<T>.Default.Equals(m_Value, value))
                    return;

                if (!m_PossibleValues.Contains(value))
                    throw new ArgumentException(string.Format("Value {0} is not present in the list of possible values", value));

                m_Value = value;
                m_Index = m_PossibleValues.IndexOf(m_Value);
                if (m_Callback != null)
                    text = m_Callback(m_Value);
                else
                    text = m_Value.ToString();
 
#if UNITY_2018_3_OR_NEWER
                MarkDirtyRepaint();
#else
                Dirty(ChangeType.Repaint);
#endif
            }
        }

        private int m_Index = -1;
        public int index
        {
            get { return m_Index; }
            set
            {
                if (value != m_Index)
                {
                    if (value >= m_PossibleValues.Count || value < 0)
                        throw new ArgumentException(string.Format("Index {0} is beyond the scope of possible value", value));
                    m_Index = value;
                    this.value = m_PossibleValues[m_Index];
                }
            }
        }

        /// <summary>
        /// Callback that will return the string to be set in the field's label.
        /// </summary>
        /// <param name="callback"></param>
        public void SetLabelCallback(Func<T, string> callback)
        {
            m_Callback = callback;
        }
        
        private PopupField(List<T> possibleValues)
        {
            if (possibleValues == null)
                throw new ArgumentNullException("possibleValues can't be null");

            m_PossibleValues = possibleValues;

            AddToClassList("popupField");
        }

        public PopupField(List<T> possibleValues, T defaultValue) :
            this(possibleValues)
        {
            if (defaultValue == null)
                throw new ArgumentNullException("defaultValue can't be null");

            if (!m_PossibleValues.Contains(defaultValue))
                throw new ArgumentException(string.Format("Default value {0} is not present in the list of possible values", defaultValue));

            // note: idx will be set when setting value
            value = defaultValue;
        }

        public PopupField(List<T> possibleValues, int defaultIndex) :
            this(possibleValues)
        {
            if (defaultIndex >= m_PossibleValues.Count || defaultIndex < 0)
                throw new ArgumentException(string.Format("Default Index {0} is beyond the scope of possible value", value));

            // note: value will be set when setting idx
            index = defaultIndex;
        }

        public void SetValueAndNotify(T newValue)
        {
            if (!EqualityComparer<T>.Default.Equals(newValue, value))
            {
                using (ChangeEvent<T> evt = ChangeEvent<T>.GetPooled(value, newValue))
                {
                    value = newValue;
                    if (m_valueCallback != null)
                        m_valueCallback(evt);
                }
            }
        }

        public void OnValueChanged(EventCallback<ChangeEvent<T>> callback)
        {
            m_valueCallback = callback;
            RegisterCallback(callback);
        }

        protected override void ExecuteDefaultAction(EventBase evt)
        {
            base.ExecuteDefaultAction(evt);

            if (evt.GetEventTypeId() == MouseDownEvent.TypeId())
                OnMouseDown();
        }

        private void OnMouseDown()
        {
            var menu = new GenericMenu();

            foreach (T item in m_PossibleValues)
            {
                bool isSelected = EqualityComparer<T>.Default.Equals(item, value);
                menu.AddItem(new GUIContent(item.ToString()), isSelected,
                    () => ChangeValueFromMenu(item));
            }

            var menuPosition = new Vector2(0.0f, layout.height);
            menuPosition = this.LocalToWorld(menuPosition);
            var menuRect = new Rect(menuPosition, Vector2.zero);
            menu.DropDown(menuRect);
        }

        private void ChangeValueFromMenu(T menuItem)
        {
            SetValueAndNotify(menuItem);
        }
    }
#endif
}
