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

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


package engine.db.handlers;

import engine.Enum.ItemContainerType;
import engine.Enum.ItemType;
import engine.Enum.OwnerType;
import engine.objects.*;
import org.pmw.tinylog.Logger;

import java.util.ArrayList;
import java.util.HashSet;


public class dbItemHandler extends dbHandlerBase {

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

	public Item ADD_ITEM(Item toAdd) {
		prepareCallable("CALL `item_CREATE`(?, ?, ?, ?, ?, ?, ?, ?, ?,?);");
		setInt(1, toAdd.getOwnerID());
		setInt(2, toAdd.getItemBaseID());
		setInt(3, toAdd.getChargesRemaining());
		setInt(4, toAdd.getDurabilityCurrent());
		setInt(5, toAdd.getDurabilityMax());
		if (toAdd.getNumOfItems() < 1)
			setInt(6, 1);
		else
			setInt(6, toAdd.getNumOfItems());

		switch (toAdd.containerType) {
			case INVENTORY:
				setString(7, "inventory");
				break;
			case EQUIPPED:
				setString(7, "equip");
				break;
			case BANK:
				setString(7, "bank");
				break;
			case VAULT:
				setString(7, "vault");
				break;
			case FORGE:
				setString(7, "forge");
				break;
				default:
					setString(7, "none"); //Shouldn't be here
					break;
		}

		setByte(8, toAdd.getEquipSlot());
		setInt(9, toAdd.getFlags());
		setString(10, toAdd.getCustomName());
		int objectUUID = (int) getUUID();

		if (objectUUID > 0)
			return GET_ITEM(objectUUID);
		return null;
	}

	public String GET_OWNER(int ownerID) {
		prepareCallable("SELECT `type` FROM `object` WHERE `UID`=?");
		setLong(1, (long) ownerID);
		return getString("type");
	}

	public boolean DO_TRADE(HashSet<Integer> from1, HashSet<Integer> from2,
			CharacterItemManager man1, CharacterItemManager man2,
			Item inventoryGold1, Item inventoryGold2, int goldFrom1, int goldFrom2) {

		AbstractCharacter ac1 = man1.getOwner();
		AbstractCharacter ac2 = man2.getOwner();
		if (ac1 == null || ac2 == null || inventoryGold1 == null || inventoryGold2 == null)
			return false;

		prepareCallable("CALL `item_TRADE`(?, ?, ?, ?, ?, ?, ?, ?)");
		setString(1, formatTradeString(from1));
		setLong(2, (long) ac1.getObjectUUID());
		setString(3, formatTradeString(from2));
		setLong(4, (long) ac2.getObjectUUID());
		setInt(5, goldFrom1);
		setLong(6, (long) inventoryGold1.getObjectUUID());
		setInt(7, goldFrom2);
		setLong(8, (long) inventoryGold2.getObjectUUID());
        return worked();
	}

	private static String formatTradeString(HashSet<Integer> list) {
		int size = list.size();

		String ret = "";
		if (size == 0)
			return ret;
		boolean start = true;
		for (int i : list) {
			if (start){
				ret += i;
				start = false;
			}
			else
			ret += "," + i;
		}
		return ret;
	}

	public ArrayList<Item> GET_EQUIPPED_ITEMS(final int targetId) {
		prepareCallable("SELECT `obj_item`.*, `object`.`parent`, `object`.`type` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE `object`.`parent`=? && `obj_item`.`item_container`='equip';");
		setLong(1, (long) targetId);
		return getObjectList();
	}

	public Item GET_ITEM(final int id) {


		prepareCallable("SELECT `obj_item`.*, `object`.`parent`, `object`.`type` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE `object`.`UID`=?;");
		setLong(1, (long) id);
		return (Item) getObjectSingle(id);
	}

	public Item GET_GOLD_FOR_PLAYER(final int playerID, final int goldID, int worldID) {
		prepareCallable("SELECT `obj_item`.*, `object`.`parent`, `object`.`type` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE `object`.`parent`=? AND `obj_item`.`item_itembaseID`=?;");
		setInt(1, playerID);
		setInt(2, goldID);
		int objectUUID = (int) getUUID();
		return (Item) getObjectSingle(objectUUID);

	}

	public ArrayList<Item> GET_ITEMS_FOR_ACCOUNT(final int accountId) {
		prepareCallable("SELECT `obj_item`.*, `object`.`parent`, `object`.`type` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE `object`.`parent`=?;");
		setLong(1, (long) accountId);
		return getObjectList();
	}

	public ArrayList<Item> GET_ITEMS_FOR_NPC(final int npcId) {
		prepareCallable("SELECT `obj_item`.*, `object`.`parent`, `object`.`type` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE `object`.`parent`=?");
		setLong(1, (long) npcId);
		return getObjectList();
	}

	public ArrayList<Item> GET_ITEMS_FOR_PC(final int id) {
		prepareCallable("SELECT `obj_item`.*, `object`.`parent`, `object`.`type` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE `object`.`parent`=?");
		setLong(1, (long) id);
		return getLargeObjectList();
	}

	public ArrayList<Item> GET_ITEMS_FOR_PLAYER_AND_ACCOUNT(final int playerID, final int accountID) {
		prepareCallable("SELECT `obj_item`.*, `object`.`parent`, `object`.`type` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE (`object`.`parent`=? OR `object`.`parent`=?)");
		setLong(1, (long) playerID);
		setLong(2, (long) accountID);
		return getLargeObjectList();
	}

	public boolean MOVE_GOLD(final Item from, final Item to, final int amt) {
		int newFromAmt = from.getNumOfItems() - amt;
		int newToAmt = to.getNumOfItems() + amt;
		prepareCallable("UPDATE `obj_item` SET `item_numberOfItems` = CASE WHEN `UID`=?  THEN ? WHEN `UID`=? THEN ? END WHERE `UID` IN (?, ?);");
		setLong(1, (long) from.getObjectUUID());
		setInt(2, newFromAmt);
		setLong(3, (long) to.getObjectUUID());
		setInt(4, newToAmt);
		setLong(5, (long) from.getObjectUUID());
		setLong(6, (long) to.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean ORPHAN_INVENTORY(final HashSet<Item> inventory) {
		boolean worked = true;
		for (Item item : inventory) {

			if (item.getItemBase().getType().equals(ItemType.GOLD))
				continue;

			prepareCallable("UPDATE `obj_item` LEFT JOIN `object` ON `object`.`UID` = `obj_item`.`UID` SET `object`.`parent`=NULL, `obj_item`.`item_container`='none' WHERE `object`.`UID`=?;");
			setLong(1, (long) item.getObjectUUID());
			if (executeUpdate() == 0)
				worked = false;
			else
				item.zeroItem();
		}
		return worked;
	}

	public Item PURCHASE_ITEM_FROM_VENDOR(final PlayerCharacter pc, final ItemBase ib) {
		Item item = null;
		byte charges = 0;
		charges = (byte) ib.getNumCharges();
		short durability = (short) ib.getDurability();

		Item temp = new Item(ib, pc.getObjectUUID(),
				OwnerType.PlayerCharacter, charges, charges, durability, durability,
				true, false,ItemContainerType.INVENTORY, (byte) 0,
                new ArrayList<>(),"");
		try {
			item = this.ADD_ITEM(temp);
		} catch (Exception e) {
			Logger.error(e);
		}
		return item;
	}

	public HashSet<Integer> GET_ITEMS_FOR_VENDOR(final int vendorID) {
		prepareCallable("SELECT ID FROM static_itembase WHERE vendorType = ?");
		setInt(1, vendorID);
		return getIntegerList(1);
	}

	public ArrayList<Item> GET_ITEMS_FOR_VENDOR_FORGING(final int npcID) {
		prepareCallable("SELECT `obj_item`.*, `object`.`parent` FROM `object` INNER JOIN `obj_item` ON `object`.`UID` = `obj_item`.`UID` WHERE `object`.`parent`=? AND `obj_item`.`item_container` =?");
		setLong(1, (long) npcID);
		setString(2, "forge");
		return getObjectList();
	}

	public String SET_PROPERTY(final Item i, String name, Object new_value) {
		prepareCallable("CALL item_SETPROP(?,?,?)");
		setLong(1, (long) i.getObjectUUID());
		setString(2, name);
		setString(3, String.valueOf(new_value));
		return getResult();
	}

	public String SET_PROPERTY(final Item i, String name, Object new_value, Object old_value) {
		prepareCallable("CALL item_GETSETPROP(?,?,?,?)");
		setLong(1, (long) i.getObjectUUID());
		setString(2, name);
		setString(3, String.valueOf(new_value));
		setString(4, String.valueOf(old_value));
		return getResult();
	}

	//Used to transfer a single item between owners or equip or vault or bank or inventory
	public boolean UPDATE_OWNER(final Item item, int newOwnerID, boolean ownerNPC, boolean ownerPlayer,
			boolean ownerAccount, ItemContainerType containerType, int slot) {

		prepareCallable("CALL `item_TRANSFER_OWNER`(?, ?, ?, ? )");
		setLong(1, (long) item.getObjectUUID());
		if (newOwnerID != 0)
			setLong(2, (long) newOwnerID);
		else
			setNULL(2, java.sql.Types.BIGINT);
		
		switch (containerType) {
			case INVENTORY:
				setString(3, "inventory");
				break;
			case EQUIPPED:
				setString(3, "equip");
				break;
			case BANK:
				setString(3, "bank");
				break;
			case VAULT:
				setString(3, "vault");
				break;
			case FORGE:
				setString(3, "forge");
				break;
			default:
				setString(3, "none"); //Shouldn't be here
				break;
		}
		setInt(4, slot);
		return worked();
	}

	public boolean SET_DURABILITY(final Item item, int value) {
		prepareCallable("UPDATE `obj_item` SET `item_durabilityCurrent`=? WHERE `UID`=? AND `item_durabilityCurrent`=?");
		setInt(1, value);
		setLong(2, (long) item.getObjectUUID());
		setInt(3, (int) item.getDurabilityCurrent());
		return (executeUpdate() != 0);

	}

	//Update an item except ownership
	public boolean UPDATE_DATABASE(final Item item) {
		prepareCallable("UPDATE `obj_item` SET `item_itembaseID`=?, `item_chargesRemaining`=?, `item_durabilityCurrent`=?, `item_durabilityMax`=?, `item_numberOfItems`=? WHERE `UID`=?");
		setInt(1, item.getItemBaseID());
		setInt(2, item.getChargesRemaining());
		setInt(3, item.getDurabilityCurrent());
		setInt(4, item.getDurabilityMax());
		setInt(5, item.getNumOfItems());
		setLong(6, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean UPDATE_ROLL_COMPLETE(final Item item) {
		prepareCallable("UPDATE `obj_item` SET `item_container` = ?, `item_dateToUpgrade` = ? WHERE `UID` = ?");
		setString(1, "forge");
		setLong(2, 0L);
		setLong(3, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean SET_DATE_TO_UPGRADE(final Item item, long date) {
		prepareCallable("UPDATE `obj_item` SET `item_dateToUPGRADE` = ? WHERE `UID` = ?");
		setLong(1, date);
		setLong(2, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean UPDATE_FORGE_TO_INVENTORY(final Item item) {
		prepareCallable("UPDATE `obj_item` SET `item_container` = ? WHERE `UID` = ? AND `item_container` = 'forge';");
		setString(1, "inventory");
		setLong(2, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	/**
	 * Attempts to update the quantity of this gold item
	 *
	 * @param value New quantity of gold
	 * @return True on success
	 */
	public boolean UPDATE_GOLD(final Item item, int value) {
		if (item == null)
			return false;
		return UPDATE_GOLD(item, value, item.getNumOfItems());
	}

	/**
	 * Attempts to update the quantity of this gold item using CAS
	 *
	 * @return True on success
	 */
	public boolean UPDATE_GOLD(final Item item, int newValue, int oldValue) {

		if (item.getItemBase().getType().equals(ItemType.GOLD) == false)
			return false;

		prepareCallable("UPDATE `obj_item` SET `item_numberOfItems`=? WHERE `UID`=?");
		setInt(1, newValue);
		setLong(2, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	/**
	 * Attempts to update the value of two Gold items simultaneously.
	 *
	 * @param value New gold quantity for this item
	 * @param otherGold Other Gold item being modified
	 * @param valueOtherGold New quantity of gold for other item
	 * @return True on success
	 */
	public boolean UPDATE_GOLD(Item gold, int value, Item otherGold, int valueOtherGold) {

		if (gold.getItemBase().getType().equals(ItemType.GOLD) == false)
			return false;

		if (otherGold.getItemBase().getType().equals(ItemType.GOLD) == false)
			return false;

		int firstOld = gold.getNumOfItems();
		int secondOld = gold.getNumOfItems();

		prepareCallable("UPDATE `obj_item` SET `item_numberOfItems` = CASE WHEN `UID`=? AND `item_numberOfItems`=? THEN ? WHEN `UID`=? AND `item_numberOfItems`=? THEN ? END WHERE `UID` IN (?, ?);");
		setLong(1, (long) gold.getObjectUUID());
		setInt(2, firstOld);
		setInt(3, value);
		setLong(4, (long) otherGold.getObjectUUID());
		setInt(5, secondOld);
		setInt(6, valueOtherGold);
		setLong(7, (long) gold.getObjectUUID());
		setLong(8, (long) otherGold.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean UPDATE_REMAINING_CHARGES(final Item item) {
		prepareCallable("UPDATE `obj_item` SET `item_chargesRemaining` = ? WHERE `UID` = ?");
		setInt(1, item.getChargesRemaining());
		setLong(2, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	// This is necessary because default number of items is 1.
	// When we create gold, we want it to start at 0 quantity.

	public boolean ZERO_ITEM_STACK(Item item) {
		prepareCallable("UPDATE `obj_item` SET `item_numberOfItems`=0 WHERE `UID` = ?");
		setLong(1, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean UPDATE_FLAGS(Item item) {
		prepareCallable("UPDATE `obj_item` SET `item_flags`=? WHERE `UID` = ?");
		setInt(1, item.getFlags());
		setLong(2, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean UPDATE_VALUE(Item item,int value) {
		prepareCallable("UPDATE `obj_item` SET `item_value`=? WHERE `UID` = ?");
		setInt(1, value);
		setLong(2, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}

	public boolean UPDATE_FLAGS(Item item, int flags) {
		prepareCallable("UPDATE `obj_item` SET `item_flags`=? WHERE `UID` = ?");
		setInt(1, flags);
		setLong(2, (long) item.getObjectUUID());
		return (executeUpdate() != 0);
	}


}