package net.minecraft.client.renderer.block.model;

import com.google.common.annotations.VisibleForTesting;
import com.google.gson.Gson;
import com.google.gson.GsonBuilder;
import com.google.gson.JsonDeserializationContext;
import com.google.gson.JsonDeserializer;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.google.gson.JsonParseException;
import java.io.Reader;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.List;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.renderer.texture.TextureAtlas;
import net.minecraft.client.resources.model.UnbakedGeometry;
import net.minecraft.client.resources.model.UnbakedModel;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.util.GsonHelper;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public record BlockModel(
	@Nullable UnbakedGeometry geometry,
	@Nullable UnbakedModel.GuiLight guiLight,
	@Nullable Boolean ambientOcclusion,
	@Nullable ItemTransforms transforms,
	TextureSlots.Data textureSlots,
	@Nullable ResourceLocation parent
) implements UnbakedModel {
	@VisibleForTesting
	static final Gson GSON = new GsonBuilder()
		.registerTypeAdapter(BlockModel.class, new BlockModel.Deserializer())
		.registerTypeAdapter(BlockElement.class, new BlockElement.Deserializer())
		.registerTypeAdapter(BlockElementFace.class, new BlockElementFace.Deserializer())
		.registerTypeAdapter(ItemTransform.class, new ItemTransform.Deserializer())
		.registerTypeAdapter(ItemTransforms.class, new ItemTransforms.Deserializer())
		.create();

	public static BlockModel fromStream(Reader reader) {
		return GsonHelper.fromJson(GSON, reader, BlockModel.class);
	}

	@Environment(EnvType.CLIENT)
	public static class Deserializer implements JsonDeserializer<BlockModel> {
		public BlockModel deserialize(JsonElement json, Type type, JsonDeserializationContext context) throws JsonParseException {
			JsonObject jsonObject = json.getAsJsonObject();
			UnbakedGeometry unbakedGeometry = this.getElements(context, jsonObject);
			String string = this.getParentName(jsonObject);
			TextureSlots.Data data = this.getTextureMap(jsonObject);
			Boolean boolean_ = this.getAmbientOcclusion(jsonObject);
			ItemTransforms itemTransforms = null;
			if (jsonObject.has("display")) {
				JsonObject jsonObject2 = GsonHelper.getAsJsonObject(jsonObject, "display");
				itemTransforms = context.deserialize(jsonObject2, ItemTransforms.class);
			}

			UnbakedModel.GuiLight guiLight = null;
			if (jsonObject.has("gui_light")) {
				guiLight = UnbakedModel.GuiLight.getByName(GsonHelper.getAsString(jsonObject, "gui_light"));
			}

			ResourceLocation resourceLocation = string.isEmpty() ? null : ResourceLocation.parse(string);
			return new BlockModel(unbakedGeometry, guiLight, boolean_, itemTransforms, data, resourceLocation);
		}

		private TextureSlots.Data getTextureMap(JsonObject json) {
			if (json.has("textures")) {
				JsonObject jsonObject = GsonHelper.getAsJsonObject(json, "textures");
				return TextureSlots.parseTextureMap(jsonObject, TextureAtlas.LOCATION_BLOCKS);
			} else {
				return TextureSlots.Data.EMPTY;
			}
		}

		private String getParentName(JsonObject json) {
			return GsonHelper.getAsString(json, "parent", "");
		}

		@Nullable
		protected Boolean getAmbientOcclusion(JsonObject json) {
			return json.has("ambientocclusion") ? GsonHelper.getAsBoolean(json, "ambientocclusion") : null;
		}

		@Nullable
		protected UnbakedGeometry getElements(JsonDeserializationContext context, JsonObject json) {
			if (!json.has("elements")) {
				return null;
			} else {
				List<BlockElement> list = new ArrayList();

				for (JsonElement jsonElement : GsonHelper.getAsJsonArray(json, "elements")) {
					list.add((BlockElement)context.deserialize(jsonElement, BlockElement.class));
				}

				return new SimpleUnbakedGeometry(list);
			}
		}
	}
}