cc-mek-scada/scada-common/log.lua

--
-- File System Logger
--

local log = {}

-- we use extra short abbreviations since computer craft screens are very small

local MODE = {
    APPEND = 0,
    NEW = 1
}

log.MODE = MODE

local LOG_DEBUG = true

local log_path = "/log.txt"
local mode = MODE.APPEND
local file_handle = nil

local _log = function (msg)
    local stamped = os.date("[%c] ") .. msg

    -- attempt to write log
    local status, result = pcall(function () 
        file_handle.writeLine(stamped)
        file_handle.flush()
    end)

    -- if we don't have much space, we need to create a new log file
    local delete_log = fs.getFreeSpace(log_path) < 100

    if not status then
        if result == "Out of space" then
            delete_log = true
        elseif result ~= nil then
            print("unknown error writing to logfile: " .. result)
        end
    end

    if delete_log then
        -- delete the old log file and open a new one
        file_handle.close()
        fs.delete(log_path)
        init(log_path, mode)

        -- leave a message
        local notif = os.date("[%c] ") .. "recycled log file"
        file_handle.writeLine(notif)
        file_handle.writeLine(stamped)
        file_handle.flush()
    end
end

log.init = function (path, write_mode)
    log_path = path
    mode = write_mode

    if mode == MODE.APPEND then
        file_handle = fs.open(path, "a")
    else
        file_handle = fs.open(path, "w+")
    end
end

log.debug = function (msg, trace)
    if LOG_DEBUG then
        local dbg_info = ""

        if trace then
            local name = ""

            if debug.getinfo(2).name ~= nil then
                name = ":" .. debug.getinfo(2).name .. "():"
            end

            dbg_info = debug.getinfo(2).short_src .. ":" .. name ..
                debug.getinfo(2).currentline .. " > "
        end

        _log("[DBG] " .. dbg_info .. msg)
    end
end

log.info = function (msg)
    _log("[INF] " .. msg)
end

log.warning = function (msg)
    _log("[WRN] " .. msg)
end

log.error = function (msg, trace)
    local dbg_info = ""
    
    if trace then
        local name = ""

        if debug.getinfo(2).name ~= nil then
            name = ":" .. debug.getinfo(2).name .. "():"
        end
        
        dbg_info = debug.getinfo(2).short_src .. ":" .. name .. 
            debug.getinfo(2).currentline .. " > "
    end

    _log("[ERR] " .. dbg_info .. msg)
end

log.fatal = function (msg)
    _log("[FTL] " .. msg)
end

return log