prestonbane/src/engine/objects/MobEquipment.java

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


package engine.objects;

import engine.exception.SerializationException;
import engine.gameManager.PowersManager;
import engine.net.ByteBufferWriter;
import engine.powers.EffectsBase;
import engine.powers.poweractions.AbstractPowerAction;
import org.pmw.tinylog.Logger;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.concurrent.atomic.AtomicInteger;

public class MobEquipment extends AbstractGameObject {

    private static AtomicInteger equipCounter = new AtomicInteger(0);
    private final ItemBase itemBase;
    private int slot;
    private int parentID;

    //effects
    private boolean enchanted;
    private boolean isID = false;
    private AbstractPowerAction prefix;
    private AbstractPowerAction suffix;
    private int pValue;
    private int sValue;
    private int magicValue;

    private float dropChance = 0;
    public int templateID;
    public ItemTemplate template;

    /**
     * No Id Constructor
     */
    public MobEquipment(ItemBase itemBase, int slot, int parentID) {
        super(MobEquipment.getNewID());
        this.itemBase = itemBase;
        this.templateID = this.itemBase.getUUID();
        this.template = ItemTemplate.itemTemplates.get(templateID);
        this.slot = slot;
        this.parentID = parentID;
        this.enchanted = false;
        this.prefix = null;
        this.suffix = null;
        this.pValue = 0;
        this.sValue = 0;
        setMagicValue();
    }

    public MobEquipment(ItemBase itemBase, int slot, int parentID, String pIDString, String sIDString, int pValue, int sValue) {
        super(MobEquipment.getNewID());
        this.itemBase = itemBase;
        this.templateID = this.itemBase.getUUID();
        this.template = ItemTemplate.itemTemplates.get(templateID);
        this.slot = slot;
        this.parentID = parentID;

        //add effects
        this.prefix = PowersManager.getPowerActionByIDString(pIDString);
        this.suffix = PowersManager.getPowerActionByIDString(sIDString);

        this.pValue = pValue;
        this.sValue = sValue;
        this.enchanted = this.prefix == null || this.suffix == null;
        setMagicValue();
    }

    /**
     * ResultSet Constructor
     */
    public MobEquipment(ResultSet rs) throws SQLException {
        super(MobEquipment.getNewID());
        int itemBaseID = rs.getInt("ItemID");
        this.itemBase = ItemBase.getItemBase(itemBaseID);
        this.templateID = this.itemBase.getUUID();
        this.template = ItemTemplate.itemTemplates.get(itemBaseID);
        this.slot = rs.getInt("slot");
        this.parentID = rs.getInt("mobID");
        setMagicValue();
    }


    public MobEquipment(int itemBaseID, float dropChance) {
        super(MobEquipment.getNewID());
        this.itemBase = ItemBase.getItemBase(itemBaseID);
        this.template = ItemTemplate.itemTemplates.get(itemBaseID);

        if (this.itemBase != null)
            this.slot = this.itemBase.getValidSlot();
        else {
            Logger.error("Failed to find Itembase for ID : " + itemBaseID);
            this.slot = 0;
        }

        this.dropChance = dropChance;

        this.parentID = 0;
        setMagicValue();
    }

    public static int getNewID() {
        return MobEquipment.equipCounter.incrementAndGet();
    }

    public static void serializeForVendor(MobEquipment mobEquipment, ByteBufferWriter writer, float percent) throws SerializationException {
        _serializeForClientMsg(mobEquipment, writer, false);
        int baseValue = mobEquipment.itemBase.getBaseValue() + mobEquipment.itemBase.getMagicValue();
        writer.putInt(mobEquipment.magicValue);
        writer.putInt(mobEquipment.magicValue);
    }

    public static void serializeForClientMsg(MobEquipment mobEquipment, ByteBufferWriter writer) throws SerializationException {
        _serializeForClientMsg(mobEquipment, writer, true);
    }

    public static void _serializeForClientMsg(MobEquipment mobEquipment, ByteBufferWriter writer, boolean useSlot) throws SerializationException {

        if (useSlot)
            writer.putInt(mobEquipment.slot);
        writer.putInt(0); // Pad
        writer.putInt(mobEquipment.itemBase.getUUID());
        writer.putInt(mobEquipment.getObjectType().ordinal());
        writer.putInt(mobEquipment.getObjectUUID());

        // Unknown statics
        for (int i = 0; i < 3; i++) {
            writer.putInt(0); // Pad
        }
        for (int i = 0; i < 4; i++) {
            writer.putInt(0x3F800000); // Static
        }
        for (int i = 0; i < 5; i++) {
            writer.putInt(0); // Pad
        }
        for (int i = 0; i < 2; i++) {
            writer.putInt(0xFFFFFFFF); // Static
        }

        writer.putInt(0);

        writer.put((byte) 1); // End Datablock byte
        writer.putInt(0); // Unknown. pad?
        writer.put((byte) 1); // End Datablock byte

        writer.putFloat(mobEquipment.template.item_health_full);
        writer.putFloat(mobEquipment.template.item_health_full);

        writer.put((byte) 1); // End Datablock byte

        writer.putInt(0); // Pad
        writer.putInt(0); // Pad

        writer.putInt(mobEquipment.itemBase.getBaseValue());
        writer.putInt(mobEquipment.magicValue);

        serializeEffects(mobEquipment, writer);

        writer.putInt(0x00000000);

        //name color, think mobEquipment is where mobEquipment goes
        if (mobEquipment.enchanted)
            if (mobEquipment.isID)
                writer.putInt(36);
            else
                writer.putInt(40);
        else
            writer.putInt(4);

        writer.putInt(0);
        writer.putInt(0); // Pad
        writer.putInt(1);
        writer.putShort((short) 0);
        writer.put((byte) 0);
    }

    public static void serializeEffects(MobEquipment mobEquipment, ByteBufferWriter writer) {

        //skip sending effects if not IDed
        if (!mobEquipment.isID) {
            writer.putInt(0);
            return;
        }

        //handle effect count
        int cnt = 0;
        EffectsBase pre = null;
        EffectsBase suf = null;
        if (mobEquipment.prefix != null) {
            pre = PowersManager.getEffectByIDString(mobEquipment.prefix.getIDString());
            if (pre != null)
                cnt++;
        }
        if (mobEquipment.suffix != null) {
            suf = PowersManager.getEffectByIDString(mobEquipment.suffix.getIDString());
            if (suf != null)
                cnt++;
        }

        writer.putInt(cnt);

        //serialize prefix
        if (pre != null)
            serializeEffect(mobEquipment, writer, pre, mobEquipment.pValue);

        //serialize suffix
        if (suf != null)
            serializeEffect(mobEquipment, writer, suf, mobEquipment.sValue);
    }

    public static void serializeEffect(MobEquipment mobEquipment, ByteBufferWriter writer, EffectsBase eb, int rank) {
        String name;
        if (eb.isPrefix()) {
            if (mobEquipment.itemBase == null)
                name = eb.getName();
            else
                name = eb.getName() + ' ' + ItemTemplate.itemTemplates.get(mobEquipment.itemBase.getUUID()).item_base_name;
        } else if (eb.isSuffix()) {
            if (mobEquipment.itemBase == null)
                name = eb.getName();
            else
                name = ItemTemplate.itemTemplates.get(mobEquipment.itemBase.getUUID()).item_base_name + ' ' + eb.getName();
        } else {
            if (mobEquipment.itemBase == null)
                name = "";
            else
                name = ItemTemplate.itemTemplates.get(mobEquipment.itemBase.getUUID()).item_base_name;
        }

        writer.putInt(eb.getToken());
        writer.putInt(rank);
        writer.putInt(1);
        writer.put((byte) 1);
        writer.putInt(mobEquipment.getObjectType().ordinal());
        writer.putInt(mobEquipment.getObjectUUID());
        writer.putString(name);
        writer.putFloat(-1000f);
    }

    public ItemBase getItemBase() {
        return itemBase;
    }

    public int getSlot() {
        return this.slot;
    }

    public void setSlot(int value) {
        this.slot = value;
    }

    public final void setMagicValue() {
        float value = 1;
        if (itemBase != null)
            value = itemBase.getBaseValue();
        if (this.prefix != null) {
            if (this.prefix.getEffectsBase() != null)
                value += this.prefix.getEffectsBase().getValue();
            if (this.prefix.getEffectsBase2() != null)
                value += this.prefix.getEffectsBase2().getValue();
        }
        if (this.suffix != null) {
            if (this.suffix.getEffectsBase() != null)
                value += this.suffix.getEffectsBase().getValue();
            if (this.suffix.getEffectsBase2() != null)
                value += this.suffix.getEffectsBase2().getValue();
        }

        if (itemBase != null)

            for (Integer token : itemBase.getBakedInStats().keySet()) {

                EffectsBase effect = PowersManager.getEffectByToken(token);

                AbstractPowerAction apa = PowersManager.getPowerActionByIDString(effect.getIDString());
                if (apa.getEffectsBase() != null)
                    if (apa.getEffectsBase().getValue() > 0) {
                        //System.out.println(apa.getEffectsBase().getValue());
                        value += apa.getEffectsBase().getValue();
                    }


                if (apa.getEffectsBase2() != null)
                    value += apa.getEffectsBase2().getValue();
            }

        this.magicValue = (int) value;
    }

    public int getMagicValue() {

        if (!this.isID) {
            return itemBase.getBaseValue();
        }
        return this.magicValue;
    }

    public void setPrefix(String pIDString, int pValue) {
        AbstractPowerAction apa = PowersManager.getPowerActionByIDString(pIDString);
        if (apa != null) {
            this.prefix = apa;
            this.pValue = pValue;
        } else
            this.prefix = null;

        this.enchanted = this.prefix != null || this.suffix != null;

        setMagicValue();
    }

    public void setSuffix(String sIDString, int sValue) {
        AbstractPowerAction apa = PowersManager.getPowerActionByIDString(sIDString);
        if (apa != null) {
            this.suffix = apa;
            this.sValue = sValue;
        } else
            this.suffix = null;

        this.enchanted = this.prefix != null || this.suffix != null;

        setMagicValue();
    }

    public void setIsID(boolean value) {
        this.isID = value;
    }

    public boolean isID() {
        return this.isID;
    }

    public void transferEnchants(Item item) {
        if (this.prefix != null) {
            String IDString = this.prefix.getIDString();
            item.addPermanentEnchantment(IDString, this.pValue);
        }
        if (this.suffix != null) {
            String IDString = this.suffix.getIDString();
            item.addPermanentEnchantment(IDString, this.sValue);
        }
        if (this.isID)
            item.setIsID(true);
    }

    /*
     * Database
     */
    @Override
    public void updateDatabase() {
    }


    public float getDropChance() {
        return dropChance;
    }

    public void setDropChance(float dropChance) {
        this.dropChance = dropChance;
    }
}