Server/src/engine/db/handlers/dbCharacterPowerHandler.java

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


package engine.db.handlers;

import engine.gameManager.DbManager;
import engine.mbEnums;
import engine.objects.CharacterPower;
import engine.objects.PlayerCharacter;
import engine.server.MBServerStatics;
import org.pmw.tinylog.Logger;

import java.sql.*;
import java.util.concurrent.ConcurrentHashMap;

public class dbCharacterPowerHandler extends dbHandlerBase {

    public dbCharacterPowerHandler() {
        this.localClass = CharacterPower.class;
        this.localObjectType = mbEnums.GameObjectType.valueOf(this.localClass.getSimpleName());
    }

    public CharacterPower ADD_CHARACTER_POWER(CharacterPower toAdd) {

        CharacterPower characterPower = null;

        if (CharacterPower.getOwner(toAdd) == null || toAdd.getPower() == null) {
            Logger.error("dbCharacterSkillHandler.ADD_Power", toAdd.getObjectUUID() + " missing owner or powersBase");
            return null;
        }

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("INSERT INTO `dyn_character_power` (`CharacterID`, `powersBaseToken`, `trains`) VALUES (?, ?, ?);", Statement.RETURN_GENERATED_KEYS)) {

            preparedStatement.setLong(1, CharacterPower.getOwner(toAdd).getObjectUUID());
            preparedStatement.setInt(2, toAdd.getPower().getToken());
            preparedStatement.setInt(3, toAdd.getTrains());

            preparedStatement.executeUpdate();
            ResultSet rs = preparedStatement.getGeneratedKeys();

            if (rs.next())
                characterPower = GET_CHARACTER_POWER(rs.getInt(1));

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

    public int DELETE_CHARACTER_POWER(final int objectUUID) {

        int rowCount = 0;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("DELETE FROM `dyn_character_power` WHERE `UID` = ?")) {

            preparedStatement.setLong(1, objectUUID);
            rowCount = preparedStatement.executeUpdate();

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

        return rowCount;
    }

    public CharacterPower GET_CHARACTER_POWER(int objectUUID) {

        CharacterPower characterPower = (CharacterPower) DbManager.getFromCache(mbEnums.GameObjectType.CharacterPower, objectUUID);

        if (characterPower != null)
            return characterPower;

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

            preparedStatement.setLong(1, objectUUID);
            ResultSet rs = preparedStatement.executeQuery();
            characterPower = (CharacterPower) getObjectFromRs(rs);

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

        return characterPower;
    }

    public ConcurrentHashMap<Integer, CharacterPower> GET_POWERS_FOR_CHARACTER(PlayerCharacter pc) {

        ConcurrentHashMap<Integer, CharacterPower> powers = new ConcurrentHashMap<>(MBServerStatics.CHM_INIT_CAP, MBServerStatics.CHM_LOAD, MBServerStatics.CHM_THREAD_LOW);
        int objectUUID = pc.getObjectUUID();

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

            preparedStatement.setLong(1, objectUUID);
            ResultSet rs = preparedStatement.executeQuery();

            while (rs.next()) {
                CharacterPower cp = new CharacterPower(rs, pc);
                if (cp.getPower() != null)
                    powers.put(cp.getPower().getToken(), cp);
            }

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

        return powers;
    }

    public void UPDATE_TRAINS(final CharacterPower characterPower) {

        //skip update if nothing changed

        if (!characterPower.isTrained())
            return;

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("UPDATE `dyn_character_power` SET `trains`=? WHERE `UID`=?")) {

            preparedStatement.setShort(1, (short) characterPower.getTrains());
            preparedStatement.setInt(2, characterPower.getObjectUUID());

            preparedStatement.execute();

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

        characterPower.setTrained(false);
    }

    public void updateDatabase(final CharacterPower pow) {

        if (pow.getPower() == null) {
            Logger.error("Failed to find powersBase for Power " + pow.getObjectUUID());
            return;
        }

        if (CharacterPower.getOwner(pow) == null) {
            Logger.error("Failed to find owner for Power " + pow.getObjectUUID());
            return;
        }

        try (Connection connection = DbManager.getConnection();
             PreparedStatement preparedStatement = connection.prepareStatement("UPDATE `dyn_character_power` SET `PowersBaseToken`=?, `CharacterID`=?, `trains`=? WHERE `UID`=?")) {

            preparedStatement.setInt(1, pow.getPower().getToken());
            preparedStatement.setInt(2, CharacterPower.getOwner(pow).getObjectUUID());
            preparedStatement.setShort(3, (short) pow.getTrains());
            preparedStatement.setInt(4, pow.getObjectUUID());
            preparedStatement.execute();

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

        pow.setTrained(false);
    }
}