yazi/yazi-plugin/preset/plugins/pdf.lua
2024-12-07 20:08:16 +08:00

55 lines
1.2 KiB
Lua

local M = {}
function M:peek(job)
local start, cache = os.clock(), ya.file_cache(job)
if not cache or self:preload(job) ~= 1 then
return
end
ya.sleep(math.max(0, 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 1
end
local output = Command("pdftoppm")
:args({
"-singlefile",
"-jpeg",
"-jpegopt",
"quality=" .. PREVIEW.image_quality,
"-f",
job.skip + 1,
tostring(job.file.url),
})
:stdout(Command.PIPED)
:stderr(Command.PIPED)
:output()
if not output then
return 0
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 0
end
return fs.write(cache, output.stdout) and 1 or 2
end
return M