More patcher tests + unit type reassign support

core/src/mindustry/mod/ContentParser.java

@@ -742,7 +742,7 @@ public class ContentParser{
         }
     );
 
-    private Prov<Unit> unitType(JsonValue value){
+    Prov<Unit> unitType(JsonValue value){
         if(value == null) return UnitEntity::create;
         return switch(value.asString()){
             case "flying" -> UnitEntity::create;

core/src/mindustry/mod/ContentPatcher.java

@@ -81,6 +81,11 @@ public class ContentPatcher{
                 currentlyApplying = null;
 
             }catch(Exception e){
+                PatchSet set = new PatchSet(patch, new JsonValue("error"));
+                set.error = true;
+                set.warnings.add(Strings.getSimpleMessage(e));
+                patches.add(set);
+
                 Log.err("Failed to apply patch: " + patch, e);
             }
         }
@@ -283,10 +288,10 @@ public class ContentPatcher{
 
             var fields = parser.getJson().getFields(actualType);
             var fdata = fields.get(field);
+            var fobj = object;
             if(fdata != null){
                 if(checkField(fdata.field)) return;
 
-                var fobj = object;
                 assignValue(object, field, new FieldData(fdata), () -> Reflect.get(fobj, fdata.field), fv -> {
                     if(fv == null && !fdata.field.isAnnotationPresent(Nullable.class) && !(Vars.headless && ContentParser.implicitNullable.contains(fdata.field.getType()))){
                         warn("Field '@' cannot be null.", fdata.field);
@@ -311,6 +316,9 @@ public class ContentPatcher{
                     Log.err(e);
                     warn("Failed to read consumers for '@': @", bl, Strings.getSimpleMessage(e));
                 }
+            }else if(value instanceof JsonValue jsv && object instanceof UnitType && field.equals("type")){
+                var fmeta = fields.get("constructor");
+                assignValue(object, "constructor", new FieldData(fmeta), () -> Reflect.get(fobj, fmeta.field), val -> Reflect.set(fobj, fmeta.field, val), parser.unitType(jsv), true);
             }else{
                 warn("Unknown field '@' for class '@'", field, actualType.getSimpleName());
             }
@@ -320,6 +328,7 @@ public class ContentPatcher{
     void assignValue(Object object, String field, FieldData metadata, Prov getter, Cons setter, Object value, boolean modify) throws Exception{
         Object prevValue = getter.get();
 
+        try{
             if(value instanceof JsonValue jsv){ //setting values from object
                if(prevValue == null || !jsv.isObject() || jsv.has("type")){
                     if(UnlockableContent.class.isAssignableFrom(metadata.type) && jsv.isObject()){
@@ -352,6 +361,9 @@ public class ContentPatcher{
 
                 setter.get(value);
             }
+        }catch(Throwable e){
+            warn("Failed to assign @.@ = @: @", object, field, value, Strings.getStackTrace(e));
+        }
     }
 
     /**
@@ -476,6 +488,7 @@ public class ContentPatcher{
         public String patch;
         public JsonValue json;
         public String name;
+        public boolean error;
         public Seq<String> warnings = new Seq<>();
 
         public PatchSet(String patch, JsonValue json){

tests/src/test/java/PatcherTests.java

@@ -3,6 +3,7 @@ import mindustry.*;
 import mindustry.content.*;
 import mindustry.entities.abilities.*;
 import mindustry.entities.bullet.*;
+import mindustry.gen.*;
 import mindustry.type.*;
 import mindustry.world.blocks.units.*;
 import mindustry.world.meta.*;
@@ -197,6 +198,20 @@ public class PatcherTests{
         assertEquals(oldLength, UnitTypes.dagger.targetFlags.length);
     }
 
+    @Test
+    void testUnitType() throws Exception{
+        Vars.state.patcher.apply(Seq.with("""
+        unit.dagger.type: legs
+        """));
+
+        assertEquals(0, Vars.state.patcher.patches.first().warnings.size);
+        assertEquals(LegsUnit.class, UnitTypes.dagger.constructor.get().getClass());
+
+        Vars.logic.reset();
+
+        assertEquals(MechUnit.class, UnitTypes.dagger.constructor.get().getClass());
+    }
+
     @Test
     void testCannotPatch() throws Exception{
         Vars.state.patcher.apply(Seq.with("""
@@ -207,6 +222,17 @@ public class PatcherTests{
         assertEquals(1, Blocks.conveyor.size);
     }
 
+    @Test
+    void testGibberish() throws Exception{
+        Vars.state.patcher.apply(Seq.with("""
+        }[35209509()jfkjhadsf,
+        ,,,,,[]
+        ]{
+        """));
+
+        assertEquals(1, Vars.state.patcher.patches.first().warnings.size);
+    }
+
     @Test
     void testUnknownFieldWarn() throws Exception{
         Vars.state.patcher.apply(Seq.with("""
@@ -218,7 +244,7 @@ public class PatcherTests{
         unit.dagger.frogs: 10
         """));
 
-        assertEquals(2, Vars.state.patcher.patches.first().warnings.size);
+        assertEquals(1, Vars.state.patcher.patches.first().warnings.size);
     }
 
     @Test