message improvements

TODO

@@ -1,2 +1,2 @@
 - look at test coverage
-- docs
+- docs

encompass-cs/MessageManager.cs

@@ -8,28 +8,25 @@ namespace Encompass
     {
         private readonly Dictionary<Type, List<IMessage>> messageTypeToMessages = new Dictionary<Type, List<IMessage>>();
 
-        internal void AddMessage<TMessage>(TMessage message) where TMessage : struct, IMessage
+        internal void RegisterMessageType(Type messageType)
         {
-            if (!messageTypeToMessages.ContainsKey(typeof(TMessage)))
+            if (!messageTypeToMessages.ContainsKey(messageType)) {
-            {
+                messageTypeToMessages.Add(messageType, new List<IMessage>());
-                messageTypeToMessages.Add(typeof(TMessage), new List<IMessage>());
             }
-
-            messageTypeToMessages[typeof(TMessage)].Add(message);
         }
 
         internal void AddMessage(IMessage message)
         {
             var type = message.GetType();
 
-            if (!messageTypeToMessages.ContainsKey(type))
-            {
-                messageTypeToMessages.Add(type, new List<IMessage>());
-            }
-
             messageTypeToMessages[type].Add(message);
         }
 
+        internal void AddMessage<TMessage>(IMessage message) where TMessage : struct, IMessage
+        {
+            messageTypeToMessages[typeof(TMessage)].Add(message);
+        }
+
         internal void ClearMessages()
         {
             foreach (var entry in messageTypeToMessages)

encompass-cs/WorldBuilder.cs

@@ -86,6 +86,11 @@ namespace Encompass
             var messageReceiveTypes = engine.receiveTypes;
             var messageSendTypes = engine.sendTypes;
 
+            foreach (var messageType in messageReceiveTypes.Union(messageSendTypes))
+            {
+                messageManager.RegisterMessageType(messageType);
+            }
+
             foreach (var messageType in messageReceiveTypes.Intersect(messageSendTypes))
             {
                 if ((messageType.IsGenericType && messageType.GetGenericTypeDefinition() == typeof(PendingComponentMessage<>)))