yazi/yazi-plugin/preset/plugins/pdf.lua

65 lines
1.6 KiB
Lua

local M = {}
function M:peek(job)
local start, cache = os.clock(), ya.file_cache(job)
if not cache then
return
end
local ok, err = self:preload(job)
if not ok or err then
return
end
ya.sleep(math.max(0, cf.preview.image_delay / 1000 + start - os.clock()))
ya.image_show(cache, job.area)
ya.preview_widgets(job, {})
end
function M:seek(job)
local h = cx.active.current.hovered
if h and h.url == job.file.url then
local step = ya.clamp(-1, job.units, 1)
ya.manager_emit("peek", { math.max(0, cx.active.preview.skip + step), only_if = job.file.url })
end
end
function M:preload(job)
local cache = ya.file_cache(job)
if not cache or fs.cha(cache) then
return true
end
-- stylua: ignore
local output, err = Command("pdftoppm")
:args({
"-f", job.skip + 1,
"-l", job.skip + 1,
"-singlefile",
"-jpeg", "-jpegopt", "quality=" .. cf.preview.image_quality,
"-scale-to-x", cf.preview.max_width, "-scale-to-y", "-1",
tostring(job.file.url),
tostring(cache),
})
:stderr(Command.PIPED)
:output()
if not output then
return true, Err("Failed to start `pdftoppm`, error: %s", err)
elseif not output.status.success then
local pages = tonumber(output.stderr:match("the last page %((%d+)%)")) or 0
if job.skip > 0 and pages > 0 then
ya.manager_emit("peek", { math.max(0, pages - 1), only_if = job.file.url, upper_bound = true })
end
return true, Err("Failed to convert PDF to image, stderr: %s", output.stderr)
end
local ok, err = os.rename(string.format("%s.jpg", cache), tostring(cache))
if ok then
return true
else
return false, Err("Failed to rename `%s.jpg` to `%s`, error: %s", cache, cache, err)
end
end
return M