主要利用到lua中的debug库中的getinfo方法。
在使用该方法之前,先了解一下lua中栈层的概念。一个栈层是一个数字,它表示某一时刻某个活动的函数,即一个正在调用还未返回的函数。调用调试库的函数是层1,调用这个函数的函数是层2。
debug.getinfo(function[,what]) 第一个参数是一个函数或是栈层,返回值是一个table,关于该函数或该栈层上函数的相关信息,来看一个简单的示例
local function sum(a, b)
return a + b
end
local info = debug.getinfo(sum)
for k,v in pairs(info) do
print(k, ':', info[k])
end
执行后获得如下信息
[root@localhost lua]# luajit /root/lua/info.lua
linedefined : 1
currentline : -1
func : function: 0x4086f228
isvararg : false
namewhat :
lastlinedefined : 3
source : @/root/lua/info.lua
nups : 0
what : Lua
nparams : 2
short_src : /root/lua/info.lua
short_src 是source的短版本,不超过60字符。
从上可知要获取当前脚本所在目录,只要获取当前脚本的source即可,如果是在当前脚本中调用的debug.getinfo, 那么当前脚本的栈层即为1。以下是完整代码
local info = debug.getinfo(1, "S") -- 第二个参数 "S" 表示仅返回 source,short_src等字段, 其他还可以 "n", "f", "I", "L"等 返回不同的字段信息
for k,v in pairs(info) do
print(k, ":", v)
end
local path = info.source
path = string.sub(path, 2, -1) -- 去掉开头的"@"
path = string.match(path, "^.*/") -- 捕获最后一个 "/" 之前的部分 就是我们最终要的目录部分
print("dir=", path)
执行结果
[root@localhost lua]# luajit /root/lua/info.lua
linedefined : 0
lastlinedefined : 16
source : @/root/lua/info.lua
what : main
short_src : /root/lua/info.lua
dir= /root/lua/