add cache and rename some files
This commit is contained in:
@@ -0,0 +1,666 @@
|
||||
local fs = require 'bee.filesystem'
|
||||
|
||||
local type = type
|
||||
local ioOpen = io.open
|
||||
local pcall = pcall
|
||||
local pairs = pairs
|
||||
local setmetatable = setmetatable
|
||||
local next = next
|
||||
local tostring = tostring
|
||||
local tableSort = table.sort
|
||||
|
||||
_ENV = nil
|
||||
|
||||
---@class fs-utility
|
||||
local m = {}
|
||||
--- 读取文件
|
||||
---@param path string|fs.path
|
||||
function m.loadFile(path, keepBom)
|
||||
if type(path) ~= 'string' then
|
||||
---@diagnostic disable-next-line: undefined-field
|
||||
path = path:string()
|
||||
end
|
||||
local f, e = ioOpen(path, 'rb')
|
||||
if not f then
|
||||
return nil, e
|
||||
end
|
||||
local text = f:read 'a'
|
||||
f:close()
|
||||
if not keepBom then
|
||||
if text:sub(1, 3) == '\xEF\xBB\xBF' then
|
||||
return text:sub(4)
|
||||
end
|
||||
if text:sub(1, 2) == '\xFF\xFE'
|
||||
or text:sub(1, 2) == '\xFE\xFF' then
|
||||
return text:sub(3)
|
||||
end
|
||||
end
|
||||
return text
|
||||
end
|
||||
|
||||
--- 写入文件
|
||||
---@param path any
|
||||
---@param content string
|
||||
function m.saveFile(path, content)
|
||||
if type(path) ~= 'string' then
|
||||
---@diagnostic disable-next-line: undefined-field
|
||||
path = path:string()
|
||||
end
|
||||
local f, e = ioOpen(path, "wb")
|
||||
|
||||
if f then
|
||||
f:write(content)
|
||||
f:close()
|
||||
return true
|
||||
else
|
||||
return false, e
|
||||
end
|
||||
end
|
||||
|
||||
function m.relative(path, base)
|
||||
local sPath = fs.absolute(path):string()
|
||||
local sBase = fs.absolute(base):string()
|
||||
--TODO 先只支持最简单的情况
|
||||
if sPath:sub(1, #sBase) == sBase
|
||||
and sPath:sub(#sBase + 1, #sBase + 1):match '^[/\\]' then
|
||||
return fs.path(sPath:sub(#sBase + 1):gsub('^[/\\]+', ''))
|
||||
end
|
||||
return nil
|
||||
end
|
||||
|
||||
local function buildOption(option)
|
||||
option = option or {}
|
||||
option.add = option.add or {}
|
||||
option.del = option.del or {}
|
||||
option.mod = option.mod or {}
|
||||
option.err = option.err or {}
|
||||
return option
|
||||
end
|
||||
|
||||
local function split(str, sep)
|
||||
local t = {}
|
||||
local current = 1
|
||||
while current <= #str do
|
||||
local s, e = str:find(sep, current)
|
||||
if not s then
|
||||
t[#t+1] = str:sub(current)
|
||||
break
|
||||
end
|
||||
if s > 1 then
|
||||
t[#t+1] = str:sub(current, s - 1)
|
||||
end
|
||||
current = e + 1
|
||||
end
|
||||
return t
|
||||
end
|
||||
|
||||
---@class dummyfs
|
||||
---@operator div(string|fs.path|dummyfs): dummyfs
|
||||
---@field files table
|
||||
local dfs = {}
|
||||
dfs.__index = dfs
|
||||
dfs.type = 'dummy'
|
||||
dfs.path = ''
|
||||
|
||||
---@return dummyfs
|
||||
function m.dummyFS(t)
|
||||
return setmetatable({
|
||||
files = t or {},
|
||||
}, dfs)
|
||||
end
|
||||
|
||||
function dfs:__tostring()
|
||||
return 'dummy:' .. tostring(self.path)
|
||||
end
|
||||
|
||||
function dfs:__div(filename)
|
||||
if type(filename) ~= 'string' then
|
||||
filename = filename:string()
|
||||
end
|
||||
local new = m.dummyFS(self.files)
|
||||
if self.path:sub(-1):match '[^/\\]' then
|
||||
new.path = self.path .. '\\' .. filename
|
||||
else
|
||||
new.path = self.path .. filename
|
||||
end
|
||||
return new
|
||||
end
|
||||
|
||||
---@package
|
||||
function dfs:_open(index)
|
||||
local paths = split(self.path, '[/\\]')
|
||||
local current = self.files
|
||||
if not index then
|
||||
index = #paths
|
||||
elseif index < 0 then
|
||||
index = #paths + index + 1
|
||||
end
|
||||
for i = 1, index do
|
||||
local path = paths[i]
|
||||
if current[path] then
|
||||
current = current[path]
|
||||
else
|
||||
return nil
|
||||
end
|
||||
end
|
||||
return current
|
||||
end
|
||||
|
||||
---@package
|
||||
function dfs:_filename()
|
||||
return self.path:match '[^/\\]+$'
|
||||
end
|
||||
|
||||
function dfs:parent_path()
|
||||
local new = m.dummyFS(self.files)
|
||||
if self.path:find('[/\\]') then
|
||||
new.path = self.path:gsub('[/\\]+[^/\\]*$', '')
|
||||
else
|
||||
new.path = ''
|
||||
end
|
||||
return new
|
||||
end
|
||||
|
||||
function dfs:filename()
|
||||
local new = m.dummyFS(self.files)
|
||||
new.path = self:_filename()
|
||||
return new
|
||||
end
|
||||
|
||||
function dfs:string()
|
||||
return self.path
|
||||
end
|
||||
|
||||
---@return fun(): dummyfs?
|
||||
function dfs:listDirectory()
|
||||
local dir = self:_open()
|
||||
if type(dir) ~= 'table' then
|
||||
return function () end
|
||||
end
|
||||
local keys = {}
|
||||
for k in pairs(dir) do
|
||||
keys[#keys+1] = k
|
||||
end
|
||||
tableSort(keys)
|
||||
local i = 0
|
||||
return function ()
|
||||
i = i + 1
|
||||
local k = keys[i]
|
||||
if not k then
|
||||
return nil
|
||||
end
|
||||
return self / k
|
||||
end
|
||||
end
|
||||
|
||||
function dfs:isDirectory()
|
||||
local target = self:_open()
|
||||
if type(target) == 'table' then
|
||||
return true
|
||||
end
|
||||
return false
|
||||
end
|
||||
|
||||
function dfs:remove()
|
||||
local dir = self:_open(-2)
|
||||
local filename = self:_filename()
|
||||
if not filename then
|
||||
return
|
||||
end
|
||||
dir[filename] = nil
|
||||
end
|
||||
|
||||
function dfs:exists()
|
||||
local target = self:_open()
|
||||
return target ~= nil
|
||||
end
|
||||
|
||||
function dfs:createDirectories(path)
|
||||
if not path then
|
||||
return false
|
||||
end
|
||||
if type(path) ~= 'string' then
|
||||
path = path:string()
|
||||
end
|
||||
local paths = split(path, '[/\\]+')
|
||||
local current = self.files
|
||||
for i = 1, #paths do
|
||||
local sub = paths[i]
|
||||
if current[sub] then
|
||||
if type(current[sub]) ~= 'table' then
|
||||
return false
|
||||
end
|
||||
else
|
||||
current[sub] = {}
|
||||
end
|
||||
current = current[sub]
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
function dfs:saveFile(path, text)
|
||||
if not path then
|
||||
return false, 'no path'
|
||||
end
|
||||
if type(path) ~= 'string' then
|
||||
path = path:string()
|
||||
end
|
||||
local temp = m.dummyFS(self.files)
|
||||
temp.path = path
|
||||
local dir = temp:_open(-2)
|
||||
if not dir then
|
||||
return false, '无法打开:' .. path
|
||||
end
|
||||
local filename = temp:_filename()
|
||||
if not filename then
|
||||
return false, '无法打开:' .. path
|
||||
end
|
||||
if type(dir[filename]) == 'table' then
|
||||
return false, '无法打开:' .. path
|
||||
end
|
||||
dir[filename] = text
|
||||
return true
|
||||
end
|
||||
|
||||
---@param path string|fs.path|dummyfs
|
||||
---@param option table
|
||||
---@return fs.path|dummyfs?
|
||||
local function fsAbsolute(path, option)
|
||||
if type(path) == 'string' then
|
||||
local suc, res = pcall(fs.path, path)
|
||||
if not suc then
|
||||
option.err[#option.err+1] = res
|
||||
return nil
|
||||
end
|
||||
path = res
|
||||
elseif type(path) == 'table' then
|
||||
return path
|
||||
end
|
||||
local suc, res = pcall(fs.absolute, path)
|
||||
if not suc then
|
||||
option.err[#option.err+1] = res
|
||||
return nil
|
||||
end
|
||||
return res
|
||||
end
|
||||
|
||||
local function fsIsDirectory(path)
|
||||
if not path then
|
||||
return false
|
||||
end
|
||||
if path.type == 'dummy' then
|
||||
return path:isDirectory()
|
||||
end
|
||||
---@cast path -dummyfs
|
||||
local status = fs.symlink_status(path):type()
|
||||
return status == 'directory'
|
||||
end
|
||||
|
||||
---@param path fs.path|dummyfs|nil
|
||||
---@param option table
|
||||
---@return fun(): fs.path|dummyfs|nil
|
||||
local function fsPairs(path, option)
|
||||
if not path then
|
||||
return function () end
|
||||
end
|
||||
if path.type == 'dummy' then
|
||||
return path:listDirectory()
|
||||
end
|
||||
local suc, res = pcall(fs.pairs, path)
|
||||
if not suc then
|
||||
option.err[#option.err+1] = res
|
||||
return function () end
|
||||
end
|
||||
return res
|
||||
end
|
||||
|
||||
local function fsRemove(path, option)
|
||||
if not path then
|
||||
return false
|
||||
end
|
||||
if path.type == 'dummy' then
|
||||
return path:remove()
|
||||
end
|
||||
local suc, res = pcall(fs.remove, path)
|
||||
if not suc then
|
||||
option.err[#option.err+1] = res
|
||||
end
|
||||
option.del[#option.del+1] = path:string()
|
||||
end
|
||||
|
||||
local function fsExists(path, option)
|
||||
if not path then
|
||||
return false
|
||||
end
|
||||
if path.type == 'dummy' then
|
||||
return path:exists()
|
||||
end
|
||||
local suc, res = pcall(fs.exists, path)
|
||||
if not suc then
|
||||
option.err[#option.err+1] = res
|
||||
return false
|
||||
end
|
||||
return res
|
||||
end
|
||||
|
||||
local function fsSave(path, text, option)
|
||||
if not path then
|
||||
return false
|
||||
end
|
||||
if path.type == 'dummy' then
|
||||
---@cast path -fs.path
|
||||
local dir = path:_open(-2)
|
||||
if not dir then
|
||||
option.err[#option.err+1] = '无法打开:' .. path:string()
|
||||
return false
|
||||
end
|
||||
local filename = path:_filename()
|
||||
if not filename then
|
||||
option.err[#option.err+1] = '无法打开:' .. path:string()
|
||||
return false
|
||||
end
|
||||
if type(dir[filename]) == 'table' then
|
||||
option.err[#option.err+1] = '无法打开:' .. path:string()
|
||||
return false
|
||||
end
|
||||
dir[filename] = text
|
||||
else
|
||||
local suc, err = m.saveFile(path, text)
|
||||
if suc then
|
||||
return true
|
||||
end
|
||||
option.err[#option.err+1] = err
|
||||
return false
|
||||
end
|
||||
end
|
||||
|
||||
local function fsLoad(path, option)
|
||||
if not path then
|
||||
return nil
|
||||
end
|
||||
if path.type == 'dummy' then
|
||||
local text = path:_open()
|
||||
if type(text) == 'string' then
|
||||
return text
|
||||
else
|
||||
option.err[#option.err+1] = '无法打开:' .. path:string()
|
||||
return nil
|
||||
end
|
||||
else
|
||||
---@cast path -dummyfs
|
||||
local text, err = m.loadFile(path)
|
||||
if text then
|
||||
return text
|
||||
else
|
||||
option.err[#option.err+1] = err
|
||||
return nil
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
local function fsCopy(source, target, option)
|
||||
if not source or not target then
|
||||
return
|
||||
end
|
||||
if source.type == 'dummy' then
|
||||
local sourceText = source:_open()
|
||||
if not sourceText then
|
||||
option.err[#option.err+1] = '无法打开:' .. source:string()
|
||||
return false
|
||||
end
|
||||
return fsSave(target, sourceText, option)
|
||||
else
|
||||
---@cast source -dummyfs
|
||||
if target.type == 'dummy' then
|
||||
local sourceText, err = m.loadFile(source)
|
||||
if not sourceText then
|
||||
option.err[#option.err+1] = err
|
||||
return false
|
||||
end
|
||||
return fsSave(target, sourceText, option)
|
||||
else
|
||||
local suc, res = pcall(fs.copy_file, source, target, fs.copy_options.overwrite_existing)
|
||||
if not suc then
|
||||
option.err[#option.err+1] = res
|
||||
return false
|
||||
end
|
||||
end
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
---@param path dummyfs|fs.path
|
||||
---@param option table
|
||||
local function fsCreateDirectories(path, option)
|
||||
if not path then
|
||||
return
|
||||
end
|
||||
if path.type == 'dummy' then
|
||||
return path:createDirectories()
|
||||
end
|
||||
local suc, res = pcall(fs.create_directories, path)
|
||||
if not suc then
|
||||
option.err[#option.err+1] = res
|
||||
return false
|
||||
end
|
||||
return true
|
||||
end
|
||||
|
||||
local function fileRemove(path, option)
|
||||
if not path then
|
||||
return
|
||||
end
|
||||
if option.onRemove and option.onRemove(path) == false then
|
||||
return
|
||||
end
|
||||
if fsIsDirectory(path) then
|
||||
for child in fsPairs(path, option) do
|
||||
fileRemove(child, option)
|
||||
end
|
||||
end
|
||||
if fsRemove(path, option) then
|
||||
option.del[#option.del+1] = path:string()
|
||||
end
|
||||
end
|
||||
|
||||
---@param source fs.path|dummyfs?
|
||||
---@param target fs.path|dummyfs?
|
||||
---@param option table
|
||||
local function fileCopy(source, target, option)
|
||||
if not source or not target then
|
||||
return
|
||||
end
|
||||
local isDir1 = fsIsDirectory(source)
|
||||
local isDir2 = fsIsDirectory(target)
|
||||
local isExists = fsExists(target, option)
|
||||
if isDir1 then
|
||||
if isDir2 or fsCreateDirectories(target, option) then
|
||||
for filePath in fsPairs(source, option) do
|
||||
local name = filePath:filename():string()
|
||||
fileCopy(filePath, target / name, option)
|
||||
end
|
||||
end
|
||||
else
|
||||
if isExists and not isDir2 then
|
||||
local buf1 = fsLoad(source, option)
|
||||
local buf2 = fsLoad(target, option)
|
||||
if buf1 and buf2 then
|
||||
if buf1 ~= buf2 then
|
||||
if fsCopy(source, target, option) then
|
||||
option.mod[#option.mod+1] = target:string()
|
||||
end
|
||||
end
|
||||
end
|
||||
else
|
||||
if fsCopy(source, target, option) then
|
||||
option.add[#option.add+1] = target:string()
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
---@param source fs.path|dummyfs?
|
||||
---@param target fs.path|dummyfs?
|
||||
---@param option table
|
||||
local function fileSync(source, target, option)
|
||||
if not source or not target then
|
||||
return
|
||||
end
|
||||
local isDir1 = fsIsDirectory(source)
|
||||
local isDir2 = fsIsDirectory(target)
|
||||
local isExists = fsExists(target, option)
|
||||
if isDir1 then
|
||||
if isDir2 then
|
||||
local fileList = m.fileList()
|
||||
if type(target) == 'table' then
|
||||
---@cast target dummyfs
|
||||
for filePath in target:listDirectory() do
|
||||
fileList[filePath] = true
|
||||
end
|
||||
else
|
||||
---@cast target fs.path
|
||||
for filePath in fs.pairs(target) do
|
||||
fileList[filePath] = true
|
||||
end
|
||||
end
|
||||
for filePath in fsPairs(source, option) do
|
||||
local name = filePath:filename():string()
|
||||
local targetPath = target / name
|
||||
fileSync(filePath, targetPath, option)
|
||||
fileList[targetPath] = nil
|
||||
end
|
||||
for path in pairs(fileList) do
|
||||
fileRemove(path, option)
|
||||
end
|
||||
else
|
||||
if isExists then
|
||||
fileRemove(target, option)
|
||||
end
|
||||
if fsCreateDirectories(target, option) then
|
||||
for filePath in fsPairs(source, option) do
|
||||
local name = filePath:filename():string()
|
||||
fileCopy(filePath, target / name, option)
|
||||
end
|
||||
end
|
||||
end
|
||||
else
|
||||
if isDir2 then
|
||||
fileRemove(target, option)
|
||||
end
|
||||
if isExists then
|
||||
local buf1 = fsLoad(source, option)
|
||||
local buf2 = fsLoad(target, option)
|
||||
if buf1 and buf2 then
|
||||
if buf1 ~= buf2 then
|
||||
if fsCopy(source, target, option) then
|
||||
option.mod[#option.mod+1] = target:string()
|
||||
end
|
||||
end
|
||||
end
|
||||
else
|
||||
if fsCopy(source, target, option) then
|
||||
option.add[#option.add+1] = target:string()
|
||||
end
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
--- 文件列表
|
||||
function m.fileList(option)
|
||||
option = option or buildOption(option)
|
||||
local keyMap = {}
|
||||
local fileList = {}
|
||||
local function computeKey(path)
|
||||
path = fsAbsolute(path, option)
|
||||
if not path then
|
||||
return nil
|
||||
end
|
||||
return path:string()
|
||||
end
|
||||
return setmetatable({}, {
|
||||
__index = function (_, path)
|
||||
local key = computeKey(path)
|
||||
return fileList[key]
|
||||
end,
|
||||
__newindex = function (_, path, value)
|
||||
local key = computeKey(path)
|
||||
if not key then
|
||||
return
|
||||
end
|
||||
if value == nil then
|
||||
keyMap[key] = nil
|
||||
else
|
||||
keyMap[key] = path
|
||||
fileList[key] = value
|
||||
end
|
||||
end,
|
||||
__pairs = function ()
|
||||
local key, path
|
||||
return function ()
|
||||
key, path = next(keyMap, key)
|
||||
return path, fileList[key]
|
||||
end
|
||||
end,
|
||||
})
|
||||
end
|
||||
|
||||
--- 删除文件（夹）
|
||||
function m.fileRemove(path, option)
|
||||
option = buildOption(option)
|
||||
path = fsAbsolute(path, option)
|
||||
|
||||
fileRemove(path, option)
|
||||
|
||||
return option
|
||||
end
|
||||
|
||||
--- 复制文件（夹）
|
||||
---@param source string|fs.path|dummyfs
|
||||
---@param target string|fs.path|dummyfs
|
||||
---@return table
|
||||
function m.fileCopy(source, target, option)
|
||||
option = buildOption(option)
|
||||
local fsSource = fsAbsolute(source, option)
|
||||
local fsTarget = fsAbsolute(target, option)
|
||||
|
||||
fileCopy(fsSource, fsTarget, option)
|
||||
|
||||
return option
|
||||
end
|
||||
|
||||
--- 同步文件（夹）
|
||||
---@param source string|fs.path|dummyfs
|
||||
---@param target string|fs.path|dummyfs
|
||||
---@return table
|
||||
function m.fileSync(source, target, option)
|
||||
option = buildOption(option)
|
||||
local fsSource = fsAbsolute(source, option)
|
||||
local fsTarget = fsAbsolute(target, option)
|
||||
|
||||
fileSync(fsSource, fsTarget, option)
|
||||
|
||||
return option
|
||||
end
|
||||
|
||||
---@param dir fs.path
|
||||
---@param callback fun(fullPath: fs.path)
|
||||
function m.scanDirectory(dir, callback)
|
||||
for fullpath in fs.pairs(dir) do
|
||||
local status = fs.symlink_status(fullpath):type()
|
||||
if status == 'directory' then
|
||||
m.scanDirectory(fullpath, callback)
|
||||
elseif status == 'regular' then
|
||||
callback(fullpath)
|
||||
end
|
||||
end
|
||||
end
|
||||
|
||||
function m.listDirectory(dir)
|
||||
if dir.type == 'dummy' then
|
||||
return dir:listDirectory()
|
||||
else
|
||||
return fs.pairs(dir)
|
||||
end
|
||||
end
|
||||
|
||||
return m
|
||||
Reference in New Issue
Block a user