using System.Collections;
using System.Collections.Generic;

namespace UnityEditor.ShaderGraph
{
    [GenerationAPI]
    internal class DefineCollection : IEnumerable<DefineCollection.Item>
    {
        public class Item : IConditional
        {
            public KeywordDescriptor descriptor { get; }
            public FieldCondition[] fieldConditions { get; }
            public string value => descriptor.ToDefineString(index);
            // KeywordType.Boolean, index 0: disable, 1: enable
            // KeywordType.Enum, index into enum entries
            public int index { get; }

            public Item(KeywordDescriptor descriptor, int index, FieldCondition[] fieldConditions)
            {
                this.descriptor = descriptor;
                this.fieldConditions = fieldConditions;
                this.index = index;
            }
        }

        readonly List<Item> m_Items;

        public DefineCollection()
        {
            m_Items = new List<Item>();
        }

        public DefineCollection(DefineCollection defines)
        {
            m_Items = new List<Item>();

            foreach (DefineCollection.Item item in defines)
            {
                m_Items.Add(item);
            }
        }

        public DefineCollection Add(DefineCollection defines)
        {
            foreach (DefineCollection.Item item in defines)
            {
                m_Items.Add(item);
            }

            return this;
        }

        public DefineCollection Add(KeywordDescriptor descriptor, int index)
        {
            m_Items.Add(new Item(descriptor, index, null));
            return this;
        }

        public DefineCollection Add(KeywordDescriptor descriptor, int index, FieldCondition fieldCondition)
        {
            m_Items.Add(new Item(descriptor, index, new FieldCondition[] { fieldCondition }));
            return this;
        }

        public DefineCollection Add(KeywordDescriptor descriptor, int index, FieldCondition[] fieldConditions)
        {
            m_Items.Add(new Item(descriptor, index, fieldConditions));
            return this;
        }

        public IEnumerator<Item> GetEnumerator()
        {
            return m_Items.GetEnumerator();
        }

        IEnumerator IEnumerable.GetEnumerator()
        {
            return GetEnumerator();
        }
    }
}