anthem/include/anthem/output/Formatting.h

#ifndef __ANTHEM_OUTPUT__FORMATTING_H
#define __ANTHEM_OUTPUT__FORMATTING_H

#include <iostream>

#include <anthem/output/ColorStream.h>

namespace anthem
{
namespace output
{

////////////////////////////////////////////////////////////////////////////////////////////////////
//
// Formatting
//
////////////////////////////////////////////////////////////////////////////////////////////////////

enum class Color
{
	Default = 39,
	Black = 30,
	Red = 31,
	Green = 32,
	Yellow = 33,
	Blue = 34,
	Magenta = 35,
	Cyan = 36,
	LightGray = 37,
	DarkGray = 90,
	LightRed = 91,
	LightGreen = 92,
	LightYellow = 93,
	LightBlue = 94,
	LightMagenta = 95,
	LightCyan = 96,
	White = 97
};

////////////////////////////////////////////////////////////////////////////////////////////////////

enum class FontWeight
{
	Normal = 21,
	Bold = 1
};

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Format
{
	Color color = Color::Default;
	FontWeight fontWeight = FontWeight::Normal;
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Format &format)
{
	if (!stream.supportsColor())
		return stream;

	const auto fontWeightCode = static_cast<int>(format.fontWeight);
	const auto colorCode = static_cast<int>(format.color);

	return (stream << "\033[" << fontWeightCode << ";" << colorCode << "m");
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct ResetFormat
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const ResetFormat &)
{
	if (!stream.supportsColor())
		return stream;

	return (stream << "\033[0m");
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Token
{
	const char *name;
};

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Function: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Function &function)
{
	return (stream
		<< Format({Color::White, FontWeight::Bold})
		<< function.name
		<< ResetFormat());
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Keyword: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Keyword &keyword)
{
	return (stream
		<< Format({Color::Blue, FontWeight::Normal})
		<< keyword.name
		<< ResetFormat());
}
////////////////////////////////////////////////////////////////////////////////////////////////////

struct Number: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Number &number)
{
	return (stream
		<< Format({Color::Yellow, FontWeight::Normal})
		<< number.name
		<< ResetFormat());
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Variable: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Variable &variable)
{
	return (stream
		<< Format({Color::Green, FontWeight::Bold})
		<< variable.name
		<< ResetFormat());
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct String: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const String &string)
{
	return (stream
		<< Format({Color::Green, FontWeight::Normal})
		<< "\"" << string.name << "\""
		<< ResetFormat());
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Boolean: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Boolean &boolean)
{
	return (stream
		<< Format({Color::Red, FontWeight::Normal})
		<< boolean.name
		<< ResetFormat());
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Reserved: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Reserved &reserved)
{
	return (stream
		<< Format({Color::White, FontWeight::Normal})
		<< reserved.name
		<< ResetFormat());
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Heading1: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Heading1 &heading1)
{
	return (stream
		<< Format({Color::Blue, FontWeight::Bold})
		<< "%---------------------------------------" << std::endl
		<< "% " << heading1.name << std::endl
		<< "%---------------------------------------"
		<< ResetFormat()
		<< std::endl);
}

////////////////////////////////////////////////////////////////////////////////////////////////////

struct Heading2: public Token
{
};

////////////////////////////////////////////////////////////////////////////////////////////////////

inline ColorStream &operator<<(ColorStream &stream, const Heading2 &heading2)
{
	return (stream
		<< Format({Color::Blue, FontWeight::Bold})
		<< "% " << heading2.name
		<< ResetFormat());
}


////////////////////////////////////////////////////////////////////////////////////////////////////

}
}

#endif