mirror of
https://github.com/AtlasMediaGroup/Scissors.git
synced 2024-12-11 16:11:02 -05:00
96 lines
4.7 KiB
Diff
96 lines
4.7 KiB
Diff
|
From 0000000000000000000000000000000000000000 Mon Sep 17 00:00:00 2001
|
||
|
From: Business Goose <arclicious@vivaldi.net>
|
||
|
Date: Mon, 4 Apr 2022 00:16:54 +0100
|
||
|
Subject: [PATCH] Prevent abusable entities from being spawned by spawners
|
||
|
|
||
|
|
||
|
diff --git a/src/main/java/net/minecraft/world/level/BaseSpawner.java b/src/main/java/net/minecraft/world/level/BaseSpawner.java
|
||
|
index 569cef8fcb1e3e1e8b66dad4fa9b956b44542bf1..f844e0b1e44f0b45b1febadf02d7b37d3099fbdb 100644
|
||
|
--- a/src/main/java/net/minecraft/world/level/BaseSpawner.java
|
||
|
+++ b/src/main/java/net/minecraft/world/level/BaseSpawner.java
|
||
|
@@ -19,7 +19,9 @@ import net.minecraft.util.StringUtil;
|
||
|
import net.minecraft.util.random.WeightedRandomList;
|
||
|
import net.minecraft.world.entity.Entity;
|
||
|
import net.minecraft.world.entity.EntityType;
|
||
|
+import net.minecraft.world.entity.LivingEntity;
|
||
|
import net.minecraft.world.entity.Mob;
|
||
|
+import net.minecraft.world.entity.MobCategory;
|
||
|
import net.minecraft.world.entity.MobSpawnType;
|
||
|
import net.minecraft.world.entity.SpawnGroupData;
|
||
|
import net.minecraft.world.entity.SpawnPlacements;
|
||
|
@@ -31,6 +33,7 @@ public abstract class BaseSpawner {
|
||
|
|
||
|
private static final Logger LOGGER = LogManager.getLogger();
|
||
|
private static final int EVENT_SPAWN = 1;
|
||
|
+ public static List<MobCategory> whitelistedCategories = List.of(MobCategory.CREATURE, MobCategory.MONSTER, MobCategory.WATER_CREATURE, MobCategory.UNDERGROUND_WATER_CREATURE, MobCategory.AMBIENT); // Scissors
|
||
|
public static WeightedRandomList<SpawnData> EMPTY_POTENTIALS = WeightedRandomList.create(); // Paper - private->public
|
||
|
public int spawnDelay = 20;
|
||
|
public WeightedRandomList<SpawnData> spawnPotentials;
|
||
|
@@ -64,6 +67,13 @@ public abstract class BaseSpawner {
|
||
|
public ResourceLocation getEntityId(@Nullable Level world, BlockPos pos) {
|
||
|
String s = this.nextSpawnData.getTag().getString("id");
|
||
|
|
||
|
+ // Scissors start
|
||
|
+ Optional<EntityType<?>> entityType = EntityType.byString(s);
|
||
|
+ if (entityType.isPresent() && !whitelistedCategories.contains(entityType.get().getCategory())) {
|
||
|
+ return null;
|
||
|
+ }
|
||
|
+ // Scissors end
|
||
|
+
|
||
|
try {
|
||
|
return StringUtil.isNullOrEmpty(s) ? null : new ResourceLocation(s);
|
||
|
} catch (ResourceLocationException resourcekeyinvalidexception) {
|
||
|
@@ -73,6 +83,12 @@ public abstract class BaseSpawner {
|
||
|
}
|
||
|
|
||
|
public void setEntityId(EntityType<?> type) {
|
||
|
+ // Scissors start - Don't allow unnatural entities to be added to spawners
|
||
|
+ if (!whitelistedCategories.contains(type.getCategory())) {
|
||
|
+ return;
|
||
|
+ }
|
||
|
+ // Scissors end
|
||
|
+
|
||
|
this.nextSpawnData.getTag().putString("id", Registry.ENTITY_TYPE.getKey(type).toString());
|
||
|
this.spawnPotentials = BaseSpawner.EMPTY_POTENTIALS; // CraftBukkit - SPIGOT-3496, MC-92282
|
||
|
}
|
||
|
@@ -121,7 +137,7 @@ public abstract class BaseSpawner {
|
||
|
CompoundTag nbttagcompound = this.nextSpawnData.getTag();
|
||
|
Optional<EntityType<?>> optional = EntityType.by(nbttagcompound);
|
||
|
|
||
|
- if (!optional.isPresent()) {
|
||
|
+ if (!(optional.isPresent() && whitelistedCategories.contains(optional.get().getCategory()))) { // Scissors - Don't try spawning in entites that aren't whitelisted
|
||
|
this.delay(world, pos);
|
||
|
return;
|
||
|
}
|
||
|
@@ -254,12 +270,16 @@ public abstract class BaseSpawner {
|
||
|
ListTag nbttaglist = nbt.getList("SpawnPotentials", 10);
|
||
|
|
||
|
for (int i = 0; i < nbttaglist.size(); ++i) {
|
||
|
- list.add(new SpawnData(nbttaglist.getCompound(i)));
|
||
|
+ // Scissors start - Filter unnatural entity types from being added
|
||
|
+ if (isValid(nbttaglist.getCompound(i))) {
|
||
|
+ list.add(new SpawnData(nbttaglist.getCompound(i)));
|
||
|
+ }
|
||
|
+ // Scissors end
|
||
|
}
|
||
|
}
|
||
|
|
||
|
this.spawnPotentials = WeightedRandomList.create((List) list);
|
||
|
- if (nbt.contains("SpawnData", 10)) {
|
||
|
+ if (nbt.contains("SpawnData", 10) && isValid(nbt.getCompound("SpawnData"))) { // Scissors - Filter unnatural entity types from being added
|
||
|
this.setNextSpawnData(world, pos, new SpawnData(1, nbt.getCompound("SpawnData")));
|
||
|
} else if (!list.isEmpty()) {
|
||
|
this.spawnPotentials.getRandom(this.random).ifPresent((mobspawnerdata) -> {
|
||
|
@@ -371,4 +391,11 @@ public abstract class BaseSpawner {
|
||
|
public double getoSpin() {
|
||
|
return this.oSpin;
|
||
|
}
|
||
|
+
|
||
|
+ // Scissors start
|
||
|
+ public boolean isValid(CompoundTag tag) {
|
||
|
+ Optional<EntityType<?>> type = EntityType.by(tag);
|
||
|
+ return type.isPresent() && whitelistedCategories.contains(type.get().getCategory());
|
||
|
+ }
|
||
|
+ // Scissors end
|
||
|
}
|