package yetAnotherCarpetAddon.mixins;

import net.minecraft.block.Block;
import net.minecraft.block.BlockState;
import net.minecraft.block.Blocks;
import net.minecraft.block.entity.BlockEntity;
import net.minecraft.block.entity.BlockEntityType;
import net.minecraft.block.entity.PistonBlockEntity;
import net.minecraft.block.piston.PistonBehavior;
import net.minecraft.entity.Entity;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Box;
import net.minecraft.util.math.Direction;
import net.minecraft.util.math.Vec3d;
import net.minecraft.util.shape.VoxelShape;

import java.util.Iterator;
import java.util.List;

import org.spongepowered.asm.mixin.Mixin;
import org.spongepowered.asm.mixin.Shadow;
import org.spongepowered.asm.mixin.injection.At;
import org.spongepowered.asm.mixin.injection.Inject;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

import com.google.common.collect.Lists;

import yetAnotherCarpetAddon.YACASettings;

@Mixin(PistonBlockEntity.class)
public abstract class PistonBlockEntity_mixinTranslocation extends BlockEntity {
	@Shadow private boolean extending;
	@Shadow private boolean source;
	@Shadow private Direction facing;
	@Shadow private BlockState pushedBlock;
	@Shadow abstract BlockState getHeadBlockState();
	@Shadow abstract Box offsetHeadBox(Box box);
	
	private static final List<Block> TRANSLOCATION_BLOCKS;
	
	private boolean checkDoTransloction(Block checkedBlock) {
		return TRANSLOCATION_BLOCKS.contains(checkedBlock);
	}

	@Inject(method = "tick()V", at = @At("TAIL"))
	private void injected(CallbackInfo ci) {
		if(YACASettings.translocation) {
			if (!this.extending && this.source) {
				doTranslocation();
			}
			if(this.checkDoTransloction(this.pushedBlock.getBlock())) {
				doTranslocation();
			}
		}
	}
	
	private void doTranslocation() {
		BlockPos translocTo = this.getPos().offset(this.extending ? this.facing : this.facing.getOpposite(), 1);
		boolean wallbelow = !this.world.getBlockState(translocTo.down()).getBlock().equals(Blocks.MOVING_PISTON);
		
		if((!this.world.getBlockState(translocTo).getBlock().equals(Blocks.MOVING_PISTON)) && wallbelow){
			VoxelShape voxelShape = this.getHeadBlockState().getCollisionShape(this.world, this.getPos());
			if(voxelShape.isEmpty())
				voxelShape = Blocks.STONE.getCollisionShape(Blocks.STONE.getDefaultState(), this.world, this.getPos(), null);
			
			Box box = this.offsetHeadBox(voxelShape.getBoundingBox());
			//System.out.println(box.toString());

			List<Entity> list = this.world.getOtherEntities((Entity) null, box);
			int mod_value = 2;
			if (!list.isEmpty()) {
				if(this.extending)
					mod_value = -2;
			
				Iterator list_iter = list.iterator();

				while (list_iter.hasNext()) {
					Entity entity;
					entity = (Entity) list_iter.next();
					if(entity.getPistonBehavior() != PistonBehavior.IGNORE) {
						Vec3d pos = entity.getPos();
						double pos_x = pos.x;
						double pos_y = pos.y;
						double pos_z = pos.z;
						switch(this.facing) {
							case NORTH:
								pos_z += mod_value;
								break;
							case SOUTH:
								pos_z -= mod_value;
								break;
							case EAST:
								pos_x -= mod_value;
								break;
							case WEST:
								pos_x += mod_value;
								break;
							case UP:
								pos_y -= mod_value;
								break;
							case DOWN:
								pos_y += mod_value;
								break;
						}
						entity.updatePosition(pos_x, pos_y, pos_z);
					}
				}
			}
		}
	}
	public PistonBlockEntity_mixinTranslocation(BlockEntityType<?> blockEntityType_1) {
		super(blockEntityType_1);
	}
	
	static {
		TRANSLOCATION_BLOCKS = Lists.newArrayList(
			Blocks.DETECTOR_RAIL,
			Blocks.OAK_FENCE_GATE,
			Blocks.SPRUCE_FENCE_GATE,
			Blocks.BIRCH_FENCE_GATE,
			Blocks.JUNGLE_FENCE_GATE,
			Blocks.ACACIA_FENCE_GATE,
			Blocks.DARK_OAK_FENCE_GATE,
			Blocks.CRIMSON_FENCE_GATE,
			Blocks.WARPED_FENCE_GATE
		);
	}
}