/*
 * Copyright (C) 2020  Christopher J. Howard
 *
 * This file is part of Antkeeper source code.
 *
 * Antkeeper source code is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Antkeeper source code 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Antkeeper source code.  If not, see <http://www.gnu.org/licenses/>.
 */

#include "logger.hpp"
#include "timestamp.hpp"
#include <iostream>

logger::logger():
	os(&std::cout),
	log_prefix(std::string()),
	log_postfix(std::string()),
	warning_prefix(std::string()),
	warning_postfix(std::string()),
	error_prefix(std::string()),
	error_postfix(std::string()),
	success_prefix(std::string()),
	success_postfix(std::string()),
	indent("|   "),
	timestamp_enabled(true)
{}

logger::~logger()
{}

void logger::redirect(std::ostream* stream)
{
	os = stream;
}

void logger::log(const std::string& text)
{
	if (os)
	{
		if (timestamp_enabled)
		{
			(*os) << timestamp() << ": ";
		}
		
		for (std::size_t i = 0; i < tasks.size(); ++i)
			(*os) << indent;

		(*os) << (log_prefix + text + log_postfix);

		os->flush();
	}
}

void logger::warning(const std::string& text)
{
	log(warning_prefix + text + warning_postfix);
}

void logger::error(const std::string& text)
{
	log(error_prefix + text + error_postfix);
}

void logger::success(const std::string& text)
{
	log(success_prefix + text + success_postfix);
}

void logger::set_log_prefix(const std::string& prefix)
{
	log_prefix = prefix;
}

void logger::set_log_postfix(const std::string& postfix)
{
	log_postfix = postfix;
}

void logger::set_warning_prefix(const std::string& prefix)
{
	warning_prefix = prefix;
}

void logger::set_warning_postfix(const std::string& postfix)
{
	warning_postfix = postfix;
}

void logger::set_error_prefix(const std::string& prefix)
{
	error_prefix = prefix;
}

void logger::set_error_postfix(const std::string& postfix)
{
	error_postfix = postfix;
}

void logger::set_success_prefix(const std::string& prefix)
{
	success_prefix = prefix;
}

void logger::set_success_postfix(const std::string& postfix)
{
	success_postfix = postfix;
}

void logger::push_task(const std::string& description)
{
	std::string message = description + "...\n";
	
	log(message);
	
	tasks.push(description);
}

void logger::pop_task(int status)
{
	if (tasks.empty())
	{
		return;
	}
	
	std::string message = tasks.top() + "... ";
	
	tasks.pop();
	
	if (status == EXIT_SUCCESS)
	{
		message += "success\n";
		log(message);
	}
	else
	{
		message += "failed (" + std::to_string(status) + ")\n";
		error(message);
	}
}

void logger::set_indent(const std::string& indent)
{
	this->indent = indent;
}

void logger::set_timestamp(bool enabled)
{
	timestamp_enabled = enabled;
}