lua中获取当前脚本所在的目录debug.getinfo()

主要利用到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

source 指代函数定义的位置，如果函数是在文件中定义的，source就是这个文件路径前加"@"

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/