Setting variable on entity instance now looks to Entity's variable fo…

…r the type fixes #1302
vchelaru · Jan 9, 2024 · 0d1e4d9 · 0d1e4d9
1 parent 092799b
commit 0d1e4d9
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/FRBDK/Glue/Glue/SaveClasses/NamedObjectSaveExtensionMethods.cs b/FRBDK/Glue/Glue/SaveClasses/NamedObjectSaveExtensionMethods.cs
@@ -510,8 +510,20 @@ public static void SetVariable(this NamedObjectSave instance, string variableNam
                 }
                 else
                 {
-                    var type = value?.GetType();
-                    instruction = instance.AddNewGenericInstructionFor(variableName, type);
+                    // If it comes from an entity, try to assign the type from the entity. This is needed if the variable
+                    // is an Entity.Type property
+                    var nosEntity = ObjectFinder.Self.GetEntitySave(instance);
+                    var variable = nosEntity?.GetCustomVariableRecursively(variableName);
+
+                    if(variable != null)
+                    {
+                        instruction = instance.AddInstruction(variableName, variable.Type);
+                    }
+                    else
+                    {
+                        var type = value?.GetType();
+                        instruction = instance.AddNewGenericInstructionFor(variableName, type);
+                    }
                 }
             }