1 files changed, 183 insertions, 0 deletions
diff --git a/Minecraft.World/MerchantContainer.cpp b/Minecraft.World/MerchantContainer.cpp
new file mode 100644
index 00000000..cadbe4e3
--- /dev/null
+++ b/Minecraft.World/MerchantContainer.cpp
@@ -0,0 +1,183 @@
+#include "stdafx.h"
+#include "net.minecraft.world.item.trading.h"
+#include "MerchantMenu.h"
+#include "MerchantContainer.h"
+
+MerchantContainer::MerchantContainer(shared_ptr<Player> player, shared_ptr<Merchant> villager)
+{
+	this->player = player;
+	merchant = villager;
+	items = ItemInstanceArray(3);
+	items[0] = nullptr;
+	items[1] = nullptr;
+	items[2] = nullptr;
+	activeRecipe = NULL;
+	selectionHint = 0;
+}
+
+MerchantContainer::~MerchantContainer()
+{
+	delete [] items.data;
+}
+
+unsigned int MerchantContainer::getContainerSize()
+{
+	return items.length;
+}
+
+shared_ptr<ItemInstance> MerchantContainer::getItem(unsigned int slot)
+{
+	return items[slot];
+}
+
+shared_ptr<ItemInstance> MerchantContainer::removeItem(unsigned int slot, int count)
+{
+	if (items[slot] != NULL)
+	{
+		if (slot == MerchantMenu::RESULT_SLOT)
+		{
+			shared_ptr<ItemInstance> item = items[slot];
+			items[slot] = nullptr;
+			return item;
+		}
+		if (items[slot]->count <= count)
+		{
+			shared_ptr<ItemInstance> item = items[slot];
+			items[slot] = nullptr;
+			if (isPaymentSlot(slot))
+			{
+				updateSellItem();
+			}
+			return item;
+		}
+		else
+		{
+			shared_ptr<ItemInstance> i = items[slot]->remove(count);
+			if (items[slot]->count == 0) items[slot] = nullptr;
+			if (isPaymentSlot(slot))
+			{
+				updateSellItem();
+			}
+			return i;
+		}
+	}
+	return nullptr;
+}
+
+bool MerchantContainer::isPaymentSlot(int slot)
+{
+	return slot == MerchantMenu::PAYMENT1_SLOT || slot == MerchantMenu::PAYMENT2_SLOT;
+}
+
+shared_ptr<ItemInstance> MerchantContainer::removeItemNoUpdate(int slot)
+{
+	if (items[slot] != NULL)
+	{
+		shared_ptr<ItemInstance> item = items[slot];
+		items[slot] = nullptr;
+		return item;
+	}
+	return nullptr;
+}
+
+void MerchantContainer::setItem(unsigned int slot, shared_ptr<ItemInstance> item)
+{
+	items[slot] = item;
+	if (item != NULL && item->count > getMaxStackSize()) item->count = getMaxStackSize();
+	if (isPaymentSlot(slot))
+	{
+		updateSellItem();
+	}
+}
+
+int MerchantContainer::getName()
+{
+	return merchant->getDisplayName();
+}
+
+int MerchantContainer::getMaxStackSize()
+{
+	return Container::LARGE_MAX_STACK_SIZE;
+}
+
+bool MerchantContainer::stillValid(shared_ptr<Player> player)
+{
+	return merchant->getTradingPlayer() == player;
+}
+
+void MerchantContainer::startOpen()
+{
+}
+
+void MerchantContainer::stopOpen()
+{
+}
+
+void MerchantContainer::setChanged()
+{
+	updateSellItem();
+}
+
+void MerchantContainer::updateSellItem()
+{
+	activeRecipe = NULL;
+
+	shared_ptr<ItemInstance> buyItem1 = items[MerchantMenu::PAYMENT1_SLOT];
+	shared_ptr<ItemInstance> buyItem2 = items[MerchantMenu::PAYMENT2_SLOT];
+
+	if (buyItem1 == NULL)
+	{
+		buyItem1 = buyItem2;
+		buyItem2 = nullptr;
+	}
+
+	if (buyItem1 == NULL)
+	{
+		setItem(MerchantMenu::RESULT_SLOT, nullptr);
+	}
+	else
+	{
+		MerchantRecipeList *offers = merchant->getOffers(player);
+		if (offers != NULL)
+		{
+			MerchantRecipe *recipeFor = offers->getRecipeFor(buyItem1, buyItem2, selectionHint);
+			if (recipeFor != NULL && !recipeFor->isDeprecated())
+			{
+				activeRecipe = recipeFor;
+				setItem(MerchantMenu::RESULT_SLOT, recipeFor->getSellItem()->copy());
+			}
+			else if (buyItem2 != NULL)
+			{
+				// try to switch
+				recipeFor = offers->getRecipeFor(buyItem2, buyItem1, selectionHint);
+				if (recipeFor != NULL && !recipeFor->isDeprecated())
+				{
+					activeRecipe = recipeFor;
+					setItem(MerchantMenu::RESULT_SLOT, recipeFor->getSellItem()->copy());
+				}
+				else
+				{
+					setItem(MerchantMenu::RESULT_SLOT, nullptr);
+				}
+
+			}
+			else
+			{
+				setItem(MerchantMenu::RESULT_SLOT, nullptr);
+			}
+		}
+	}
+
+	merchant->notifyTradeUpdated(getItem(MerchantMenu::RESULT_SLOT));
+}
+
+MerchantRecipe *MerchantContainer::getActiveRecipe()
+{
+	return activeRecipe;
+}
+
+void MerchantContainer::setSelectionHint(int selectionHint)
+{
+	this->selectionHint = selectionHint;
+	updateSellItem();
+}
+\ No newline at end of file