#pragma once

#include "Misc/OutputDevice.h"
#include "Delegates/Delegate.h"
#include "Logging/LogVerbosity.h"
#include "Runtime/Launch/Resources/Version.h"

using FOnSerializeMessage =
#if ENGINE_MAJOR_VERSION == 4 && ENGINE_MINOR_VERSION < 26
	TBaseDelegate<void, const TCHAR*, ELogVerbosity::Type, const FName&, TOptional<double>>;
#elif ENGINE_MAJOR_VERSION < 5 || (ENGINE_MAJOR_VERSION == 5 && ENGINE_MINOR_VERSION < 3)
	TDelegate<void(const TCHAR*, ELogVerbosity::Type, const FName&, TOptional<double>)>;
#else
	TDelegate<void(const TCHAR*, ELogVerbosity::Type, const FName&, TOptional<double>), FDefaultTSDelegateUserPolicy>;
#endif

class FRiderOutputDevice : public FOutputDevice {
public:
	~FRiderOutputDevice();
	void Setup(TFunction<FOnSerializeMessage::TFuncType>);
	virtual void TearDown() override;

protected:
	virtual void Serialize(const TCHAR* V, ELogVerbosity::Type Verbosity, const FName& Category) override;
	virtual void Serialize(const TCHAR* V, ELogVerbosity::Type Verbosity, const FName& Category, double Time) override;
	
private:
	FOnSerializeMessage onSerializeMessage;
	FCriticalSection CriticalSection;
};