// Copyright Epic Games, Inc. All Rights Reserved.

#include <zencore/zencore.h>

#if ZEN_PLATFORM_WINDOWS

#	include <zencore/logging/helpers.h>
#	include <zencore/logging/messageonlyformatter.h>
#	include <zencore/logging/msvcsink.h>

ZEN_THIRD_PARTY_INCLUDES_START
#	include <Windows.h>
ZEN_THIRD_PARTY_INCLUDES_END

namespace zen::logging {

// Default formatter for MSVC debug output: [level] message\n
class DefaultMsvcFormatter : public Formatter
{
public:
	void Format(const LogMessage& Msg, MemoryBuffer& Dest) override
	{
		Dest.push_back('[');
		helpers::AppendStringView(helpers::LevelToShortString(Msg.Level), Dest);
		Dest.push_back(']');
		Dest.push_back(' ');
		helpers::AppendStringView(Msg.Payload, Dest);
		Dest.push_back('\n');
	}

	std::unique_ptr<Formatter> Clone() const override { return std::make_unique<DefaultMsvcFormatter>(); }
};

MsvcSink::MsvcSink() : m_Formatter(std::make_unique<DefaultMsvcFormatter>())
{
}

void
MsvcSink::Log(const LogMessage& Msg)
{
	std::lock_guard<std::mutex> Lock(m_Mutex);

	MemoryBuffer Formatted;
	m_Formatter->Format(Msg, Formatted);

	// Null-terminate for OutputDebugStringA
	Formatted.push_back('\0');

	OutputDebugStringA(Formatted.data());
}

void
MsvcSink::Flush()
{
	// Nothing to flush for OutputDebugString
}

void
MsvcSink::SetFormatter(std::unique_ptr<Formatter> InFormatter)
{
	std::lock_guard<std::mutex> Lock(m_Mutex);
	m_Formatter = std::move(InFormatter);
}

}  // namespace zen::logging

#endif	// ZEN_PLATFORM_WINDOWS