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

local SUPPORTED_TYPES = "application/audio/biosig/chemical/font/image/inode/message/model/rinex/text/vector/video/x-epoc/"

local M = {}

local function match_mimetype(s)
	local type, sub = s:match("([-a-z]+/)([+-.a-zA-Z0-9]+)%s*$")
	if type and sub and string.find(SUPPORTED_TYPES, type, 1, true) then
		return type .. sub
	end
end

function M:preload()
	local urls = {}
	for _, file in ipairs(self.files) do
		urls[#urls + 1] = tostring(file.url)
	end

	local cmd = os.getenv("YAZI_FILE_ONE") or "file"
	local child, code = Command(cmd):args({ "-bL", "--mime-type" }):args(urls):stdout(Command.PIPED):spawn()
	if not child then
		ya.err(string.format("Spawn `%s` command returns %s", cmd, code))
		return 0
	end

	local mimes, last = {}, ya.time()
	local flush = function(force)
		if not force and ya.time() - last < 0.3 then
			return
		end
		if next(mimes) then
			ya.manager_emit("update_mimetype", {}, mimes)
			mimes, last = {}, ya.time()
		end
	end

	local i, j, mime = 1, 0, 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

		mime = match_mimetype(line)
		if mime and string.find(line, mime, 1, true) ~= 1 then
			goto continue
		elseif mime then
			j, mimes[urls[i]] = j + 1, mime
			flush(false)
		end

		i = i + 1
		::continue::
	until i > #urls

	flush(true)
	return j == #urls and 3 or 2
end

return M