From 1ac1ac29651c00c2be3064599f38e5f65a09a6d7 Mon Sep 17 00:00:00 2001
From: Chris Cameron <chris.cameron@wedoresi.com>
Date: Wed, 7 Feb 2018 16:56:18 -0500
Subject: [PATCH] GetCustomAttribute/s extension methods

---
 .../Extensions/ReflectionExtensions.cs        | 46 +++++++++++++++++++
 ICD.Common.Utils/ReflectionUtils.cs           |  5 +-
 2 files changed, 47 insertions(+), 4 deletions(-)

diff --git a/ICD.Common.Utils/Extensions/ReflectionExtensions.cs b/ICD.Common.Utils/Extensions/ReflectionExtensions.cs
index f22b19b..79f344b 100644
--- a/ICD.Common.Utils/Extensions/ReflectionExtensions.cs
+++ b/ICD.Common.Utils/Extensions/ReflectionExtensions.cs
@@ -14,6 +14,21 @@ namespace ICD.Common.Utils.Extensions
 	/// </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>(this ICustomAttributeProvider extends)
+			where T : Attribute
+		{
+			if (extends == null)
+				throw new ArgumentNullException("extends");
+
+			return extends.GetCustomAttributes<T>(false);
+		}
+
 		/// <summary>
 		/// Returns the custom attributes attached to the member.
 		/// </summary>
@@ -29,5 +44,36 @@ namespace ICD.Common.Utils.Extensions
 
 			return extends.GetCustomAttributes(typeof(T), inherits).Cast<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>(this ICustomAttributeProvider extends)
+			where T : Attribute
+		{
+			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>(this ICustomAttributeProvider extends, bool inherits)
+			where T : Attribute
+		{
+			if (extends == null)
+				throw new ArgumentNullException("extends");
+
+			return extends.GetCustomAttributes<T>(inherits).First();
+		}
 	}
 }
diff --git a/ICD.Common.Utils/ReflectionUtils.cs b/ICD.Common.Utils/ReflectionUtils.cs
index 82ab92c..0aaeab0 100644
--- a/ICD.Common.Utils/ReflectionUtils.cs
+++ b/ICD.Common.Utils/ReflectionUtils.cs
@@ -2,6 +2,7 @@
 using System.Collections.Generic;
 using System.Linq;
 using ICD.Common.Properties;
+using ICD.Common.Utils.Extensions;
 using ICD.Common.Utils.IO;
 #if SIMPLSHARP
 using Crestron.SimplSharp.CrestronIO;
@@ -251,9 +252,6 @@ namespace ICD.Common.Utils
 			if (assembly == null)
 				throw new ArgumentNullException("assembly");
 
-#if SIMPLSHARP
-			return assembly.GetCustomAttributes(typeof(T), false).Cast<T>();
-#else
 			try
 			{
 				return assembly.GetCustomAttributes<T>();
@@ -262,7 +260,6 @@ namespace ICD.Common.Utils
 			{
 				return Enumerable.Empty<T>();
 			}
-#endif
 		}
 
 		/// <summary>