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
|
local array = require "prosody.util.array";
local utf8 = rawget(_G, "utf8") or require"prosody.util.encodings".utf8;
local len = utf8.len or function(s)
local _, count = s:gsub("[%z\001-\127\194-\253][\128-\191]*", "");
return count;
end;
local function getchar(n)
local stty_ret = os.execute("stty raw -echo 2>/dev/null");
local ok, char;
if stty_ret then
ok, char = pcall(io.read, n or 1);
os.execute("stty sane");
else
ok, char = pcall(io.read, "*l");
if ok then
char = char:sub(1, n or 1);
end
end
if ok then
return char;
end
end
local function getline()
local ok, line = pcall(io.read, "*l");
if ok then
return line;
end
end
local function getpass()
local stty_ret = os.execute("stty -echo 2>/dev/null");
if not stty_ret then
io.write("\027[08m"); -- ANSI 'hidden' text attribute
end
local ok, pass = pcall(io.read, "*l");
if stty_ret then
os.execute("stty sane");
else
io.write("\027[00m");
end
io.write("\n");
if ok then
return pass;
end
end
local function show_yesno(prompt)
io.write(prompt, " ");
local choice = getchar():lower();
io.write("\n");
if not choice:match("%a") then
choice = prompt:match("%[.-(%U).-%]$");
if not choice then return nil; end
end
return (choice == "y");
end
local function read_password()
local password;
while true do
io.write("Enter new password: ");
password = getpass();
if not password then
print("No password - cancelled");
return;
end
io.write("Retype new password: ");
if getpass() ~= password then
if not show_yesno [=[Passwords did not match, try again? [Y/n]]=] then
return;
end
else
break;
end
end
return password;
end
local function show_prompt(prompt)
io.write(prompt, " ");
local line = getline();
line = line and line:gsub("\n$","");
return (line and #line > 0) and line or nil;
end
local function printf(fmt, ...)
print(fmt:format(...));
end
local function padright(s, width)
return s..string.rep(" ", width-len(s));
end
local function padleft(s, width)
return string.rep(" ", width-len(s))..s;
end
local pat = "[%z\001-\127\194-\253][\128-\191]*";
local function utf8_cut(s, pos)
return s:match("^"..pat:rep(pos)) or s;
end
if utf8.len and utf8.offset then
function utf8_cut(s, pos)
return s:sub(1, utf8.offset(s, pos+1)-1);
end
end
local function term_width(default)
local env_cols = os.getenv "COLUMNS";
if env_cols then return env_cols; end
local stty = io.popen("stty -a");
if not stty then return default; end
local result = stty:read("*a");
if result then
result = result:match("%f[%w]columns[ =]*(%d+)");
end
stty:close();
return tonumber(result or default);
end
local function ellipsis(s, width)
if len(s) <= width then return s; end
if width == 1 then return "…"; end
return utf8_cut(s, width - 1) .. "…";
end
local function new_table(col_specs, max_width)
max_width = max_width or term_width(os.getenv("COLUMNS") or 80);
local separator = " | ";
local widths = {};
local total_width = max_width - #separator * (#col_specs-1);
local free_width = total_width;
-- Calculate width of fixed-size columns
for i = 1, #col_specs do
local width = col_specs[i].width or "0";
if not (type(width) == "string" and width:match("[p%%]$")) then
local title = col_specs[i].title;
width = math.max(tonumber(width), title and (#title+1) or 0);
widths[i] = width;
free_width = free_width - width;
if i > 1 then
free_width = free_width - #separator;
end
end
end
-- Calculate width of proportional columns
local total_proportional_width = 0;
for i = 1, #col_specs do
if not widths[i] then
local width_spec = col_specs[i].width:match("([%d%.]+)[p%%]");
total_proportional_width = total_proportional_width + tonumber(width_spec);
end
end
for i = 1, #col_specs do
if not widths[i] then
local width_spec = col_specs[i].width:match("([%d%.]+)[p%%]");
local rel_width = tonumber(width_spec);
widths[i] = math.floor(free_width*(rel_width/total_proportional_width));
end
end
return function (row)
local titles;
if not row then
titles, row = true, array.pluck(col_specs, "title", "");
end
local output = {};
for i, column in ipairs(col_specs) do
local width = widths[i];
local v = row[not titles and column.key or i];
if not titles and column.mapper then
v = column.mapper(v, row, width, column);
end
if v == nil then
v = column.default or "";
else
v = tostring(v);
end
if len(v) < width then
if column.align == "right" then
v = padleft(v, width);
else
v = padright(v, width);
end
elseif len(v) > width then
v = (column.ellipsis or ellipsis)(v, width);
end
table.insert(output, v);
end
return table.concat(output, separator);
end, max_width;
end
return {
getchar = getchar;
getline = getline;
getpass = getpass;
show_yesno = show_yesno;
read_password = read_password;
show_prompt = show_prompt;
printf = printf;
padleft = padleft;
padright = padright;
term_width = term_width;
ellipsis = ellipsis;
table = new_table;
};
|
