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
|
local format, rep = string.format, string.rep;
local io_write = io.write;
local pcall = pcall;
local debug = debug;
local tostring = tostring;
local math_max = math.max;
local logger = require "util.logger";
local default_logging = { { to = "console" } };
-- Global log function, because some people are too
-- lazy to get their own...
_G.log = logger.init("general");
local log_sink_types = {};
local get_levels;
local logging_levels = { "debug", "info", "warn", "error", "critical" }
--- Main function to read config, create the appropriate sinks and tell logger module
function setup_logging(log)
log = log or config.get("*", "core", "log") or default_logging;
-- Set default logger
if type(log) == "string" then
if not log:match("^%*") then
end
elseif type(log) == "table" then
-- Advanced configuration chosen
for i, sink_config in ipairs(log) do
local sink_maker = log_sink_types[sink_config.to];
if sink_maker then
if sink_config.levels and not sink_config.source then
-- Create sink
local sink = sink_maker(sink_config);
-- Set sink for all chosen levels
for level in pairs(get_levels(sink_config.levels)) do
logger.add_level_sink(level, sink);
end
elseif sink_config.source and not sink_config.levels then
logger.add_name_sink(sink_config.source, sink_maker(sink_config));
elseif sink_config.source and sink_config.levels then
local levels = get_levels(sink_config.levels);
local sink = sink_maker(sink_config);
logger.add_name_sink(sink_config.source,
function (name, level, ...)
if levels[level] then
return sink(name, level, ...);
end
end);
else
-- All sources
-- Create sink
local sink = sink_maker(sink_config);
-- Set sink for all levels
for _, level in pairs(logging_levels) do
logger.add_level_sink(level, sink);
end
end
else
-- No such sink type
end
end
end
end
--- Definition of built-in logging sinks ---
local math_max, rep = math.max, string.rep;
-- Column width for "source" (used by stdout and console)
function log_sink_types.nowhere()
return function () return false; end;
end
local sourcewidth = 20;
function log_sink_types.stdout()
return function (name, level, message, ...)
sourcewidth = math_max(#name+2, sourcewidth);
local namelen = #name;
if ... then
io_write(name, rep(" ", sourcewidth-namelen), level, "\t", format(message, ...), "\n");
else
io_write(name, rep(" ", sourcewidth-namelen), level, "\t", message, "\n");
end
end
end
do
local getstyle, getstring = require "util.termcolours".getstyle, require "util.termcolours".getstring;
local do_pretty_printing = not os.getenv("WINDIR");
local logstyles = {};
if do_pretty_printing then
logstyles["info"] = getstyle("bold");
logstyles["warn"] = getstyle("bold", "yellow");
logstyles["error"] = getstyle("bold", "red");
end
function log_sink_types.console(config)
-- Really if we don't want pretty colours then just use plain stdout
if not do_pretty_printing then
return log_sink_types.stdout(config);
end
return function (name, level, message, ...)
sourcewidth = math_max(#name+2, sourcewidth);
local namelen = #name;
if ... then
io_write(name, rep(" ", sourcewidth-namelen), getstring(logstyles[level], level), "\t", format(message, ...), "\n");
else
io_write(name, rep(" ", sourcewidth-namelen), getstring(logstyles[level], level), "\t", message, "\n");
end
end
end
end
function log_sink_types.file(config)
local log = config.filename;
local logfile = io.open(log, "a+");
if not logfile then
return function () end
end
local write, format, flush = logfile.write, string.format, logfile.flush;
return function (name, level, message, ...)
if ... then
write(logfile, name, "\t", level, "\t", format(message, ...), "\n");
else
write(logfile, name, "\t" , level, "\t", message, "\n");
end
flush(logfile);
end;
end
function log_sink_types.syslog()
end
--- Helper function to get a set of levels given a "criteria" table
function get_levels(criteria, set)
set = set or {};
if type(criteria) == "string" then
set[criteria] = true;
return set;
end
local min, max = criteria.min, criteria.max;
if min or max then
local in_range;
for _, level in ipairs(logging_levels) do
if min == level then
set[level] = true;
in_range = true;
elseif max == level then
set[level] = true;
return set;
elseif in_range then
set[level] = true;
end
end
end
for _, level in ipairs(criteria) do
set[level] = true;
end
return set;
end
--- Set up logging
setup_logging();
|
