Arcturus-Community/src/main/java/com/eu/habbo/habbohotel/items/CrackableReward.java

package com.eu.habbo.habbohotel.items;

import com.eu.habbo.Emulator;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.AbstractMap;
import java.util.HashMap;
import java.util.Map;

public class CrackableReward {
    public final int itemId;
    public final int count;
    public final Map<Integer, Map.Entry<Integer, Integer>> prizes;
    public final String achievementTick;
    public final String achievementCracked;
    public final int requiredEffect;
    public final int subscriptionDuration;
    public final RedeemableSubscriptionType subscriptionType;
    public int totalChance;

    public CrackableReward(ResultSet set) throws SQLException {
        this.itemId = set.getInt("item_id");
        this.count = set.getInt("count");
        this.achievementTick = set.getString("achievement_tick");
        this.achievementCracked = set.getString("achievement_cracked");
        this.requiredEffect = set.getInt("required_effect");
        this.subscriptionDuration = set.getInt("subscription_duration");
        this.subscriptionType = RedeemableSubscriptionType.fromString(set.getString("subscription_type"));


        String[] prizes = set.getString("prizes").split(";");
        this.prizes = new HashMap<>();

        if (set.getString("prizes").isEmpty()) return;

        this.totalChance = 0;
        for (String prize : prizes) {
            try {
                int itemId = 0;
                int chance = 100;

                if (prize.contains(":") && prize.split(":").length == 2) {
                    itemId = Integer.valueOf(prize.split(":")[0]);
                    chance = Integer.valueOf(prize.split(":")[1]);
                } else if (prize.contains(":")) {
                    Emulator.getLogging().logErrorLine("Invalid configuration of crackable prizes (item id: " + this.itemId + "). '" + prize + "' format should be itemId:chance.");
                } else {
                    itemId = Integer.valueOf(prize.replace(":", ""));
                }

                this.prizes.put(itemId, new AbstractMap.SimpleEntry<>(this.totalChance, this.totalChance + chance));
                this.totalChance += chance;
            } catch (Exception e) {
                logger.error("Caught exception", e);
            }
        }
    }

    public int getRandomReward() {
        if (this.prizes.size() == 0) return 0;

        int random = Emulator.getRandom().nextInt(this.totalChance);

        int notFound = 0;
        for (Map.Entry<Integer, Map.Entry<Integer, Integer>> set : this.prizes.entrySet()) {
            notFound = set.getKey();
            if (random >= set.getValue().getKey() && random < set.getValue().getValue()) {
                return set.getKey();
            }
        }

        return notFound;
    }
}