Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/crafting/CraftingManager.java

package com.eu.habbo.habbohotel.crafting;

import com.eu.habbo.Emulator;
import com.eu.habbo.habbohotel.items.Item;
import gnu.trove.map.hash.THashMap;
import gnu.trove.procedure.TObjectProcedure;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;

public class CraftingManager {
    private static final Logger LOGGER = LoggerFactory.getLogger(CraftingManager.class);

    private final THashMap<Item, CraftingAltar> altars;

    public CraftingManager() {
        this.altars = new THashMap<>();

        this.reload();
    }

    public void reload() {
        this.dispose();

        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("SELECT * FROM crafting_altars_recipes " +
                "INNER JOIN crafting_recipes ON crafting_altars_recipes.recipe_id = crafting_recipes.id " +
                "INNER JOIN crafting_recipes_ingredients ON crafting_recipes.id = crafting_recipes_ingredients.recipe_id " +
                "WHERE crafting_recipes.enabled = ? ORDER BY altar_id ASC")) {
            statement.setString(1, "1");
            try (ResultSet set = statement.executeQuery()) {
                while (set.next()) {
                    Item item = Emulator.getGameEnvironment().getItemManager().getItem(set.getInt("altar_id"));

                    if (item != null) {
                        if (!this.altars.containsKey(item)) {
                            this.altars.put(item, new CraftingAltar(item));
                        }

                        CraftingAltar altar = this.altars.get(item);

                        if (altar != null) {
                            CraftingRecipe recipe = altar.getRecipe(set.getInt("crafting_recipes_ingredients.recipe_id"));

                            if (recipe == null) {
                                recipe = new CraftingRecipe(set);
                                altar.addRecipe(recipe);
                            }

                            Item ingredientItem = Emulator.getGameEnvironment().getItemManager().getItem(set.getInt("crafting_recipes_ingredients.item_id"));

                            if (ingredientItem != null) {
                                recipe.addIngredient(ingredientItem, set.getInt("crafting_recipes_ingredients.amount"));
                                altar.addIngredient(ingredientItem);
                            } else {
                                LOGGER.error("Unknown ingredient item " + set.getInt("crafting_recipes_ingredients.item_id"));
                            }
                        }
                    }
                }
            }
        } catch (SQLException e) {
            LOGGER.error("Caught SQL exception", e);
        }
    }

    public int getRecipesWithItemCount(final Item item) {
        final int[] i = {0};

        synchronized (this.altars) {
            this.altars.forEachValue(new TObjectProcedure<CraftingAltar>() {
                @Override
                public boolean execute(CraftingAltar altar) {
                    if (altar.hasIngredient(item)) {
                        i[0]++;
                    }

                    return true;
                }
            });
        }

        return i[0];
    }

    public CraftingRecipe getRecipe(String recipeName) {
        CraftingRecipe recipe;
        for (CraftingAltar altar : this.altars.values()) {
            recipe = altar.getRecipe(recipeName);

            if (recipe != null) {
                return recipe;
            }
        }

        return null;
    }

    public CraftingAltar getAltar(Item item) {
        return this.altars.get(item);
    }

    public void dispose() {
        try (Connection connection = Emulator.getDatabase().getDataSource().getConnection(); PreparedStatement statement = connection.prepareStatement("UPDATE crafting_recipes SET remaining = ? WHERE id = ? LIMIT 1")) {
            for (CraftingAltar altar : this.altars.values()) {
                for (CraftingRecipe recipe : altar.getRecipes()) {
                    if (recipe.isLimited()) {
                        statement.setInt(1, recipe.getRemaining());
                        statement.setInt(2, recipe.getId());

                        statement.addBatch();
                    }
                }
            }
            statement.executeBatch();
        } catch (SQLException e) {
            LOGGER.error("Caught SQL exception", e);
        }

        this.altars.clear();
    }
}