feat(predicates): Add movementAffectedBy field in Entity.

Ayfri · Aug 15, 2024 · d3e3769 · d3e3769
1 parent 2b1b959
commit d3e3769
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/kore/src/main/kotlin/io/github/ayfri/kore/features/predicates/sub/Entity.kt b/kore/src/main/kotlin/io/github/ayfri/kore/features/predicates/sub/Entity.kt
@@ -22,6 +22,7 @@ data class Entity(
 	var flags: EntityFlags? = null,
 	var location: Location? = null,
 	var movement: Movement? = null,
+	var movementAffectedBy: Location? = null,
 	@Serializable(NbtAsJsonSerializer::class) var nbt: NbtCompound? = null,
 	var passenger: Entity? = null,
 	var periodicTicks: Int? = null,
@@ -86,6 +87,10 @@ fun Entity.movement(init: Movement.() -> Unit = {}) {
 	movement = Movement().apply(init)
 }
 
+fun Entity.movementAffectedBy(init: Location.() -> Unit = {}) {
+	movementAffectedBy = Location().apply(init)
+}
+
 fun Entity.nbt(block: NbtCompoundBuilder.() -> Unit) {
 	nbt = buildNbtCompound(block)
 }

diff --git a/kore/src/test/kotlin/io/github/ayfri/kore/features/PredicateTests.kt b/kore/src/test/kotlin/io/github/ayfri/kore/features/PredicateTests.kt
@@ -103,6 +103,10 @@ fun DataPack.predicateTests() {
 				horizontalSpeed(1.0)
 			}
 
+			movementAffectedBy {
+				canSeeSky = true
+			}
+
 			nbt {
 				this["foo"] = "bar"
 			}
@@ -161,6 +165,9 @@ fun DataPack.predicateTests() {
 					},
 					"horizontal_speed": 1.0
 				},
+				"movement_affected_by": {
+					"can_see_sky": true
+				},
 				"nbt": {
 					"foo": "bar"
 				},