util/termcolours.lua

   1 -- Prosody IM v0.2
   2 -- Copyright (C) 2008 Matthew Wild
   3 -- Copyright (C) 2008 Waqas Hussain
   4 --
   5 -- This program is free software; you can redistribute it and/or
   6 -- modify it under the terms of the GNU General Public License
   7 -- as published by the Free Software Foundation; either version 2
   8 -- of the License, or (at your option) any later version.
   9 --
  10 -- This program is distributed in the hope that it will be useful,
  11 -- but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 -- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 -- GNU General Public License for more details.
  14 --
  15 -- You should have received a copy of the GNU General Public License
  16 -- along with this program; if not, write to the Free Software
  17 -- Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  18 --
  19
  20
  21 local t_concat, t_insert = table.concat, table.insert;
  22 local char, format = string.char, string.format;
  23 local ipairs = ipairs;
  24 module "termcolours"
  25
  26 local stylemap = {
  27                         reset = 0; bright = 1, dim = 2, underscore = 4, blink = 5, reverse = 7, hidden = 8;
  28                         black = 30; red = 31; green = 32; yellow = 33; blue = 34; magenta = 35; cyan = 36; white = 37;
  29                         ["black background"] = 40; ["red background"] = 41; ["green background"] = 42; ["yellow background"] = 43; ["blue background"] = 44; ["magenta background"] = 45; ["cyan background"] = 46; ["white background"] = 47;
  30                         bold = 1, dark = 2, underline = 4, underlined = 4, normal = 0;
  31                 }
  32
  33 local fmt_string = char(0x1B).."[%sm%s"..char(0x1B).."[0m";
  34 function getstring(style, text)
  35         if style then
  36                 return format(fmt_string, style, text);
  37         else
  38                 return text;
  39         end
  40 end
  41
  42 function getstyle(...)
  43         local styles, result = { ... }, {};
  44         for i, style in ipairs(styles) do
  45                 style = stylemap[style];
  46                 if style then
  47                         t_insert(result, style);
  48                 end
  49         end
  50         return t_concat(result, ";");
  51 end
  52
  53 return _M;