모듈:GameText

This module is intended to provide functionality to the {{gameText}} template.

---

local trim = mw.text.trim

local args_table -- cache
-- helper function
local getArg = function(key)
	local value = trim(args_table[key] or '')
	return (value ~= '') and value or nil
end

local db = {}
local loadDatabase = function(lang)
	if not db[lang] then
		if lang == 'de' or lang == 'es' or lang == 'fr' or lang == 'it' or lang == 'pl' or lang == 'pt' or lang == 'ru' or lang == 'uk' or lang == 'zh' then
			---luacache.get() is relative slow, therefore we load data from luacache only once per page.
			db[lang] = mw.loadData("Module:GameText/loaddata-"..lang)
		else
			db[lang] = mw.loadData("Module:GameText/loaddata-en")
		end
	end
	return db[lang]
end


-- key: e.g.  "BuffName.ManaRegeneration".
-- return string or table, nil when key or data is invalid.
local get -- for recursion
get = function(key, lang)
	local data = db[lang] or loadDatabase(lang or 'en')
	local key = trim(key)
	local pos = 1
	for st,sp in function() return string.find(key, '.', pos, true) end do
		index = string.sub(key, pos, st-1)
		data = data[tonumber(index) or index] -- convert to number for pure number index.
		if not data then
			return nil
		end
		pos = sp + 1
	end
	if pos ~= 0 then 
		key = string.sub(key, pos)
	end
	local result = data[tonumber(key) or key]
	if result then
		if type(result) == 'table' then
			arr = {}
			for _, v in pairs(result) do
				arr[#arr+1] = v
			end
			result = table.concat(arr, '₪') -- we must use a separator that is guaranteed not to be contained in any string. We can easily replace it in wikitext with any that we like
		else
			-- for ref marks such as {$CommonItemTooltip.RightClickToOpen}
			result = string.gsub(result, "({%$(.-)})", function(_, ref) return get(ref, lang) end)
		end
	end
	return result
end

-- args: replacement table, e.g.: { ["{0}"] = "Party Girl", ["<right>"] = "Right click" }
local getText = function(key, lang, args)
	if not key then
		return
	end
	local str = get(key, lang) or get(key, 'en')
	if not str then
		return
	end
	str = string.gsub(str, "{%?.-}", "") -- strip condition marks, e.g.: "{?Homeless}"" -> ""
	if args then
		str = string.gsub(str, "%b{}", args)
		str = string.gsub(str, "%b<>", args)
	end
	return str
end

-- bulid replacement table from template arguments.
-- e.g.: |x_0=Party Girl|y_right=Right click -> { ["{0}"] = "Party Girl", ["<right>"] = "Right click" }
local replacementArgs = function(argsTable)
	local args
	for k, v in pairs(argsTable) do
		string.gsub(k, '^x_(.+)', function(n)
			if not args then args = {} end
			args['{'..n..'}'] = v
		end)
		string.gsub(k, '^y_(.+)', function(n)
			if not args then args = {} end
			args['<'..n..'>'] = v
		end)
	end
	return args
end


--------------------------------------------------------------------------------
return {
	---------- for template ----------------
	get = function(frame)
		args_table = frame:getParent().args -- cache
		local str = getText(getArg(1), getArg('lang') or frame.args['lang'], replacementArgs(args_table))
		if str then
			return '<span class="gameText">' .. string.gsub(str, '\n', '<br/>') .. '</span>'
		end
	end,

	getRaw = function(frame)
		return (frame.args['prefix'] or '')
		     ..(getText(frame.args[1], frame.args['lang'] or frame:expandTemplate{title = 'lang'}, replacementArgs(frame.args)) or '')
		     ..(frame.args['postfix'] or '')
	end,

	listAll = function(frame)
		local lang = frame.args['lang'] or 'en'
		loadDatabase(lang)
		local arr = {}
		if frame.args[1] then
			if not db[lang][frame.args[1]] then
				return
			end
			for k, v in pairs(db[lang][frame.args[1]]) do
				arr[#arr + 1] = k .. '₪' .. v
			end

		else
			for k, v in pairs(db[lang]) do
				arr[#arr + 1] = k
			end
		end
		return table.concat(arr, '¦')
	end,

	listKeys = function(frame)
		local lang = frame.args['lang'] or 'en'
		loadDatabase(lang)
		local arr = {}
		for k, v in pairs(frame.args[1] and (db[lang][frame.args[1]] or {}) or db[lang]) do
			arr[#arr + 1] = k
		end
		return table.concat(arr, '¦')
	end,

	---------- for module ----------------
	getText = getText,
	replacementArgs = replacementArgs, -- used by Module:Random_world_names
}