/*************************************************************************
*    UrBackup - Client/Server backup system
*    Copyright (C) 2011-2016 Martin Raiber
*
*    This program is free software: you can redistribute it and/or modify
*    it under the terms of the GNU Affero General Public License as published by
*    the Free Software Foundation, either version 3 of the License, or
*    (at your option) any later version.
*
*    This program is distributed in the hope that it will be useful,
*    but WITHOUT ANY WARRANTY; without even the implied warranty of
*    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
*    GNU Affero General Public License for more details.
*
*    You should have received a copy of the GNU Affero General Public License
*    along with this program.  If not, see <http://www.gnu.org/licenses/>.
**************************************************************************/

#include "server_log.h"
#include "../stringtools.h"
#include "../Interface/Database.h"
#include "../Interface/Query.h"
#include "database.h"
#include "../stringtools.h"
#include <limits.h>

std::map<logid_t, SLogData> ServerLogger::logdata;
IMutex *ServerLogger::mutex=NULL;
std::map<int, SCircularData> ServerLogger::circular_logdata;
logid_t ServerLogger::logid_gen;
std::map<logid_t, int> ServerLogger::logid_client;

const size_t circular_logdata_buffersize=20;
const size_t max_memory_logdata_size = 2 * 1024 * 1024;

void ServerLogger::Log(logid_t logid, const std::string &pStr, int LogLevel)
{
	Server->Log(pStr, LogLevel);

	IScopedLock lock(mutex);

	logCircular(logid_client[logid], logid, pStr, LogLevel);

	if(LogLevel<0)
		return;

	logMemory(Server->getTimeSeconds(), logid, pStr, LogLevel);
}

void ServerLogger::Log(int64 times, logid_t logid, const std::string &pStr, int LogLevel)
{
	Server->Log(pStr, LogLevel);

	IScopedLock lock(mutex);

	logCircular(logid_client[logid], logid, pStr, LogLevel);

	if(LogLevel<0)
		return;

	logMemory(times, logid, pStr, LogLevel);
}

void ServerLogger::logMemory(int64 times, logid_t logid, const std::string &pStr, int LogLevel)
{
	std::map<logid_t, int>::iterator it = logid_client.find(logid);

	if (it != logid_client.end() && it->second <= 0)
	{
		return;
	}

	SLogEntry le;
	le.data=pStr;
	le.loglevel=LogLevel;
	le.time=times;

	std::map<logid_t, SLogData>::iterator iter=logdata.find(logid);
	size_t old_memory_used = 0;
	size_t new_memory_used;
	if( iter==logdata.end() )
	{
		SLogData log_data;
		log_data.data.push_back(le);
		log_data.memory_used = logEntrySize(le);
		new_memory_used = log_data.memory_used;
		logdata.insert(std::pair<logid_t, SLogData>(logid, log_data) );		
	}
	else
	{
		if (iter->second.memory_used >= max_memory_logdata_size)
		{
			return;
		}

		old_memory_used = iter->second.memory_used;
		iter->second.memory_used += logEntrySize(le);
		new_memory_used = iter->second.memory_used;
		iter->second.data.push_back(le);		
	}

	if (old_memory_used < max_memory_logdata_size
		&& new_memory_used >= max_memory_logdata_size)
	{
		SLogEntry le;
		le.data = "Max log size reached (" + PrettyPrintBytes(max_memory_logdata_size) + "). Not logging more.";
		le.loglevel = LL_WARNING;
		le.time = Server->getTimeSeconds();
		logdata[logid].data.push_back(le);
	}
}

size_t ServerLogger::logEntrySize(const SLogEntry& entry)
{
	//short string optimization on 64bit
	if (entry.data.size() > 22)
	{
		return sizeof(void*) + sizeof(SLogEntry) + entry.data.capacity();
	}
	else
	{
		return sizeof(void*) + sizeof(SLogEntry);
	}
}

std::vector<SCircularLogEntry> ServerLogger::stripLogIdFilter(const std::vector<SCircularLogEntryWithId>& data, logid_t logid)
{
	std::vector<SCircularLogEntry> ret;
	ret.reserve(data.size());
	for (size_t i = 0; i < data.size(); ++i)
	{
		if (logid == logid_t()
			|| logid == data[i].logid)
		{
			SCircularLogEntry ne;
			ne.id = data[i].id;
			ne.loglevel = data[i].loglevel;
			ne.time = data[i].time;
			ne.utf8_msg = data[i].utf8_msg;
			ret.push_back(ne);
		}		
	}
	return ret;
}

void ServerLogger::logCircular(int clientid, logid_t logid, const std::string &pStr, int LogLevel)
{
	std::map<int, SCircularData>::iterator iter=circular_logdata.find(clientid);
	SCircularData *data;
	if(iter==circular_logdata.end())
	{
		data=&circular_logdata[clientid];
		data->data.resize(circular_logdata_buffersize);
		data->idx=0;
		data->id=1;
	}
	else
	{
		data=&iter->second;
	}

	SCircularLogEntryWithId& entry=data->data[data->idx];
	entry.id=data->id++;
	entry.loglevel=LogLevel;
	entry.time=Server->getTimeSeconds();
	entry.utf8_msg=pStr;
	entry.logid = logid;

	data->idx=(data->idx+1)%circular_logdata_buffersize;
}

void ServerLogger::init_mutex(void)
{
	mutex=Server->createMutex();
}

void ServerLogger::destroy_mutex(void)
{
	Server->destroy(mutex);
}

std::string ServerLogger::getLogdata(logid_t logid, int &errors, int &warnings, int &infos)
{
	IScopedLock lock(mutex);

	std::string ret;

	std::map<logid_t, SLogData >::iterator iter=logdata.find(logid);
	if( iter!=logdata.end() )
	{
		for(size_t i=0;i<iter->second.data.size();++i)
		{
			SLogEntry &le=iter->second.data[i];
			
			if(le.loglevel==LL_ERROR)
				++errors;
			else if(le.loglevel==LL_WARNING)
				++warnings;
			else if(le.loglevel==LL_INFO)
				++infos;
			
			ret+=convert(le.loglevel);
			ret+="-";
			ret+=convert(le.time);
			ret+="-";
			ret+=(le.data);
			ret+="\n";
		}
		
		return ret;
	}
	else
	{
		return "";
	}
}

std::string ServerLogger::getWarningLevelTextLogdata(logid_t logid)
{
	IScopedLock lock(mutex);

	std::string ret;
	std::map<logid_t, SLogData >::iterator iter=logdata.find(logid);
	if( iter!=logdata.end() )
	{
		for(size_t i=0;i<iter->second.data.size();++i)
		{
			SLogEntry &le=iter->second.data[i];
			
			if(le.loglevel>=LL_WARNING)
			{
				if(le.loglevel==LL_WARNING)
					ret+="WARNING: ";
				else if(le.loglevel==LL_ERROR)
					ret+="ERROR: ";

				ret+=le.data;
				ret+="\r\n";
			}
		}
		
		return ret;
	}
	else
	{
		return "";
	}
}

void ServerLogger::reset(logid_t id)
{
	IScopedLock lock(mutex);

	std::map<logid_t, SLogData >::iterator iter=logdata.find(id);
	if( iter!=logdata.end() )
	{
		std::vector<SLogEntry>().swap(iter->second.data);
		iter->second.memory_used = 0;
	}
}

void ServerLogger::reset( int clientid )
{
	IScopedLock lock(mutex);

	for(std::map<logid_t, int>::iterator it=logid_client.begin();
		it!=logid_client.end();++it)
	{
		if(it->second==clientid)
		{
			reset(it->first);
		}
	}
}

std::vector<SCircularLogEntry> ServerLogger::getCircularLogdata( int clientid, size_t minid, logid_t logid)
{
	IScopedLock lock(mutex);

	std::map<int, SCircularData>::const_iterator iter=circular_logdata.find(clientid);
	if(iter!=circular_logdata.end())
	{
		if(minid==std::string::npos)
			return stripLogIdFilter(iter->second.data, logid);

		for(size_t i=0;i<iter->second.data.size();++i)
		{
			if (iter->second.data[i].id > minid &&
				iter->second.data[i].id != std::string::npos &&
				(logid == logid_t() || iter->second.data[i].logid == logid) )
			{
				return stripLogIdFilter(iter->second.data, logid);
			}
		}
		return std::vector<SCircularLogEntry>();
	}
	else
	{
		if(minid==std::string::npos)
		{
			std::vector<SCircularLogEntry> ret;
			SCircularLogEntry entry;
			entry.id=0;
			entry.loglevel=LL_INFO;
			entry.time=Server->getTimeSeconds();
			entry.utf8_msg="No log entries yet";
			ret.push_back(entry);
			return ret;
		}
		return std::vector<SCircularLogEntry>();
	}
}

logid_t ServerLogger::getLogId( int clientid )
{
	IScopedLock lock(mutex);

	logid_t ret= std::make_pair(++logid_gen.first, 0);

	logid_client[ret] = clientid;

	return ret;
}

bool ServerLogger::hasClient( logid_t id, int clientid )
{
	IScopedLock lock(mutex);

	return logid_client[id] == clientid;
}