PlaceableItemFrame.java

package joshie.harvest.buildings.placeable.entities;

import com.google.gson.annotations.Expose;
import joshie.harvest.buildings.LootHelper;
import joshie.harvest.core.helpers.EntityHelper;
import net.minecraft.block.state.IBlockState;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityHanging;
import net.minecraft.entity.item.EntityItemFrame;
import net.minecraft.item.ItemStack;
import net.minecraft.util.EnumFacing;
import net.minecraft.util.ResourceLocation;
import net.minecraft.util.Rotation;
import net.minecraft.util.math.BlockPos;
import net.minecraft.world.World;

public class PlaceableItemFrame extends PlaceableHanging {
    @Expose
    private ResourceLocation chestType;
    @Expose
    private ItemStack stack;
    @Expose
    private int rotation;

    public PlaceableItemFrame() {}
    public PlaceableItemFrame(ResourceLocation chestType, ItemStack stack, int rotation, EnumFacing facing, int x, int y, int z) {
        super(facing, x, y, z);
        this.chestType = chestType;
        this.stack = stack;
        this.rotation = rotation;
    }

    @Override
    public void remove(World world, BlockPos pos, Rotation rotation, ConstructionStage stage, IBlockState replacement) {
        if (canPlace(stage)) {
            BlockPos transformed = getTransformedPosition(pos, rotation);
            EntityHelper.getEntities(EntityItemFrame.class, world, transformed, 0.5D, 0.5D).stream().forEach(Entity::setDead);
        }
    }

    @Override
    public EntityHanging getEntityHanging(World world, BlockPos pos, EnumFacing facing) {
        EntityItemFrame frame = new EntityItemFrame(world, new BlockPos(pos.getX(), pos.getY(), pos.getZ()), facing);
        ItemStack loot = null;

        if (stack != null) loot = stack.copy();
        if (chestType != null) {
            loot = LootHelper.getStack(world, null, chestType);
        }

        frame.setDisplayedItem(loot);
        frame.setItemRotation(this.rotation);
        return frame;
    }

    @Override
    public PlaceableItemFrame getCopyFromEntity(Entity e, int x, int y, int z) {
        EntityItemFrame frame = (EntityItemFrame) e;

        ResourceLocation chestType = null;
        ItemStack stack = frame.getDisplayedItem();
        if (stack != null && stack.hasDisplayName()) {
            chestType = new ResourceLocation("harvestfestival", "frames/" + stack.getDisplayName());
            stack = new ItemStack(stack.getItem(), 1, stack.getItemDamage());
        }

        return new PlaceableItemFrame(chestType, stack, frame.getRotation(), frame.facingDirection, x, y, z);
    }
}