i2pd/Log.h

#ifndef LOG_H__
#define LOG_H__

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

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

class Log;
struct LogMsg
{
	std::stringstream s;
	Log * log;	
	LogLevel level;

	LogMsg (Log * l = nullptr, LogLevel lv = eLogInfo): log (l), level (lv) {};
	
	void Process();
};

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

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

		void SetLogFile (const std::string& fullFilePath, bool truncate = true);
		void ReopenLogFile ();
		void SetLogLevel (const std::string& level);
		void SetLogStream (std::shared_ptr<std::ostream> logStream);
		std::shared_ptr<std::ostream> GetLogStream () const { return m_LogStream; };	
		const std::string& GetTimestamp ();
		LogLevel GetLogLevel () { return m_MinLevel; };
	 	const std::string& GetFullFilePath () const { return m_FullFilePath; };
		/** start logging to syslog */
		void StartSyslog(const std::string & ident, const int facility = LOG_USER);
		/** stop logging to syslog */
		void StopSyslog();
		/** are we logging to syslog right now? */
		bool SyslogEnabled();
	private:

		void Flush ();

	private:

		std::string m_FullFilePath; // empty if stream
		std::shared_ptr<std::ostream> m_LogStream;
		enum LogLevel m_MinLevel;
		std::string m_Timestamp;
#if (__GNUC__ == 4) && (__GNUC_MINOR__ <= 6) && !defined(__clang__) // gcc 4.6
		std::chrono::monotonic_clock::time_point m_LastTimestampUpdate;
#else		
		std::chrono::steady_clock::time_point m_LastTimestampUpdate;	
#endif
		std::string m_Ident;
};

extern Log * g_Log;

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

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

inline void StopLog ()
{
	if (g_Log)
	{
		auto log = g_Log;
		g_Log = nullptr;
		log->Stop ();
		delete log;
	}		
}

inline void SetLogLevel (const std::string& level)
{
	if (g_Log)	
		g_Log->SetLogLevel(level);
}

inline void ReopenLogFile ()
{
	if (g_Log)	
		g_Log->ReopenLogFile ();
}

inline bool IsLogToFile ()
{
	return g_Log ? !g_Log->GetFullFilePath ().empty () : false;
}	

inline void StartSyslog()
{
	StartLog("");
	g_Log->StartSyslog("i2pd");
}

inline void StopSyslog()
{
	if(g_Log)
		g_Log->StopSyslog();
}

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) 
{
	if (g_Log && level > g_Log->GetLogLevel ())
		return;
	LogMsg * msg = new LogMsg (g_Log, level);
	LogPrint (msg->s, args...);
	msg->s << std::endl;
	if (g_Log) {
		g_Log->Put (msg);
	} else {
		msg->Process ();
		delete msg;
	}
}

#endif