Add trial grief whitelist block tag (part of #141)

CafeteriaGuild · Nov 11, 2023 · 6f05fad · 6f05fad
1 parent 53c50de
commit 6f05fad
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 7 deletions.
diff --git a/base/src/main/kotlin/dev/nathanpb/dml/misc/Tags.kt b/base/src/main/kotlin/dev/nathanpb/dml/misc/Tags.kt
@@ -0,0 +1,10 @@
+package dev.nathanpb.dml.misc
+
+import dev.nathanpb.dml.identifier
+import net.minecraft.block.Block
+import net.minecraft.registry.RegistryKeys
+
+import net.minecraft.registry.tag.TagKey
+
+
+val TRIAL_GRIEF_WHITELIST: TagKey<Block> = TagKey.of(RegistryKeys.BLOCK, identifier("trial_grief_whitelist"))
diff --git a/base/src/main/kotlin/dev/nathanpb/dml/trial/TrialGriefPrevention.kt b/base/src/main/kotlin/dev/nathanpb/dml/trial/TrialGriefPrevention.kt
@@ -20,13 +20,13 @@
 package dev.nathanpb.dml.trial
 
 import dev.nathanpb.dml.config
+import dev.nathanpb.dml.misc.TRIAL_GRIEF_WHITELIST
 import dev.nathanpb.dml.utils.runningTrials
 import dev.nathanpb.dml.utils.squared
 import dev.nathanpb.dml.utils.toBlockPos
 import dev.nathanpb.dml.utils.toVec3i
 import net.fabricmc.fabric.api.event.player.AttackBlockCallback
 import net.fabricmc.fabric.api.event.player.UseBlockCallback
-import net.minecraft.block.Blocks
 import net.minecraft.entity.Entity
 import net.minecraft.entity.damage.DamageSource
 import net.minecraft.entity.mob.EndermanEntity
@@ -39,7 +39,6 @@ import net.minecraft.util.math.BlockPos
 import net.minecraft.util.math.Direction
 import net.minecraft.util.math.Vec3d
 import net.minecraft.world.World
-import net.minecraft.world.explosion.Explosion
 import net.minecraft.world.explosion.ExplosionBehavior
 
 class TrialGriefPrevention : AttackBlockCallback, UseBlockCallback {
@@ -65,11 +64,8 @@ class TrialGriefPrevention : AttackBlockCallback, UseBlockCallback {
             !world.isClient
             && config.trial.interactGriefPrevention
             && isBlockProtected(world, pos)
-            && world.getBlockState(pos)?.run {
-                block !== Blocks.CHEST
-                && block !== Blocks.TRAPPED_CHEST
-                && "grave" !in Registries.BLOCK.getId(block).path
-            } != false
+            && !world.getBlockState(pos).isIn(TRIAL_GRIEF_WHITELIST)
+            && "grave" !in Registries.BLOCK.getId(world.getBlockState(pos).block).path
         ) {
             ActionResult.FAIL
         } else ActionResult.PASS

diff --git a/base/src/main/resources/data/dml-refabricated/tags/blocks/trial_grief_whitelist.json b/base/src/main/resources/data/dml-refabricated/tags/blocks/trial_grief_whitelist.json
@@ -0,0 +1,7 @@
+{
+  "values": [
+    "#c:chests",
+    {"id": "yigd:grave", "required": false},
+    {"id": "gravestones:gravestone", "required": false}
+  ]
+}