package com.anthonyhilyard.iceberg.mixin;

import java.util.Objects;

import com.anthonyhilyard.iceberg.events.EntityFluidEvents;

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.At.Shift;
import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;

import net.minecraft.world.entity.Entity;
import net.minecraft.world.level.material.Fluid;
import net.minecraft.tags.Tag;

@Mixin(Entity.class)
public class EntityMixin
{
	private Fluid previousFluidOnEyes = null;

	@Shadow
	protected Tag<Fluid> fluidOnEyes;

	@Inject(method = "updateFluidOnEyes", at = @At(value = "RETURN"))
	public void onUpdateFluidOnEyes(CallbackInfo callbackInfo)
	{
		if (fluidOnEyes != null && fluidOnEyes.getValues().size() > 0)
		{
			previousFluidOnEyes = fluidOnEyes.getValues().get(0);
		}
		else if (previousFluidOnEyes != null)
		{
			// We were submerged in a fluid that we no longer are.
			if (previousFluidOnEyes != null)
			{
				EntityFluidEvents.EXITED.invoker().onExited((Entity)(Object)this, previousFluidOnEyes);
			}
			previousFluidOnEyes = null;
		}
	}

	@Inject(method = "updateFluidOnEyes",
			at = @At(value = "FIELD", target = "Lnet/minecraft/world/entity/Entity;fluidOnEyes:Lnet/minecraft/tags/Tag;", ordinal = 1, shift = Shift.AFTER))
	public void onUpdateFluidOnEyeAssign(CallbackInfo callbackInfo)
	{
		Fluid currentFluid = null;
		if (fluidOnEyes != null && fluidOnEyes.getValues().size() > 0)
		{
			currentFluid = fluidOnEyes.getValues().get(0);
		}

		if (!Objects.equals(previousFluidOnEyes, currentFluid))
		{
			// We are now submerged in a fluid that doesn't match the previous one.
			if (currentFluid != null)
			{
				EntityFluidEvents.ENTERED.invoker().onEntered((Entity)(Object)this, currentFluid);
			}
		}
	}
}