dotfiles/yazi_1/.config/yazi/plugins/ouch.yazi/main.lua

local M = {}

-- Extract the tree prefix (if any) from a line
local function get_tree_prefix(line)
  local _, prefix_len = line:find("─ ", 1, true)
  if prefix_len then
    return line:sub(1, prefix_len)
  else
    return ""
  end
end

-- Add a filetype icon to a line
local function line_with_icon(line)
  line = line:gsub("[\r\n]+$", "") -- Trailing newlines mess with filetype detection

  local tree_prefix = get_tree_prefix(line)
  local url = line:sub(#tree_prefix + 1)
  local icon = File({
    url = Url(url),
    cha = Cha {
      mode = tonumber(url:sub(-1) == "/" and "40700" or "100644", 8),
      kind = url:sub(-1) == "/" and 1 or 0, -- For Yazi <25.9.x compatibility
    }
  }):icon()

  if icon then
    line =  ui.Line { tree_prefix, ui.Span(icon.text .. " "):style(icon.style), url }
  end

  return line
end

function M:peek(job)
  local cmd = Command("ouch"):arg("l")
  if not job.args.list_view then
    cmd:arg("-t")
  end
  cmd:arg({ "-y", tostring(job.file.url) })
      :stdout(Command.PIPED)
      :stderr(Command.PIPED)

  local child = cmd:spawn()
  local limit = job.area.h
  local archive_icon = job.args.archive_icon or "\u{1f4c1} "
  local file_name = string.match(tostring(job.file.url), ".*[/\\](.*)")
  local lines = { string.format(" %s%s", archive_icon, file_name) }
  local num_skip = 0
  repeat
    local line, event = child:read_line()
    if event == 1 then
      ya.err(tostring(event))
    elseif event ~= 0 then
      break
    end

    if line:find('Archive', 1, true) ~= 1 and line:find('[INFO]', 1, true) ~= 1 then
      if num_skip >= job.skip then
        if job.args.show_file_icons then
          if line:find ('[ERROR]', 1, true) == 1 then
            -- On error, disable file icons for the rest of the output
            job.args.show_file_icons = false
          elseif line:find ('[WARNING]', 1, true) ~= 1 then
            -- Show icons for non-warning lines only
            line = line_with_icon(line)
          end
        end

        line = ui.Line { " ", line } -- One space padding
        table.insert(lines, line)
      else
        num_skip = num_skip + 1
      end
    end
  until #lines >= limit

  child:start_kill()
  if job.skip > 0 and #lines < limit then
    ya.emit(
      "peek",
      { tostring(math.max(0, job.skip - (limit - #lines))), only_if = tostring(job.file.url), upper_bound = "" }
    )
  else
    ya.preview_widget(job, { ui.Text(lines):area(job.area) })
  end
end

function M:seek(job)
  local h = cx.active.current.hovered
  if h and h.url == job.file.url then
    local step = math.floor(job.units * job.area.h / 10)
    ya.emit("peek", {
      math.max(0, cx.active.preview.skip + step),
      only_if = tostring(job.file.url),
    })
  end
end

-- Check if file exists
local function file_exists(name)
  local f = io.open(name, "r")
  if f ~= nil then
    io.close(f)
    return true
  else
    return false
  end
end

-- Get the files that need to be compressed and infer a default archive name
local get_compression_target = ya.sync(function()
  local tab = cx.active
  local default_name
  local paths = {}
  if #tab.selected == 0 then
    if tab.current.hovered then
      local name = tab.current.hovered.name
      default_name = name
      table.insert(paths, name)
    else
      return
    end
  else
    default_name = tab.current.cwd.name
    for _, url in pairs(tab.selected) do
      table.insert(paths, tostring(url))
    end
    -- The compression targets are aquired, now unselect them
    ya.emit("escape", {})
  end
  return paths, default_name
end)

local function invoke_compress_command(paths, name)
  local cmd_output, err_code = Command("ouch")
      :arg({ "c", "-y" })
      :arg(paths)
      :arg(name)
      :stderr(Command.PIPED)
      :output()
  if err_code ~= nil then
    ya.notify({
      title = "Failed to run ouch command",
      content = "Status: " .. err_code,
      timeout = 5.0,
      level = "error",
    })
  elseif not cmd_output.status.success then
    ya.notify({
      title = "Compression failed: status code " .. cmd_output.status.code,
      content = cmd_output.stderr,
      timeout = 5.0,
      level = "error",
    })
  end
end

function M:entry(job)
  local default_fmt = job.args[1]
  if default_fmt == nil then
    default_fmt = "zip"
  end

  ya.emit("escape", { visual = true })

  -- Get the files that need to be compressed and infer a default archive name
  local paths, default_name = get_compression_target()

  -- Get archive name from user
  local output_name, name_event = ya.input({
    title = "Create archive:",
    value = default_name .. "." .. default_fmt,
    position = { "top-center", y = 3, w = 40 },
    pos = { "top-center", y = 3, w = 40 },
  })
  if name_event ~= 1 then
    return
  end

  -- Get confirmation if file exists
  if file_exists(output_name) then
    local confirm, confirm_event = ya.input({
      title = "Overwrite " .. output_name .. "? (y/N)",
      position = { "top-center", y = 3, w = 40 },
      pos = { "top-center", y = 3, w = 40 },
    })
    if not (confirm_event == 1 and confirm:lower() == "y") then
      return
    end
  end

  invoke_compress_command(paths, output_name)
end

return M