canola.nvim/lua/oil/fs.lua

local M = {}

---@type boolean
M.is_windows = vim.loop.os_uname().version:match("Windows")

---@type string
M.sep = M.is_windows and "\\" or "/"

---@param ... string
M.join = function(...)
  return table.concat({ ... }, M.sep)
end

---Check if OS path is absolute
---@param dir string
---@return boolean
M.is_absolute = function(dir)
  if M.is_windows then
    return dir:match("^%a:\\")
  else
    return vim.startswith(dir, "/")
  end
end

---@param path string
---@param cb fun(err: nil|string)
M.touch = function(path, cb)
  vim.loop.fs_open(path, "a", 420, function(err, fd) -- 0644
    if err then
      cb(err)
    else
      vim.loop.fs_close(fd, cb)
    end
  end)
end

---@param path string
---@return string
M.posix_to_os_path = function(path)
  if M.is_windows then
    if vim.startswith(path, "/") then
      local drive, rem = path:match("^/([^/]+)/(.*)$")
      return string.format("%s:\\%s", drive, rem:gsub("/", "\\"))
    else
      return path:gsub("/", "\\")
    end
  else
    return path
  end
end

---@param path string
---@return string
M.os_to_posix_path = function(path)
  if M.is_windows then
    if M.is_absolute(path) then
      local drive, rem = path:match("^([^:]+):\\(.*)$")
      return string.format("/%s/%s", drive, rem:gsub("\\", "/"))
    else
      return path:gsub("\\", "/")
    end
  else
    return path
  end
end

local home_dir = vim.loop.os_homedir()

---@param path string
---@return string
M.shorten_path = function(path)
  local cwd = vim.fn.getcwd()
  if vim.startswith(path, cwd) then
    return path:sub(cwd:len() + 2)
  end
  if vim.startswith(path, home_dir) then
    return "~" .. path:sub(home_dir:len() + 1)
  end
  return path
end

M.mkdirp = function(dir)
  local mod = ""
  local path = dir
  while vim.fn.isdirectory(path) == 0 do
    mod = mod .. ":h"
    path = vim.fn.fnamemodify(dir, mod)
  end
  while mod ~= "" do
    mod = mod:sub(3)
    path = vim.fn.fnamemodify(dir, mod)
    vim.loop.fs_mkdir(path, 493)
  end
end

---@param dir string
---@param cb fun(err: nil|string, entries: nil|{type: oil.EntryType, name: string})
M.listdir = function(dir, cb)
  vim.loop.fs_opendir(dir, function(open_err, fd)
    if open_err then
      return cb(open_err)
    end
    local read_next
    read_next = function()
      vim.loop.fs_readdir(fd, function(err, entries)
        if err then
          vim.loop.fs_closedir(fd, function()
            cb(err)
          end)
          return
        elseif entries then
          cb(nil, entries)
          read_next()
        else
          vim.loop.fs_closedir(fd, function(close_err)
            if close_err then
              cb(close_err)
            else
              cb()
            end
          end)
        end
      end)
    end
    read_next()
  end, 100) -- TODO do some testing for this
end

---@param entry_type oil.EntryType
---@param path string
---@param cb fun(err: nil|string)
M.recursive_delete = function(entry_type, path, cb)
  if entry_type ~= "directory" then
    return vim.loop.fs_unlink(path, cb)
  end
  vim.loop.fs_opendir(path, function(open_err, fd)
    if open_err then
      return cb(open_err)
    end
    local poll
    poll = function(inner_cb)
      vim.loop.fs_readdir(fd, function(err, entries)
        if err then
          return inner_cb(err)
        elseif entries then
          local waiting = #entries
          local complete
          complete = function(err2)
            if err then
              complete = function() end
              return inner_cb(err2)
            end
            waiting = waiting - 1
            if waiting == 0 then
              poll(inner_cb)
            end
          end
          for _, entry in ipairs(entries) do
            M.recursive_delete(entry.type, path .. M.sep .. entry.name, complete)
          end
        else
          inner_cb()
        end
      end)
    end
    poll(function(err)
      vim.loop.fs_closedir(fd)
      if err then
        return cb(err)
      end
      vim.loop.fs_rmdir(path, cb)
    end)
  end, 100) -- TODO do some testing for this
end

---@param entry_type oil.EntryType
---@param src_path string
---@param dest_path string
---@param cb fun(err: nil|string)
M.recursive_copy = function(entry_type, src_path, dest_path, cb)
  if entry_type ~= "directory" then
    vim.loop.fs_copyfile(src_path, dest_path, { excl = true }, cb)
    return
  end
  vim.loop.fs_stat(src_path, function(stat_err, src_stat)
    if stat_err then
      return cb(stat_err)
    end
    vim.loop.fs_mkdir(dest_path, src_stat.mode, function(mkdir_err)
      if mkdir_err then
        return cb(mkdir_err)
      end
      vim.loop.fs_opendir(src_path, function(open_err, fd)
        if open_err then
          return cb(open_err)
        end
        local poll
        poll = function(inner_cb)
          vim.loop.fs_readdir(fd, function(err, entries)
            if err then
              return inner_cb(err)
            elseif entries then
              local waiting = #entries
              local complete
              complete = function(err2)
                if err then
                  complete = function() end
                  return inner_cb(err2)
                end
                waiting = waiting - 1
                if waiting == 0 then
                  poll(inner_cb)
                end
              end
              for _, entry in ipairs(entries) do
                M.recursive_copy(
                  entry.type,
                  src_path .. M.sep .. entry.name,
                  dest_path .. M.sep .. entry.name,
                  complete
                )
              end
            else
              inner_cb()
            end
          end)
        end
        poll(cb)
      end, 100) -- TODO do some testing for this
    end)
  end)
end

---@param entry_type oil.EntryType
---@param src_path string
---@param dest_path string
---@param cb fun(err: nil|string)
M.recursive_move = function(entry_type, src_path, dest_path, cb)
  vim.loop.fs_rename(src_path, dest_path, function(err)
    if err then
      -- fs_rename fails for cross-partition or cross-device operations.
      -- We then fall back to a copy + delete
      M.recursive_copy(entry_type, src_path, dest_path, function(err2)
        if err2 then
          cb(err2)
        else
          M.recursive_delete(entry_type, src_path, cb)
        end
      end)
    else
      cb()
    end
  end)
end

return M