package wayoftime.bloodmagic.api;

import javax.annotation.Nonnull;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonSyntaxException;
import com.mojang.brigadier.exceptions.CommandSyntaxException;

import net.minecraft.fluid.Fluid;
import net.minecraft.fluid.Fluids;
import net.minecraft.item.ItemStack;
import net.minecraft.item.crafting.ShapedRecipe;
import net.minecraft.nbt.CompoundNBT;
import net.minecraft.nbt.JsonToNBT;
import net.minecraft.util.JSONUtils;
import net.minecraft.util.ResourceLocation;
import net.minecraftforge.fluids.FluidStack;
import net.minecraftforge.registries.ForgeRegistries;
import wayoftime.bloodmagic.util.Constants;

/**
 * Copied liberally from Mekanism. Thanks, pupnewfster!
 *
 */
public class SerializerHelper
{
	private SerializerHelper()
	{
	}

	private static final Gson GSON = new GsonBuilder().setPrettyPrinting().disableHtmlEscaping().create();

	private static void validateKey(@Nonnull JsonObject json, @Nonnull String key)
	{
		if (!json.has(key))
		{
			throw new JsonSyntaxException("Missing '" + key + "', expected to find an object");
		}
		if (!json.get(key).isJsonObject())
		{
			throw new JsonSyntaxException("Expected '" + key + "' to be an object");
		}
	}

	public static ItemStack getItemStack(@Nonnull JsonObject json, @Nonnull String key)
	{
		validateKey(json, key);
		return ShapedRecipe.deserializeItem(JSONUtils.getJsonObject(json, key));
	}

	public static JsonElement serializeItemStack(@Nonnull ItemStack stack)
	{
		JsonObject json = new JsonObject();
		json.addProperty(Constants.JSON.ITEM, stack.getItem().getRegistryName().toString());
		if (stack.getCount() > 1)
		{
			json.addProperty(Constants.JSON.COUNT, stack.getCount());
		}
		if (stack.hasTag())
		{
			json.addProperty(Constants.JSON.NBT, stack.getTag().toString());
		}
		return json;
	}

	public static FluidStack getFluidStack(@Nonnull JsonObject json, @Nonnull String key)
	{
		validateKey(json, key);
		return deserializeFluid(JSONUtils.getJsonObject(json, key));
	}

	public static FluidStack deserializeFluid(@Nonnull JsonObject json)
	{
		if (!json.has(Constants.JSON.AMOUNT))
		{
			throw new JsonSyntaxException("Expected to receive a amount that is greater than zero");
		}
		JsonElement count = json.get(Constants.JSON.AMOUNT);
		if (!JSONUtils.isNumber(count))
		{
			throw new JsonSyntaxException("Expected amount to be a number greater than zero.");
		}
		int amount = count.getAsJsonPrimitive().getAsInt();
		if (amount < 1)
		{
			throw new JsonSyntaxException("Expected amount to be greater than zero.");
		}
		ResourceLocation resourceLocation = new ResourceLocation(JSONUtils.getString(json, Constants.JSON.FLUID));
		Fluid fluid = ForgeRegistries.FLUIDS.getValue(resourceLocation);
		if (fluid == null || fluid == Fluids.EMPTY)
		{
			throw new JsonSyntaxException("Invalid fluid type '" + resourceLocation + "'");
		}
		CompoundNBT nbt = null;
		if (json.has(Constants.JSON.NBT))
		{
			JsonElement jsonNBT = json.get(Constants.JSON.NBT);
			try
			{
				if (jsonNBT.isJsonObject())
				{
					nbt = JsonToNBT.getTagFromJson(GSON.toJson(jsonNBT));
				} else
				{
					nbt = JsonToNBT.getTagFromJson(JSONUtils.getString(jsonNBT, Constants.JSON.NBT));
				}
			} catch (CommandSyntaxException e)
			{
				throw new JsonSyntaxException("Invalid NBT entry for fluid '" + resourceLocation + "'");
			}
		}
		return new FluidStack(fluid, amount, nbt);
	}

	public static JsonElement serializeFluidStack(@Nonnull FluidStack stack)
	{
		JsonObject json = new JsonObject();
		json.addProperty(Constants.JSON.FLUID, stack.getFluid().getRegistryName().toString());
		json.addProperty(Constants.JSON.AMOUNT, stack.getAmount());
		if (stack.hasTag())
		{
			json.addProperty(Constants.JSON.NBT, stack.getTag().toString());
		}
		return json;
	}
}