You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
86 lines
2.3 KiB
86 lines
2.3 KiB
|
|
local storm = require 'ffi.storm'
|
|
local stormlib = require 'ffi.stormlib'
|
|
|
|
mpq_util = {}
|
|
|
|
-- 从地图中解压出文件来然后调用回调函数更新
|
|
-- map_path - 地图路径，fs.path
|
|
-- path_in_archive - 地图压缩包中的路径，string
|
|
-- process_function - 函数，必须接收一个fs.path对象，返回一个fs.path对象
|
|
-- 形如 function (in_path) return out_path end
|
|
-- 返回值：true表示成功，false表示失败
|
|
function mpq_util:update_file(map_path, path_in_archive, process_function)
|
|
-- 结果
|
|
local result = false
|
|
log.trace("mpq_util.update_file.")
|
|
|
|
-- 打开MPQ（地图）
|
|
local mpq = stormlib.open(map_path)
|
|
if mpq then
|
|
-- 确定解压路径
|
|
local extract_file_path = fs.ydwe_path() / "logs" / "file.out"
|
|
-- 将文件解压
|
|
if mpq:has_file(path_in_archive) and
|
|
mpq:extract(path_in_archive, extract_file_path)
|
|
then
|
|
log.trace(path_in_archive .. " has been extracted from " .. map_path:filename():string())
|
|
|
|
-- 调用处理函数处理
|
|
local success, out_file_path = pcall(process_function, mpq, extract_file_path)
|
|
-- 如果函数正常结束（没有出错）
|
|
if success then
|
|
-- 如果函数成功完成任务
|
|
if out_file_path then
|
|
-- 替换文件
|
|
result = mpq:add_file(path_in_archive, out_file_path)
|
|
else
|
|
-- 出现了错误
|
|
log.error("Processor function cannot complete its task.")
|
|
end
|
|
else
|
|
-- 记录出错原因
|
|
log.error(out_file_path)
|
|
end
|
|
|
|
-- 删除临时文件
|
|
--pcall(fs.remove_all, extract_file_path)
|
|
else
|
|
log.error("Cannot extract " .. path_in_archive)
|
|
end
|
|
|
|
-- 关闭地图
|
|
mpq:close()
|
|
else
|
|
log.error("Cannot open map archive " .. map_path:string())
|
|
end
|
|
|
|
return result
|
|
end
|
|
|
|
-- 从主程序的mpq目录下载入MPQ
|
|
-- mpqname - MPQ的文件名
|
|
-- 返回值：MPQ句柄
|
|
function mpq_util:load_mpq(mpqname, priority)
|
|
local result = 0
|
|
local mpq = fs.ydwe_path() / "share" / "mpq" / mpqname
|
|
|
|
if fs.exists(mpq) then
|
|
virtual_mpq.open_path(mpq, priority)
|
|
return result
|
|
end
|
|
|
|
mpq = fs.ydwe_path() / "share" / "mpq" / (mpqname .. ".mpq")
|
|
if fs.exists(mpq) then
|
|
result = storm.open(mpq, priority)
|
|
if result then
|
|
log.debug("Loaded " .. mpq:filename():string())
|
|
else
|
|
log.error("Cannot load " .. mpq:filename():string())
|
|
end
|
|
else
|
|
log.error("Cannot find " .. mpq:filename():string())
|
|
end
|
|
|
|
return result
|
|
end
|
|
|