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

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

local p = {}

local function trim(s)
    if not s then return s end
    return (s:gsub("^%s*(.-)%s*$", "%1"))
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

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 pos then
            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 lcfirst(s)
    if not s or s == "" then return s end
    return string.lower(s:sub(1, 1)) .. (s:sub(2) or "")
end

local function makeTplCall(tplPath, sw, key, id, extra)
    local tplStr = "{{" .. tplPath .. "|" .. sw .. "|" .. key
    tplStr = tplStr .. "|id=" .. tostring(id)
    if extra and extra ~= "" then tplStr = tplStr .. "|" .. extra end
    tplStr = tplStr .. "}}"
    return tplStr
end

local function makeSourceLink(s)
    local className = s.name .. (s.kind == "component" and "Component" or "Prototype")
    local tplLabel = "Template:" .. s.tplPath
    return "[[" .. tplLabel .. "|" .. className .. "]]"
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, 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 switches = { "card", "title" }
    local switchKeyOrder, switchKeyToTemplates, switchKeySources = {}, {}, {}
    for _, sw in ipairs(switches) do
        switchKeyOrder[sw] = {}; switchKeyToTemplates[sw] = {}; switchKeySources[sw] = {}
    end

    local switchConfigs = {
        card = {
            wrapper = function(key, tplCalls, sources)
                if not tplCalls or #tplCalls == 0 then return "" end
                local calls = table.concat(tplCalls, " ")
                local srcStr = ""
                if sources and #sources > 0 then
                    local srcParts = {}
                    for _, s in ipairs(sources) do table.insert(srcParts, makeSourceLink(s)) end
                    srcStr = " " .. table.concat(srcParts, " ")
                end
                return "{{карточка/Сущность|" .. mw.text.encode(key) .. "|" .. calls .. srcStr .. "}}"
            end
        },
        title = {
            wrapper = function(key, tplCalls, sources)
                local parts = {}
                table.insert(parts, "<h2>" .. mw.text.encode(key) .. "</h2>")
                if tplCalls and #tplCalls > 0 then
                    for i, tpl in ipairs(tplCalls) do
                        local line = tpl
                        local src = sources and sources[i]
                        if src then line = line .. " " .. makeSourceLink(src) end
                        table.insert(parts, "<p>" .. line .. "</p>")
                    end
                end
                return table.concat(parts, "\n")
            end
        }
    }

    local errors = {}
    local function processEntity(kind, name)
        local pathName = lcfirst(name)
        local tplPath = kind .. "/" .. pathName
        local content = load_template_content(tplPath)
        if not content then
            local classType = (kind == "component" and "Component" or "Prototype")
            local className = name .. classType
            local tplLabel = "Template:" .. tplPath
            table.insert(errors,
                "{{сущность/infobox|тип=" .. classType .. "|название=" .. className .. "|ссылка=" .. tplLabel .. "}}")
            return
        end
        local parsed = parse_keys_from_template(content, switches)
        local ok, dp = pcall(require, "Module:GetField")
        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] = {}
                    switchKeySources[sw][key] = {}
                    table.insert(switchKeyOrder[sw], key)
                end
                local extra = ""
                if ok and dp and dp.flattenField then
                    local dataPage = tplPath .. ".json"
                    extra = dp.flattenField({ args = { id, dataPage } })
                end
                local tplStr = makeTplCall(tplPath, sw, key, id, extra)
                table.insert(switchKeyToTemplates[sw][key], tplStr)
                table.insert(switchKeySources[sw][key],
                    { kind = kind, name = name, pathName = pathName, tplPath = tplPath })
            end
        end
    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 _, it in ipairs(items) do processEntity(it.kind, it.name) end

    local out = {}
    for _, e in ipairs(errors) do table.insert(out, e) 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 {}
            local sources = switchKeySources[sw][key] or {}
            if cfg.wrapper then
                local outStr = cfg.wrapper(key, tplCalls, sources)
                if outStr and outStr ~= "" then table.insert(out, outStr) end
            end
        end
    end

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

return p