105 lines
2.5 KiB
JavaScript
105 lines
2.5 KiB
JavaScript
|
'use strict';
|
||
|
|
|
||
|
|
const { FORCE_COLOR, NODE_DISABLE_COLORS, TERM } = process.env;
|
||
|
|
|
||
|
|
const $ = {
|
||
|
|
enabled: !NODE_DISABLE_COLORS && TERM !== 'dumb' && FORCE_COLOR !== '0',
|
||
|
|
|
||
|
|
// modifiers
|
||
|
|
reset: init(0, 0),
|
||
|
|
bold: init(1, 22),
|
||
|
|
dim: init(2, 22),
|
||
|
|
italic: init(3, 23),
|
||
|
|
underline: init(4, 24),
|
||
|
|
inverse: init(7, 27),
|
||
|
|
hidden: init(8, 28),
|
||
|
|
strikethrough: init(9, 29),
|
||
|
|
|
||
|
|
// colors
|
||
|
|
black: init(30, 39),
|
||
|
|
red: init(31, 39),
|
||
|
|
green: init(32, 39),
|
||
|
|
yellow: init(33, 39),
|
||
|
|
blue: init(34, 39),
|
||
|
|
magenta: init(35, 39),
|
||
|
|
cyan: init(36, 39),
|
||
|
|
white: init(37, 39),
|
||
|
|
gray: init(90, 39),
|
||
|
|
grey: init(90, 39),
|
||
|
|
|
||
|
|
// background colors
|
||
|
|
bgBlack: init(40, 49),
|
||
|
|
bgRed: init(41, 49),
|
||
|
|
bgGreen: init(42, 49),
|
||
|
|
bgYellow: init(43, 49),
|
||
|
|
bgBlue: init(44, 49),
|
||
|
|
bgMagenta: init(45, 49),
|
||
|
|
bgCyan: init(46, 49),
|
||
|
|
bgWhite: init(47, 49)
|
||
|
|
};
|
||
|
|
|
||
|
|
function run(arr, str) {
|
||
|
|
let i=0, tmp, beg='', end='';
|
||
|
|
for (; i < arr.length; i++) {
|
||
|
|
tmp = arr[i];
|
||
|
|
beg += tmp.open;
|
||
|
|
end += tmp.close;
|
||
|
|
if (str.includes(tmp.close)) {
|
||
|
|
str = str.replace(tmp.rgx, tmp.close + tmp.open);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return beg + str + end;
|
||
|
|
}
|
||
|
|
|
||
|
|
function chain(has, keys) {
|
||
|
|
let ctx = { has, keys };
|
||
|
|
|
||
|
|
ctx.reset = $.reset.bind(ctx);
|
||
|
|
ctx.bold = $.bold.bind(ctx);
|
||
|
|
ctx.dim = $.dim.bind(ctx);
|
||
|
|
ctx.italic = $.italic.bind(ctx);
|
||
|
|
ctx.underline = $.underline.bind(ctx);
|
||
|
|
ctx.inverse = $.inverse.bind(ctx);
|
||
|
|
ctx.hidden = $.hidden.bind(ctx);
|
||
|
|
ctx.strikethrough = $.strikethrough.bind(ctx);
|
||
|
|
|
||
|
|
ctx.black = $.black.bind(ctx);
|
||
|
|
ctx.red = $.red.bind(ctx);
|
||
|
|
ctx.green = $.green.bind(ctx);
|
||
|
|
ctx.yellow = $.yellow.bind(ctx);
|
||
|
|
ctx.blue = $.blue.bind(ctx);
|
||
|
|
ctx.magenta = $.magenta.bind(ctx);
|
||
|
|
ctx.cyan = $.cyan.bind(ctx);
|
||
|
|
ctx.white = $.white.bind(ctx);
|
||
|
|
ctx.gray = $.gray.bind(ctx);
|
||
|
|
ctx.grey = $.grey.bind(ctx);
|
||
|
|
|
||
|
|
ctx.bgBlack = $.bgBlack.bind(ctx);
|
||
|
|
ctx.bgRed = $.bgRed.bind(ctx);
|
||
|
|
ctx.bgGreen = $.bgGreen.bind(ctx);
|
||
|
|
ctx.bgYellow = $.bgYellow.bind(ctx);
|
||
|
|
ctx.bgBlue = $.bgBlue.bind(ctx);
|
||
|
|
ctx.bgMagenta = $.bgMagenta.bind(ctx);
|
||
|
|
ctx.bgCyan = $.bgCyan.bind(ctx);
|
||
|
|
ctx.bgWhite = $.bgWhite.bind(ctx);
|
||
|
|
|
||
|
|
return ctx;
|
||
|
|
}
|
||
|
|
|
||
|
|
function init(open, close) {
|
||
|
|
let blk = {
|
||
|
|
open: `\x1b[${open}m`,
|
||
|
|
close: `\x1b[${close}m`,
|
||
|
|
rgx: new RegExp(`\\x1b\\[${close}m`, 'g')
|
||
|
|
};
|
||
|
|
return function (txt) {
|
||
|
|
if (this !== void 0 && this.has !== void 0) {
|
||
|
|
this.has.includes(open) || (this.has.push(open),this.keys.push(blk));
|
||
|
|
return txt === void 0 ? this : $.enabled ? run(this.keys, txt+'') : txt+'';
|
||
|
|
}
|
||
|
|
return txt === void 0 ? chain([open], [blk]) : $.enabled ? run([blk], txt+'') : txt+'';
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = $;
|