Модуль:Песочница/Pok

Для документации этого модуля может быть создана страница Модуль:Песочница/Pok/doc

local p = {}

local function trim(s)
	if not s then return s end
	return (s:gsub("^%s*(.-)%s*$", "%1"))
end

local function split(s, sep)
	if s == nil then return {} end
	local parts = {}
	sep = sep or "%."
	for part in string.gmatch(s, "([^" .. sep .. "]+)") do
		table.insert(parts, part)
	end
	return parts
end

local function parse_keys_from_template(content)
	if not content then return {} end
	local keys = {}
	local titlePos = string.find(content, "|%s*Title%s*=")
	if not titlePos then
		return keys
	end
	local substr = content:sub(titlePos)
	local endPos = substr:find("}}")
	local region = substr
	if endPos then
		region = substr:sub(1, endPos)
	end
	for key in string.gmatch(region, "|%s*([^=|%}]-)%s*=") do
		local k = trim(key)
		if k ~= "" then
			table.insert(keys, k)
		end
	end
	return keys
end

local function load_module_data(page)
	local baseUser = "IanComradeBot/"
	local moduleName = "Module:" .. baseUser .. page .. "/data"
	local ok, data = pcall(mw.loadData, moduleName)
	if not ok then return nil end
	return data
end

local function load_template_content(path)
	local title = mw.title.new("Template:" .. path)
	if not title then return nil end
	local ok, content = pcall(function() return title:getContent() end)
	if not ok then return nil end
	return content
end

function p.get(frame)
	local args = frame.args or {}
	local id = args[1] or ""
	if id == "" then return "" end

	local componentDefs = load_module_data("component.json")
	local prototypeDefs = load_module_data("prototype.json")
	if not componentDefs or not prototypeDefs then
		return ""
	end

	local foundComponents = {}
	local foundPrototypes = {}

	-- If id directly matches a prototype or component key, include it
	if prototypeDefs[id] ~= nil then
		foundPrototypes[id] = true
	end
	if componentDefs[id] ~= nil then
		foundComponents[id] = true
	end

	-- Support comma-separated list of ids (e.g. "absorbent,action")
	for name in string.gmatch(id, "[^,]+") do
		local n = trim(name)
		if n ~= "" then
			if prototypeDefs[n] ~= nil then foundPrototypes[n] = true end
			if componentDefs[n] ~= nil then foundComponents[n] = true end
		end
	end

	local out = {}

	for compName,_ in pairs(foundComponents) do
		local tplPath = "component/" .. compName
		local content = load_template_content(tplPath)
		if not content then
			table.insert(out, "<h2>" .. mw.text.encode(compName) .. "</h2>")
			table.insert(out, "Template: " .. tplPath)
		else
			local keys = parse_keys_from_template(content)
			for _, key in ipairs(keys) do
				table.insert(out, "<h2>" .. key .. "</h2>")
				table.insert(out, "{{" .. tplPath .. "|title|" .. key .. "}}")
			end
		end
	end

	for protoName,_ in pairs(foundPrototypes) do
		local tplPath = "prototype/" .. protoName
		local content = load_template_content(tplPath)
		if not content then
			table.insert(out, "<h2>" .. mw.text.encode(protoName) .. "</h2>")
			table.insert(out, "Template: " .. tplPath)
		else
			local keys = parse_keys_from_template(content)
			for _, key in ipairs(keys) do
				table.insert(out, "<h2>" .. key .. "</h2>")
				table.insert(out, "{{" .. tplPath .. "|title|" .. key .. "}}")
			end
		end
	end

	return table.concat(out, "\n\n")
end

return p