util.termcolours: Silence luacheck warning
[prosody.git] / util / termcolours.lua
1 -- Prosody IM
2 -- Copyright (C) 2008-2010 Matthew Wild
3 -- Copyright (C) 2008-2010 Waqas Hussain
4 --
5 -- This project is MIT/X11 licensed. Please see the
6 -- COPYING file in the source package for more information.
7 --
8 --
9 -- luacheck: ignore 213/i
10
11
12 local t_concat, t_insert = table.concat, table.insert;
13 local char, format = string.char, string.format;
14 local tonumber = tonumber;
15 local ipairs = ipairs;
16 local io_write = io.write;
17
18 local windows;
19 if os.getenv("WINDIR") then
20         windows = require "util.windows";
21 end
22 local orig_color = windows and windows.get_consolecolor and windows.get_consolecolor();
23
24 local _ENV = nil;
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 winstylemap = {
34         ["0"] = orig_color, -- reset
35         ["1"] = 7+8, -- bold
36         ["1;33"] = 2+4+8, -- bold yellow
37         ["1;31"] = 4+8 -- bold red
38 }
39
40 local cssmap = {
41         [1] = "font-weight: bold", [2] = "opacity: 0.5", [4] = "text-decoration: underline", [8] = "visibility: hidden",
42         [30] = "color:black", [31] = "color:red", [32]="color:green", [33]="color:#FFD700",
43         [34] = "color:blue", [35] = "color: magenta", [36] = "color:cyan", [37] = "color: white",
44         [40] = "background-color:black", [41] = "background-color:red", [42]="background-color:green",
45         [43]="background-color:yellow", [44] = "background-color:blue", [45] = "background-color: magenta",
46         [46] = "background-color:cyan", [47] = "background-color: white";
47 };
48
49 local fmt_string = char(0x1B).."[%sm%s"..char(0x1B).."[0m";
50 local function getstring(style, text)
51         if style then
52                 return format(fmt_string, style, text);
53         else
54                 return text;
55         end
56 end
57
58 local function getstyle(...)
59         local styles, result = { ... }, {};
60         for i, style in ipairs(styles) do
61                 style = stylemap[style];
62                 if style then
63                         t_insert(result, style);
64                 end
65         end
66         return t_concat(result, ";");
67 end
68
69 local last = "0";
70 local function setstyle(style)
71         style = style or "0";
72         if style ~= last then
73                 io_write("\27["..style.."m");
74                 last = style;
75         end
76 end
77
78 if windows then
79         function setstyle(style)
80                 style = style or "0";
81                 if style ~= last then
82                         windows.set_consolecolor(winstylemap[style] or orig_color);
83                         last = style;
84                 end
85         end
86         if not orig_color then
87                 function setstyle(style) end
88         end
89 end
90
91 local function ansi2css(ansi_codes)
92         if ansi_codes == "0" then return "</span>"; end
93         local css = {};
94         for code in ansi_codes:gmatch("[^;]+") do
95                 t_insert(css, cssmap[tonumber(code)]);
96         end
97         return "</span><span style='"..t_concat(css, ";").."'>";
98 end
99
100 local function tohtml(input)
101         return input:gsub("\027%[(.-)m", ansi2css);
102 end
103
104 return {
105         getstring = getstring;
106         getstyle = getstyle;
107         setstyle = setstyle;
108         tohtml = tohtml;
109 };