NMSHandler.java

package me.desht.dhutils.nms.v1_7_R2;

import me.desht.dhutils.nms.api.NMSAbstraction;
import net.minecraft.server.v1_7_R2.*;
import org.bukkit.World;
import org.bukkit.block.BlockFace;
import org.bukkit.craftbukkit.v1_7_R2.CraftChunk;
import org.bukkit.craftbukkit.v1_7_R2.CraftWorld;
import org.bukkit.craftbukkit.v1_7_R2.entity.CraftPlayer;
import org.bukkit.entity.Player;
import org.bukkit.util.Vector;

import java.lang.reflect.Method;

public class NMSHandler implements NMSAbstraction {

	@Override
    public boolean setBlockFast(World world, int x, int y, int z, int blockId, byte data) {
        net.minecraft.server.v1_7_R2.World w = ((CraftWorld) world).getHandle();
        Chunk chunk = w.getChunkAt(x >> 4, z >> 4);
        return a(chunk, x & 0x0f, y, z & 0x0f, Block.e(blockId), data);
    }

    private boolean a(Chunk that, int i, int j, int k, Block block, int l) {
        int i1 = k << 4 | i;

        if (j >= that.b[i1] - 1) {
            that.b[i1] = -999;
        }

        int j1 = that.heightMap[i1];
        Block block1 = that.getType(i, j, k);
        int k1 = that.getData(i, j, k);

        if (block1 == block && k1 == l) {
            return false;
        } else {
            boolean flag = false;
            ChunkSection chunksection = that.i()[j >> 4];

            if (chunksection == null) {
                if (block == Blocks.AIR) {
                    return false;
                }

                chunksection = that.i()[j >> 4] = new ChunkSection(j >> 4 << 4, !that.world.worldProvider.g);
                flag = j >= j1;
            }

            int l1 = that.locX * 16 + i;
            int i2 = that.locZ * 16 + k;

            if (!that.world.isStatic) {
                block1.f(that.world, l1, j, i2, k1);
            }

            // CraftBukkit start - Delay removing containers until after they're cleaned up
            if (!(block1 instanceof IContainer)) {
                chunksection.setTypeId(i, j & 15, k, block);
            }
            // CraftBukkit end

            if (!that.world.isStatic) {
                block1.remove(that.world, l1, j, i2, block1, k1);
            } else if (block1 instanceof IContainer && block1 != block) {
                that.world.p(l1, j, i2);
            }

            // CraftBukkit start - Remove containers now after cleanup
            if (block1 instanceof IContainer) {
                chunksection.setTypeId(i, j & 15, k, block);
            }
            // CraftBukkit end

            if (chunksection.getTypeId(i, j & 15, k) != block) {
                return false;
            } else {
                chunksection.setData(i, j & 15, k, l);
                if (flag) {
                    that.initLighting();
                }
                TileEntity tileentity;

                if (block1 instanceof IContainer) {
                    tileentity = that.e(i, j, k);
                    if (tileentity != null) {
                        tileentity.u();
                    }
                }

                // CraftBukkit - Don't place while processing the BlockPlaceEvent, unless it's a BlockContainer
                if (!that.world.isStatic && (!that.world.callingPlaceEvent || (block instanceof BlockContainer))) {
                    block.onPlace(that.world, l1, j, i2);
                }

                if (block instanceof IContainer) {
                    // CraftBukkit start - Don't create tile entity if placement failed
                    if (that.getType(i, j, k) != block) {
                        return false;
                    }
                    // CraftBukkit end

                    tileentity = that.e(i, j, k);
                    if (tileentity == null) {
                        tileentity = ((IContainer) block).a(that.world, l);
                        that.world.setTileEntity(l1, j, i2, tileentity);
                    }

                    if (tileentity != null) {
                        tileentity.u();
                    }
                }

                that.n = true;
                return true;
            }
        }
    }

	@Override
	public void forceBlockLightLevel(World world, int x, int y, int z, int level) {
		net.minecraft.server.v1_7_R2.World w = ((CraftWorld) world).getHandle();
		w.b(EnumSkyBlock.BLOCK, x, y, z, level);
	}

	@Override
	public int getBlockLightEmission(int blockId) {
		return Block.e(blockId).m();
	}

	@Override
	public int getBlockLightBlocking(int blockId) {
		return Block.e(blockId).k();
	}

	@SuppressWarnings("unchecked")
	@Override
	public void queueChunkForUpdate(Player player, int cx, int cz) {
		((CraftPlayer) player).getHandle().chunkCoordIntPairQueue.add(new ChunkCoordIntPair(cx, cz));
	}

	@Override
	public Vector[] getBlockHitbox(org.bukkit.block.Block block) {
		net.minecraft.server.v1_7_R2.World w = ((CraftWorld)block.getWorld()).getHandle();
		net.minecraft.server.v1_7_R2.Block b = w.getType(block.getX(), block.getY(), block.getZ());
		b.updateShape(w, block.getX(), block.getY(), block.getZ());
		return new Vector[] {
				new Vector(block.getX() + b.x(), block.getY() + b.z(), block.getZ() + b.B()),
				new Vector(block.getX() + b.y(), block.getY() + b.A(), block.getZ() + b.C())
		};
	}

    @Override
    public void recalculateBlockLighting(World world, int x, int y, int z) {
        // Don't consider blocks that are completely surrounded by other non-transparent blocks
        if (!canAffectLighting(world, x, y, z)) {
            return;
        }

        int i = x & 0x0F;
        int j = y & 0xFF;
        int k = z & 0x0F;
        CraftChunk craftChunk = (CraftChunk)world.getChunkAt(x >> 4, z >> 4);
        Chunk nmsChunk = craftChunk.getHandle();

        int i1 = k << 4 | i;
        int maxY = nmsChunk.heightMap[i1];

        Block block = nmsChunk.getType(i, j, k);
        int j2 = block.k();

        if (j2 > 0) {
            if (j >= maxY) {
                invokeNmsH(nmsChunk, i, j + 1, k);
            }
        } else if (j == maxY - 1) {
            invokeNmsH(nmsChunk,i, j, k);
        }

        if (nmsChunk.getBrightness(EnumSkyBlock.SKY, i, j, k) > 0 || nmsChunk.getBrightness(EnumSkyBlock.BLOCK, i, j, k) > 0) {
            invokeNmsE(nmsChunk, i, k);
        }

        net.minecraft.server.v1_7_R2.World w = ((CraftWorld) world).getHandle();
        w.c(EnumSkyBlock.BLOCK, i, j, k);
    }

    private Method h;
    private void invokeNmsH(Chunk nmsChunk, int i, int j, int k) {
        try {
            if (h == null) {
                Class[] classes = {int.class, int.class, int.class};
                h = Chunk.class.getDeclaredMethod("h", classes);
                h.setAccessible(true);
            }
            h.invoke(nmsChunk, i, j, k);
        } catch (Exception e) {
            System.out.println("Reflection exception: " + e);
        }
    }

    private Method e;
    private void invokeNmsE(Chunk nmsChunk, int i, int j) {
        try {
            if (e == null) {
                Class[] classes = {int.class, int.class};
                e = Chunk.class.getDeclaredMethod("e", classes);
                e.setAccessible(true);
            }
            e.invoke(nmsChunk, i, j);
        } catch (Exception e) {
            System.out.println("Reflection exception: " + e);
        }
    }

    private boolean canAffectLighting(World world, int x, int y, int z) {
        org.bukkit.block.Block base  = world.getBlockAt(x, y, z);
        org.bukkit.block.Block east  = base.getRelative(BlockFace.EAST);
        org.bukkit.block.Block west  = base.getRelative(BlockFace.WEST);
        org.bukkit.block.Block up    = base.getRelative(BlockFace.UP);
        org.bukkit.block.Block down  = base.getRelative(BlockFace.DOWN);
        org.bukkit.block.Block south = base.getRelative(BlockFace.SOUTH);
        org.bukkit.block.Block north = base.getRelative(BlockFace.NORTH);

        return east.getType().isTransparent() ||
                west.getType().isTransparent() ||
                up.getType().isTransparent() ||
                down.getType().isTransparent() ||
                south.getType().isTransparent() ||
                north.getType().isTransparent();
    }
}