prestonbane/src/engine/db/handlers/dbRuneBaseHandler.java

// • ▌ ▄ ·.  ▄▄▄·  ▄▄ • ▪   ▄▄· ▄▄▄▄·  ▄▄▄·  ▐▄▄▄  ▄▄▄ .
// ·██ ▐███▪▐█ ▀█ ▐█ ▀ ▪██ ▐█ ▌▪▐█ ▀█▪▐█ ▀█ •█▌ ▐█▐▌·
// ▐█ ▌▐▌▐█·▄█▀▀█ ▄█ ▀█▄▐█·██ ▄▄▐█▀▀█▄▄█▀▀█ ▐█▐ ▐▌▐▀▀▀
// ██ ██▌▐█▌▐█ ▪▐▌▐█▄▪▐█▐█▌▐███▌██▄▪▐█▐█ ▪▐▌██▐ █▌▐█▄▄▌
// ▀▀  █▪▀▀▀ ▀  ▀ ·▀▀▀▀ ▀▀▀·▀▀▀ ·▀▀▀▀  ▀  ▀ ▀▀  █▪ ▀▀▀
//      Magicbane Emulator Project © 2013 - 2022
//                www.magicbane.com


package engine.db.handlers;

import engine.gameManager.DbManager;
import engine.objects.RuneBase;
import engine.powers.RunePowerEntry;
import engine.powers.RuneSkillAdjustEntry;
import org.pmw.tinylog.Logger;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.HashMap;

public class dbRuneBaseHandler extends dbHandlerBase {

    public dbRuneBaseHandler() {
        this.localClass = RuneBase.class;
        this.localObjectType = engine.Enum.GameObjectType.valueOf(this.localClass.getSimpleName());
    }

    public static HashMap<Integer, ArrayList<RunePowerEntry>> LOAD_RUNE_POWERS() {

        HashMap<Integer, ArrayList<RunePowerEntry>> mobPowers = new HashMap<>();
        RunePowerEntry runePowerEntry;

        int rune_id;
        int recordsRead = 0;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM static_rune_powers")) {

            ResultSet rs = preparedStatement.executeQuery();

            while (rs.next()) {

                recordsRead++;

                rune_id = rs.getInt("rune_id");
                runePowerEntry = new RunePowerEntry(rs);

                if (mobPowers.get(rune_id) == null) {
                    ArrayList<RunePowerEntry> runePowerList = new ArrayList<>();
                    runePowerList.add(runePowerEntry);
                    mobPowers.put(rune_id, runePowerList);
                } else {
                    ArrayList<RunePowerEntry> powerList = mobPowers.get(rune_id);
                    powerList.add(runePowerEntry);
                    mobPowers.put(rune_id, powerList);
                }
            }
        } catch (SQLException e) {
            Logger.error(e);
            return mobPowers;
        }

        Logger.info("read: " + recordsRead + " cached: " + mobPowers.size());
        return mobPowers;
    }

    public static HashMap<Integer, ArrayList<RuneSkillAdjustEntry>> LOAD_RUNE_SKILL_ADJUSTS() {

        HashMap<Integer, ArrayList<RuneSkillAdjustEntry>> runeSkillAdjusts = new HashMap<>();
        RuneSkillAdjustEntry runeSkillAdjustEntry;

        int rune_id;
        int recordsRead = 0;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM static_rune_skill_adjusts")) {

            ResultSet rs = preparedStatement.executeQuery();

            while (rs.next()) {

                recordsRead++;

                rune_id = rs.getInt("rune_id");
                runeSkillAdjustEntry = new RuneSkillAdjustEntry(rs);

                if (runeSkillAdjusts.get(rune_id) == null) {
                    ArrayList<RuneSkillAdjustEntry> skillAdjustList = new ArrayList<>();
                    skillAdjustList.add(runeSkillAdjustEntry);
                    runeSkillAdjusts.put(rune_id, skillAdjustList);
                } else {
                    ArrayList<RuneSkillAdjustEntry> powerList = runeSkillAdjusts.get(rune_id);
                    powerList.add(runeSkillAdjustEntry);
                    runeSkillAdjusts.put(rune_id, powerList);
                }
            }
        } catch (SQLException e) {
            Logger.error(e);
            return runeSkillAdjusts;
        }

        Logger.info("read: " + recordsRead + " cached: " + runeSkillAdjusts.size());
        return runeSkillAdjusts;
    }

    public void GET_RUNE_REQS(final RuneBase rb) {

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM `static_rune_runereq` WHERE `runeID` = ?")) {

            preparedStatement.setInt(1, rb.getObjectUUID());

            ResultSet rs = preparedStatement.executeQuery();

            while (rs.next()) {
                int type = rs.getInt("type");

                switch (type) {
                    case 1:
                        rb.getRace().put(rs.getInt("requiredRuneID"), rs.getBoolean("isAllowed"));
                        break;
                    case 2:
                        rb.getBaseClass().put(rs.getInt("requiredRuneID"), rs.getBoolean("isAllowed"));
                        break;
                    case 3:
                        rb.getPromotionClass().put(rs.getInt("requiredRuneID"), rs.getBoolean("isAllowed"));
                        break;
                    case 4:
                        rb.getDiscipline().put(rs.getInt("requiredRuneID"), rs.getBoolean("isAllowed"));
                        break;
                    case 5:
                        rb.getOverwrite().add(rs.getInt("requiredRuneID"));
                        break;
                    case 6:
                        rb.setLevelRequired(rs.getInt("requiredRuneID"));
                        break;
                }
            }

        } catch (SQLException e) {
            Logger.error(e);
        }

    }

    public RuneBase GET_RUNEBASE(final int id) {

        RuneBase runeBase = null;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM `static_rune_runebase` WHERE `ID` = ?")) {

            preparedStatement.setInt(1, id);

            ResultSet rs = preparedStatement.executeQuery();
            runeBase = (RuneBase) getObjectFromRs(rs);

        } catch (SQLException e) {
            Logger.error(e);
        }
        return runeBase;
    }

    public ArrayList<RuneBase> LOAD_ALL_RUNEBASES() {

        ArrayList<RuneBase> runeBasesList = new ArrayList<>();

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM `static_rune_runebase`;")) {

            ResultSet rs = preparedStatement.executeQuery();
            runeBasesList = getObjectsFromRs(rs, 1000);

        } catch (SQLException e) {
            Logger.error(e);
        }

        return runeBasesList;
    }

    public HashMap<Integer, ArrayList<Integer>> LOAD_ALLOWED_STARTING_RUNES_FOR_BASECLASS() {

        HashMap<Integer, ArrayList<Integer>> runeSets = new HashMap<>();

        int recordsRead = 0;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM static_rune_baseclassrune")) {

            ResultSet rs = preparedStatement.executeQuery();

            while (rs.next()) {

                recordsRead++;

                int baseClassID = rs.getInt("BaseClassesID");
                int runeBaseID = rs.getInt("RuneBaseID");

                if (runeSets.get(baseClassID) == null) {
                    ArrayList<Integer> runeList = new ArrayList<>();
                    runeList.add(runeBaseID);
                    runeSets.put(baseClassID, runeList);
                } else {
                    ArrayList<Integer> runeList = runeSets.get(baseClassID);
                    runeList.add(runeBaseID);
                    runeSets.put(baseClassID, runeList);
                }
            }

        } catch (SQLException e) {
            Logger.error(e);
        }

        Logger.info("read: " + recordsRead + " cached: " + runeSets.size());

        return runeSets;
    }

    public HashMap<Integer, ArrayList<Integer>> LOAD_ALLOWED_STARTING_RUNES_FOR_RACE() {

        HashMap<Integer, ArrayList<Integer>> runeSets;

        runeSets = new HashMap<>();
        int recordsRead = 0;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("SELECT * FROM static_rune_racerune")) {

            ResultSet rs = preparedStatement.executeQuery();

            while (rs.next()) {

                recordsRead++;

                int raceID = rs.getInt("RaceID");
                int runeBaseID = rs.getInt("RuneBaseID");

                if (runeSets.get(raceID) == null) {
                    ArrayList<Integer> runeList = new ArrayList<>();
                    runeList.add(runeBaseID);
                    runeSets.put(raceID, runeList);
                } else {
                    ArrayList<Integer> runeList = runeSets.get(raceID);
                    runeList.add(runeBaseID);
                    runeSets.put(raceID, runeList);
                }
            }
        } catch (SQLException e) {
            Logger.error(e);
        }

        Logger.info("read: " + recordsRead + " cached: " + runeSets.size());
        return runeSets;
    }
}