summaryrefslogtreecommitdiff
path: root/server/src/log.lua
blob: eb34505a06ee259466475138d55071d00d9ebd65 (plain)
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
local fs = require 'bee.filesystem'

local log = {}

log.file = nil
log.start_time = os.time() - os.clock()
log.size = 0
log.max_size = 100 * 1024 * 1024

local function trim_src(src)
    src = src:sub(log.prefix_len, -5)
    src = src:gsub('[\\/]+', '.')
    return src
end

local function push_log(level, ...)
    if not log.path then
        return
    end
    if log.size > log.max_size then
        return
    end
    local t = table.pack(...)
    for i = 1, t.n do
        t[i] = tostring(t[i])
    end
    local str = table.concat(t, '\t')
    if level == 'error' then
        str = str .. '\n' .. debug.traceback(nil, 3)
    end
    if not log.file then
        log.file = io.open(log.path, 'w')
        if not log.file then
            return
        end
        log.file:write('')
        log.file:close()
        log.file = io.open(log.path, 'ab')
        if not log.file then
            return
        end
        log.file:setvbuf 'no'
    end
    local sec, ms = math.modf(log.start_time + os.clock())
    local timestr = os.date('%Y-%m-%d %H:%M:%S', sec)
    local info = debug.getinfo(3, 'Sl')
    local buf
    if info and info.currentline > 0 then
        buf = ('[%s.%03.f][%s]: [%s:%s]%s\n'):format(timestr, ms * 1000, level, trim_src(info.source), info.currentline, str)
    else
        buf = ('[%s.%03.f][%s]: %s\n'):format(timestr, ms * 1000, level, str)
    end
    log.file:write(buf)
    log.size = log.size + #buf
    if log.size > log.max_size then
        log.file:write('[REACH MAX SIZE]')
    end
    return str
end

function log.info(...)
    push_log('info', ...)
end

function log.debug(...)
    push_log('debug', ...)
end

function log.trace(...)
    push_log('trace', ...)
end

function log.warn(...)
    push_log('warn', ...)
end

function log.error(...)
    push_log('error', ...)
end

function log.init(root, path)
    log.path = path:string()
    log.prefix_len = #root:string() + 3
    log.file = nil
    log.size = 0
    if not fs.exists(path:parent_path()) then
        fs.create_directories(path:parent_path())
    end
end

return log