canola.nvim/lua/oil/mutator/progress.lua

local config = require("oil.config")
local columns = require("oil.columns")
local layout = require("oil.layout")
local loading = require("oil.loading")
local util = require("oil.util")
local Progress = {}

local FPS = 20

function Progress.new()
  local bufnr = vim.api.nvim_create_buf(false, true)
  vim.bo[bufnr].bufhidden = "wipe"
  return setmetatable({
    lines = { "", "" },
    count = "",
    spinner = "",
    bufnr = bufnr,
    winid = nil,
    min_bufnr = nil,
    min_winid = nil,
    autocmds = {},
    closing = false,
  }, {
    __index = Progress,
  })
end

---@param opts nil|table
---    cancel fun()
function Progress:show(opts)
  opts = opts or {}
  if self.winid and vim.api.nvim_win_is_valid(self.winid) then
    return
  end
  self.closing = false
  self.cancel = opts.cancel
  local loading_iter = loading.get_bar_iter()
  local spinner = loading.get_iter("dots")
  self.timer = vim.loop.new_timer()
  self.timer:start(
    0,
    math.floor(1000 / FPS),
    vim.schedule_wrap(function()
      self.lines[2] = string.format("%s %s", self.count, loading_iter())
      self.spinner = spinner()
      self:_render()
    end)
  )
  local width, height = layout.calculate_dims(120, 10, config.progress)
  self.winid = vim.api.nvim_open_win(self.bufnr, true, {
    relative = "editor",
    width = width,
    height = height,
    row = math.floor((layout.get_editor_height() - height) / 2),
    col = math.floor((layout.get_editor_width() - width) / 2),
    zindex = 152, -- render on top of the floating window title
    style = "minimal",
    border = config.progress.border,
  })
  vim.bo[self.bufnr].filetype = "oil_progress"
  for k, v in pairs(config.preview.win_options) do
    vim.wo[self.winid][k] = v
  end
  table.insert(
    self.autocmds,
    vim.api.nvim_create_autocmd("VimResized", {
      callback = function()
        self:_reposition()
      end,
    })
  )
  table.insert(
    self.autocmds,
    vim.api.nvim_create_autocmd("WinLeave", {
      callback = function()
        self:minimize()
      end,
    })
  )
  local cancel = self.cancel or function() end
  local minimize = function()
    if self.winid and vim.api.nvim_win_is_valid(self.winid) then
      vim.api.nvim_win_close(self.winid, true)
    end
  end
  vim.keymap.set("n", "c", cancel, { buffer = self.bufnr, nowait = true })
  vim.keymap.set("n", "C", cancel, { buffer = self.bufnr, nowait = true })
  vim.keymap.set("n", "m", minimize, { buffer = self.bufnr, nowait = true })
  vim.keymap.set("n", "M", minimize, { buffer = self.bufnr, nowait = true })
end

function Progress:_render()
  if self.bufnr and vim.api.nvim_buf_is_valid(self.bufnr) then
    util.render_text(
      self.bufnr,
      self.lines,
      { winid = self.winid, actions = { "[M]inimize", "[C]ancel" } }
    )
  end
  if self.min_bufnr and vim.api.nvim_buf_is_valid(self.min_bufnr) then
    util.render_text(
      self.min_bufnr,
      { string.format("%sOil: %s", self.spinner, self.count) },
      { winid = self.min_winid, h_align = "left" }
    )
  end
end

function Progress:_reposition()
  if self.winid and vim.api.nvim_win_is_valid(self.winid) then
    local min_width = 120
    local line_width = vim.api.nvim_strwidth(self.lines[1])
    if line_width > min_width then
      min_width = line_width
    end
    local width, height = layout.calculate_dims(min_width, 10, config.progress)
    vim.api.nvim_win_set_config(self.winid, {
      relative = "editor",
      width = width,
      height = height,
      row = math.floor((layout.get_editor_height() - height) / 2),
      col = math.floor((layout.get_editor_width() - width) / 2),
      zindex = 152, -- render on top of the floating window title
    })
  end
end

function Progress:_cleanup_main_win()
  if self.winid then
    if vim.api.nvim_win_is_valid(self.winid) then
      vim.api.nvim_win_close(self.winid, true)
    end
    self.winid = nil
  end
  for _, id in ipairs(self.autocmds) do
    vim.api.nvim_del_autocmd(id)
  end
  self.autocmds = {}
  self.bufnr = nil
end

function Progress:minimize()
  if self.closing then
    return
  end
  self:_cleanup_main_win()
  local bufnr = vim.api.nvim_create_buf(false, true)
  vim.bo[bufnr].bufhidden = "wipe"
  local winid = vim.api.nvim_open_win(bufnr, false, {
    relative = "editor",
    width = 16,
    height = 1,
    anchor = "SE",
    row = layout.get_editor_height(),
    col = layout.get_editor_width(),
    zindex = 152, -- render on top of the floating window title
    style = "minimal",
    border = config.progress.minimized_border,
  })
  self.min_bufnr = bufnr
  self.min_winid = winid
  self:_render()
end

---@param action oil.Action
---@param idx integer
---@param total integer
function Progress:set_action(action, idx, total)
  local adapter = util.get_adapter_for_action(action)
  local change_line
  if action.type == "change" then
    change_line = columns.render_change_action(adapter, action)
  else
    change_line = adapter.render_action(action)
  end
  self.lines[1] = change_line
  self.count = string.format("%d/%d", idx, total)
  self:_reposition()
  self:_render()
end

function Progress:close()
  self.closing = true
  if self.timer then
    self.timer:close()
    self.timer = nil
  end
  self:_cleanup_main_win()
  if self.min_winid and vim.api.nvim_win_is_valid(self.min_winid) then
    vim.api.nvim_win_close(self.min_winid, true)
  end
  self.min_winid = nil
  self.min_bufnr = nil
end

return Progress