Improve extended piston replacement

pmdevita · pmdevita · commit f175d0598f70 · 2021-02-13T22:59:07.000-05:00
diff --git a/src/main/kotlin/net/pdevita/creeperheal2/constants/DependentBlocks.kt b/src/main/kotlin/net/pdevita/creeperheal2/constants/DependentBlocks.kt
@@ -203,11 +203,13 @@ private class Blocks13: VersionBlocks() {
             Material.LIGHT_GRAY_WALL_BANNER to Behind,
             Material.LIME_WALL_BANNER to Behind,
             Material.MAGENTA_WALL_BANNER to Behind,
-            Material.MOVING_PISTON to Behind,
             Material.ORANGE_WALL_BANNER to Behind,
             Material.PAINTING to Behind,
             Material.PINK_WALL_BANNER to Behind,
-            Material.PISTON_HEAD to Behind,
+//            Material.PISTON_HEAD to Behind,
+//            Material.MOVING_PISTON to Behind,
+            Material.PISTON to Piston,
+            Material.STICKY_PISTON to Piston,
             Material.PLAYER_WALL_HEAD to Behind,
             Material.PURPLE_WALL_BANNER to Behind,
             Material.RED_WALL_BANNER to Behind,
diff --git a/src/main/kotlin/net/pdevita/creeperheal2/utils/FindDependentBlock.kt b/src/main/kotlin/net/pdevita/creeperheal2/utils/FindDependentBlock.kt
@@ -9,6 +9,7 @@ import org.bukkit.block.data.Directional
 import org.bukkit.block.data.FaceAttachable
 import org.bukkit.block.data.MultipleFacing
 import org.bukkit.block.data.type.Bed
+import org.bukkit.block.data.type.Piston
 import org.bukkit.block.data.type.Switch
 
 interface FindDependentBlock {
@@ -108,4 +109,16 @@ object Door:FindDependentBlock {
     }
 }
 
+object Piston:FindDependentBlock {
+    override fun reorient(state: BlockState): Location? {
+        if (state.blockData is org.bukkit.block.data.type.Piston) {
+            val piston = state.blockData as org.bukkit.block.data.type.Piston
+            if (piston.isExtended) {
+                return state.block.getRelative(piston.facing).location
+            }
+        }
+        return null
+    }
+}
+
 
