i2pd/Log.h

#ifndef LOG_H__
#define LOG_H__

#include <string>
#include <iostream>
#include <sstream>
#include <fstream>
#include <functional>
#include "Queue.h"

enum LogLevel
{
	eLogError = 0,
	eLogWarning,
	eLogInfo,
	eLogDebug,	
	eNumLogLevels
};

struct LogMsg
{
	std::stringstream s;
	std::ostream& output;	
	LogLevel level;

	LogMsg (std::ostream& o = std::cout, LogLevel l = eLogInfo): output (o), level (l) {};
	
	void Process();
};

class Log: public i2p::util::MsgQueue<LogMsg>
{
	public:

		Log (): m_LogStream (nullptr) { SetOnEmpty (std::bind (&Log::Flush, this)); };
		~Log () { delete m_LogStream; };

		void SetLogFile (const std::string& fullFilePath);
		void SetLogStream (std::ostream * logStream);
		std::ostream * GetLogStream () const { return m_LogStream; };	

	private:

		void Flush ();

	private:
		
		std::ostream * m_LogStream;
};

extern Log * g_Log;

inline void StartLog (const std::string& fullFilePath)
{
	if (!g_Log)
	{	
		g_Log = new Log ();
		if (fullFilePath.length () > 0)
			g_Log->SetLogFile (fullFilePath);
	}	
}

inline void StartLog (std::ostream * s)
{
	if (!g_Log)
	{	
		g_Log = new Log ();
		if (s)
			g_Log->SetLogStream (s);
	}	
}

inline void StopLog ()
{
	if (g_Log)
	{
		delete g_Log;
		g_Log = nullptr;
	}		
}

template<typename TValue>
void LogPrint (std::stringstream& s, TValue arg) 
{
	s << arg;
}
 
template<typename TValue, typename... TArgs>
void LogPrint (std::stringstream& s, TValue arg, TArgs... args) 
{
	LogPrint (s, arg);
	LogPrint (s, args...);
}

template<typename... TArgs>
void LogPrint (LogLevel level, TArgs... args) 
{
	LogMsg * msg = (g_Log && g_Log->GetLogStream ()) ? new LogMsg (*g_Log->GetLogStream (), level) : 
		new LogMsg (std::cout, level);
	LogPrint (msg->s, args...);
	msg->s << std::endl;
	if (g_Log)	
		g_Log->Put (msg);
	else
	{
		msg->Process ();
		delete msg;
	}
}

template<typename... TArgs>
void LogPrint (TArgs... args) 
{
	LogPrint (eLogInfo, args...);
}	

#endif