init

Common/DTS.CommonCore/.svn/pristine/b8/b87e3700d1c318e691d8f2a2e5bb2fdd24ca848f.svn-base

@@ -0,0 +1,54 @@
+using System;
+using System.Windows.Markup;
+
+namespace DTS.Common.Enums
+{
+    public class EnumBindingSourceExtension : MarkupExtension
+    {
+        private Type _enumType;
+        public Type EnumType
+        {
+            get => _enumType;
+            set
+            {
+                if (value == _enumType) { return; }
+                if (null != value)
+                {
+                    var enumType = Nullable.GetUnderlyingType(value) ?? value;
+                    if (!enumType.IsEnum)
+                    {
+                        throw new ArgumentException("Type must be for an Enum.");
+                    }
+                }
+                _enumType = value;
+            }
+        }
+
+        public EnumBindingSourceExtension() { }
+
+        public EnumBindingSourceExtension(Type enumType)
+        {
+            EnumType = enumType;
+        }
+
+        public override object ProvideValue(IServiceProvider serviceProvider)
+        {
+            if (null == _enumType)
+            {
+                throw new InvalidOperationException("The EnumType must be specified.");
+            }
+
+            var actualEnumType = Nullable.GetUnderlyingType(_enumType) ?? _enumType;
+            var enumValues = Enum.GetValues(actualEnumType);
+
+            if (actualEnumType == _enumType)
+            {
+                return enumValues;
+            }
+
+            var tempArray = Array.CreateInstance(actualEnumType, enumValues.Length + 1);
+            enumValues.CopyTo(tempArray, 1);
+            return tempArray;
+        }
+    }
+}