feat: add a logger based on plenary.log

lua/nvim-lsp-installer/log.lua

@@ -1,18 +1,175 @@
-local M = {}
+--
+-- The following is extracted and modified from plenary.log by TJ Devries.
 
--- TODO
+-- User configuration section
+local default_config = {
+    -- Name of the plugin. Prepended to log messages
+    name = "lsp-installer",
 
-function M.debug(...)
-    -- print("[debug]", vim.inspect(...))
-end
-function M.error(...)
-    -- print("[error]", vim.inspect(...))
-end
-function M.warn(...)
-    -- print("[warn]", vim.inspect(...))
-end
-function M.info(...)
-    -- print("[info]", vim.inspect(...))
+    -- Should print the output to neovim while running
+    -- values: 'sync','async',false
+    use_console = "async",
+
+    -- Should highlighting be used in console (using echohl)
+    highlights = true,
+
+    -- Should write to a file
+    use_file = true,
+
+    -- Any messages above this level will be logged.
+    level = "warn",
+
+    -- Level configuration
+    modes = {
+        { name = "trace", hl = "Comment" },
+        { name = "debug", hl = "Comment" },
+        { name = "info", hl = "None" },
+        { name = "warn", hl = "WarningMsg" },
+        { name = "error", hl = "ErrorMsg" },
+        { name = "fatal", hl = "ErrorMsg" },
+    },
+
+    -- Can limit the number of decimals displayed for floats
+    float_precision = 0.01,
+}
+
+-- {{{ NO NEED TO CHANGE
+local log = {}
+
+local unpack = unpack or table.unpack
+
+log.new = function(config, standalone)
+    config = vim.tbl_deep_extend("force", default_config, config)
+
+    local outfile = string.format("%s/%s.log", vim.api.nvim_call_function("stdpath", { "cache" }), config.name)
+
+    local obj
+    if standalone then
+        obj = log
+    else
+        obj = config
+    end
+
+    local levels = {}
+    for i, v in ipairs(config.modes) do
+        levels[v.name] = i
+    end
+
+    local round = function(x, increment)
+        increment = increment or 1
+        x = x / increment
+        return (x > 0 and math.floor(x + 0.5) or math.ceil(x - 0.5)) * increment
+    end
+
+    local make_string = function(...)
+        local t = {}
+        for i = 1, select("#", ...) do
+            local x = select(i, ...)
+
+            if type(x) == "number" and config.float_precision then
+                x = tostring(round(x, config.float_precision))
+            elseif type(x) == "table" then
+                x = vim.inspect(x)
+            else
+                x = tostring(x)
+            end
+
+            t[#t + 1] = x
+        end
+        return table.concat(t, " ")
+    end
+
+    local log_at_level = function(level, level_config, message_maker, ...)
+        -- Return early if we're below the config.level
+        if level < levels[config.level] then
+            return
+        end
+        local nameupper = level_config.name:upper()
+
+        local msg = message_maker(...)
+        local info = debug.getinfo(config.info_level or 2, "Sl")
+        local lineinfo = info.short_src .. ":" .. info.currentline
+
+        -- Output to console
+        if config.use_console then
+            local log_to_console = function()
+                local console_string = string.format("[%-6s%s] %s: %s", nameupper, os.date "%H:%M:%S", lineinfo, msg)
+
+                if config.highlights and level_config.hl then
+                    vim.cmd(string.format("echohl %s", level_config.hl))
+                end
+
+                local split_console = vim.split(console_string, "\n")
+                for _, v in ipairs(split_console) do
+                    local formatted_msg = string.format("[%s] %s", config.name, vim.fn.escape(v, [["\]]))
+
+                    local ok = pcall(vim.cmd, string.format([[echom "%s"]], formatted_msg))
+                    if not ok then
+                        vim.api.nvim_out_write(msg .. "\n")
+                    end
+                end
+
+                if config.highlights and level_config.hl then
+                    vim.cmd "echohl NONE"
+                end
+            end
+            if config.use_console == "sync" and not vim.in_fast_event() then
+                log_to_console()
+            else
+                vim.schedule(log_to_console)
+            end
+        end
+
+        -- Output to log file
+        if config.use_file then
+            local fp = assert(io.open(outfile, "a"))
+            local str = string.format("[%-6s%s] %s: %s\n", nameupper, os.date(), lineinfo, msg)
+            fp:write(str)
+            fp:close()
+        end
+    end
+
+    for i, x in ipairs(config.modes) do
+        -- log.info("these", "are", "separated")
+        obj[x.name] = function(...)
+            return log_at_level(i, x, make_string, ...)
+        end
+
+        -- log.fmt_info("These are %s strings", "formatted")
+        obj[("fmt_%s"):format(x.name)] = function(...)
+            return log_at_level(i, x, function(...)
+                local passed = { ... }
+                local fmt = table.remove(passed, 1)
+                local inspected = {}
+                for _, v in ipairs(passed) do
+                    table.insert(inspected, vim.inspect(v))
+                end
+                return string.format(fmt, unpack(inspected))
+            end, ...)
+        end
+
+        -- log.lazy_info(expensive_to_calculate)
+        obj[("lazy_%s"):format(x.name)] = function()
+            return log_at_level(i, x, function(f)
+                return f()
+            end)
+        end
+
+        -- log.file_info("do not print")
+        obj[("file_%s"):format(x.name)] = function(vals, override)
+            local original_console = config.use_console
+            config.use_console = false
+            config.info_level = override.info_level
+            log_at_level(i, x, make_string, unpack(vals))
+            config.use_console = original_console
+            config.info_level = nil
+        end
+    end
+
+    return obj
 end
 
-return M
+log.new(default_config, true)
+-- }}}
+
+return log

lua/nvim-lsp-installer/notify.lua

@@ -8,6 +8,12 @@ return function(msg, level)
             title = TITLE,
         })
     else
-        vim.notify(("[%s] %s"):format(TITLE, msg), level)
+        local vim_log_levels = vim.tbl_keys(vim.log.levels)
+        local log_levels = vim.tbl_map(function(lv)
+            return lv:lower()
+        end, vim_log_levels)
+
+        local Log = require "nvim-lsp-installer.log"
+        Log[log_levels[level]](msg)
     end
 end