From 8b6e15457cad90684ea9abbdf56fe4ddc8ccadef Mon Sep 17 00:00:00 2001 From: Auxilor Date: Sat, 18 Feb 2023 12:26:32 +0000 Subject: [PATCH] Fixed clearing entity goals on 1.19+ --- .../proxy/v1_19_R1/EntityControllerFactory.kt | 2 +- .../v1_19_R1/entity/EcoEntityController.kt | 95 +++++++++++++++++++ .../proxy/v1_19_R2/EntityControllerFactory.kt | 2 +- .../v1_19_R2/entity/EcoEntityController.kt | 95 +++++++++++++++++++ 4 files changed, 192 insertions(+), 2 deletions(-) create mode 100644 eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/entity/EcoEntityController.kt create mode 100644 eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/entity/EcoEntityController.kt diff --git a/eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/EntityControllerFactory.kt b/eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/EntityControllerFactory.kt index a34ca473..e4585b21 100644 --- a/eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/EntityControllerFactory.kt +++ b/eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/EntityControllerFactory.kt @@ -2,7 +2,7 @@ package com.willfp.eco.internal.spigot.proxy.v1_19_R1 import com.willfp.eco.core.entities.ai.EntityController import com.willfp.eco.internal.spigot.proxy.EntityControllerFactoryProxy -import com.willfp.eco.internal.spigot.proxy.common.ai.EcoEntityController +import com.willfp.eco.internal.spigot.proxy.v1_19_R1.entity.EcoEntityController import org.bukkit.entity.Mob class EntityControllerFactory : EntityControllerFactoryProxy { diff --git a/eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/entity/EcoEntityController.kt b/eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/entity/EcoEntityController.kt new file mode 100644 index 00000000..10d9ff65 --- /dev/null +++ b/eco-core/core-nms/v1_19_R1/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R1/entity/EcoEntityController.kt @@ -0,0 +1,95 @@ +package com.willfp.eco.internal.spigot.proxy.v1_19_R1.entity + +import com.willfp.eco.core.entities.ai.CustomGoal +import com.willfp.eco.core.entities.ai.EntityController +import com.willfp.eco.core.entities.ai.EntityGoal +import com.willfp.eco.core.entities.ai.TargetGoal +import com.willfp.eco.internal.spigot.proxy.common.ai.CustomGoalFactory +import com.willfp.eco.internal.spigot.proxy.common.ai.getGoalFactory +import com.willfp.eco.internal.spigot.proxy.common.toPathfinderMob +import net.minecraft.world.entity.PathfinderMob +import net.minecraft.world.entity.ai.goal.Goal +import org.bukkit.entity.Mob + +class EcoEntityController( + private val handle: T +) : EntityController { + override fun addEntityGoal(priority: Int, goal: EntityGoal): EntityController { + val nms = getNms() ?: return this + + nms.goalSelector.addGoal( + priority, + goal.getGoalFactory()?.create(goal, nms) ?: return this + ) + + return this + } + + override fun removeEntityGoal(goal: EntityGoal): EntityController { + val nms = getNms() ?: return this + + val predicate: (Goal) -> Boolean = if (goal is CustomGoal<*>) { + { CustomGoalFactory.isGoalOfType(it, goal) } + } else { + { goal.getGoalFactory()?.isGoalOfType(it) == true } + } + + for (wrapped in nms.goalSelector.availableGoals.toSet()) { + if (predicate(wrapped.goal)) { + nms.goalSelector.removeGoal(wrapped.goal) + } + } + + return this + } + + override fun clearEntityGoals(): EntityController { + val nms = getNms() ?: return this + nms.goalSelector.availableGoals.clear() + return this + } + + override fun addTargetGoal(priority: Int, goal: TargetGoal): EntityController { + val nms = getNms() ?: return this + + nms.targetSelector.addGoal( + priority, goal.getGoalFactory()?.create(goal, nms) ?: return this + ) + + nms.targetSelector + + return this + } + + override fun removeTargetGoal(goal: TargetGoal): EntityController { + val nms = getNms() ?: return this + + val predicate: (Goal) -> Boolean = if (goal is CustomGoal<*>) { + { CustomGoalFactory.isGoalOfType(it, goal) } + } else { + { goal.getGoalFactory()?.isGoalOfType(it) == true } + } + + for (wrapped in nms.targetSelector.availableGoals.toSet()) { + if (predicate(wrapped.goal)) { + nms.targetSelector.removeGoal(wrapped.goal) + } + } + + return this + } + + override fun clearTargetGoals(): EntityController { + val nms = getNms() ?: return this + nms.targetSelector.availableGoals.clear() + return this + } + + private fun getNms(): PathfinderMob? { + return handle.toPathfinderMob() + } + + override fun getEntity(): T { + return handle + } +} \ No newline at end of file diff --git a/eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/EntityControllerFactory.kt b/eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/EntityControllerFactory.kt index fd542d9d..da056a11 100644 --- a/eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/EntityControllerFactory.kt +++ b/eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/EntityControllerFactory.kt @@ -2,7 +2,7 @@ package com.willfp.eco.internal.spigot.proxy.v1_19_R2 import com.willfp.eco.core.entities.ai.EntityController import com.willfp.eco.internal.spigot.proxy.EntityControllerFactoryProxy -import com.willfp.eco.internal.spigot.proxy.common.ai.EcoEntityController +import com.willfp.eco.internal.spigot.proxy.v1_19_R2.entity.EcoEntityController import org.bukkit.entity.Mob class EntityControllerFactory : EntityControllerFactoryProxy { diff --git a/eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/entity/EcoEntityController.kt b/eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/entity/EcoEntityController.kt new file mode 100644 index 00000000..4b31c630 --- /dev/null +++ b/eco-core/core-nms/v1_19_R2/src/main/kotlin/com/willfp/eco/internal/spigot/proxy/v1_19_R2/entity/EcoEntityController.kt @@ -0,0 +1,95 @@ +package com.willfp.eco.internal.spigot.proxy.v1_19_R2.entity + +import com.willfp.eco.core.entities.ai.CustomGoal +import com.willfp.eco.core.entities.ai.EntityController +import com.willfp.eco.core.entities.ai.EntityGoal +import com.willfp.eco.core.entities.ai.TargetGoal +import com.willfp.eco.internal.spigot.proxy.common.ai.CustomGoalFactory +import com.willfp.eco.internal.spigot.proxy.common.ai.getGoalFactory +import com.willfp.eco.internal.spigot.proxy.common.toPathfinderMob +import net.minecraft.world.entity.PathfinderMob +import net.minecraft.world.entity.ai.goal.Goal +import org.bukkit.entity.Mob + +class EcoEntityController( + private val handle: T +) : EntityController { + override fun addEntityGoal(priority: Int, goal: EntityGoal): EntityController { + val nms = getNms() ?: return this + + nms.goalSelector.addGoal( + priority, + goal.getGoalFactory()?.create(goal, nms) ?: return this + ) + + return this + } + + override fun removeEntityGoal(goal: EntityGoal): EntityController { + val nms = getNms() ?: return this + + val predicate: (Goal) -> Boolean = if (goal is CustomGoal<*>) { + { CustomGoalFactory.isGoalOfType(it, goal) } + } else { + { goal.getGoalFactory()?.isGoalOfType(it) == true } + } + + for (wrapped in nms.goalSelector.availableGoals.toSet()) { + if (predicate(wrapped.goal)) { + nms.goalSelector.removeGoal(wrapped.goal) + } + } + + return this + } + + override fun clearEntityGoals(): EntityController { + val nms = getNms() ?: return this + nms.goalSelector.availableGoals.clear() + return this + } + + override fun addTargetGoal(priority: Int, goal: TargetGoal): EntityController { + val nms = getNms() ?: return this + + nms.targetSelector.addGoal( + priority, goal.getGoalFactory()?.create(goal, nms) ?: return this + ) + + nms.targetSelector + + return this + } + + override fun removeTargetGoal(goal: TargetGoal): EntityController { + val nms = getNms() ?: return this + + val predicate: (Goal) -> Boolean = if (goal is CustomGoal<*>) { + { CustomGoalFactory.isGoalOfType(it, goal) } + } else { + { goal.getGoalFactory()?.isGoalOfType(it) == true } + } + + for (wrapped in nms.targetSelector.availableGoals.toSet()) { + if (predicate(wrapped.goal)) { + nms.targetSelector.removeGoal(wrapped.goal) + } + } + + return this + } + + override fun clearTargetGoals(): EntityController { + val nms = getNms() ?: return this + nms.targetSelector.availableGoals.clear() + return this + } + + private fun getNms(): PathfinderMob? { + return handle.toPathfinderMob() + } + + override fun getEntity(): T { + return handle + } +} \ No newline at end of file