Fix entities not saving when chunk was unloaded

common/src/main/java/org/waste/of/time/mixin/EntityMixin.java

@@ -0,0 +1,16 @@
+package org.waste.of.time.mixin;
+
+import net.minecraft.entity.Entity;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.waste.of.time.WorldTools;
+
+@Mixin(Entity.class)
+public class EntityMixin {
+    @Inject(method = "setRemoved", at = @At("HEAD"), cancellable = true)
+    private void cancelMarkRemoved(Entity.RemovalReason reason, CallbackInfo ci) {
+        if (WorldTools.INSTANCE.getCaching() && reason == Entity.RemovalReason.DISCARDED) ci.cancel();
+    }
+}

common/src/main/kotlin/org/waste/of/time/storage/StorageManager.kt

@@ -315,6 +315,8 @@ object StorageManager {
                                     putByte("Invulnerable", 1)
                                     putByte("Silent", 1)
                                 }
+
+                                it.setRemoved(Entity.RemovalReason.DISCARDED)
                             })
 
                             cachedEntities.remove(it)

common/src/main/resources/worldtools.mixins.common.json

@@ -7,6 +7,7 @@
     "BossBarHudMixin",
     "ClientChunkManagerMixin",
     "ClientPlayNetworkHandlerMixin",
+    "EntityMixin",
     "MinecraftClientMixin",
     "WorldChunkMixin",
     "accessor.AdvancementProgressesAccessor",