refactor: split util

Author: folke

lua/lazy/core/util.lua

@@ -20,64 +20,13 @@ function M.track(name, time)
     end
     return entry
   else
+    ---@type LazyProfile
     local entry = table.remove(M._profiles)
     entry.time = vim.loop.hrtime() - entry.time
     return entry
   end
 end
 
-function M.file_exists(file)
-  return vim.loop.fs_stat(file) ~= nil
-end
-
-function M.open(uri)
-  if M.file_exists(uri) then
-    return vim.cmd.view(uri)
-  end
-  local cmd
-  if vim.fn.has("win32") == 1 then
-    cmd = { "cmd.exe", "/c", "start", '""', vim.fn.shellescape(uri) }
-  elseif vim.fn.has("macunix") == 1 then
-    cmd = { "open", uri }
-  else
-    cmd = { "xdg-open", uri }
-  end
-
-  local ret = vim.fn.system(cmd)
-  if vim.v.shell_error ~= 0 then
-    local msg = {
-      "Failed to open uri",
-      ret,
-      vim.inspect(cmd),
-    }
-    vim.notify(table.concat(msg, "\n"), vim.log.levels.ERROR)
-  end
-end
-
----@param ms number
----@param fn fun()
-function M.throttle(ms, fn)
-  local timer = vim.loop.new_timer()
-  local running = false
-  local first = true
-
-  return function()
-    if not running then
-      if first then
-        fn()
-        first = false
-      end
-
-      timer:start(ms, 0, function()
-        running = false
-        vim.schedule(fn)
-      end)
-
-      running = true
-    end
-  end
-end
-
 function M.very_lazy()
   local function _load()
     vim.defer_fn(function()
@@ -103,102 +52,58 @@ function M.very_lazy()
   })
 end
 
+---@alias FileType "file"|"directory"|"link"
+---@alias DirEntry {name: string, path: string, type: FileType}[]
 ---@param path string
-function M.scandir(path)
-  ---@type {name: string, path: string, type: "file"|"directory"|"link"}[]
-  local ret = {}
-
+---@param fn fun(path: string, name:string, type:FileType)
+function M.scandir(path, fn)
   local dir = vim.loop.fs_opendir(path, nil, 100)
-
   if dir then
-    ---@type {name: string, path: string, type: "file"|"directory"|"link"}[]
-    local entries = vim.loop.fs_readdir(dir)
+    local entries = vim.loop.fs_readdir(dir) --[[@as DirEntry[]]
     while entries do
       for _, entry in ipairs(entries) do
         entry.path = path .. "/" .. entry.name
-        table.insert(ret, entry)
+        fn(path .. "/" .. entry.name, entry.name, entry.type)
       end
       entries = vim.loop.fs_readdir(dir)
     end
     vim.loop.fs_closedir(dir)
   end
-
-  return ret
 end
-
-function M.profile()
-  local lines = { "# Profile" }
-
-  ---@param entry LazyProfile
-  local function _profile(entry, depth)
-    if entry.time < 0.5 then
-      -- Nothing
-    end
-
-    table.insert(
-      lines,
-      ("  "):rep(depth) .. "- " .. entry.name .. ": **" .. math.floor((entry.time or 0) / 1e6 * 100) / 100 .. "ms**"
-    )
-
-    for _, child in ipairs(entry) do
-      _profile(child, depth + 1)
+---@param path string
+---@param fn fun(path: string, name:string, type:FileType)
+function M.ls(path, fn)
+  local handle = vim.loop.fs_scandir(path)
+  while handle do
+    local name, t = vim.loop.fs_scandir_next(handle)
+    if not name then
+      break
     end
-  end
-
-  for _, entry in ipairs(M._profiles[1]) do
-    _profile(entry, 1)
-  end
-
-  M.markdown(lines)
-end
-
----@return string?
-function M.head(file)
-  local f = io.open(file)
-  if f then
-    local ret = f:read()
-    f:close()
-    return ret
+    fn(path .. "/" .. name, name, t)
   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),
-      }
+---@param path string
+---@param fn fun(path: string, name:string, type:FileType)
+function M.walk(path, fn)
+  M.ls(path, function(child, name, type)
+    if type == "directory" then
+      M.walk(child, fn)
     end
-  end
+    fn(child, name, type)
+  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 {})
-  )
+---@param root string
+---@param fn fun(modname:string, modpath:string)
+function M.lsmod(root, fn)
+  M.ls(root, function(path, name, type)
+    if type == "file" and name:sub(-4) == ".lua" then
+      fn(name:sub(1, -5), path)
+    elseif type == "directory" and vim.loop.fs_stat(path .. "/init.lua") then
+      fn(name, path .. "/init.lua")
+    end
+  end)
 end
 
 function M.error(msg)
@@ -213,37 +118,4 @@ function M.info(msg)
   })
 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
-
 return M