Feature/collection generators

Scripts/Editor/CustomEditors/CollectionCustomEditor.cs

@@ -31,10 +31,39 @@ public class CollectionCustomEditor : BaseEditor<CollectionCustomEditor>
         private readonly Dictionary<int, Rect> itemIndexToRect = new();
         private float? reorderableListYPosition;
         private Dictionary<Object, SerializedObject> collectionItemSerializedObjectCache = new();
+
+        [NonSerialized] private Type generatorType;
+        [NonSerialized] private IScriptableObjectCollectionGeneratorBase generator;
+
+        private bool IsAutoGenerated => generatorType != null;
 
         protected virtual bool CanBeReorderable => true;
-        protected virtual bool DisplayAddButton => true;
-        protected virtual bool DisplayRemoveButton => true;
+
+        protected virtual bool DisplayAddButton
+        {
+            get
+            {
+                // If this is a generated collection and it's set to remove items that aren't re-generated, then it
+                // doesn't make sense for you to add items because they will be removed next time you generate.
+                if (IsAutoGenerated && generator.ShouldRemoveNonGeneratedItems)
+                    return false;
+                return true;
+            }
+        }
+
+        protected virtual bool DisplayRemoveButton
+        {
+            get
+            {
+                // If this is a generated collection and it's set to remove items that aren't re-generated, then it
+                // doesn't make sense for you to remove items because they will be added back next time you generate.
+                if (IsAutoGenerated && generator.ShouldRemoveNonGeneratedItems)
+                    return false;
+
+                return true;
+            }
+        }
+
         protected virtual bool AllowCustomTypeCreation => true;
 
         private static bool IsWaitingForNewTypeBeCreated
@@ -51,6 +80,10 @@ protected virtual void OnEnable()
                 CollectionsRegistry.Instance.ReloadCollections();
 
             itemsSerializedProperty = serializedObject.FindProperty("items");
+
+            // Need to cache this before the reorderable list is created, because it affects how the list is displayed.
+            generatorType = CollectionGenerators.GetGeneratorTypeForCollection(collection.GetType());
+            generator = generatorType == null ? null : CollectionGenerators.GetGenerator(generatorType);
 
             ValidateCollectionItems();
 
@@ -560,15 +593,23 @@ private void ValidateCollectionItems()
 
         private void DrawBottomMenu()
         {
-            using (new EditorGUILayout.HorizontalScope("Box"))
+            using (new EditorGUILayout.VerticalScope("Box"))
             {
-                if (GUILayout.Button($"Generate Static Access File", EditorStyles.miniButtonRight))
+                if (GUILayout.Button($"Generate Static Access File", EditorStyles.miniButton))
                 {
                     EditorApplication.delayCall += () =>
                     {
                         CodeGenerationUtility.GenerateStaticCollectionScript(collection);
                     };
                 }
+
+                if (generatorType != null && GUILayout.Button($"Generate Items", GUILayout.Height(40)))
+                {
+                    EditorApplication.delayCall += () =>
+                    {
+                        CollectionGenerators.RunGenerator(generatorType);
+                    };
+                }
             }
         }
 

Scripts/Editor/Extensions/SerializedPropertyExtensions.cs

@@ -1,6 +1,7 @@
 using System;
 using System.Reflection;
 using UnityEditor;
+using UnityEngine;
 
 namespace BrunoMikoski.ScriptableObjectCollections
 {
@@ -23,5 +24,89 @@ public static FieldInfo GetFieldInfoFromPathByType(this SerializedProperty prope
 
             return fieldInfo;
         }
+
+        public static void SetValue(this SerializedProperty serializedProperty, object value)
+        {
+            switch (serializedProperty.propertyType)
+            {
+                case SerializedPropertyType.Integer:
+                    serializedProperty.intValue = (int)value;
+                    break;
+                case SerializedPropertyType.Boolean:
+                    serializedProperty.boolValue = (bool)value;
+                    break;
+                case SerializedPropertyType.Float:
+                    serializedProperty.floatValue = (float)value;
+                    break;
+                case SerializedPropertyType.String:
+                    serializedProperty.stringValue = (string)value;
+                    break;
+                case SerializedPropertyType.Color:
+                    serializedProperty.colorValue = (Color)value;
+                    break;
+                case SerializedPropertyType.ObjectReference:
+                    serializedProperty.objectReferenceValue = (UnityEngine.Object)value;
+                    break;
+                case SerializedPropertyType.LayerMask:
+                    serializedProperty.intValue = (LayerMask)value;
+                    break;
+                case SerializedPropertyType.Enum:
+                    serializedProperty.intValue = (int)value;
+                    break;
+                case SerializedPropertyType.Vector2:
+                    serializedProperty.vector2Value = (Vector2)value;
+                    break;
+                case SerializedPropertyType.Vector3:
+                    serializedProperty.vector3Value = (Vector3)value;
+                    break;
+                case SerializedPropertyType.Vector4:
+                    serializedProperty.vector4Value = (Vector4)value;
+                    break;
+                case SerializedPropertyType.Rect:
+                    serializedProperty.rectValue = (Rect)value;
+                    break;
+                //case SerializedPropertyType.ArraySize:
+                //break;
+                case SerializedPropertyType.Character:
+                    serializedProperty.stringValue = ((char)value).ToString();
+                    break;
+                case SerializedPropertyType.AnimationCurve:
+                    serializedProperty.animationCurveValue = (AnimationCurve)value;
+                    break;
+                case SerializedPropertyType.Bounds:
+                    serializedProperty.boundsValue = (Bounds)value;
+                    break;
+                //case SerializedPropertyType.Gradient:
+                //break;
+                case SerializedPropertyType.Quaternion:
+                    serializedProperty.quaternionValue = (Quaternion)value;
+                    break;
+                //case SerializedPropertyType.ExposedReference:
+                //break;
+                //case SerializedPropertyType.FixedBufferSize:
+                //break;
+                case SerializedPropertyType.Vector2Int:
+                    serializedProperty.vector2IntValue = (Vector2Int)value;
+                    break;
+                case SerializedPropertyType.Vector3Int:
+                    serializedProperty.vector3IntValue = (Vector3Int)value;
+                    break;
+                case SerializedPropertyType.RectInt:
+                    serializedProperty.rectIntValue = (RectInt)value;
+                    break;
+                case SerializedPropertyType.BoundsInt:
+                    serializedProperty.boundsIntValue = (BoundsInt)value;
+                    break;
+                //case SerializedPropertyType.ManagedReference:
+                //break;
+                case SerializedPropertyType.Hash128:
+                    serializedProperty.hash128Value = (Hash128)value;
+                break;
+                default:
+                    Debug.LogWarning(
+                        $"Tried to copy value '{value}' from a template to an SOC item but apparently that's not supported.");
+                    break;
+            }
+        }
     }
 }