Change host component back to field

Author: D3TONAT0R

Editor/PropertyDrawers/ComponentStackDrawer.cs

@@ -221,20 +221,27 @@ private void FixDuplicateReferences(SerializedProperty stack)
 
 		private float DrawItem(Rect position, SerializedProperty arrayProp, SerializedProperty stackProp, int i)
 		{
-			var elem = arrayProp.GetArrayElementAtIndex(i);
-			var obj = elem.GetValue() as StackComponent;
-			if(obj != null) elem.FindPropertyRelative("hostComponent").objectReferenceValue = stackProp.serializedObject.targetObject as MonoBehaviour;
-			var drawer = GetDrawerFor(obj?.GetType() ?? typeof(StackElementDrawer));
+			try
+			{
+				var elem = arrayProp.GetArrayElementAtIndex(i);
+				var obj = elem.GetValue() as StackComponent;
+				if(obj != null) elem.FindPropertyRelative("hostComponent").objectReferenceValue = stackProp.serializedObject.targetObject as MonoBehaviour;
+				var drawer = GetDrawerFor(obj?.GetType() ?? typeof(StackElementDrawer));
 
-			position.height = ItemHeaderHeight;
+				position.height = ItemHeaderHeight;
 
-			drawer.OnHeaderGUI(position, i, elem, obj, stackProp, arrayProp);
+				drawer.OnHeaderGUI(position, i, elem, obj, stackProp, arrayProp);
 
-			position.xMin += 2;
-			position.xMax -= 2;
-			if(elem.isExpanded && obj != null)
+				position.xMin += 2;
+				position.xMax -= 2;
+				if(elem.isExpanded && obj != null)
+				{
+					drawer.OnGUI(ref position, elem, obj, stackProp);
+				}
+			}
+			catch (Exception e)
 			{
-				drawer.OnGUI(ref position, elem, obj, stackProp);
+				e.LogException();
 			}
 			return position.yMax;
 		}

Runtime/Collections/ComponentStack.cs

@@ -13,7 +13,9 @@ public abstract class ComponentStack<T> : IEnumerable<T> where T : StackComponen
 		[SerializeField, SerializeReference]
 		protected List<T> stack = new List<T>();
 
-		public MonoBehaviour HostComponent { get; private set; }
+		[SerializeField, HideInInspector]
+		private MonoBehaviour hostComponent;
+		public MonoBehaviour HostComponent => hostComponent;
 
 		public T this[int index]
 		{
@@ -23,9 +25,9 @@ public T this[int index]
 
 		public int Count => stack.Count;
 
-		public void SetHost(MonoBehaviour hostComponent)
+		public void SetHost(MonoBehaviour host)
 		{
-			HostComponent = hostComponent;
+			hostComponent = host;
 		}
 
 		public void Add(T item) => stack.Add(item);