Software /
code /
prosody
Annotate
util/termcolours.lua @ 7203:be8b88ad35a3
util.termcolours: Add 256 color support
author | Kim Alvefur <zash@zash.se> |
---|---|
date | Sat, 27 Feb 2016 16:47:12 +0100 |
parent | 7199:5846e0bca4ff |
child | 7204:ab2b7496a617 |
rev | line source |
---|---|
1523
841d61be198f
Remove version number from copyright headers
Matthew Wild <mwild1@gmail.com>
parents:
896
diff
changeset
|
1 -- Prosody IM |
2923
b7049746bd29
Update copyright headers for 2010
Matthew Wild <mwild1@gmail.com>
parents:
1523
diff
changeset
|
2 -- Copyright (C) 2008-2010 Matthew Wild |
b7049746bd29
Update copyright headers for 2010
Matthew Wild <mwild1@gmail.com>
parents:
1523
diff
changeset
|
3 -- Copyright (C) 2008-2010 Waqas Hussain |
5776
bd0ff8ae98a8
Remove all trailing whitespace
Florian Zeitz <florob@babelmonkeys.de>
parents:
4698
diff
changeset
|
4 -- |
758 | 5 -- This project is MIT/X11 licensed. Please see the |
6 -- COPYING file in the source package for more information. | |
519
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
262
diff
changeset
|
7 -- |
7198
48d167f652ad
util.termcolours: Silence luacheck warning
Kim Alvefur <zash@zash.se>
parents:
6777
diff
changeset
|
8 -- |
48d167f652ad
util.termcolours: Silence luacheck warning
Kim Alvefur <zash@zash.se>
parents:
6777
diff
changeset
|
9 -- luacheck: ignore 213/i |
519
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
262
diff
changeset
|
10 |
cccd610a0ef9
Insert copyright/license headers
Matthew Wild <mwild1@gmail.com>
parents:
262
diff
changeset
|
11 |
262 | 12 local t_concat, t_insert = table.concat, table.insert; |
13 local char, format = string.char, string.format; | |
4698
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
14 local tonumber = tonumber; |
262 | 15 local ipairs = ipairs; |
3749
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
16 local io_write = io.write; |
7203
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
17 local m_floor = math.floor; |
3749
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
18 |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
19 local windows; |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
20 if os.getenv("WINDIR") then |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
21 windows = require "util.windows"; |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
22 end |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
23 local orig_color = windows and windows.get_consolecolor and windows.get_consolecolor(); |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
24 |
6777
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
25 local _ENV = nil; |
262 | 26 |
27 local stylemap = { | |
28 reset = 0; bright = 1, dim = 2, underscore = 4, blink = 5, reverse = 7, hidden = 8; | |
29 black = 30; red = 31; green = 32; yellow = 33; blue = 34; magenta = 35; cyan = 36; white = 37; | |
30 ["black background"] = 40; ["red background"] = 41; ["green background"] = 42; ["yellow background"] = 43; ["blue background"] = 44; ["magenta background"] = 45; ["cyan background"] = 46; ["white background"] = 47; | |
31 bold = 1, dark = 2, underline = 4, underlined = 4, normal = 0; | |
32 } | |
33 | |
3749
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
34 local winstylemap = { |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
35 ["0"] = orig_color, -- reset |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
36 ["1"] = 7+8, -- bold |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
37 ["1;33"] = 2+4+8, -- bold yellow |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
38 ["1;31"] = 4+8 -- bold red |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
39 } |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
40 |
4698
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
41 local cssmap = { |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
42 [1] = "font-weight: bold", [2] = "opacity: 0.5", [4] = "text-decoration: underline", [8] = "visibility: hidden", |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
43 [30] = "color:black", [31] = "color:red", [32]="color:green", [33]="color:#FFD700", |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
44 [34] = "color:blue", [35] = "color: magenta", [36] = "color:cyan", [37] = "color: white", |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
45 [40] = "background-color:black", [41] = "background-color:red", [42]="background-color:green", |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
46 [43]="background-color:yellow", [44] = "background-color:blue", [45] = "background-color: magenta", |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
47 [46] = "background-color:cyan", [47] = "background-color: white"; |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
48 }; |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
49 |
262 | 50 local fmt_string = char(0x1B).."[%sm%s"..char(0x1B).."[0m"; |
6777
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
51 local function getstring(style, text) |
262 | 52 if style then |
53 return format(fmt_string, style, text); | |
54 else | |
55 return text; | |
56 end | |
57 end | |
58 | |
7203
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
59 local function gray(n) |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
60 return m_floor(n*3/32)+0xe8; |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
61 end |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
62 local function color(r,g,b) |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
63 if r == g and g == b then |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
64 return gray(r); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
65 end |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
66 r = m_floor(r*3/128); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
67 g = m_floor(g*3/128); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
68 b = m_floor(b*3/128); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
69 return 0x10 + ( r * 36 ) + ( g * 6 ) + ( b ); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
70 end |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
71 local function hex2rgb(hex) |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
72 local r = tonumber(hex:sub(1,2),16); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
73 local g = tonumber(hex:sub(3,4),16); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
74 local b = tonumber(hex:sub(5,6),16); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
75 return r,g,b; |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
76 end |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
77 |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
78 setmetatable(stylemap, { __index = function(_, style) |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
79 local g = style:sub(7) == " background" and "48;5;" or "38;5;"; |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
80 return g .. color(hex2rgb(style)); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
81 end } ); |
be8b88ad35a3
util.termcolours: Add 256 color support
Kim Alvefur <zash@zash.se>
parents:
7199
diff
changeset
|
82 |
6777
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
83 local function getstyle(...) |
262 | 84 local styles, result = { ... }, {}; |
85 for i, style in ipairs(styles) do | |
86 style = stylemap[style]; | |
87 if style then | |
88 t_insert(result, style); | |
89 end | |
90 end | |
91 return t_concat(result, ";"); | |
92 end | |
93 | |
3749
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
94 local last = "0"; |
6777
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
95 local function setstyle(style) |
3749
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
96 style = style or "0"; |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
97 if style ~= last then |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
98 io_write("\27["..style.."m"); |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
99 last = style; |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
100 end |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
101 end |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
102 |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
103 if windows then |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
104 function setstyle(style) |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
105 style = style or "0"; |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
106 if style ~= last then |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
107 windows.set_consolecolor(winstylemap[style] or orig_color); |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
108 last = style; |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
109 end |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
110 end |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
111 if not orig_color then |
7199
5846e0bca4ff
util.termcolours: Remove argument from noop function [luacheck]
Kim Alvefur <zash@zash.se>
parents:
7198
diff
changeset
|
112 function setstyle() end |
3749
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
113 end |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
114 end |
588c09d7903c
util.termcolours: Added setstyle(str), which works on Windows too.
Waqas Hussain <waqas20@gmail.com>
parents:
2923
diff
changeset
|
115 |
4698
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
116 local function ansi2css(ansi_codes) |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
117 if ansi_codes == "0" then return "</span>"; end |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
118 local css = {}; |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
119 for code in ansi_codes:gmatch("[^;]+") do |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
120 t_insert(css, cssmap[tonumber(code)]); |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
121 end |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
122 return "</span><span style='"..t_concat(css, ";").."'>"; |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
123 end |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
124 |
6777
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
125 local function tohtml(input) |
4698
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
126 return input:gsub("\027%[(.-)m", ansi2css); |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
127 end |
739f7ae1a01e
util.termcolours: tohtml() for converting output to HTML. I don't know.
Matthew Wild <mwild1@gmail.com>
parents:
3749
diff
changeset
|
128 |
6777
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
129 return { |
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
130 getstring = getstring; |
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
131 getstyle = getstyle; |
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
132 setstyle = setstyle; |
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
133 tohtml = tohtml; |
5de6b93d0190
util.*: Remove use of module() function, make all module functions local and return them in a table at the end
Kim Alvefur <zash@zash.se>
parents:
5776
diff
changeset
|
134 }; |