rspamd/lualib/ansicolors.lua

local colormt = {}
local ansicolors = {}

local rspamd_util = require "rspamd_util"
local isatty = rspamd_util.isatty()

function colormt:__tostring()
  return self.value
end

function colormt:__concat(other)
  return tostring(self) .. tostring(other)
end

function colormt:__call(s)
  return self .. s .. ansicolors.reset
end

colormt.__metatable = {}
local function makecolor(value)
  if isatty then
    return setmetatable({
      value = string.char(27) .. '[' .. tostring(value) .. 'm'
    }, colormt)
  else
    return setmetatable({
      value = ''
    }, colormt)
  end
end

local colors = {
  -- attributes
  reset = 0,
  clear = 0,
  bright = 1,
  dim = 2,
  underscore = 4,
  blink = 5,
  reverse = 7,
  hidden = 8,

  -- foreground
  black = 30,
  red = 31,
  green = 32,
  yellow = 33,
  blue = 34,
  magenta = 35,
  cyan = 36,
  white = 37,

  -- background
  onblack = 40,
  onred = 41,
  ongreen = 42,
  onyellow = 43,
  onblue = 44,
  onmagenta = 45,
  oncyan = 46,
  onwhite = 47,
}

for c, v in pairs(colors) do
  ansicolors[c] = makecolor(v)
end

return ansicolors