From d72dde6886bb6e3087631711dffa6a3983d441fd Mon Sep 17 00:00:00 2001
From: Cyberboss <Cyberboss@users.noreply.github.com>
Date: Wed, 18 Oct 2017 13:00:31 -0400
Subject: [PATCH] Reflection method for determining valid interfaces

---
 TGServiceInterface/Server.cs | 25 ++++++++++++++++++++++---
 1 file changed, 22 insertions(+), 3 deletions(-)

diff --git a/TGServiceInterface/Server.cs b/TGServiceInterface/Server.cs
index a0bf48284b..0a09c88ed1 100644
--- a/TGServiceInterface/Server.cs
+++ b/TGServiceInterface/Server.cs
@@ -1,10 +1,13 @@
 using System;
 using System.Collections.Generic;
 using System.Diagnostics;
+using System.Linq;
 using System.Net;
 using System.Net.Security;
+using System.Reflection;
 using System.Security.Principal;
 using System.ServiceModel;
+
 namespace TGServiceInterface
 {
 	public class Server
@@ -12,7 +15,7 @@ namespace TGServiceInterface
 		/// <summary>
 		/// List of types that can be used with GetComponen
 		/// </summary>
-		public static readonly IList<Type> ValidInterfaces = new List<Type> { typeof(ITGByond), typeof(ITGChat), typeof(ITGCompiler), typeof(ITGConfig), typeof(ITGDreamDaemon), typeof(ITGRepository), typeof(ITGSService), typeof(ITGConnectivity), typeof(ITGAdministration), typeof(ITGInterop) };
+		public static readonly IList<Type> ValidInterfaces = CollectComponents();
 
 		/// <summary>
 		/// The maximum message size to and from a local server 
@@ -51,6 +54,22 @@ namespace TGServiceInterface
 		static string HTTPSPassword;
 
 		static Dictionary<Type, ChannelFactory> ChannelFactoryCache = new Dictionary<Type, ChannelFactory>();
+
+		/// <summary>
+		/// Returns a <see cref="IList{T}"/> of <see langword="interface"/> <see cref="Type"/>s that can be used with the service
+		/// </summary>
+		/// <returns>A <see cref="IList{T}"/> of <see langword="interface"/> <see cref="Type"/>s that can be used with the service</returns>
+		static IList<Type> CollectComponents()
+		{
+			//find all interfaces in this assembly in this namespace that have the service contract attribute
+			var query = from t in Assembly.GetExecutingAssembly().GetTypes()
+						where t.IsInterface 
+						&& t.Namespace == typeof(Server).Namespace
+						&& t.GetCustomAttribute(typeof(ServiceContractAttribute)) != null
+						select t;
+			return query.ToList();
+		}
+
 		public static void SetBadCertificateHandler(Func<string, bool> handler)
 		{
 			ServicePointManager.ServerCertificateValidationCallback = (sender, cert, chain, error) =>
@@ -97,8 +116,8 @@ namespace TGServiceInterface
 
 		public static bool VersionMismatch(out string errorMessage)
 		{
-			var splits = Server.GetComponent<ITGSService>().Version().Split(' ');
-			var theirs = new Version(splits[splits.Length - 1].Substring(1));
+			var splits = Server.GetComponent<ITGSService>().Version().Split(' ');
+			var theirs = new Version(splits[splits.Length - 1].Substring(1));
 			var ours = new Version(FileVersionInfo.GetVersionInfo(System.Reflection.Assembly.GetExecutingAssembly().Location).FileVersion);
 			if(theirs != ours)
 			{