Module:Arguments/Fallback

-- A simple index fallback implementation for tables. -- Useful for template argument aliasing. -- Hmm, what about __newindex?

local function makeFallback(args, arg_aliases) local oldArgsMeta = getmetatable(args) or {} local newArgsMeta = {} -- Forget about thread-safety. -- States kept to avoid strange loops. local referencedKeys = {} -- dataType:Set/hashtable-impl local attemptDepth = 0   -- useful for "last nil" case -- Start the new metatable as a copy of the old metatable. for k, v in ipairs(oldArgsMeta) do		newArgsMeta[k] = v	end

-- Change the __index metamethod to our implementation. -- See https://www.lua.org/pil/13.4.1.html. newArgsMeta.__index = function (t, k) -- My friend, why are you here again? if referencedKeys[k] then -- You have to be drunk. Go home. return nil end referencedKeys[k] = true -- Try the old metamethod first. -- Thanks to closures, this whole oldArgsMeta object will stay. if oldArgsMeta.__index ~= nil then local val = oldArgsMeta.__index(t, k)			if val ~= nil then referencedKeys = {} return val end end attemptDepth = attemptDepth + 1 -- Now try use the aliases given. for _, v in ipairs(arg_aliases[k] or {}) do -- If a working value is found, use it. -- Note: mw-argument-specific empty str chk. if t[v] ~= nil and t[v] ~= '' then referencedKeys = {} return t[v] end end attemptDepth = attemptDepth - 1 if attemptDepth == 0 then referencedKeys = {} end return nil end setmetatable(args, newArgsMeta) return args -- just in case someone wants a return value. end

return makeFallback

--[[ P.S. Don't blame me for writing these obj-states. Sure, we can use an accumulator and do some "index with acc" ourselves, but that sounds like too much work done just to avoid states.

Here is that custom index function in case anyone is curious about it:

function index_w_acc(t, k, acc) local v = rawget(t, k)	   if v ~= nil return v	   else return getmetatable(t).__index(k, acc) end end

My guess is that it will certainly be slower than the native t[k] one. Perhaps not getting the metatable every time will help a bit. ]]--