aistudio-wpf-diagram/Others/Dragablz/Dragablz/Core/CollectionTeaser.cs

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

namespace Dragablz.Core
{
    internal class CollectionTeaser
    {
        private readonly Action<object> _addMethod;
        private readonly Action<object> _removeMethod;

        private CollectionTeaser(Action<object> addMethod, Action<object> removeMethod)
        {            
            _addMethod = addMethod;
            _removeMethod = removeMethod;
        }

        public static bool TryCreate(object items, out CollectionTeaser collectionTeaser)
        {
            collectionTeaser = null;

            var list = items as IList;
            if (list != null)
            {
                collectionTeaser = new CollectionTeaser(i => list.Add(i), list.Remove);                
            }
            else if (items != null)
            {
                var itemsType = items.GetType();
                var genericCollectionType = typeof (ICollection<>);

                //TODO, *IF* we really wanted to we could get the consumer to inform us of the correct type
                //if there are multiple impls.  havent got time for this edge case right now though
                var collectionImplType = itemsType.GetInterfaces().SingleOrDefault(x =>
                    x.IsGenericType &&
                    x.GetGenericTypeDefinition() == genericCollectionType);

                if (collectionImplType != null)
                {
                    var genericArgType = collectionImplType.GetGenericArguments().First();

                    var addMethodInfo = collectionImplType.GetMethod("Add", new[] {genericArgType});
                    var removeMethodInfo = collectionImplType.GetMethod("Remove", new[] { genericArgType });

                    collectionTeaser = new CollectionTeaser(
                        i => addMethodInfo.Invoke(items, new[] {i}),
                        i => removeMethodInfo.Invoke(items, new[] {i}));
                }
            }
            
            return collectionTeaser != null;
        }

        public void Add(object item)
        {
            _addMethod(item);
        }

        public void Remove(object item)
        {
            _removeMethod(item);
        }
    }
}