lazy.nvim/lua/lazy/util.lua

173 lines
3.5 KiB
Lua
Raw Normal View History

---@class LazyUtil: LazyUtilCore
2022-11-25 05:03:00 +08:00
local M = setmetatable({}, { __index = require("lazy.core.util") })
function M.file_exists(file)
return vim.loop.fs_stat(file) ~= nil
end
function M.open(uri)
if M.file_exists(uri) then
vim.cmd.split()
2022-11-25 05:03:00 +08:00
return vim.cmd.view(uri)
end
local cmd
if vim.fn.has("win32") == 1 then
cmd = { "explorer", uri }
-- cmd = { 'cmd.exe', '/c', 'start', '""', uri }
2022-11-25 05:03:00 +08:00
elseif vim.fn.has("macunix") == 1 then
cmd = { "open", uri }
else
cmd = { "xdg-open", uri }
end
local ret = vim.fn.jobstart(cmd, { detach = true })
if ret <= 0 then
2022-11-25 05:03:00 +08:00
local msg = {
"Failed to open uri",
ret,
vim.inspect(cmd),
}
vim.notify(table.concat(msg, "\n"), vim.log.levels.ERROR)
end
end
function M.read_file(file)
local fd = assert(io.open(file, "r"))
---@type string
local data = fd:read("*a")
fd:close()
return data
end
function M.write_file(file, contents)
local fd = assert(io.open(file, "w+"))
fd:write(contents)
fd:close()
end
2022-12-23 17:18:19 +08:00
---@generic F: fun()
2022-11-25 05:03:00 +08:00
---@param ms number
2022-12-23 17:18:19 +08:00
---@param fn F
---@return F
2022-11-25 05:03:00 +08:00
function M.throttle(ms, fn)
local timer = vim.loop.new_timer()
local running = false
local first = true
2022-12-23 17:18:19 +08:00
return function(...)
local args = { ... }
local wrapped = function()
fn(unpack(args))
end
2022-11-25 05:03:00 +08:00
if not running then
if first then
2022-12-23 17:18:19 +08:00
wrapped()
2022-11-25 05:03:00 +08:00
first = false
end
timer:start(ms, 0, function()
running = false
2022-12-23 17:18:19 +08:00
vim.schedule(wrapped)
2022-11-25 05:03:00 +08:00
end)
running = true
end
end
end
---@return string?
function M.head(file)
local f = io.open(file)
if f then
local ret = f:read()
f:close()
return ret
end
end
---@return {branch: string, hash:string}?
function M.git_info(dir)
local line = M.head(dir .. "/.git/HEAD")
if line then
---@type string, string
local ref, branch = line:match("ref: (refs/heads/(.*))")
if ref then
return {
branch = branch,
hash = M.head(dir .. "/.git/" .. ref),
}
end
end
end
---@param msg string|string[]
---@param opts? table
function M.markdown(msg, opts)
if type(msg) == "table" then
msg = table.concat(msg, "\n") or msg
end
vim.notify(
msg,
vim.log.levels.INFO,
vim.tbl_deep_extend("force", {
title = "lazy.nvim",
on_open = function(win)
vim.wo[win].conceallevel = 3
vim.wo[win].concealcursor = "n"
vim.wo[win].spell = false
vim.treesitter.start(vim.api.nvim_win_get_buf(win), "markdown")
end,
}, opts or {})
)
end
function M._dump(value, result)
local t = type(value)
if t == "number" or t == "boolean" then
table.insert(result, tostring(value))
elseif t == "string" then
table.insert(result, ("%q"):format(value))
elseif t == "table" then
table.insert(result, "{")
local i = 1
---@diagnostic disable-next-line: no-unknown
for k, v in pairs(value) do
if k == i then
elseif type(k) == "string" then
table.insert(result, ("[%q]="):format(k))
else
table.insert(result, k .. "=")
end
M._dump(v, result)
table.insert(result, ",")
i = i + 1
end
table.insert(result, "}")
else
error("Unsupported type " .. t)
end
end
function M.dump(value)
local result = {}
M._dump(value, result)
return table.concat(result, "")
end
2022-12-05 21:46:11 +08:00
---@generic V
---@param t table<string, V>
---@param fn fun(key:string, value:V)
function M.foreach(t, fn)
---@type string[]
local keys = vim.tbl_keys(t)
pcall(table.sort, keys)
2022-12-05 21:46:11 +08:00
for _, key in ipairs(keys) do
fn(key, t[key])
end
end
2022-11-25 05:03:00 +08:00
return M