如何改写Lua枚举函数实现enum eItemType{...}式的全局变量创建？

你可以通过Lua的debug库获取调用代码行的内容，解析出目标全局变量名，再将处理后的枚举表赋值给该全局变量。以下是具体实现：

function enum(...)
    -- 获取调用栈信息定位调用位置
    local callInfo = debug.getinfo(2, "lS")
    -- 读取调用enum的代码行
    local callLine = debug.getline(callInfo.source, callInfo.currentline)
    -- 从代码行中提取变量名（匹配enum后的合法Lua标识符）
    local varName = callLine:match("enum%s+([%a_][%w_]*)")
    
    -- 处理传入的枚举表：兼容参数传递的不同写法
    local tbl = ...
    if type(tbl) ~= "table" then
        tbl = select(2, ...)
    end
    
    -- 原有的枚举逻辑：给表添加反向映射
    local length = #tbl
    for i = 1, length do
        local v = tbl[i]
        tbl[v] = i
    end
    
    -- 将枚举表赋值为全局变量
    _G[varName] = tbl
end

-- 按你期望的方式调用
enum eItemType
{
    "wpn",
    "outf",
    "helm",
    "art",
    "boost",
    "bkpk",
    "dev",
    "ammo",
    "none"
}

-- 测试验证
print(eItemType.wpn) -- 输出1
print(eItemType[5]) -- 输出"boost"

关键逻辑说明 #

• debug库的作用：debug.getinfo(2, "lS")获取调用enum的上层栈信息，2代表跳过当前enum函数栈，直接取调用者的栈数据；"lS"指定获取行号和源码路径。debug.getline则读取对应行的代码文本。
• 变量名解析：通过正则表达式匹配enum关键字后的合法标识符，确保符合Lua变量命名规则（字母/下划线开头，后跟字母、数字或下划线）。
• 全局变量赋值：Lua的全局变量都存储在_G表中，通过_G[varName] = tbl可以直接创建或覆盖全局变量。

注意事项 #

• 该实现依赖调用代码的格式规范：enum关键字后必须紧跟目标变量名，变量名与后续的枚举表可以换行（Lua允许函数调用参数换行）。
• 如果调用行包含多余注释或特殊格式，正则匹配可能失效，需根据实际场景调整匹配规则。

内容的提问来源于stack exchange，提问作者liner