286 lines
7.3 KiB
Lua
286 lines
7.3 KiB
Lua
local lfs = require("lfs")
|
||
|
||
local M = {}
|
||
|
||
|
||
-- 创建目录
|
||
M.create_dir = function(dirpath)
|
||
-- 检查操作系统类型
|
||
local os_type = package.config:sub(1,1)
|
||
if os_type == "\\" then
|
||
-- Windows 系统
|
||
os.execute("mkdir \"" .. dirpath:gsub("/", "\\") .. "\" /p")
|
||
else
|
||
-- Unix/Linux/Mac 系统
|
||
os.execute("mkdir -p \"" .. dirpath .. "\"")
|
||
end
|
||
end
|
||
-- 转换为整数
|
||
M.tointeger = function(data)
|
||
local num = tonumber(data)
|
||
if num == nil then
|
||
return nil
|
||
end
|
||
return math.floor(num)
|
||
end
|
||
-- 转为时间戳
|
||
M.to_timestamp = function(time_str,pattern)
|
||
if pattern == nil then
|
||
pattern = "(%d+)%-(%d+)%-(%d+)%s+(%d+):(%d+):(%d+)"
|
||
end
|
||
local y, m, d, h, min, s = time_str:match(pattern)
|
||
local timestamp = os.time({
|
||
year = tonumber(y),
|
||
month = tonumber(m),
|
||
day = tonumber(d),
|
||
hour = tonumber(h),
|
||
min = tonumber(min),
|
||
sec = tonumber(s)
|
||
})
|
||
return timestamp
|
||
end
|
||
-- 取扩展名
|
||
M.ext = function(filepath)
|
||
-- 取扩展名
|
||
return string.match(filepath,"%.([^.]+)$")
|
||
end
|
||
-- 复制文件
|
||
-- 增加第三个参数 replace，是否替换目标文件，默认为 false
|
||
M.copy_file = function(src, dst, replace)
|
||
replace = replace or false
|
||
-- 检查目标文件是否存在
|
||
local dst_file = io.open(dst, "r")
|
||
if dst_file ~= nil then
|
||
dst_file:close()
|
||
if not replace then
|
||
return true
|
||
end
|
||
end
|
||
-- 打开源文件
|
||
local file = io.open(src, "rb")
|
||
if not file then
|
||
print("ERR 2,src:",src)
|
||
return false
|
||
end
|
||
local content = file:read("*all")
|
||
file:close()
|
||
-- 写入目标文件
|
||
local file = io.open(dst, "wb")
|
||
if not file then
|
||
print("ERR 3")
|
||
return false
|
||
end
|
||
file:write(content)
|
||
file:close()
|
||
return true
|
||
end
|
||
-- 取路径文件名(带扩展名)
|
||
M.filename = function(filepath)
|
||
return string.match(filepath,"[^/]+$")
|
||
end
|
||
-- 删除文件
|
||
M.delete_file = function(filepath)
|
||
if os.remove(fw.website_dir()..filepath) ~= true then
|
||
return false
|
||
end
|
||
return true
|
||
end
|
||
-- 读取文件内容
|
||
M.read_file = function(filepath)
|
||
local file = io.open(filepath, "rb")
|
||
if not file then
|
||
return nil, "无法打开文件: " .. tostring(filepath)
|
||
end
|
||
local content = file:read("*all")
|
||
file:close()
|
||
return content
|
||
end
|
||
|
||
-- 保存内容到文件
|
||
M.save_file = function(filepath, content)
|
||
local file = io.open(filepath, "wb")
|
||
if not file then
|
||
return false, "无法打开文件: " .. tostring(filepath)
|
||
end
|
||
file:write(content)
|
||
file:close()
|
||
return true
|
||
end
|
||
-- 是否存在文件
|
||
M.exists_file = function(filepath)
|
||
local file = io.open(filepath, "rb")
|
||
if file then
|
||
file:close()
|
||
return true
|
||
else
|
||
return false
|
||
end
|
||
end
|
||
-- 取近N个月时间
|
||
M.recent_months = function(n)
|
||
local mons = {}
|
||
local function format_time(y, m, d, h, i, s)
|
||
return string.format("%04d-%02d-%02d %02d:%02d:%02d", y, m, d, h, i, s)
|
||
end
|
||
local now = os.time()
|
||
local cur = os.date("*t", now)
|
||
for i = n-1, 0, -1 do
|
||
local year = cur.year
|
||
local month = cur.month - i
|
||
while month <= 0 do
|
||
month = month + 12
|
||
year = year - 1
|
||
end
|
||
-- 获取该月第一天与最后一天
|
||
local first_day = format_time(year, month, 1, 0, 0, 0)
|
||
local next_month = month + 1
|
||
local next_year = year
|
||
if next_month > 12 then
|
||
next_month = 1
|
||
next_year = year + 1
|
||
end
|
||
-- next_month 1号的前一天就是当前月最后一天
|
||
local last_day_ts = os.time{year=next_year, month=next_month, day=1, hour=0, min=0, sec=0} - 1
|
||
local last_day_tm = os.date("*t", last_day_ts)
|
||
local last_day = format_time(last_day_tm.year, last_day_tm.month, last_day_tm.day, 23, 59, 59)
|
||
local month_str = string.format("%04d-%02d", year, month)
|
||
table.insert(mons, month_str)
|
||
end
|
||
return mons
|
||
end
|
||
M.recent_months2 = function(n)
|
||
local mons = {}
|
||
local function format_time(y, m, d, h, i, s)
|
||
return string.format("%04d-%02d-%02d %02d:%02d:%02d", y, m, d, h, i, s)
|
||
end
|
||
local now = os.time()
|
||
local cur = os.date("*t", now)
|
||
for i = n-1, 0, -1 do
|
||
local year = cur.year
|
||
local month = cur.month - i
|
||
while month <= 0 do
|
||
month = month + 12
|
||
year = year - 1
|
||
end
|
||
-- 获取该月第一天与最后一天
|
||
local first_day = format_time(year, month, 1, 0, 0, 0)
|
||
local next_month = month + 1
|
||
local next_year = year
|
||
if next_month > 12 then
|
||
next_month = 1
|
||
next_year = year + 1
|
||
end
|
||
-- next_month 1号的前一天就是当前月最后一天
|
||
local last_day_ts = os.time{year=next_year, month=next_month, day=1, hour=0, min=0, sec=0} - 1
|
||
local last_day_tm = os.date("*t", last_day_ts)
|
||
local last_day = format_time(last_day_tm.year, last_day_tm.month, last_day_tm.day, 23, 59, 59)
|
||
local month_str = string.format("%04d-%02d", year, month)
|
||
table.insert(mons, {
|
||
month = month_str,
|
||
start_time = first_day,
|
||
end_time = last_day
|
||
})
|
||
end
|
||
return mons
|
||
end
|
||
-- 十六进制转字节数组
|
||
M.hex_to_bytes = function(hex_str)
|
||
-- 移除所有非十六进制字符
|
||
hex_str = hex_str:gsub("[^%x]", ""):upper()
|
||
|
||
-- 补0使长度为偶数
|
||
if #hex_str % 2 == 1 then
|
||
hex_str = "0" .. hex_str
|
||
end
|
||
|
||
-- 使用 gsub 一次性转换
|
||
return (hex_str:gsub("(%x%x)", function(hex)
|
||
return string.char(tonumber(hex, 16))
|
||
end))
|
||
end
|
||
--[[
|
||
还原 Redis 字符串（去转义）
|
||
@param value 被转义的字符串
|
||
@return 返回还原后的字符串
|
||
]]
|
||
M.unescape_value = function(value)
|
||
if value == nil then
|
||
return nil
|
||
end
|
||
local str = tostring(value)
|
||
-- 如果是用双引号包住的，去掉包裹并恢复转义
|
||
if #str >= 2 and string.sub(str,1,1) == "\"" and string.sub(str,-1,-1) == "\"" then
|
||
str = string.sub(str,2,-2)
|
||
str = string.gsub(str, "\\\"", "\"")
|
||
str = string.gsub(str, "\\\\", "\\")
|
||
return str
|
||
end
|
||
return str
|
||
end
|
||
-- 是否为静态资源扩展名
|
||
M.is_static_ext_not_html = function(ext)
|
||
local static_exts = {
|
||
"jpg",
|
||
"jpeg",
|
||
"png",
|
||
"gif",
|
||
"bmp",
|
||
"ico",
|
||
"pdf",
|
||
"doc",
|
||
"docx",
|
||
"xls",
|
||
"xlsx",
|
||
"ppt",
|
||
"pptx",
|
||
"txt",
|
||
"css",
|
||
"js",
|
||
"json",
|
||
"xml",
|
||
"yaml",
|
||
"yml",
|
||
"zip",
|
||
"rar",
|
||
"7z",
|
||
"tar",
|
||
"gz",
|
||
"bz2",
|
||
"xz",
|
||
"mp3",
|
||
"wav",
|
||
"ogg",
|
||
"aac",
|
||
"m4a",
|
||
"mp4",
|
||
"avi",
|
||
"mov",
|
||
"wmv",
|
||
"flv",
|
||
"webm",
|
||
"mkv",
|
||
"avi",
|
||
"mov",
|
||
"wmv",
|
||
"flv",
|
||
"webm",
|
||
"mkv",
|
||
}
|
||
for _, v in ipairs(static_exts) do
|
||
if v == ext then
|
||
return true
|
||
end
|
||
end
|
||
return false
|
||
end
|
||
-- 遍历目录
|
||
M.traverse_dir = function(dirpath)
|
||
local files = {}
|
||
for file in lfs.dir(dirpath) do
|
||
if file ~= "." and file ~= ".." then
|
||
table.insert(files, file)
|
||
end
|
||
end
|
||
return files
|
||
end
|
||
return M |