mirror of
https://github.com/yamadapc/d-colorize
synced 2024-11-15 05:24:00 +01:00
fc2a2dd537
I think this can be simplified with templates. I'm wondering if the colorize function could be made compile-time when that's possible.
146 lines
2.9 KiB
D
146 lines
2.9 KiB
D
/**
|
|
* Authors: Pedro Tacla Yamada
|
|
* Date: June 9, 2014
|
|
* License: Licensed under the MIT license. See LICENSE for more information
|
|
* Version: 0.0.2
|
|
*/
|
|
|
|
import std.string : format;
|
|
|
|
// Foreground colors
|
|
static enum fg : int
|
|
{
|
|
init = 39,
|
|
|
|
black = 30,
|
|
red = 31,
|
|
green = 32,
|
|
yellow = 33,
|
|
blue = 34,
|
|
magenta = 35,
|
|
cyan = 36,
|
|
white = 37,
|
|
|
|
light_black = 90,
|
|
light_red = 91,
|
|
light_green = 92,
|
|
light_yellow = 93,
|
|
light_blue = 94,
|
|
light_magenta = 95,
|
|
light_cyan = 96,
|
|
light_white = 97
|
|
}
|
|
|
|
// Background colors
|
|
static enum bg : int
|
|
{
|
|
init = 49,
|
|
|
|
black = 40,
|
|
red = 41,
|
|
green = 42,
|
|
yellow = 43,
|
|
blue = 44,
|
|
magenta = 45,
|
|
cyan = 46,
|
|
white = 47,
|
|
|
|
light_black = 100,
|
|
light_red = 101,
|
|
light_green = 102,
|
|
light_yellow = 103,
|
|
light_blue = 104,
|
|
light_magenta = 105,
|
|
light_cyan = 106,
|
|
light_white = 107
|
|
}
|
|
|
|
// Text modes
|
|
static enum mode : int
|
|
{
|
|
init = 0,
|
|
bold = 1,
|
|
underline = 4,
|
|
blink = 5,
|
|
swap = 7,
|
|
hide = 8
|
|
}
|
|
|
|
/**
|
|
* Wraps a string around color escape sequences.
|
|
*
|
|
* Params:
|
|
* str = The string to wrap with colors and modes
|
|
* c = The foreground color (see the fg enum type)
|
|
* b = The background color (see the bg enum type)
|
|
* m = The text mode (see the mode enum type)
|
|
* Example:
|
|
* ---
|
|
* writeln("This is blue".colorize(fg.blue));
|
|
* writeln(
|
|
* colorize("This is red over green blinking", fg.blue, bg.green, mode.blink)
|
|
* );
|
|
* ---
|
|
*/
|
|
|
|
string colorize(
|
|
const string str,
|
|
const fg c=fg.init,
|
|
const bg b=bg.init,
|
|
const mode m=mode.init
|
|
) pure
|
|
{
|
|
return format("\033[%d;%d;%dm%s\033[0m", m, c, b, str);
|
|
}
|
|
|
|
unittest
|
|
{
|
|
import std.stdio;
|
|
string ret;
|
|
|
|
ret = "This is yellow".colorize(fg.yellow);
|
|
writeln(ret);
|
|
assert(ret == "\033[0;33;49mThis is yellow\033[0m");
|
|
|
|
ret = "This is light green".colorize(fg.light_green);
|
|
writeln(ret);
|
|
assert(ret == "\033[0;92;49mThis is light green\033[0m");
|
|
|
|
ret = "This is light blue with red background".colorize(fg.light_blue, bg.red);
|
|
writeln(ret);
|
|
assert(ret == "\033[0;94;41mThis is light blue with red background\033[0m");
|
|
|
|
ret = "This is red on blue blinking".colorize(fg.red, bg.blue, mode.blink);
|
|
writeln(ret);
|
|
assert(ret == "\033[5;31;44mThis is red on blue blinking\033[0m");
|
|
}
|
|
|
|
string background(const string str, const bg b=bg.init)
|
|
{
|
|
return format("\033[%dm%s\033[0m", b, str);
|
|
}
|
|
|
|
string foreground(const string str, const fg c=fg.init)
|
|
{
|
|
return format("\033[%dm%s\033[0m", c, str);
|
|
}
|
|
|
|
string style(const string str, const mode m=mode.init)
|
|
{
|
|
return format("\033[%dm%s\033[0m", m, str);
|
|
}
|
|
|
|
unittest
|
|
{
|
|
import std.stdio;
|
|
string ret;
|
|
|
|
ret = "This is red on blue blinking"
|
|
.foreground(fg.red)
|
|
.background(bg.blue)
|
|
.style(mode.blink);
|
|
|
|
writeln(ret);
|
|
assert(ret == "\033[5m\033[44m\033[31mThis is red on blue blinking\033[0m\033[0m\033[0m");
|
|
}
|