feat: Added ability to assign the created asset after assembly recompilation

Author: SolidAlloy

Editor/ScriptableObjects/GenericSOCreator.cs

@@ -1,7 +1,6 @@
 namespace GenericUnityObjects.Editor.ScriptableObjects
 {
     using System;
-    using System.Linq;
     using Cysharp.Threading.Tasks;
     using GenericUnityObjects;
     using GenericUnityObjects.Util;
@@ -60,7 +59,15 @@ async UniTaskVoid CreateAssetImpl()
                     (Type genericSOType, string path) = PersistentStorage.GetGenericSODetails();
                     var concreteType = BehavioursDatabase.GetConcreteType(genericSOType);
                     await WaitUntilEditorInitialized();
-                    CreateAssetFromConcreteType(concreteType, asset => AssetDatabase.CreateAsset(asset, path));
+                    var createdAsset = CreateAssetFromConcreteType(concreteType, asset => AssetDatabase.CreateAsset(asset, path));
+
+                    var property = PersistentStorage.GetSavedProperty();
+
+                    if (property == null)
+                        return;
+
+                    property.objectReferenceValue = createdAsset;
+                    property.serializedObject.ApplyModifiedProperties();
                 }
                 finally
                 {
@@ -99,18 +106,20 @@ public static void CreateAssetInteractively(Type genericTypeWithoutArgs, Type[]
             AssetDatabase.Refresh();
         }
 
-        public static ScriptableObject CreateAssetAtPath(Type genericType, string path)
+        public static ScriptableObject CreateAssetAtPath(SerializedProperty property, Type genericType, string path)
         {
             if (ScriptableObjectsDatabase.TryGetConcreteType(genericType, out var concreteType))
             {
                 return CreateAssetFromConcreteType(concreteType, asset => AssetDatabase.CreateAsset(asset, path));
             }
 
             PersistentStorage.SaveForScriptsReload(genericType, path);
+            PersistentStorage.SaveForScriptsReload(property);
             PersistentStorage.ExecuteOnScriptsReload(FinishSOCreationAtPath);
 
             ConcreteClassCreator<GenericScriptableObject>.CreateConcreteClass(genericType.GetGenericTypeDefinition(), genericType.GenericTypeArguments);
             AssetDatabase.Refresh();
+            // At this point the domain is reloaded, so the method doesn't return.
             return null;
         }
 

Editor/Util/PersistentStorage.cs

@@ -21,6 +21,9 @@ internal class PersistentStorage : EditorOnlySingletonSO<PersistentStorage>, ICa
         [SerializeField] private GameObject _gameObject;
         [SerializeField] private TypeReference _genericBehaviourType;
 
+        [SerializeField] private int _instanceID;
+        [SerializeField] private string _propertyPath;
+
         [SerializeField] private MenuItemMethod[] _menuItemMethods = { };
 
         public static MenuItemMethod[] MenuItemMethods
@@ -76,6 +79,12 @@ public static void SaveForScriptsReload(Type genericTypeToCreate, string fileNam
             Instance._fileName = fileName;
         }
 
+        public static void SaveForScriptsReload(SerializedProperty property)
+        {
+            Instance._instanceID = property.serializedObject.targetObject.GetInstanceID();
+            Instance._propertyPath = property.propertyPath;
+        }
+
         public static void SaveForScriptsReload(GameObject gameObject, Type genericType)
         {
             Instance._gameObject = gameObject;
@@ -92,12 +101,31 @@ public static (GameObject gameObject, Type genericBehaviourType) GetGenericBehav
             return (Instance._gameObject, Instance._genericBehaviourType);
         }
 
+        public static SerializedProperty GetSavedProperty()
+        {
+            var targetObject = EditorUtility.InstanceIDToObject(Instance._instanceID);
+
+            if (targetObject == null)
+                return null;
+
+            var serializedObject = new SerializedObject(targetObject);
+            return serializedObject.FindProperty(Instance._propertyPath);
+        }
+
+        public static void DisableFirstCompilation()
+        {
+            Instance._firstCompilation = false;
+            EditorUtility.SetDirty(Instance);
+        }
+
         public static void Clear()
         {
             Instance._genericSOType = null;
             Instance._fileName = null;
             Instance._gameObject = null;
             Instance._genericBehaviourType = null;
+            Instance._instanceID = 0;
+            Instance._propertyPath = null;
         }
 
         public void Initialize() { }