package ru.betterend.mixin.common;

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

import net.minecraft.block.AbstractBlock;
import net.minecraft.block.Block;
import net.minecraft.block.BlockState;
import net.minecraft.block.Blocks;
import net.minecraft.block.ChorusPlantBlock;
import net.minecraft.item.ItemPlacementContext;
import net.minecraft.state.StateManager;
import net.minecraft.state.property.Properties;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.Direction;
import net.minecraft.world.BlockView;
import net.minecraft.world.World;
import net.minecraft.world.WorldAccess;
import net.minecraft.world.WorldView;
import ru.betterend.registry.EndBlocks;
import ru.betterend.registry.EndTags;
import ru.betterend.util.BlocksHelper;

@Mixin(value = ChorusPlantBlock.class, priority = 100)
public abstract class ChorusPlantBlockMixin extends Block {
	public ChorusPlantBlockMixin(Settings settings) {
		super(settings);
	}

	@Inject(method = "<init>*", at = @At("TAIL"))
	private void beOnInit(AbstractBlock.Settings settings, CallbackInfo info) {
		this.setDefaultState(this.getDefaultState().with(BlocksHelper.ROOTS, false));
	}
	
	@Inject(method = "appendProperties", at = @At("TAIL"))
	private void beAddProperties(StateManager.Builder<Block, BlockState> builder, CallbackInfo info) {
		builder.add(BlocksHelper.ROOTS);
	}
	
	@Inject(method = "withConnectionProperties", at = @At("RETURN"), cancellable = true)
	private void beConnectionProperties(BlockView world, BlockPos pos, CallbackInfoReturnable<BlockState> info) {
		BlockState plant = info.getReturnValue();
		if (plant.isOf(Blocks.CHORUS_PLANT)) {
			if (world.getBlockState(pos.down()).isIn(EndTags.END_GROUND)) {
				info.setReturnValue(plant.with(Properties.DOWN, true).with(BlocksHelper.ROOTS, true));
				info.cancel();
			}
			else {
				info.setReturnValue(plant.with(BlocksHelper.ROOTS, false));
				info.cancel();
			}
		}
	}
	
	@Inject(method = "canPlaceAt", at = @At("HEAD"), cancellable = true)
	private void beCanPlace(BlockState state, WorldView world, BlockPos pos, CallbackInfoReturnable<Boolean> info) {
		BlockState down = world.getBlockState(pos.down());
		if (down.isOf(EndBlocks.CHORUS_NYLIUM) || down.isOf(Blocks.END_STONE)) {
			info.setReturnValue(true);
			info.cancel();
		}
	}
	
	@Inject(method = "getStateForNeighborUpdate", at = @At("RETURN"), cancellable = true)
	private void beStateForNeighborUpdate(BlockState state, Direction direction, BlockState newState, WorldAccess world, BlockPos pos, BlockPos posFrom, CallbackInfoReturnable<BlockState> info) {
		BlockState plant = info.getReturnValue();
		if (plant.isOf(Blocks.CHORUS_PLANT)) {
			if (world.getBlockState(pos.down()).isIn(EndTags.END_GROUND)) {
				plant = plant.with(Properties.DOWN, true).with(BlocksHelper.ROOTS, true);
				info.cancel();
			}
			else {
				plant = plant.with(BlocksHelper.ROOTS, false);
				info.cancel();
			}
			info.setReturnValue(plant);
			info.cancel();
		}
	}
	
	@Inject(method = "getPlacementState", at = @At("RETURN"), cancellable = true)
	private void beGetPlacementState(ItemPlacementContext ctx, CallbackInfoReturnable<BlockState> info) {
		BlockPos pos = ctx.getBlockPos();
		World world = ctx.getWorld();
		BlockState plant = info.getReturnValue();
		if (ctx.canPlace() && plant.isOf(Blocks.CHORUS_PLANT) && world.getBlockState(pos.down()).isIn(EndTags.END_GROUND)) {
			info.setReturnValue(plant.with(BlocksHelper.ROOTS, true).with(Properties.DOWN, true));
			info.cancel();
		}
	}
}