Svelto.ECS/Svelto.ECS/EntityBuilder.CheckFields.cs

#if !DEBUG || PROFILER
#define DISABLE_CHECKS
using System.Diagnostics;
#endif    
using System;
using System.Reflection;

namespace Svelto.ECS
{
    public static class EntityBuilderUtilities
    {
#if DISABLE_CHECKS        
        [Conditional("_CHECKS_DISABLED")]
#endif
        public static void CheckFields(Type type, bool needsReflection)
        {
            if (type == ENTITY_STRUCT_INFO_VIEW || type == EGIDType || type == ECLUSIVEGROUPSTRUCTTYPE) 
                return;

            if (needsReflection == false)
            {
                if (type.IsClass)
                    throw new EntityStructException("EntityStructs must be structs.", ENTITY_VIEW_TYPE, type);

                var fields = type.GetFields(BindingFlags.Public | BindingFlags.Instance);

                for (int i = fields.Length - 1; i >= 0; --i)
                {
                    var field = fields[i];
                    var fieldFieldType = field.FieldType;
                    
                    if (fieldFieldType == STRINGTYPE) continue;

                    SubCheckFields(fieldFieldType);
                }
            }
            else
            {
                var fields = type.GetFields(BindingFlags.Public | BindingFlags.Instance);

                if (fields.Length < 1)
                    ProcessError("Entity View Structs must hold only entity components interfaces.", type);
                
                for (int i = fields.Length - 1; i >= 0; --i)
                {
                    var field = fields[i];
                    
                    if (field.FieldType.IsInterfaceEx() == false)
                        ProcessError("Entity View Structs must hold only entity components interfaces.", type);
                    
                    var properties = field.FieldType.GetProperties(BindingFlags.Public |
                                                        BindingFlags.Instance | BindingFlags.DeclaredOnly);

                    for (int j = properties.Length - 1; j >= 0; --j)
                    {
                        if (properties[j].PropertyType.IsGenericType == true)
                        {
                            var genericTypeDefinition = properties[j].PropertyType.GetGenericTypeDefinition();
                            if (genericTypeDefinition == DISPATCHONSETTYPE ||
                                genericTypeDefinition == DISPATCHONCHANGETYPE) continue;
                        }

                        var propertyType = properties[j].PropertyType;
                        if (propertyType != STRINGTYPE)
                            SubCheckFields(propertyType);
                    }
                }
            }
        }

        static void SubCheckFields(Type fieldFieldType)
        {
            if (fieldFieldType.IsPrimitive == true || fieldFieldType.IsValueType == true)
            {
                if (fieldFieldType.IsValueType == true && !fieldFieldType.IsEnum && fieldFieldType.IsPrimitive == false)
                {
                    CheckFields(fieldFieldType, false);
                }

                return;
            }
            
            ProcessError("Entity Structs field and Entity View Struct components must hold value types.", 
                         fieldFieldType);
        }
        
        static void ProcessError(string message, Type type)
        {
#if !RELAXED_ECS
            Type ENTITY_VIEW_TYPE = typeof(Type);
            throw new EntityStructException(message, ENTITY_VIEW_TYPE, type);
#endif
        }
        
        static readonly Type EGIDType                = typeof(EGID);
        static readonly Type ECLUSIVEGROUPSTRUCTTYPE = typeof(ExclusiveGroup.ExclusiveGroupStruct);
        static readonly Type DISPATCHONSETTYPE       = typeof(DispatchOnSet<>);
        static readonly Type DISPATCHONCHANGETYPE    = typeof(DispatchOnChange<>);
        static readonly Type STRINGTYPE              = typeof(String);
        static readonly Type ENTITY_VIEW_TYPE        = typeof(Type);
        static readonly Type ENTITY_STRUCT_INFO_VIEW  = typeof(EntityStructInfoView);
    }
    
    public class EntityStructException : Exception
    {
        public EntityStructException(string message, Type entityViewType, Type type):
            base(message.FastConcat(" entity view: ", entityViewType.ToString(), " field: ", type.ToString()))
        {}
    }
}