#include "ElementLog.h"
#include "../../Include/RmlUi/Core/Context.h"
#include "../../Include/RmlUi/Core/Factory.h"
#include "BeaconSource.h"
#include "CommonSource.h"
#include "LogSource.h"
#include <limits.h>

namespace Rml {
namespace Debugger {

const int MAX_LOG_MESSAGES = 50;

ElementLog::ElementLog(const String& tag) : ElementDebugDocument(tag)
{
	dirty_logs = false;
	beacon = nullptr;
	current_beacon_level = Log::LT_MAX;
	auto_scroll = true;
	message_content = nullptr;
	current_index = 0;

	// Set up the log type buttons.
	log_types[Log::LT_ALWAYS].visible = true;
	log_types[Log::LT_ALWAYS].class_name = "error";
	log_types[Log::LT_ALWAYS].alert_contents = "A";

	log_types[Log::LT_ERROR].visible = true;
	log_types[Log::LT_ERROR].class_name = "error";
	log_types[Log::LT_ERROR].alert_contents = "!";
	log_types[Log::LT_ERROR].button_name = "error_button";

	log_types[Log::LT_ASSERT].visible = true;
	log_types[Log::LT_ASSERT].class_name = "error";
	log_types[Log::LT_ASSERT].alert_contents = "!";

	log_types[Log::LT_WARNING].visible = true;
	log_types[Log::LT_WARNING].class_name = "warning";
	log_types[Log::LT_WARNING].alert_contents = "!";
	log_types[Log::LT_WARNING].button_name = "warning_button";

	log_types[Log::LT_INFO].visible = false;
	log_types[Log::LT_INFO].class_name = "info";
	log_types[Log::LT_INFO].alert_contents = "i";
	log_types[Log::LT_INFO].button_name = "info_button";

	log_types[Log::LT_DEBUG].visible = true;
	log_types[Log::LT_DEBUG].class_name = "debug";
	log_types[Log::LT_DEBUG].alert_contents = "?";
	log_types[Log::LT_DEBUG].button_name = "debug_button";
}

ElementLog::~ElementLog()
{
	RemoveEventListener(EventId::Click, this);

	if (beacon && beacon->GetFirstChild())
		beacon->GetFirstChild()->RemoveEventListener(EventId::Click, this);

	if (beacon && beacon->GetParentNode())
		beacon->GetParentNode()->RemoveChild(beacon);

	if (message_content)
	{
		message_content->RemoveEventListener(EventId::Resize, this);
	}
}

bool ElementLog::Initialise()
{
	SetInnerRML(log_rml);
	SetId("rmlui-debug-log");

	message_content = GetElementById("content");
	if (message_content)
	{
		message_content->AddEventListener(EventId::Resize, this);
	}

	SharedPtr<StyleSheetContainer> style_sheet = Factory::InstanceStyleSheetString(String(common_rcss) + String(log_rcss));
	if (!style_sheet)
		return false;

	SetStyleSheetContainer(std::move(style_sheet));

	AddEventListener(EventId::Click, this);

	// Create the log beacon.
	beacon = GetContext()->CreateDocument("debug-document");
	RMLUI_ASSERT(rmlui_dynamic_cast<ElementDebugDocument*>(beacon));
	if (!beacon)
		return false;

	beacon->SetId("rmlui-debug-log-beacon");
	beacon->SetProperty(PropertyId::Visibility, Property(Style::Visibility::Hidden));
	beacon->SetInnerRML(beacon_rml);

	Element* button = beacon->GetFirstChild();
	if (button)
		beacon->GetFirstChild()->AddEventListener(EventId::Click, this);

	style_sheet = Factory::InstanceStyleSheetString(String(common_rcss) + String(beacon_rcss));
	if (!style_sheet)
	{
		GetContext()->UnloadDocument(beacon);
		beacon = nullptr;
		return false;
	}

	beacon->SetStyleSheetContainer(style_sheet);

	return true;
}

void ElementLog::AddLogMessage(Log::Type type, const String& message)
{
	LogMessageList& log_message_list = log_types[type].log_messages;
	if (log_message_list.size() >= MAX_LOG_MESSAGES)
	{
		log_message_list.erase(log_message_list.begin());
	}

	// Add the message to the list of messages for the specified log type.
	LogMessage log_message;
	log_message.index = current_index++;
	log_message.message = StringUtilities::EncodeRml(message);
	log_message_list.push_back(log_message);

	// If this log type is invisible, and there is a button for this log type, then change its text from
	// "Off" to "Off*" to signal that there are unread logs.
	if (!log_types[type].visible)
	{
		if (!log_types[type].button_name.empty())
		{
			Element* button = GetElementById(log_types[type].button_name);
			if (button)
			{
				button->SetInnerRML("Off*");
			}
		}
	}
	// Trigger the beacon if we're hidden. Override any lower-level log type if it is already visible.
	else
	{
		if (beacon != nullptr)
		{
			if (type < current_beacon_level)
			{
				beacon->SetProperty(PropertyId::Visibility, Property(Style::Visibility::Visible));

				current_beacon_level = type;
				Element* beacon_button = beacon->GetFirstChild();
				if (beacon_button)
				{
					beacon_button->SetClassNames(log_types[type].class_name);
					beacon_button->SetInnerRML(log_types[type].alert_contents);
				}

				// We need to update the document manually in case the beacon appears during context update.
				beacon->UpdateDocument();
			}
		}
	}

	// Force a refresh of the RML.
	dirty_logs = true;
}

void ElementLog::OnUpdate()
{
	ElementDocument::OnUpdate();

	if (dirty_logs)
	{
		// Set the log content:
		String messages;
		if (message_content)
		{
			unsigned int log_pointers[Log::LT_MAX];
			for (int i = 0; i < Log::LT_MAX; i++)
				log_pointers[i] = 0;
			int next_type = FindNextEarliestLogType(log_pointers);
			int num_messages = 0;
			while (next_type != -1 && num_messages < MAX_LOG_MESSAGES)
			{
				messages += CreateString("<div class=\"log-entry\"><div class=\"icon %s\">%s</div><p class=\"message\">",
					log_types[next_type].class_name.c_str(), log_types[next_type].alert_contents.c_str());
				messages += log_types[next_type].log_messages[log_pointers[next_type]].message;
				messages += "</p></div>";

				log_pointers[next_type]++;
				next_type = FindNextEarliestLogType(log_pointers);
				num_messages++;
			}

			if (message_content->HasChildNodes())
			{
				float last_element_top = message_content->GetLastChild()->GetAbsoluteTop();
				auto_scroll = message_content->GetAbsoluteTop() + message_content->GetAbsoluteTop() > last_element_top;
			}
			else
				auto_scroll = true;

			message_content->SetInnerRML(messages);

			dirty_logs = false;
		}
	}
}

void ElementLog::ProcessEvent(Event& event)
{
	// Only process events if we're visible
	if (beacon != nullptr)
	{
		if (event == EventId::Click)
		{
			if (event.GetTargetElement() == beacon->GetFirstChild())
			{
				Show();
				beacon->SetProperty(PropertyId::Visibility, Property(Style::Visibility::Hidden));
				current_beacon_level = Log::LT_MAX;
			}
			else if (event.GetTargetElement()->GetId() == "close_button")
			{
				Hide();
			}
			else if (event.GetTargetElement()->GetId() == "clear_button")
			{
				for (int i = 0; i < Log::LT_MAX; i++)
				{
					log_types[i].log_messages.clear();
					if (!log_types[i].visible)
					{
						if (Element* button = GetElementById(log_types[i].button_name))
							button->SetInnerRML("Off");
					}
				}
				dirty_logs = true;
			}
			else
			{
				for (int i = 0; i < Log::LT_MAX; i++)
				{
					if (!log_types[i].button_name.empty() && event.GetTargetElement()->GetId() == log_types[i].button_name)
					{
						log_types[i].visible = !log_types[i].visible;
						if (log_types[i].visible)
							event.GetTargetElement()->SetInnerRML("On");
						else
							event.GetTargetElement()->SetInnerRML("Off");
						dirty_logs = true;
					}
				}
			}
		}
	}

	if (event == EventId::Resize && auto_scroll)
	{
		if (message_content != nullptr && message_content->HasChildNodes())
			message_content->GetLastChild()->ScrollIntoView();
	}
}

int ElementLog::FindNextEarliestLogType(unsigned int log_pointers[Log::LT_MAX])
{
	int log_channel = -1;
	unsigned int index = UINT_MAX;

	for (int i = 0; i < Log::LT_MAX; i++)
	{
		if (log_types[i].visible)
		{
			if (log_pointers[i] < log_types[i].log_messages.size())
			{
				if (log_types[i].log_messages[log_pointers[i]].index < index)
				{
					index = log_types[i].log_messages[log_pointers[i]].index;
					log_channel = i;
				}
			}
		}
	}

	return log_channel;
}

} // namespace Debugger
} // namespace Rml