package ru.bclib.recipes;

import java.util.Arrays;
import java.util.Map;

import com.google.common.collect.Maps;

import net.minecraft.core.NonNullList;
import net.minecraft.resources.ResourceLocation;
import net.minecraft.tags.Tag;
import net.minecraft.world.item.Item;
import net.minecraft.world.item.ItemStack;
import net.minecraft.world.item.crafting.CraftingRecipe;
import net.minecraft.world.item.crafting.Ingredient;
import net.minecraft.world.item.crafting.RecipeType;
import net.minecraft.world.item.crafting.ShapedRecipe;
import net.minecraft.world.item.crafting.ShapelessRecipe;
import net.minecraft.world.level.ItemLike;
import ru.bclib.BCLib;

public class GridRecipe {
	private static final GridRecipe INSTANCE = new GridRecipe();
	
	private ResourceLocation id;
	private ItemLike output;
	
	private String group;
	private RecipeType<?> type;
	private boolean shaped;
	private String[] shape;
	private Map<Character, Ingredient> materialKeys = Maps.newHashMap();
	private int count;
	private boolean exist = true;
	
	private GridRecipe() {}
	
	public static GridRecipe make(String modID, String name, ItemLike output) {
		INSTANCE.id = new ResourceLocation(modID, name);
		INSTANCE.output = output;
		
		INSTANCE.group = "";
		INSTANCE.type = RecipeType.CRAFTING;
		INSTANCE.shaped = true;
		INSTANCE.shape = new String[] {"#"};
		INSTANCE.materialKeys.clear();
		INSTANCE.count = 1;
		
		INSTANCE.exist = BCLRecipeManager.exists(output);
		
		return INSTANCE;
	}

	public GridRecipe setGroup(String group) {
		this.group = group;
		return this;
	}
	
	public GridRecipe setShape(String... shape) {
		this.shape = shape;
		return this;
	}
	
	public GridRecipe setList(String shape) {
		this.shape = new String[] { shape };
		this.shaped = false;
		return this;
	}
	
	public GridRecipe addMaterial(char key, Tag<Item> value) {
		return addMaterial(key, Ingredient.of(value));
	}
	
	public GridRecipe addMaterial(char key, ItemStack... value) {
		return addMaterial(key, Ingredient.of(Arrays.stream(value)));
	}
	
	public GridRecipe addMaterial(char key, ItemLike... values) {
		for (ItemLike item: values) {
			exist &= BCLRecipeManager.exists(item);
		}
		return addMaterial(key, Ingredient.of(values));
	}
	
	private GridRecipe addMaterial(char key, Ingredient value) {
		materialKeys.put(key, value);
		return this;
	}
	
	public GridRecipe setOutputCount(int count) {
		this.count = count;
		return this;
	}
	
	private NonNullList<Ingredient> getMaterials(int width, int height) {
		NonNullList<Ingredient> materials = NonNullList.withSize(width * height, Ingredient.EMPTY);
		int pos = 0;
		for (String line: shape) {
			for (int i = 0; i < width; i++) {
				char c = line.charAt(i);
				Ingredient material = materialKeys.get(c);
				materials.set(pos ++, material == null ? Ingredient.EMPTY : material);
			}
		}
		return materials;
	}
	
	public void build() {
		if (exist) {
			int height = shape.length;
			int width = shape[0].length();
			ItemStack result = new ItemStack(output, count);
			NonNullList<Ingredient> materials = this.getMaterials(width, height);
			
			CraftingRecipe recipe = shaped ? new ShapedRecipe(id, group, width, height, materials, result) : new ShapelessRecipe(id, group, result, materials);
			BCLRecipeManager.addRecipe(type, recipe);
		} else {
			BCLib.LOGGER.debug("Recipe {} couldn't be added", id);
		}
	}
}