ICD.Common.Utils/ICD.Common.Utils/Extensions/ReflectionExtensions.cs

using System;
using System.Collections.Generic;
using System.Linq;
using ICD.Common.Properties;
#if SIMPLSHARP
using Crestron.SimplSharp.Reflection;
#else
using System.Reflection;
#endif

namespace ICD.Common.Utils.Extensions
{
	/// <summary>
	/// Extension methods for use with reflection objects.
	/// </summary>
	public static class ReflectionExtensions
	{
		/// <summary>
		/// Returns the custom attributes attached to the member.
		/// </summary>
		/// <typeparam name="T"></typeparam>
		/// <param name="extends"></param>
		/// <returns></returns>
		public static IEnumerable<T> GetCustomAttributes<T>([NotNull] this ICustomAttributeProvider extends)
		{
			if (extends == null)
				throw new ArgumentNullException("extends");

			return extends.GetCustomAttributes<T>(false);
		}

		/// <summary>
		/// Returns the custom attributes attached to the member.
		/// </summary>
		/// <typeparam name="T"></typeparam>
		/// <param name="extends"></param>
		/// <param name="inherits"></param>
		/// <returns></returns>
		public static IEnumerable<T> GetCustomAttributes<T>([NotNull] this ICustomAttributeProvider extends,
		                                                    bool inherits)
		{
			if (extends == null)
				throw new ArgumentNullException("extends");

			try
			{
				return extends.GetCustomAttributes(typeof(T), inherits).Cast<T>();
			}
			// Crestron bug?
			catch (ArgumentNullException)
			{
				return Enumerable.Empty<T>();
			}
		}

		/// <summary>
		/// Returns the custom attribute attached to the member.
		/// </summary>
		/// <typeparam name="T"></typeparam>
		/// <param name="extends"></param>
		/// <returns></returns>
		public static T GetCustomAttribute<T>([NotNull] this ICustomAttributeProvider extends)
		{
			if (extends == null)
				throw new ArgumentNullException("extends");

			return extends.GetCustomAttribute<T>(false);
		}

		/// <summary>
		/// Returns the custom attribute attached to the member.
		/// </summary>
		/// <typeparam name="T"></typeparam>
		/// <param name="extends"></param>
		/// <param name="inherits"></param>
		/// <returns></returns>
		public static T GetCustomAttribute<T>([NotNull] this ICustomAttributeProvider extends, bool inherits)
		{
			if (extends == null)
				throw new ArgumentNullException("extends");

			return extends.GetCustomAttributes<T>(inherits).First();
		}

#if NETSTANDARD
		/// <summary>
		/// Returns the custom attributes attached to the member.
		/// </summary>
		/// <typeparam name="T"></typeparam>
		/// <param name="extends"></param>
		/// <returns></returns>
		public static IEnumerable<T> GetCustomAttributesIncludingBaseInterfaces<T>([NotNull] this Type extends)
			where T : Attribute
		{
			if (extends == null)
				throw new ArgumentNullException("extends");

			return extends.GetCustomAttributes<T>(true)
			              .Union(extends.GetInterfaces()
			                            .SelectMany(interfaceType => interfaceType
				                                        .GetCustomAttributes<T>(true)))
			              .Distinct();
		}

		/// <summary>
		/// Returns the custom attributes attached to the member.
		/// </summary>
		/// <typeparam name="T"></typeparam>
		/// <param name="extends"></param>
		/// <returns></returns>
		public static IEnumerable<T> GetCustomAttributesIncludingBaseInterfaces<T>([NotNull] this MemberInfo extends)
			where T : Attribute
		{
			if (extends == null)
				throw new ArgumentNullException("extends");

			return extends.GetCustomAttributes<T>(true)
			              .Union(extends.DeclaringType?
			                            .GetInterfaces()
			                            .SelectMany(interfaceType => interfaceType
			                                                         .GetMember(extends.Name,
			                                                                    extends.MemberType,
			                                                                    BindingFlags.Instance)
			                                                         .FirstOrDefault()?
			                                                         .GetCustomAttributes<T>(true) ??
			                                                         Enumerable.Empty<T>())?
			                            .Except(null) ?? Enumerable.Empty<T>())
			              .Distinct();
		}

		/// <summary>
		/// Sets the value of a property
		/// Traverses the path to access properties nested in other properties
		/// </summary>
		/// <param name="extends"></param>
		/// <param name="value"></param>
		/// <param name="path"></param>
		/// <returns>true if property was set, false if property was not found</returns>
		public static bool SetProperty([NotNull] this object extends, [CanBeNull] object value, [NotNull] params string[] path)
		{
			if (extends == null)
				throw new ArgumentNullException(nameof(extends));
			if (path == null)
				throw new ArgumentNullException(nameof(path));

			object currentObject = extends;

			//Grab property values until the last item in the path
			for (int i = 0; i < path.Length - 1; i++)
			{
				PropertyInfo info = extends.GetType().GetProperty(path[i]);
				if (info == null)
					return false;
				currentObject = info.GetValue(currentObject);
			}

			//Set the property to the value
			PropertyInfo finalPath = currentObject.GetType().GetProperty(path[path.Length - 1]);
			if (finalPath == null)
				return false;

			finalPath.SetValue(currentObject, value);
			return true;
		}

		/// <summary>
		/// Gets the value of a property
		/// Traverses the path to access properties nested in other properties
		/// </summary>
		/// <param name="extends"></param>
		/// <param name="value"></param>
		/// <param name="path"></param>
		/// <returns>true if property was set, false if property was not found</returns>
		public static bool GetProperty([NotNull] this object extends, [CanBeNull] out object value, [NotNull] params string[] path)
		{
			if (extends == null)
				throw new ArgumentNullException(nameof(extends));
			if (path == null)
				throw new ArgumentNullException(nameof(path));

			value = null;

			object currentObject = extends;

			//Grab property values
			foreach (string node in path)
			{
				PropertyInfo info = extends.GetType().GetProperty(node);
				if (info == null)
					return false;
				currentObject = info.GetValue(currentObject);
			}

			//set the last value and return
			value = currentObject;
			return true;
		}
#endif

		/// <summary>
		/// Gets the EventArgs Type for the given event.
		/// </summary>
		/// <param name="extends"></param>
		/// <returns></returns>
		public static Type GetEventArgsType([NotNull] this EventInfo extends)
		{
			if (extends == null)
				throw new ArgumentNullException("extends");

			Type eventHandlerType = extends.EventHandlerType;
			return eventHandlerType == typeof(EventHandler)
				       ? typeof(EventArgs)
				       : eventHandlerType.GetInnerGenericTypes(typeof(EventHandler<>))
				                         .First();
		}
	}
}