using System;
using System.Collections.Generic;
using System.Linq;
using System.Reflection;
using System.ComponentModel;

namespace Microsoft.Windows.Controls.PropertyGrid.Editors
{
    public class EnumComboBoxEditor : ComboBoxEditor
    {
        protected override IList<object> CreateItemsSource(PropertyItem propertyItem)
        {
            return GetValues(propertyItem.PropertyType);
        }

        private static object[] GetValues(Type enumType)
        {
            List<object> values = new List<object>();            

            var fields = enumType.GetFields().Where(x => x.IsLiteral);
            foreach (FieldInfo field in fields)
            {
                // Get array of BrowsableAttribute attributes
                object[] attrs = field.GetCustomAttributes(typeof(BrowsableAttribute), false);
                if (attrs.Length == 1)
                {
                    // If attribute exists and its value is false continue to the next field...
                    BrowsableAttribute brAttr = (BrowsableAttribute)attrs[0];
                    if (brAttr.Browsable == false) continue;
                }

                values.Add(field.GetValue(enumType));
            }

            return values.ToArray();
        }
    }
}