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

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

local p = {}
local getArgs = require('Module:Arguments').getArgs

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

local function strip_trailing_digits(s)
    return s:gsub('%d+$', '')
end

local function split_first_underscore(k)
    local pos = k:find('_', 1, true)
    if not pos or pos == 1 or pos == #k then
        return nil, nil
    end
    return k:sub(1, pos - 1), k:sub(pos + 1)
end

local function each_compound_arg(args, fn)
    for k, v in pairs(args) do
        if type(k) == 'string' and k:find('_', 1, true) then
            local base, rem = split_first_underscore(k)
            if base and rem then
                fn(base, rem, v)
            end
        end
    end
end

local function collect_labels_from_args(args)
    local meta = {}
    local seen = {}

    each_compound_arg(args, function(base, rem)
        local label = trim(strip_trailing_digits(rem))
        if label ~= '' then
            meta[base] = meta[base] or {}
            if not seen[base] then seen[base] = {} end
            if not seen[base][label] then
                table.insert(meta[base], label)
                seen[base][label] = true
            end
        end
    end)

    for k, v in pairs(args) do
        if type(k) == 'string' and not k:find('_', 1, true) then
            meta[k] = tostring(v)
        end
    end

    return meta
end

local function render_from_args(args)
    local field = args[1]
    local label = args[2]
    field = field and trim(field) or ''
    label = label and trim(label) or ''

    -- map base -> label -> value
    local map = {}
    each_compound_arg(args, function(base, rem, v)
        local lab = trim(strip_trailing_digits(rem))
        map[base] = map[base] or {}
        local cur = map[base][lab]
        if cur then
            map[base][lab] = cur .. '\n' .. tostring(v)
        else
            map[base][lab] = tostring(v)
        end
    end)

    if field ~= '' and label ~= '' then
        local base = map[field]
        if base then return base[label] or '' end
    end

    if field ~= '' and args[field] ~= nil then
        return trim(tostring(args[field]))
    end

    return ''
end

function p.main(frame)
    local args = getArgs(frame, { removeBlanks = false })
    local field = args[1]

    if field == "json" then
        local json = collect_labels_from_args(args)
        return mw.text.jsonEncode(json)
    end

    return render_from_args(args)
end

return p