Add SetFireTimeEntityModifier

Author: ErrorCraft

src/main/java/errorcraft/entitymodifiers/entity/modifier/EntityModifierTypes.java

@@ -1,9 +1,6 @@
 package errorcraft.entitymodifiers.entity.modifier;
 
-import errorcraft.entitymodifiers.entity.modifier.modifiers.SetAirTimeEntityModifier;
-import errorcraft.entitymodifiers.entity.modifier.modifiers.SetHealthEntityModifier;
-import errorcraft.entitymodifiers.entity.modifier.modifiers.SetHungerEntityModifier;
-import errorcraft.entitymodifiers.entity.modifier.modifiers.SetInvulnerableEntityModifier;
+import errorcraft.entitymodifiers.entity.modifier.modifiers.*;
 import errorcraft.entitymodifiers.mixin.registry.RegistryAccessor;
 import net.minecraft.util.Identifier;
 import net.minecraft.util.JsonSerializing;
@@ -18,6 +15,7 @@ public class EntityModifierTypes {
 	public static final EntityModifierType SET_HEALTH = register("set_health", new SetHealthEntityModifier.Serialiser());
 	public static final EntityModifierType SET_HUNGER = register("set_hunger", new SetHungerEntityModifier.Serialiser());
 	public static final EntityModifierType SET_AIR_TIME = register("set_air_time", new SetAirTimeEntityModifier.Serialiser());
+	public static final EntityModifierType SET_FIRE_TIME = register("set_fire_time", new SetFireTimeEntityModifier.Serialiser());
 
 	public static Object createGsonAdapter() {
 		return JsonSerializing.createSerializerBuilder(ENTITY_MODIFIER_TYPE, "function", "function", EntityModifier::getType).build();

src/main/java/errorcraft/entitymodifiers/entity/modifier/modifiers/SetFireTimeEntityModifier.java

@@ -0,0 +1,49 @@
+package errorcraft.entitymodifiers.entity.modifier.modifiers;
+
+import com.google.gson.JsonDeserializationContext;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonSerializationContext;
+import errorcraft.entitymodifiers.entity.modifier.EntityModifier;
+import errorcraft.entitymodifiers.entity.modifier.EntityModifierType;
+import errorcraft.entitymodifiers.entity.modifier.EntityModifierTypes;
+import net.minecraft.entity.Entity;
+import net.minecraft.loot.context.LootContext;
+import net.minecraft.loot.provider.number.LootNumberProvider;
+import net.minecraft.util.JsonHelper;
+
+public class SetFireTimeEntityModifier implements EntityModifier {
+	private final LootNumberProvider fireTimeProvider;
+	private final boolean add;
+
+	public SetFireTimeEntityModifier(LootNumberProvider fireTimeProvider, boolean add) {
+		this.fireTimeProvider = fireTimeProvider;
+		this.add = add;
+	}
+
+	@Override
+	public EntityModifierType getType() {
+		return EntityModifierTypes.SET_FIRE_TIME;
+	}
+
+	@Override
+	public Entity apply(Entity entity, LootContext lootContext) {
+		int newFireTime = this.add ? entity.getFireTicks() : 0;
+		entity.setFireTicks(newFireTime + this.fireTimeProvider.nextInt(lootContext));
+		return entity;
+	}
+
+	public static class Serialiser implements EntityModifier.Serialiser<SetFireTimeEntityModifier> {
+		@Override
+		public void toJson(JsonObject json, SetFireTimeEntityModifier object, JsonSerializationContext context) {
+			json.add("time", context.serialize(object.fireTimeProvider));
+			json.addProperty("add", object.add);
+		}
+
+		@Override
+		public SetFireTimeEntityModifier fromJson(JsonObject json, JsonDeserializationContext context) {
+			LootNumberProvider fireTimeProvider = JsonHelper.deserialize(json, "time", context, LootNumberProvider.class);
+			boolean add = JsonHelper.getBoolean(json, "add", false);
+			return new SetFireTimeEntityModifier(fireTimeProvider, add);
+		}
+	}
+}