Модуль:Сущность

Материал из Space Station 14 Вики

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

local p = {}

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

local function parse_keys_from_template(content, switches)
	if not content then return {} end
	local lower = content:lower()
	local result = {}
	for _, sw in ipairs(switches) do
		result[sw] = {}
		local swLower = sw:lower()
		local pos = lower:find("|%s*" .. swLower .. "%s*=")
		if not pos then
			-- no keys for this switch
		else
			local startBrace = content:find("{{", pos)
			local region = nil
			if startBrace then
				local len = #content
				local i = startBrace
				local depth = 0
				while i <= len - 1 do
					local two = content:sub(i, i+1)
					if two == "{{" then
						depth = depth + 1
						i = i + 2
					elseif two == "}}" then
						depth = depth - 1
						i = i + 2
						if depth == 0 then
							region = content:sub(startBrace, i-1)
							break
						end
					else
						i = i + 1
					end
				end
			end
			if not region then
				local substr = content:sub(pos)
				local endPos = substr:find("}}")
				if endPos then
					region = substr:sub(1, endPos)
				else
					region = substr
				end
			end
			for key in string.gmatch(region, "|%s*([^=|%}]-)%s*=") do
				local k = trim(key)
				if k ~= "" then
					table.insert(result[sw], k)
				end
			end
		end
	end
	return result
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 = {}
	local compList = componentDefs[id]
	if type(compList) == "table" then
		for _, v in ipairs(compList) do
			if type(v) == "string" then
				foundComponents[v] = true
			end
		end
	end

	local protoList = prototypeDefs[id]
	if type(protoList) == "table" then
		for _, v in ipairs(protoList) do
			if type(v) == "string" then
				foundPrototypes[v] = true
			end
		end
	end

	for name in string.gmatch(id, "[^,]+") do
		local n = trim(name)
		if n ~= "" then
			if componentDefs[n] ~= nil then
				foundComponents[n] = true
			end
			if prototypeDefs[n] ~= nil then
				foundPrototypes[n] = true
			end
			if componentDefs[n] == nil and prototypeDefs[n] == nil then
				foundComponents[n] = true
			end
		end
	end

	local out = {}
	local errors = {}
	local switches = { "card", "title" }
	local switchKeyOrder = {}
	local switchKeyToTemplates = {}
	for _, sw in ipairs(switches) do
		switchKeyOrder[sw] = {}
		switchKeyToTemplates[sw] = {}
	end
	local switchConfigs = {
		card = {
			wrapper = function(key, tplCalls)
				if not tplCalls or #tplCalls == 0 then return "" end
				return "{{card|" .. mw.text.encode(key) .. "|" .. table.concat(tplCalls, " ") .. "}}"
			end
		},
		title = {
			wrapper = function(key, tplCalls)
				local parts = {}
				table.insert(parts, "<h2>" .. mw.text.encode(key) .. "</h2>")
				if tplCalls and #tplCalls > 0 then
					for _, tpl in ipairs(tplCalls) do
						table.insert(parts, "<p>" .. tpl .. "</p>")
					end
				end
				return table.concat(parts, "\n")
			end
		}
	}

	local function lcfirst(s)
		if not s or s == "" then return s end
		return string.lower(s:sub(1,1)) .. (s:sub(2) or "")
	end

	local items = {}
	for compName,_ in pairs(foundComponents) do
		table.insert(items, { kind = "component", name = compName })
	end
	for protoName,_ in pairs(foundPrototypes) do
		table.insert(items, { kind = "prototype", name = protoName })
	end

	for _, item in ipairs(items) do
		local kind = item.kind
		local name = item.name
		local pathName = lcfirst(name)
		local tplPath = kind .. "/" .. pathName
		local content = load_template_content(tplPath)
		if not content then
			table.insert(errors, "Ошибка: не найден шаблон " .. kind .. "/" .. pathName)
		else
			local parsed = parse_keys_from_template(content, switches)
			for _, sw in ipairs(switches) do
				local keys = parsed[sw] or {}
				for _, key in ipairs(keys) do
					if not switchKeyToTemplates[sw][key] then
						switchKeyToTemplates[sw][key] = {}
						table.insert(switchKeyOrder[sw], key)
					end
					local param = sw
					local extra = ""
					local ok, dp = pcall(require, "Module:GetField")
					if ok and dp and dp.flattenField then
						local dataPage = kind .. "/" .. pathName .. ".json"
						extra = dp.flattenField({ args = { id, dataPage } })
					end
					local tplStr = "{{" .. tplPath .. "|" .. param .. "|" .. key
					tplStr = tplStr .. "|id=" .. tostring(id)
					if extra and extra ~= "" then
						tplStr = tplStr .. "|" .. extra
					end
					tplStr = tplStr .. "}}"
					table.insert(switchKeyToTemplates[sw][key], tplStr)
				end
			end
		end
	end

	for _, e in ipairs(errors) do
		table.insert(out, "<div class=\"error\">" .. mw.text.encode(e) .. "</div>")
	end

	for _, sw in ipairs(switches) do
		local cfg = switchConfigs[sw] or {}
		for _, key in ipairs(switchKeyOrder[sw]) do
			local tplCalls = switchKeyToTemplates[sw][key] or {}
			if cfg.wrapper then
				local outStr = cfg.wrapper(key, tplCalls)
				if outStr and outStr ~= "" then
					table.insert(out, outStr)
				end
			end
		end
	end

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

return p