hopeful fix for blocks not being pistons when they should be

By: Tahg <tahgtahv@gmail.com>

paper-api/src/main/java/org/bukkit/event/block/BlockPistonEvent.java

@@ -8,9 +8,11 @@ import org.bukkit.material.PistonBaseMaterial;
 
 public abstract class BlockPistonEvent extends BlockEvent implements Cancellable                                                                                                                                                              {
     private boolean cancelled;
+    private BlockFace direction;
 
-    public BlockPistonEvent(Type type, Block block) {
+    public BlockPistonEvent(Type type, Block block, BlockFace direction) {
         super(type, block);
+        this.direction = direction;
     }
 
     public boolean isCancelled() {
@@ -38,6 +40,7 @@ public abstract class BlockPistonEvent extends BlockEvent implements Cancellable
     public BlockFace getDirection() {
         // Both are meh!
         // return ((PistonBaseMaterial) block.getType().getNewData(block.getData())).getFacing();
-        return ((PistonBaseMaterial) block.getState().getData()).getFacing();
+        // return ((PistonBaseMaterial) block.getState().getData()).getFacing();
+        return direction;
     }
 }

paper-api/src/main/java/org/bukkit/event/block/BlockPistonExtendEvent.java

@@ -5,13 +5,14 @@ import java.util.Collections;
 import java.util.List;
 
 import org.bukkit.block.Block;
+import org.bukkit.block.BlockFace;
 
 public class BlockPistonExtendEvent extends BlockPistonEvent {
     private int length;
     private List<Block> blocks;
 
-    public BlockPistonExtendEvent(Block block, int length) {
-        super(Type.BLOCK_PISTON_EXTEND, block);
+    public BlockPistonExtendEvent(Block block, int length, BlockFace direction) {
+        super(Type.BLOCK_PISTON_EXTEND, block, direction);
 
         this.length = length;
     }

paper-api/src/main/java/org/bukkit/event/block/BlockPistonRetractEvent.java

@@ -2,10 +2,11 @@ package org.bukkit.event.block;
 
 import org.bukkit.Location;
 import org.bukkit.block.Block;
+import org.bukkit.block.BlockFace;
 
 public class BlockPistonRetractEvent extends BlockPistonEvent {
-    public BlockPistonRetractEvent(Block block) {
-        super(Type.BLOCK_PISTON_RETRACT, block);
+    public BlockPistonRetractEvent(Block block, BlockFace direction) {
+        super(Type.BLOCK_PISTON_RETRACT, block, direction);
     }
 
     /**