net.dns: Ensure all pending requests get notified of a timeout when looking up a...

[prosody.git] / util / termcolours.lua

diff --git a/util/termcolours.lua b/util/termcolours.lua
index 4e267bee438fe2e7b4970b2c750d3d340001db36..6ef3b689e20dc893eb04c2970033bc97ec11d1f9 100644 (file)
--- a/util/termcolours.lua
+++ b/util/termcolours.lua
@@ -9,7 +9,16 @@
 
 local t_concat, t_insert = table.concat, table.insert;
 local char, format = string.char, string.format;
+local tonumber = tonumber;
 local ipairs = ipairs;
+local io_write = io.write;
+
+local windows;
+if os.getenv("WINDIR") then
+       windows = require "util.windows";
+end
+local orig_color = windows and windows.get_consolecolor and windows.get_consolecolor();
+
 module "termcolours"
 
 local stylemap = {
@@ -19,6 +28,22 @@ local stylemap = {
                        bold = 1, dark = 2, underline = 4, underlined = 4, normal = 0;
                }
 
+local winstylemap = {
+       ["0"] = orig_color, -- reset
+       ["1"] = 7+8, -- bold
+       ["1;33"] = 2+4+8, -- bold yellow
+       ["1;31"] = 4+8 -- bold red
+}
+
+local cssmap = {
+       [1] = "font-weight: bold", [2] = "opacity: 0.5", [4] = "text-decoration: underline", [8] = "visibility: hidden",
+       [30] = "color:black", [31] = "color:red", [32]="color:green", [33]="color:#FFD700",
+       [34] = "color:blue", [35] = "color: magenta", [36] = "color:cyan", [37] = "color: white",
+       [40] = "background-color:black", [41] = "background-color:red", [42]="background-color:green",
+       [43]="background-color:yellow", [44] = "background-color:blue", [45] = "background-color: magenta",
+       [46] = "background-color:cyan", [47] = "background-color: white";
+};
+
 local fmt_string = char(0x1B).."[%sm%s"..char(0x1B).."[0m";
 function getstring(style, text)
        if style then
@@ -39,4 +64,39 @@ function getstyle(...)
        return t_concat(result, ";");
 end
 
+local last = "0";
+function setstyle(style)
+       style = style or "0";
+       if style ~= last then
+               io_write("\27["..style.."m");
+               last = style;
+       end
+end
+
+if windows then
+       function setstyle(style)
+               style = style or "0";
+               if style ~= last then
+                       windows.set_consolecolor(winstylemap[style] or orig_color);
+                       last = style;
+               end
+       end
+       if not orig_color then
+               function setstyle(style) end
+       end
+end
+
+local function ansi2css(ansi_codes)
+       if ansi_codes == "0" then return "</span>"; end
+       local css = {};
+       for code in ansi_codes:gmatch("[^;]+") do
+               t_insert(css, cssmap[tonumber(code)]);
+       end
+       return "</span><span style='"..t_concat(css, ";").."'>";
+end
+
+function tohtml(input)
+       return input:gsub("\027%[(.-)m", ansi2css);
+end
+
 return _M;