/*
 * Format.cpp -- convenient function for formatting text with escape sequences
 *
 * Copyright (c) 2015 David Demelier <markand@malikania.fr>
 *
 * Permission to use, copy, modify, and/or distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <array>
#include <sstream>
#include <unordered_map>

#include "Format.h"

namespace fmt {

namespace {

const std::array<unsigned char, 17> colorsTable{
	0,	// Default
	30,	// Black
	31,	// Red
	32,	// Green
	33,	// Yellow
	34,	// Blue
	35,	// Magenta
	36,	// Cyan
	37,	// Light gray
	90,	// Dark gray
	91,	// Light red
	92,	// Light green
	93,	// Light yellow
	94,	// Light blue
	95,	// Light cyan
	96,	// White
	97	
};

const std::unordered_map<int, int> attributesTable{
	{ Bold,		1 },
	{ Dim,		2 },
	{ Underline,	4 },
	{ Blink,	5 },
	{ Reverse,	7 },
	{ Hidden,	8 }
};

} // !namespace

std::string convert(std::string input, unsigned short flags)
{
	std::ostringstream oss;

#if !defined(_WIN32)
	// Attributes
	for (const auto &pair : attributesTable) {
		if (flags & pair.first) {
			oss << "\033[" << pair.second << "m";
		}
	}

	// Background
	if (((flags >> 11) & 0x3f) != Default) {
		oss << "\033[" << std::to_string(colorsTable[static_cast<unsigned char>((flags >> 11) & 0x3f)] + 10) << "m";
	}

	// Foreground
	if (((flags >> 6) & 0x3f) != Default) {
		oss << "\033[" << std::to_string(colorsTable[static_cast<unsigned char>((flags >> 6) & 0x3f)]) << "m";
	}

	oss << std::move(input) << "\033[0m";
#else
	oss << std::move(input);
#endif

	return oss.str();
}

} // !fmt