mirror of
https://github.com/sxyazi/yazi.git
synced 2026-07-24 00:01:03 +00:00
91 lines
2.1 KiB
Lua
91 lines
2.1 KiB
Lua
-- stylua: ignore
|
|
local TYPE_PATS = { "text", "image", "video", "application", "audio", "font", "inode", "message", "model", "vector", "biosig", "chemical", "rinex", "x%-epoc" }
|
|
|
|
local M = {}
|
|
|
|
local function match_mimetype(line)
|
|
for _, pat in ipairs(TYPE_PATS) do
|
|
local typ, sub = line:match(string.format("(%s/)([+-.a-zA-Z0-9]+)%%s+$", pat))
|
|
if not sub then
|
|
elseif line:find(typ .. sub, 1, true) == 1 then
|
|
return typ:gsub("^x%-", "", 1) .. sub:gsub("^x%-", "", 1):gsub("^vnd%.", "", 1)
|
|
else
|
|
return nil, true
|
|
end
|
|
end
|
|
end
|
|
|
|
local function spawn_file1(paths)
|
|
local bin = os.getenv("YAZI_FILE_ONE") or "file"
|
|
local windows = ya.target_family() == "windows"
|
|
|
|
local cmd = Command(bin):arg({ "-bL", "--mime-type" }):stdout(Command.PIPED)
|
|
if windows then
|
|
cmd:arg({ "-f", "-" }):stdin(Command.PIPED)
|
|
else
|
|
cmd:arg("--"):arg(paths)
|
|
end
|
|
|
|
local child, err = cmd:spawn()
|
|
if not child then
|
|
return nil, Err("Failed to start `%s`, error: %s", bin, err)
|
|
elseif windows then
|
|
child:write_all(table.concat(paths, "\n"))
|
|
child:flush()
|
|
ya.drop(child:take_stdin())
|
|
end
|
|
|
|
return child
|
|
end
|
|
|
|
function M:fetch(job)
|
|
local urls, paths = {}, {}
|
|
for i, file in ipairs(job.files) do
|
|
if file.cache then
|
|
urls[i], paths[i] = tostring(file.url), tostring(file.cache)
|
|
else
|
|
paths[i] = tostring(file.url)
|
|
end
|
|
end
|
|
|
|
local child, err = spawn_file1(paths)
|
|
if not child then
|
|
return true, err
|
|
end
|
|
|
|
local updates, last = {}, ya.time()
|
|
local flush = function(force)
|
|
if not force and ya.time() - last < 0.3 then
|
|
return
|
|
end
|
|
if next(updates) then
|
|
ya.emit("update_mimes", { updates = updates })
|
|
updates, last = {}, ya.time()
|
|
end
|
|
end
|
|
|
|
local i, state, match, ignore = 1, {}, nil, nil
|
|
repeat
|
|
local line, event = child:read_line_with { timeout = 300 }
|
|
if event == 3 then
|
|
flush(true)
|
|
goto continue
|
|
elseif event ~= 0 then
|
|
break
|
|
end
|
|
|
|
match, ignore = match_mimetype(line)
|
|
if match then
|
|
updates[urls[i] or paths[i]], state[i], i = match, true, i + 1
|
|
flush(false)
|
|
elseif not ignore then
|
|
state[i], i = false, i + 1
|
|
end
|
|
::continue::
|
|
until i > #paths
|
|
|
|
flush(true)
|
|
return state
|
|
end
|
|
|
|
return M
|