#include "Globals.h"  // NOTE: MSVC stupidness requires this to be the same across all modules

#include "BrewingstandEntity.h"
#include "../Bindings/PluginManager.h"
#include "../UI/BrewingstandWindow.h"
#include "../Entities/Player.h"
#include "../Chunk.h"





cBrewingstandEntity::cBrewingstandEntity(
	BLOCKTYPE a_BlockType,
	NIBBLETYPE a_BlockMeta,
	Vector3i a_Pos,
	cWorld * a_World
) :
	Super(a_BlockType, a_BlockMeta, a_Pos, ContentsWidth, ContentsHeight, a_World),
	m_IsBrewing(false),
	m_TimeBrewed(0),
	m_RemainingFuel(0)
{
	m_Contents.AddListener(*this);
}





void cBrewingstandEntity::CopyFrom(const cBlockEntity & a_Src)
{
	Super::CopyFrom(a_Src);
	auto & src = static_cast<const cBrewingstandEntity &>(a_Src);
	m_IsBrewing = src.m_IsBrewing;
	m_CurrentBrewingRecipes = src.m_CurrentBrewingRecipes;
	m_Results = src.m_Results;
	m_TimeBrewed = src.m_TimeBrewed;
	m_RemainingFuel = src.m_RemainingFuel;
}





void cBrewingstandEntity::OnRemoveFromWorld()
{
	const auto Window = GetWindow();
	if (Window != nullptr)
	{
		// Tell window its owner is destroyed:
		Window->OwnerDestroyed();
	}
}





void cBrewingstandEntity::SendTo(cClientHandle & a_Client)
{
	// Nothing needs to be sent
	UNUSED(a_Client);
}





bool cBrewingstandEntity::Tick(std::chrono::milliseconds a_Dt, cChunk & a_Chunk)
{
	UNUSED(a_Dt);

	if (!m_IsBrewing)
	{
		return false;
	}

	// The necessary brewing time, has been reached
	if (m_TimeBrewed >= m_NeedBrewingTime)
	{
		BroadcastProgress(0, 0);
		m_IsBrewing = false;
		m_TimeBrewed = 0;

		// Return if the hook has been cancelled
		if (cPluginManager::Get()->CallHookBrewingCompleting(*m_World, *this))
		{
			return false;
		}

		// Decrease item count, full stacks are allowed in the ingredient slot
		cItem Ingredient = m_Contents.GetSlot(bsIngredient);
		Ingredient.m_ItemCount -= 1;
		m_Contents.SetSlot(bsIngredient, Ingredient);

		// Fuel slot
		m_RemainingFuel--;
		if (m_RemainingFuel <= 0)
		{
			if (!m_Contents.GetSlot(bsFuel).IsEmpty())
			{
				cItem Fuel = m_Contents.GetSlot(bsFuel);
				Fuel.m_ItemCount -= 1;
				m_Contents.SetSlot(bsFuel, Fuel);
				m_RemainingFuel = 20;
				BroadcastProgress(1, m_RemainingFuel);
			}
		}
		else
		{
			BroadcastProgress(1, m_RemainingFuel);
		}


		// Loop over all bottle slots and update available bottles
		const cBrewingRecipes::cRecipe * Recipe = nullptr;
		for (std::size_t i = 0; i < 3; i++)
		{
			if (m_Contents.GetSlot(static_cast<int>(i)).IsEmpty() || (m_CurrentBrewingRecipes[i] == nullptr))
			{
				continue;
			}

			Recipe = m_CurrentBrewingRecipes[i];
			m_Contents.SetSlot(static_cast<int>(i), Recipe->Output.CopyOne());
		}

		// Brewing process completed
		m_World->BroadcastSoundEffect("block.brewing_stand.brew", m_Pos, 1.0f, 1.0f);
		cPluginManager::Get()->CallHookBrewingCompleted(*m_World, *this);
		return true;
	}

	m_TimeBrewed++;
	UpdateProgressBars(false);
	return false;
}





bool cBrewingstandEntity::UsedBy(cPlayer * a_Player)
{
	a_Player->GetStatistics().Custom[CustomStatistic::InteractWithBrewingstand]++;

	cWindow * Window = GetWindow();
	if (Window == nullptr)
	{
		OpenWindow(new cBrewingstandWindow(this));
		Window = GetWindow();
	}

	if (Window != nullptr)
	{
		if (a_Player->GetWindow() != Window)
		{
			a_Player->OpenWindow(*Window);
		}
	}

	if (m_IsBrewing)
	{
		BroadcastProgress(0, m_NeedBrewingTime - m_TimeBrewed);
	}
	else
	{
		BroadcastProgress(0, 0);
	}
	BroadcastProgress(1, m_RemainingFuel);
	return true;
}





void cBrewingstandEntity::BroadcastProgress(size_t a_ProgressbarID, short a_Value)
{
	cWindow * Window = GetWindow();
	if (Window != nullptr)
	{
		Window->SetProperty(a_ProgressbarID, a_Value);
	}
}





void cBrewingstandEntity::OnSlotChanged(cItemGrid * a_ItemGrid, int a_SlotNum)
{
	Super::OnSlotChanged(a_ItemGrid, a_SlotNum);

	ASSERT(a_ItemGrid == &m_Contents);

	// Check for fuel
	if (m_RemainingFuel <= 0)
	{
		if (GetSlot(bsFuel).IsEmpty())
		{
			// No remaining fuel stop brewing and bail out
			m_IsBrewing = false;
			return;
		}
		else
		{
			// Fuel is available, refill
			m_RemainingFuel = 20;
			BroadcastProgress(1, m_RemainingFuel);
			cItem Fuel = m_Contents.GetSlot(bsFuel);
			Fuel.m_ItemCount -= 1;
			m_Contents.SetSlot(bsFuel, Fuel);
		}
	}

	// Check if still a item is in the ingredient slot
	if (GetSlot(bsIngredient).IsEmpty())
	{
		if (m_IsBrewing)
		{
			// Cancel brewing
			BroadcastProgress(0, 0);
			m_IsBrewing = false;
			m_TimeBrewed = 0;
		}
		return;
	}

	// Recheck the bottles
	cBrewingRecipes * BR = cRoot::Get()->GetBrewingRecipes();
	const cBrewingRecipes::cRecipe * Recipe = nullptr;
	bool Stop = true;
	for (std::size_t i = 0; i < 3; i++)
	{
		if (GetSlot(static_cast<int>(i)).IsEmpty())
		{
			m_CurrentBrewingRecipes[i] = nullptr;
			m_Results[i].Clear();
			continue;
		}

		if (m_CurrentBrewingRecipes[i] != nullptr)
		{
			Recipe = m_CurrentBrewingRecipes[i];
			if (Recipe->Ingredient.IsEqual(GetSlot(bsIngredient)) &&
				Recipe->Input.IsEqual(GetSlot(static_cast<int>(i))))
			{
				Stop = false;
				continue;
			}
		}

		Recipe = BR->GetRecipeFrom(m_Contents.GetSlot(static_cast<int>(i)), m_Contents.GetSlot(bsIngredient));
		if (Recipe != nullptr)
		{
			// Found a brewing recipe for the items
			m_CurrentBrewingRecipes[i] = Recipe;
			m_Results[i] = Recipe->Output.CopyOne();
			Stop = false;
		}
	}

	if (Stop)
	{
		if (m_IsBrewing)
		{
			// Cancel brewing
			BroadcastProgress(0, 0);
			m_IsBrewing = false;
			m_TimeBrewed = 0;
		}
		return;
	}

	// Start brewing process, if not running
	if (!m_IsBrewing)
	{
		m_IsBrewing = true;
	}
}





void cBrewingstandEntity::UpdateProgressBars(bool a_ForceUpdate)
{
	// Send an update every 3rd tick, using a higher value makes the progressbar look ugly:
	if (!a_ForceUpdate && ((m_World->GetWorldTickAge() % 3_tick) != 0_tick))
	{
		return;
	}

	BroadcastProgress(0, m_NeedBrewingTime - m_TimeBrewed);
}





void cBrewingstandEntity::ContinueBrewing(void)
{
	// Continue brewing if number is greater than 0
	if ((m_TimeBrewed > 0) && (m_RemainingFuel > 0))
	{
		m_IsBrewing = true;
	}
}





void cBrewingstandEntity::LoadRecipes(void)
{
	if (GetSlot(bsIngredient).IsEmpty())
	{
		return;
	}

	cBrewingRecipes * BR = cRoot::Get()->GetBrewingRecipes();
	const cBrewingRecipes::cRecipe * Recipe = nullptr;
	for (std::size_t i = 0; i < 3; i++)
	{
		if (GetSlot(static_cast<int>(i)).IsEmpty())
		{
			continue;
		}
		Recipe = BR->GetRecipeFrom(GetSlot(static_cast<int>(i)), GetSlot(bsIngredient));
		if (Recipe != nullptr)
		{
			m_CurrentBrewingRecipes[i] = Recipe;
			m_Results[i] = Recipe->Output.CopyOne();
		}
	}
}