41 lines
972 B
Lua
41 lines
972 B
Lua
|
|
local _M = {
|
|
root_path = nil
|
|
}
|
|
|
|
|
|
local function get_script_dir()
|
|
local lfs_ok, lfs = pcall(require, "lfs")
|
|
|
|
local source
|
|
if debug and debug.getinfo then
|
|
-- OpenResty 或标准 Lua 都适用
|
|
source = debug.getinfo(2, "S").source:sub(2)
|
|
elseif arg and arg[0] then
|
|
source = arg[0]
|
|
else
|
|
return nil
|
|
end
|
|
|
|
-- 如果是相对路径，转换为绝对路径
|
|
if not source:match("^/") then
|
|
if lfs_ok then
|
|
source = lfs.currentdir() .. "/" .. source
|
|
else
|
|
-- 没有 lfs，使用 io.popen("pwd") 获取当前目录
|
|
local handle = io.popen("pwd")
|
|
local cwd = handle:read("*l")
|
|
handle:close()
|
|
source = cwd .. "/" .. source
|
|
end
|
|
end
|
|
|
|
-- 提取目录
|
|
local dir = source:match("(.*/)")
|
|
-- 降末尾的/lua/module/去掉
|
|
return dir:sub(1,-13)
|
|
end
|
|
|
|
_M.root_path = get_script_dir()
|
|
print(_M.root_path)
|
|
return _M |