ItemScriptHelper.java

package com.denizenscript.denizen.scripts.containers.core;

import com.denizenscript.denizen.Denizen;
import com.denizenscript.denizen.events.BukkitScriptEvent;
import com.denizenscript.denizen.nms.util.jnbt.CompoundTag;
import com.denizenscript.denizen.utilities.Utilities;
import com.denizenscript.denizen.utilities.debugging.Debug;
import com.denizenscript.denizen.events.bukkit.ScriptReloadEvent;
import com.denizenscript.denizen.nms.NMSHandler;
import com.denizenscript.denizen.objects.ItemTag;
import com.denizenscript.denizen.tags.BukkitTagContext;
import com.denizenscript.denizencore.events.ScriptEvent;
import com.denizenscript.denizencore.objects.core.DurationTag;
import com.denizenscript.denizencore.objects.core.ListTag;
import com.denizenscript.denizencore.objects.core.ScriptTag;
import com.denizenscript.denizencore.scripts.ScriptBuilder;
import com.denizenscript.denizencore.tags.TagContext;
import com.denizenscript.denizencore.tags.TagManager;
import com.denizenscript.denizencore.utilities.CoreUtilities;
import com.denizenscript.denizencore.utilities.Deprecations;
import com.denizenscript.denizencore.utilities.YamlConfiguration;
import com.denizenscript.denizencore.utilities.text.StringHolder;
import org.bukkit.*;
import org.bukkit.event.EventHandler;
import org.bukkit.event.EventPriority;
import org.bukkit.event.Listener;
import org.bukkit.event.block.BlockCookEvent;
import org.bukkit.event.inventory.BrewEvent;
import org.bukkit.event.inventory.BrewingStandFuelEvent;
import org.bukkit.event.inventory.CraftItemEvent;
import org.bukkit.event.inventory.PrepareItemCraftEvent;
import org.bukkit.inventory.*;

import java.math.BigInteger;
import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;
import java.util.*;

public class ItemScriptHelper implements Listener {

    public static final Map<String, ItemScriptContainer> item_scripts = new HashMap<>();
    public static final Map<String, ItemScriptContainer> item_scripts_by_hash_id = new HashMap<>();
    public static final Map<String, ItemScriptContainer> recipeIdToItemScript = new HashMap<>();

    public ItemScriptHelper() {
        Denizen.getInstance().getServer().getPluginManager().registerEvents(this, Denizen.getInstance());
    }

    public static void removeDenizenRecipes() {
        recipeIdToItemScript.clear();
        NMSHandler.getItemHelper().clearDenizenRecipes();
    }

    public String getIdFor(ItemScriptContainer container, String type, int id) {
        String basicId = type + "_" + Utilities.cleanseNamespaceID(container.getName()) + "_" + id;
        if (!recipeIdToItemScript.containsKey(basicId)) {
            recipeIdToItemScript.put("denizen:" + basicId, container);
            return basicId;
        }
        int newNumber = 1;
        String newId = basicId + "_1";
        while (recipeIdToItemScript.containsKey(newId)) {
            newId = basicId + "_" + newNumber++;
        }
        recipeIdToItemScript.put("denizen:" + newId, container);
        return newId;
    }

    public static List<String> splitByNonBracketedSlashes(String str) {
        boolean brackets = false;
        int start = 0;
        List<String> output = new ArrayList<>(4);
        for (int i = 0; i < str.length(); i++) {
            char c = str.charAt(i);
            if (c == '[') {
                brackets = true;
            }
            else if (c == ']') {
                brackets = false;
            }
            else if (c == '/' && !brackets) {
                output.add(str.substring(start, i));
                start = i + 1;
            }
        }
        output.add(str.substring(start));
        return output;
    }

    public ItemStack[] textToItemArray(ItemScriptContainer container, String text, boolean exact) {
        if (CoreUtilities.toLowerCase(text).equals("air")) {
            return new ItemStack[0];
        }
        List<String> ingredientText = splitByNonBracketedSlashes(text);
        List<ItemStack> outputItems = new ArrayList<>(ingredientText.size());
        for (int i = 0; i < ingredientText.size(); i++) {
            String entry = ingredientText.get(i);
            if (ScriptEvent.isAdvancedMatchable(entry)) {
                boolean any = false;
                for (Material material : Material.values()) {
                    if (BukkitScriptEvent.tryMaterial(material, entry)) {
                        outputItems.add(new ItemStack(material, 1));
                        any = true;
                    }
                }
                if (exact) {
                    for (ItemScriptContainer possibleContainer : ItemScriptHelper.item_scripts.values()) {
                        if (BukkitScriptEvent.tryItem(possibleContainer.getCleanReference(), entry)) {
                            outputItems.add(possibleContainer.getCleanReference().getItemStack());
                            any = true;
                        }
                    }
                }
                if (!any) {
                    Debug.echoError("Invalid ItemTag ingredient (empty advanced matcher), recipe will not be registered for item script '" + container.getName() + "': " + entry);
                    return null;
                }
            }
            else {
                ItemTag ingredient = ItemTag.valueOf(entry, container);
                if (ingredient == null) {
                    Debug.echoError("Invalid ItemTag ingredient, recipe will not be registered for item script '" + container.getName() + "': " + entry);
                    return null;
                }
                outputItems.add(ingredient.getItemStack().clone());
            }
        }
        return outputItems.toArray(new ItemStack[0]);
    }

    public void registerShapedRecipe(ItemScriptContainer container, ItemStack item, List<String> recipeList, String internalId, String group) {
        for (int n = 0; n < recipeList.size(); n++) {
            recipeList.set(n, TagManager.tag(ScriptBuilder.stripLinePrefix(recipeList.get(n)), new BukkitTagContext(null, null, new ScriptTag(container))));
        }
        List<ItemStack[]> ingredients = new ArrayList<>();
        List<Boolean> exacts = new ArrayList<>();
        int width = 1;
        for (String recipeRow : recipeList) {
            String[] elements = recipeRow.split("\\|", 3);
            if (width < 3 && elements.length == 3) {
                width = 3;
            }
            if (width < 2 && elements.length >= 2) {
                width = 2;
            }
            for (String element : elements) {
                String itemText = element;
                boolean isExact = !itemText.startsWith("material:");
                if (!isExact) {
                    itemText = itemText.substring("material:".length());
                }
                exacts.add(isExact);
                ItemStack[] items = textToItemArray(container, itemText, isExact);
                if (items == null) {
                    return;
                }
                ingredients.add(items);
            }
        }
        NamespacedKey key = new NamespacedKey("denizen", internalId);
        ShapedRecipe recipe = new ShapedRecipe(key, item);
        recipe.setGroup(group);
        String shape1 = "ABC".substring(0, width);
        String shape2 = "DEF".substring(0, width);
        String shape3 = "GHI".substring(0, width);
        String itemChars = shape1 + shape2 + shape3;
        if (recipeList.size() == 3) {
            recipe = recipe.shape(shape1, shape2, shape3);
        }
        else if (recipeList.size() == 2) {
            recipe = recipe.shape(shape1, shape2);
        }
        else {
            recipe = recipe.shape(shape1);
        }
        for (int i = 0; i < ingredients.size(); i++) {
            if (ingredients.get(i).length != 0) {
                NMSHandler.getItemHelper().setShapedRecipeIngredient(recipe, itemChars.charAt(i), ingredients.get(i), exacts.get(i));
            }
        }
        Bukkit.addRecipe(recipe);
    }

    public void registerShapelessRecipe(ItemScriptContainer container, ItemStack item, String shapelessString, String internalId, String group) {
        TagContext context = new BukkitTagContext(null, null, new ScriptTag(container));
        String list = TagManager.tag(shapelessString, context);
        List<ItemStack[]> ingredients = new ArrayList<>();
        List<Boolean> exacts = new ArrayList<>();
        for (String element : ListTag.valueOf(list, context)) {
            String itemText = element;
            boolean isExact = !itemText.startsWith("material:");
            if (!isExact) {
                itemText = itemText.substring("material:".length());
            }
            exacts.add(isExact);
            ItemStack[] items = textToItemArray(container, itemText, isExact);
            if (items == null) {
                return;
            }
            ingredients.add(items);
        }
        boolean[] bools = new boolean[exacts.size()];
        for (int i = 0; i < exacts.size(); i++) {
            bools[i] = exacts.get(i);
        }
        NMSHandler.getItemHelper().registerShapelessRecipe(internalId, group, item, ingredients, bools);
    }

    public void registerFurnaceRecipe(ItemScriptContainer container, ItemStack item, String furnaceItemString, float exp, int time, String type, String internalId, String group) {
        boolean exact = true;
        if (furnaceItemString.startsWith("material:")) {
            exact = false;
            furnaceItemString = furnaceItemString.substring("material:".length());
        }
        ItemStack[] items = textToItemArray(container, furnaceItemString, exact);
        if (items == null) {
            return;
        }
        NMSHandler.getItemHelper().registerFurnaceRecipe(internalId, group, item, items, exp, time, type, exact);
    }

    public void registerStonecuttingRecipe(ItemScriptContainer container, ItemStack item, String inputItemString, String internalId, String group) {
        boolean exact = true;
        if (inputItemString.startsWith("material:")) {
            exact = false;
            inputItemString = inputItemString.substring("material:".length());
        }
        ItemStack[] items = textToItemArray(container, inputItemString, exact);
        if (items == null) {
            return;
        }
        NMSHandler.getItemHelper().registerStonecuttingRecipe(internalId, group, item, items, exact);
    }

    public void registerSmithingRecipe(ItemScriptContainer container, ItemStack item, String baseItemString, String upgradeItemString, String internalId) {
        boolean baseExact = true;
        if (baseItemString.startsWith("material:")) {
            baseExact = false;
            baseItemString = baseItemString.substring("material:".length());
        }
        ItemStack[] baseItems = textToItemArray(container, baseItemString, baseExact);
        if (baseItems == null) {
            return;
        }
        boolean upgradeExact = true;
        if (upgradeItemString.startsWith("material:")) {
            upgradeExact = false;
            upgradeItemString = upgradeItemString.substring("material:".length());
        }
        ItemStack[] upgradeItems = textToItemArray(container, upgradeItemString, upgradeExact);
        if (upgradeItems == null) {
            return;
        }
        NMSHandler.getItemHelper().registerSmithingRecipe(internalId, item, baseItems, baseExact, upgradeItems, upgradeExact);
    }

    public void rebuildRecipes() {
        for (ItemScriptContainer container : item_scripts.values()) {
            try {
                if (container.contains("recipes", Map.class)) {
                    YamlConfiguration section = container.getConfigurationSection("recipes");
                    int id = 0;
                    for (StringHolder key : section.getKeys(false)) {
                        id++;
                        YamlConfiguration subSection = section.getConfigurationSection(key.str);
                        String type = CoreUtilities.toLowerCase(subSection.getString("type"));
                        String internalId = subSection.contains("recipe_id") ? subSection.getString("recipe_id") : getIdFor(container, type + "_recipe", id);
                        String group = subSection.contains("group") ? subSection.getString("group") : "";
                        ItemStack item = container.getCleanReference().getItemStack().clone();
                        if (subSection.contains("output_quantity")) {
                            item.setAmount(Integer.parseInt(subSection.getString("output_quantity")));
                        }
                        switch (type) {
                            case "shaped":
                                registerShapedRecipe(container, item, subSection.getStringList("input"), internalId, group);
                                break;
                            case "shapeless":
                                registerShapelessRecipe(container, item, subSection.getString("input"), internalId, group);
                                break;
                            case "stonecutting":
                                registerStonecuttingRecipe(container, item, subSection.getString("input"), internalId, group);
                                break;
                            case "furnace":
                            case "blast":
                            case "smoker":
                            case "campfire":
                                float exp = 0;
                                int cookTime = 40;
                                if (subSection.contains("experience")) {
                                    exp = Float.parseFloat(subSection.getString("experience"));
                                }
                                if (subSection.contains("cook_time")) {
                                    cookTime = DurationTag.valueOf(subSection.getString("cook_time"), new BukkitTagContext(container)).getTicksAsInt();
                                }
                                registerFurnaceRecipe(container, item, subSection.getString("input"), exp, cookTime, type, internalId, group);
                                break;
                            case "smithing":
                                registerSmithingRecipe(container, item, subSection.getString("base"), subSection.getString("upgrade"), internalId);
                                break;
                        }
                    }
                }
                // Old script style
                if (container.contains("RECIPE", List.class)) {
                    Deprecations.oldRecipeScript.warn(container);
                    registerShapedRecipe(container, container.getCleanReference().getItemStack().clone(), container.getStringList("RECIPE"), getIdFor(container, "old_recipe", 0), "custom");
                }
                if (container.contains("SHAPELESS_RECIPE", String.class)) {
                    Deprecations.oldRecipeScript.warn(container);
                    registerShapelessRecipe(container, container.getCleanReference().getItemStack().clone(), container.getString("SHAPELESS_RECIPE"), getIdFor(container, "old_shapeless", 0), "custom");
                }
                if (container.contains("FURNACE_RECIPE", String.class)) {
                    Deprecations.oldRecipeScript.warn(container);
                    registerFurnaceRecipe(container, container.getCleanReference().getItemStack().clone(), container.getString("FURNACE_RECIPE"), 0, 40, "furnace", getIdFor(container, "old_furnace", 0), "custom");
                }
            }
            catch (Exception ex) {
                Debug.echoError("Error while rebuilding item script recipes for '" + container.getName() + "'...");
                Debug.echoError(ex);
            }
        }
    }

    @EventHandler
    public void scriptReload(ScriptReloadEvent event) {
        rebuildRecipes();
    }

    public static boolean isItemscript(ItemStack item) {
        return getItemScriptContainer(item) != null;
    }

    public static ItemScriptContainer getItemScriptContainer(ItemStack item) {
        if (item == null) {
            return null;
        }
        CompoundTag tag = NMSHandler.getItemHelper().getNbtData(item);
        String scriptName = tag.getString("DenizenItemScript");
        if (scriptName != null && !scriptName.equals("")) {
            return item_scripts.get(scriptName);
        }
        // TODO: Legacy hashed format
        String nbt = tag.getString("Denizen Item Script");
        if (nbt != null && !nbt.equals("")) {
            return item_scripts_by_hash_id.get(nbt);
        }
        return null;
    }

    public static String ItemScriptHashID = ChatColor.RED.toString() + ChatColor.BLUE + ChatColor.BLACK;

    public static String createItemScriptID(ItemScriptContainer container) {
        String colors = createItemScriptID(container.getName());
        container.setHashID(colors);
        return colors;
    }

    public static String createItemScriptID(String name) {
        String script = name.toUpperCase();
        StringBuilder colors = new StringBuilder();
        colors.append(ItemScriptHashID);
        try {
            MessageDigest md = MessageDigest.getInstance("MD5");
            byte[] bytes = script.getBytes(StandardCharsets.UTF_8);
            md.update(bytes, 0, bytes.length);
            String hash = new BigInteger(1, md.digest()).toString(16);
            for (int i = 0; i < 16; i++) {
                colors.append(ChatColor.COLOR_CHAR).append(hash.charAt(i));
            }
        }
        catch (Exception ex) {
            Debug.echoError(ex);
            colors.append(ChatColor.BLUE);
        }
        return colors.toString();
    }

    public static boolean isAllowedChoice(ItemScriptContainer script, RecipeChoice choice) {
        if (choice instanceof RecipeChoice.ExactChoice) {
            for (ItemStack choiceOpt : ((RecipeChoice.ExactChoice) choice).getChoices()) {
                ItemScriptContainer choiceOptContainer = getItemScriptContainer(choiceOpt);
                if (script == choiceOptContainer) {
                    return true;
                }
            }
        }
        return false;
    }

    private static ItemStack AIR = new ItemStack(Material.AIR);

    public enum DenyCraftReason {
        ALLOWED,
        IMPOSSIBLE,
        NOT_ALLOWED
    }

    public static DenyCraftReason shouldDenyCraft(ItemStack[] items, Recipe recipe) {
        int width = items.length == 9 ? 3 : 2;
        int shapeStartX = 0, shapeStartY = 0;
        if (recipe instanceof ShapedRecipe) {
            String[] shape = ((ShapedRecipe) recipe).getShape();
            if (shape.length != width || shape[0].length() != width) {
                if (shape.length > width || shape[0].length() > width) {
                    return DenyCraftReason.ALLOWED; // Already impossible regardless
                }
                loopStart:
                for (shapeStartX = 0; shapeStartX <= width - shape[0].length(); shapeStartX++) {
                    for (shapeStartY = 0; shapeStartY <= width - shape.length; shapeStartY++) {
                        boolean hasAnyInvalid = false;
                        for (int x = 0; x < shape[0].length(); x++) {
                            for (int y = 0; y < shape.length; y++) {
                                ItemStack item = items[(y + shapeStartY) * width + (x + shapeStartX)];
                                RecipeChoice choice = ((ShapedRecipe) recipe).getChoiceMap().get(shape[y].charAt(x));
                                if (choice != null && !choice.test(item == null ? AIR : item)) {
                                    hasAnyInvalid = true;
                                    break;
                                }
                            }
                        }
                        if (!hasAnyInvalid) {
                            break loopStart;
                        }
                    }
                }
            }
        }
        for (int i = 0; i < items.length; i++) {
            ItemStack item = items[i];
            if (item == null || item.getType() == Material.AIR) {
                continue;
            }
            ItemScriptContainer container = getItemScriptContainer(item);
            if (container == null || container.allowInMaterialRecipes) {
                continue;
            }
            boolean allowed = false;
            if (recipe instanceof ShapelessRecipe) {
                for (RecipeChoice choice : ((ShapelessRecipe) recipe).getChoiceList()) {
                    if (isAllowedChoice(container, choice)) {
                        allowed = true;
                        break;
                    }
                }
            }
            else if (recipe instanceof ShapedRecipe) {
                int x = i % width - shapeStartX;
                int y = i / width - shapeStartY;
                if (x < 0 || y < 0) {
                    return DenyCraftReason.IMPOSSIBLE;
                }
                String[] shape = ((ShapedRecipe) recipe).getShape();
                if (y < shape.length && x < shape[y].length()) {
                    char c = shape[y].charAt(x);
                    RecipeChoice choice = ((ShapedRecipe) recipe).getChoiceMap().get(c);
                    if (isAllowedChoice(container, choice)) {
                        allowed = true;
                    }
                }
            }
            else if (recipe instanceof CookingRecipe) {
                allowed = isAllowedChoice(container, ((CookingRecipe) recipe).getInputChoice());
            }
            else {
                allowed = true; // Shouldn't be possible?
            }
            if (!allowed) {
                return DenyCraftReason.NOT_ALLOWED;
            }
        }
        return DenyCraftReason.ALLOWED;
    }

    public static boolean hasAlternateValidRecipe(Recipe recipe, ItemStack[] items) {
        // Workaround for Spigot bug with the wrong recipe ID getting grabbed
        if (recipe instanceof ShapedRecipe) {
            ItemStack result = recipe.getResult();
            if (isItemscript(result)) {
                for (Recipe altRecipe : Bukkit.getRecipesFor(result)) {
                    if (altRecipe instanceof ShapedRecipe) {
                        if (shouldDenyCraft(items, altRecipe) == DenyCraftReason.ALLOWED) {
                            return true;
                        }
                    }
                }
            }
        }
        return false;
    }

    @EventHandler(priority = EventPriority.LOW)
    public void onCraftPrepared(PrepareItemCraftEvent event) {
        Recipe recipe = event.getRecipe();
        if (recipe == null) {
            return;
        }
        ItemStack[] items = event.getInventory().getMatrix();
        if (shouldDenyCraft(items, recipe) != DenyCraftReason.ALLOWED && !hasAlternateValidRecipe(recipe, items)) {
            event.getInventory().setResult(null);
        }
    }

    @EventHandler(priority = EventPriority.LOW, ignoreCancelled = true)
    public void onItemCrafted(CraftItemEvent event) {
        Recipe recipe = event.getRecipe();
        ItemStack[] items = event.getInventory().getMatrix();
        if (shouldDenyCraft(items, recipe) != DenyCraftReason.ALLOWED && !hasAlternateValidRecipe(recipe, items)) {
            event.setCancelled(true);
        }
    }

    @EventHandler(priority = EventPriority.LOW, ignoreCancelled = true)
    public void onItemCooked(BlockCookEvent event) {
        ItemScriptContainer container = getItemScriptContainer(event.getSource());
        if (container == null || container.allowInMaterialRecipes) {
            return;
        }
        ItemStack[] stacks = new ItemStack[] { event.getSource() };
        for (Recipe recipe : Bukkit.getRecipesFor(event.getResult())) {
            if (recipe instanceof CookingRecipe && shouldDenyCraft(stacks, recipe) == DenyCraftReason.ALLOWED) {
                return;
            }
        }
        event.setCancelled(true);
    }

    public boolean isAllowedToCraftWith(ItemStack item) {
        if (item == null || item.getType() == Material.AIR) {
            return true;
        }
        ItemScriptContainer container = getItemScriptContainer(item);
        if (container == null) {
            return true;
        }
        return container.allowInMaterialRecipes;
    }

    @EventHandler(priority = EventPriority.LOW)
    public void onBrewingStandBrews(BrewEvent event) {
        for (ItemStack item : event.getContents()) {
            if (!isAllowedToCraftWith(item)) {
                event.setCancelled(true);
                return;
            }
        }
    }

    @EventHandler(priority = EventPriority.LOW)
    public void onBrewingStandFuel(BrewingStandFuelEvent event) {
        if (!isAllowedToCraftWith(event.getFuel())) {
            event.setCancelled(true);
        }
    }
}