/*
 *  (C) Copyright Gennadiy Rozental 2005-2008.
 *  Distributed under the Boost Software License, Version 1.0.
 *  (See accompanying file LICENSE_1_0.txt or copy at
 *  http://www.boost.org/LICENSE_1_0.txt)
 *
 *  See http://www.boost.org/libs/test for the library home page.
 */
/*
 * @file        colour_log_formatter.cpp
 * @author      Zofia Abramowska (z.abramowska@samsung.com)
 * @version
 * @brief
 */
// Boost.Test
#include <colour_log_formatter.h>
#include <boost/test/impl/execution_monitor.ipp>
#if BOOST_VERSION >= 105900
#include <boost/test/tree/test_unit.hpp>
#else
#include <boost/test/unit_test_suite_impl.hpp>
#endif
#include <boost/test/framework.hpp>
#include <boost/test/utils/basic_cstring/io.hpp>
#include <boost/test/utils/lazy_ostream.hpp>

// Boost
#include <boost/version.hpp>

// STL
#include <iostream>
#include <string>

#include <cchecker/dpl/colors.h>

// ************************************************************************** //
// **************            colour_log_formatter            ************** //
// ************************************************************************** //

using namespace boost::unit_test;
namespace CCHECKER {

namespace {

const_string
test_unit_type_name(const test_unit &tu)
{
#if BOOST_VERSION >= 105900
	return const_string(tu.p_type_name);
#else
	return tu.p_type_name.get();
#endif
}

const_string
test_unit_name(const test_unit &tu)
{
#if BOOST_VERSION >= 105900
	return const_string(tu.p_name);
#else
	return tu.p_name.get();
#endif
}

const_string
test_phase_identifier()
{
    return test_unit_name(framework::current_test_case());
}

const_string
get_basename(const const_string &file_name)
{
	return basename(file_name.begin());
}

std::string
get_basename(const std::string &file_name)
{
	return basename(file_name.c_str());
}

bool
test_unit_type_name_contains(const test_unit &tu, const std::string &substr)
{
	return test_unit_type_name(tu).find(const_string(substr)) == 0;
}

} // local namespace

//____________________________________________________________________________//

void
colour_log_formatter::log_start(
	std::ostream &output,
	counter_t test_cases_amount)
{
	if (test_cases_amount > 0)
		output  << "Running " << test_cases_amount << " test "
				<< (test_cases_amount > 1 ? "cases" : "case") << "...\n";
}

//____________________________________________________________________________//

void
colour_log_formatter::log_finish(std::ostream &ostr)
{
	ostr.flush();
}

//____________________________________________________________________________//

void
colour_log_formatter::log_build_info(std::ostream &output)
{
	output  << "Platform: " << BOOST_PLATFORM            << '\n'
			<< "Compiler: " << BOOST_COMPILER            << '\n'
			<< "STL     : " << BOOST_STDLIB              << '\n'
			<< "Boost   : " << BOOST_VERSION / 100000      << "."
			<< BOOST_VERSION / 100 % 1000  << "."
			<< BOOST_VERSION % 100       << std::endl;
}

//____________________________________________________________________________//

void
colour_log_formatter::log_build_info(std::ostream &output, bool log_build_info)
{
	if (log_build_info)
		output  << "Platform: " << BOOST_PLATFORM            << '\n'
				<< "Compiler: " << BOOST_COMPILER            << '\n'
				<< "STL     : " << BOOST_STDLIB              << '\n';
	output  << "Boost   : " << BOOST_VERSION / 100000      << '.'
			<< BOOST_VERSION / 100 % 1000  << '.'
			<< BOOST_VERSION % 100       << std::endl;
}

//____________________________________________________________________________//

void
colour_log_formatter::test_unit_start(
	std::ostream &output,
	test_unit const &tu)
{
	if (test_unit_type_name_contains(tu, "suite")) {
		output << "Starting test "; 
	} else {
		output << "Running test ";
	}
	output <<  test_unit_type_name(tu) << " \"" << test_unit_name(tu) << "\"" << std::endl;
}

//____________________________________________________________________________//

void
colour_log_formatter::test_unit_finish(
	std::ostream &output,
	test_unit const &tu,
	unsigned long elapsed)
{
	if (test_unit_type_name_contains(tu, "suite")) {
		output << "Finished test " << test_unit_type_name(tu) << " \"" << test_unit_name(tu) << "\"" << std::endl;
		return;
	}

	std::string color = CCHECKER::Colors::Text::GREEN_BEGIN;
	std::string status = "OK";

	if (m_isTestCaseFailed) {
		color = CCHECKER::Colors::Text::RED_BEGIN;
		status = "FAIL";
	}

	output << "\t" << "[   " << color << status << CCHECKER::Colors::Text::COLOR_END << "   ]";
	output << ", " << CCHECKER::Colors::Text::CYAN_BEGIN << "time: ";

	if (elapsed > 0) {
		if (elapsed % 1000 == 0)
			output << elapsed / 1000 << "ms";
		else
			output << elapsed << "mks";
	} else {
		output << "N/A";
	}

	output << CCHECKER::Colors::Text::COLOR_END << std::endl;
	m_isTestCaseFailed = false;
}

//____________________________________________________________________________//

void
colour_log_formatter::test_unit_skipped(
	std::ostream &output,
	test_unit const &tu)
{
	output  << "Test " << test_unit_type_name(tu) << " \"" << test_unit_name(tu) << "\"" << "is skipped" << std::endl;
}

//____________________________________________________________________________//

void
colour_log_formatter::log_exception_start(
	std::ostream &output,
	log_checkpoint_data const &checkpoint_data,
	boost::execution_exception const &ex)
{
	boost::execution_exception::location const &loc = ex.where();
	output << '\t' << CCHECKER::Colors::Text::BOLD_YELLOW_BEGIN << get_basename(loc.m_file_name)
		   << '(' << loc.m_line_num << "), ";
	output << "fatal error in \""
		   << (loc.m_function.is_empty() ? test_phase_identifier() : loc.m_function) << "\": ";
	output << CCHECKER::Colors::Text::COLOR_END << ex.what();

	if (!checkpoint_data.m_file_name.is_empty()) {
		output << '\n';
		output << "\tlast checkpoint : " << get_basename(checkpoint_data.m_file_name)
			   << '(' << checkpoint_data.m_line_num << ")";

		if (!checkpoint_data.m_message.empty())
			output << ": " << checkpoint_data.m_message;
	}

	output << std::endl;
	m_isTestCaseFailed = true;
}

void
colour_log_formatter::log_exception_finish(std::ostream &os)
{
	(void)os;
}
//____________________________________________________________________________//

void
colour_log_formatter::log_entry_start(
	std::ostream &output,
	log_entry_data const &entry_data,
	log_entry_types let)
{
	switch (let) {
	case BOOST_UTL_ET_INFO:
		output << '\t' << entry_data.m_file_name << '(' << entry_data.m_line_num << "), ";
		output << "info: ";
		break;

	case BOOST_UTL_ET_MESSAGE:
		break;

	case BOOST_UTL_ET_WARNING:
		output << '\t' << get_basename(entry_data.m_file_name) << '(' << entry_data.m_line_num << "), ";
		output << "warning in \"" << test_phase_identifier() << "\": ";
		break;

	case BOOST_UTL_ET_ERROR:
		output << '\t' << CCHECKER::Colors::Text::BOLD_YELLOW_BEGIN <<  get_basename(entry_data.m_file_name)
			   << '(' << entry_data.m_line_num << "), ";
		output << "error in \"" << test_phase_identifier() << "\": ";
		m_isTestCaseFailed = true;
		break;

	case BOOST_UTL_ET_FATAL_ERROR:
		output << '\t' << CCHECKER::Colors::Text::BOLD_YELLOW_BEGIN <<  get_basename(entry_data.m_file_name)
			   << '(' << entry_data.m_line_num << "),  ";
		output <<  " fatal error in \"" << test_phase_identifier() << "\": ";
		m_isTestCaseFailed = true;
		break;
	}

	output << CCHECKER::Colors::Text::COLOR_END;
}

//____________________________________________________________________________//

void
colour_log_formatter::log_entry_value(
	std::ostream &output,
	const_string value)
{
	output << value;
}

//____________________________________________________________________________//

void
colour_log_formatter::log_entry_value(
	std::ostream &output,
	lazy_ostream const &value)
{
	output << value;
}

//____________________________________________________________________________//

void
colour_log_formatter::log_entry_finish(
	std::ostream &output)
{
	output << std::endl;
}

//____________________________________________________________________________//

void
colour_log_formatter::entry_context_start(std::ostream& os, boost::unit_test::log_level l)
{
	(void)os;
	(void)l;
}
void
colour_log_formatter::log_entry_context(std::ostream& os, boost::unit_test::const_string value)
{
	(void)os;
	(void)value;
}
void
colour_log_formatter::entry_context_finish(std::ostream& os)
{
	(void)os;
}
//____________________________________________________________________________//

#if BOOST_VERSION >= 106500
void
colour_log_formatter::log_entry_context(std::ostream& os, log_level l, const_string value)
{
	(void)os;
	(void)l;
	(void)value;
}
void
colour_log_formatter::entry_context_finish(std::ostream& os, log_level l)
{
	(void)os;
	(void)l;
}
#endif

//____________________________________________________________________________//

} // namespace CCHECKER

//____________________________________________________________________________//