-
Notifications
You must be signed in to change notification settings - Fork 0
/
warna.lua
405 lines (334 loc) · 11.7 KB
/
warna.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
#!/usr/bin/env lua
---@class warna
local warna = {}
local ffi_ok, ffi = pcall(require, "ffi")
local on_windows = (not ffi_ok and package.config:sub(1, 1) == "\\") or (ffi_ok and ffi.os == "Windows")
local ver, winver, buildver
if on_windows then
local fd, err = io.popen("ver", "r")
if fd and not err then
local output = fd:read("*a"):gsub("^%s+", ""):gsub("%s+$", "")
_, ver = output:match("(%[Version (.+)%])$")
winver, buildver = ver:match("^(%d+%.%d+)%.(%d+)")
winver = tonumber(winver)
fd:close()
end
end
local tbl_unpack = (unpack or table.unpack)
---@param cmd string
---@return integer
local function execute_cmd(cmd)
local code
if _VERSION == "Lua 5.1" and not jit then
code = os.execute(cmd)
else
_, _, code = os.execute(cmd)
end
return code ---@diagnostic disable-line:return-type-mismatch
end
---@param hex string
---@return number
---@return number
---@return number
local function hex2rgb(hex)
if not hex:match("^#?%x%x%x%x%x%x$") then error(hex .. " is not a hex!") end
hex = hex:gsub("^#", "")
---@diagnostic disable-next-line:return-type-mismatch
return tonumber("0x" .. hex:sub(1, 2)), tonumber("0x" .. hex:sub(3, 4)), tonumber("0x" .. hex:sub(5, 6))
end
---@return number
local function detect_colors()
local term = os.getenv("TERM")
local colorterm = os.getenv("COLORTERM")
---@cast term string
---@cast colorterm string
local min = 0
local no_color = os.getenv("NO_COLOR")
if no_color and no_color ~= "" then return 0 end
local force_color = os.getenv("FORCE_COLOR")
if not (on_windows and force_color) and execute_cmd("[ -t 1 ]") ~= 0 then return -1 end
if force_color and force_color ~= "" then
local level = tonumber(force_color)
return level and math.min(3, level) or 1
end
if os.getenv("TF_BUILD") and os.getenv("AGENT_NAME") then return 1 end
if term == "dumb" then return min end
if on_windows and winver >= 10 and buildver > "10586" then return buildver >= "14931" and 3 or 2 end
if os.getenv("CI") then
if os.getenv("GITHUB_ACTIONS") or os.getenv("GITEA_ACTIONS") then return 3 end
local is_ci = false
for _, sign in pairs({ "TRAVIS", "CIRCLECI", "APPVEYOR", "GITLAB_CI", "BUILDKITE", "DRONE" }) do
if os.getenv(sign) then
is_ci = true
break
end
end
if is_ci or os.getenv("CI_NAME") == "codeship" then return 1 end
end
local teamcity_env = os.getenv("TEAMCITY_VERSION")
if teamcity_env then
return (teamcity_env:match("^9%.(0*[1-9]%d*)%.") or teamcity_env:match("%d%d+%.")) and 1 or 0
end
if colorterm == "truecolor" then return 3 end
if term == "xterm-kitty" then return 3 end
local term_program = os.getenv("TERM_PROGRAM")
if term_program then
local version = tonumber((os.getenv("TERM_PROGRAM_VERSION") or ""):match("^(%d+)") or 0)
if term_program == "iTerm.app" then
return version >= 3 and 3 or 2
elseif term_program == "Apple_Terminal" then
return 2
end
end
if term:match("%-256$") or term:match("%-256color$") then return 2 end
for _, pattern in pairs({ "^screen", "^xterm", "^vt100", "^vt220", "^rxvt", "color", "ansi", "cygwin", "linux" }) do
if term:match(pattern) then return 1 end
end
if colorterm then return 1 end
return min
end
---@param str string
---@return string
local function attributes_to_escsequence(str)
local buff = ""
for str_attr in str:gmatch("([^ ]+)") do
local attr, args = str_attr:match("^([^:]+):-([^:]*)$")
local fn_args = {}
for arg_attr in (args or ""):gmatch("([^,;]+)") do
fn_args[#fn_args + 1] = arg_attr
end
local attr_value = warna.options.level == 0 and warna.attributes[attr]
or warna.options.level > 0 and (warna.attributes.colors[attr] or warna.attributes[attr])
local type_attr_value = type(attr_value)
if type_attr_value == "function" then
buff = buff .. (attr_value(tbl_unpack(fn_args)) or "")
elseif attr_value and type_attr_value ~= "table" then
buff = buff .. ("\27[%sm"):format(tostring(attr_value))
end
end
return buff
end
---@class warna.options
warna.options = {
---
--- Specifies the level of color support.
--- * `-1` — Disable escape sequences completely.
--- * `0` — Disable color support.
--- * `1` — Basic color support (8-16 colors).
--- * `2` — 256 colors support.
--- * `3` — Truecolor support (16 million colors).
---
--- Can be overridden by NO_COLOR and FORCE_COLOR.
---
level = detect_colors(),
}
---@alias warna.attributes_function fun(...: string): string?
---@class warna.attributes
warna.attributes = {
reset = 0,
bright = 1,
bold = 1,
dim = 2,
dark = 2,
italic = 3,
underline = 4,
underl = 4,
blink = 5,
inverse = 7,
reverse = 7,
hidden = 8,
invisible = 8,
strikethrough = 9,
strike = 9,
colors = {
black = 30,
red = 31,
green = 32,
yellow = 33,
blue = 34,
magenta = 35,
cyan = 36,
white = 37,
default = 39,
["bg-black"] = 40,
["bg-red"] = 41,
["bg-green"] = 42,
["bg-yellow"] = 43,
["bg-blue"] = 44,
["bg-magenta"] = 45,
["bg-cyan"] = 46,
["bg-white"] = 47,
["bg-default"] = 49,
---@type warna.attributes_function
color256 = function(n)
if warna.options.level < 2 then return end
return ("\27[38;5;%sm"):format(n)
end,
---@type warna.attributes_function
rgb = function(r, g, b)
if warna.options.level < 3 then return end
return ("\27[38;2;%s;%s;%sm"):format(r, g, b)
end,
---@type warna.attributes_function
hex = function(hex)
if warna.options.level < 3 then return end
return ("\27[38;2;%d;%d;%dm"):format(hex2rgb(hex))
end,
---@type warna.attributes_function
["bg-color256"] = function(n)
if warna.options.level < 2 then return end
return ("\27[48;5;%sm"):format(n)
end,
---@type warna.attributes_function
["bg-rgb"] = function(r, g, b)
if warna.options.level < 3 then return end
return ("\27[48;2;%s;%s;%sm"):format(r, g, b)
end,
---@type warna.attributes_function
["bg-hex"] = function(hex)
if warna.options.level < 3 then return end
return ("\27[48;2;%d;%d;%dm"):format(hex2rgb(hex))
end,
},
}
---@param str string
---@param attrs string[]
---@return string
---
--- Apply attributes to a string.
---
function warna.raw_apply(str, attrs)
return attributes_to_escsequence(table.concat(attrs, " ")) .. str
end
---@param str string
---@param attrs string[]
---@return string
---
--- Similar to `warna.raw_apply`, except the string has `reset` attribute appended.
---
function warna.apply(str, attrs)
return warna.raw_apply(str, attrs) .. (warna.options.level < 0 and "" or "\27[0m")
end
---@param fmt string
---@return string
---
--- Format a string with format attributes.
---
function warna.raw_format(fmt)
return (
fmt:gsub("(%%?%%{(.-)})", function(f, s)
if f:sub(1, 2) == "%%" then return end
return attributes_to_escsequence(s)
end)
)
end
---@param fmt string
---@return string
---
--- Similar to `warna.raw_format`, except the string has `reset` attribute appended.
---
function warna.format(fmt)
return warna.raw_format(fmt) .. (warna.options.level < 0 and "" or "\27[0m")
end
---@param str any
---@return unknown
---
--- Strip a string containing escape sequences (`ESC[...m`).
---
function warna.strip_escapes(str)
return (str:gsub("\27%[.-m", ""))
end
---@param skip_registry boolean?
---@return boolean
---@return string
---
--- Patch the Windows VTE problem.
---
--- Requires Windows 10 build after 14393 (Anniversary update) and `ffi` or [`cffi`](https://github.com/q66/cffi-lua) library to patch.
--- If not fallbacks to editing registry.
---
--- For Windows 10 before build 14393 (Anniversary update) or before Windows 10, requires [ANSICON](https://github.com/adoxa/ansicon) to patch.
---
function warna.windows_patch_vte(skip_registry)
if not on_windows then return false, "not windows" end
if (winver >= 10 and buildver < "14393") or winver < 10 then
return execute_cmd((os.getenv("ANSICON") or "ansicon") .. " -p 2>1 1>NUL") == 0, "ansicon method"
end
if not ffi and on_windows then
if skip_registry then return false, "registry method" end
if execute_cmd("reg query HKCU\\CONSOLE /v VirtualTerminalLevel 2>1 1>NUL") == 0 then
return true, "registry method"
end
io.stderr:write("This script will attempt to edit Registry to enable SGR, allow? [Yy/n...] ")
io.stderr:flush()
local yn = io.read(1)
return yn:lower() == "y" and execute_cmd(
"reg add HKCU\\CONSOLE /f /v VirtualTerminalLevel /t REG_DWORD /d 1 2>1 1>NUL"
) == 0,
"registry method"
end
-- stylua: ignore
ffi.cdef [[
/* borrowed from https://github.com/malkia/luajit-winapi */
typedef uintptr_t UINT_PTR;
typedef int32_t BOOL;
typedef uint32_t DWORD;
typedef UINT_PTR HANDLE;
typedef HANDLE WINAPI_FILE_HANDLE;
typedef DWORD WINAPI_ConsoleModeFlags;
typedef DWORD WINAPI_StdHandle;
BOOL SetConsoleMode(HANDLE hConsoleHandle, WINAPI_ConsoleModeFlags dwMode);
WINAPI_FILE_HANDLE GetStdHandle(WINAPI_StdHandle nStdHandle);
]]
local winapi = ffi.load("kernel32.dll")
return winapi.SetConsoleMode(winapi.GetStdHandle(-11), 7) ~= 0 and winapi.SetConsoleMode(
winapi.GetStdHandle(-12),
7
) ~= 0,
"winapi method"
end
if pcall(debug.getlocal, 4, 1) then
return warna
else
local len_arg = #arg
local text = table.remove(arg, 1) or ""
local flags = ""
if text:sub(1, 1) == "-" and text:find("[hbfar]", 1) then
flags = text:sub(2)
elseif text:sub(1, 1) == "-" and text:find("[^-hbfar]", 1) then
io.stderr:write(("Error: Unknown flag '-%s'\n"):format(text:match("[^-hbfar]")))
os.exit(1)
end
local help_flag = flags:find("h", 1)
if help_flag or len_arg == 0 then
local prog = arg[0]:gsub(".-/", "")
print((([[
Usage: %s <flags>
%s <fmt|text> [<attributes...>]
%s -b <fmt|text> [<attributes...>]
%s -f <fmt>
%s -a <text> [<attributes...>]
Flags: * -h -- Prints the command usage
* -r -- Don't skip editing registry when enabling VTE in Windows.
* -b -- Both format the text and apply attributes to the text (Default if a flag isn't supplied)
* -f -- Format the text
* -a -- Apply the text with attributes
<flags> can be stack up as many as you want. e.g `-ar 'My Text' red`.
Accepts NO_COLOR and FORCE_COLOR to manipulate color support.]]):gsub("%%s", prog)))
os.exit(help_flag and 0 or 1)
end
if flags ~= "" and #arg == 0 then
io.stderr:write("Error: Cannot process a flag, at least supply an input!\n")
os.exit(1)
end
warna.windows_patch_vte(flags:find("r", 1) == nil)
if flags == "" or flags:find("b", 1) then
text = flags ~= "" and table.remove(arg, 1) or text
io.stdout:write(warna.format(warna.apply(text, { table.concat(arg, " ") })))
elseif flags:find("f", 1) then
io.stdout:write(warna.format(arg[1]))
elseif flags:find("a", 1) then
text = table.remove(arg, 1)
io.stdout:write(warna.apply(text, { table.concat(arg, " ") }))
end
end