package me.Danker.handlers;

import io.netty.channel.ChannelDuplexHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelPromise;
import me.Danker.commands.ToggleCommand;
import me.Danker.utils.Utils;
import net.minecraft.client.Minecraft;
import net.minecraft.entity.Entity;
import net.minecraft.entity.item.EntityItemFrame;
import net.minecraft.init.Blocks;
import net.minecraft.init.Items;
import net.minecraft.item.ItemStack;
import net.minecraft.network.Packet;
import net.minecraft.network.play.client.C02PacketUseEntity;
import net.minecraft.network.play.server.S04PacketEntityEquipment;
import net.minecraft.util.BlockPos;

import java.lang.reflect.Field;

public class PacketHandler extends ChannelDuplexHandler {
	
	// Spirit boots fix
	@Override
	public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
		if (Utils.inSkyblock && msg instanceof Packet && msg.getClass().getName().endsWith("S04PacketEntityEquipment")) { // Inventory packet name
			S04PacketEntityEquipment packet = (S04PacketEntityEquipment) msg;
			if (packet.getEntityID() == Minecraft.getMinecraft().thePlayer.getEntityId()) {
				Field slot = packet.getClass().getDeclaredField("field_149392_b"); // equipmentSlot
				slot.setAccessible(true);
				slot.setInt(packet, slot.getInt(packet) + 1);
				msg = packet;
			}
		}
		
		super.channelRead(ctx, msg);
	}
	
	// Ignore item frames with arrows on sea lanterns
	@Override
	public void write(ChannelHandlerContext ctx, Object msg, ChannelPromise promise) throws Exception {
		if (ToggleCommand.itemFrameOnSeaLanternsToggled && Utils.inDungeons && msg instanceof Packet && msg.getClass().getName().endsWith("C02PacketUseEntity")) {
			Minecraft mc = Minecraft.getMinecraft();
			C02PacketUseEntity packet = (C02PacketUseEntity) msg;
			Entity entityHit = packet.getEntityFromWorld(mc.theWorld);
			if (entityHit instanceof EntityItemFrame) {
				EntityItemFrame itemFrame = (EntityItemFrame) entityHit;
	    		ItemStack item = itemFrame.getDisplayedItem();
	    		if (item != null && item.getItem() == Items.arrow) {
		    		BlockPos blockPos = Utils.getBlockUnderItemFrame(itemFrame);
		    		if (mc.theWorld.getBlockState(blockPos).getBlock() == Blocks.sea_lantern) {
		    			return;
		    		}
	    		}
			}
		}
		
		super.write(ctx, msg, promise);
	}
	
}