Модуль:MenuBar

Документацію для цього модуля можна створити у Модуль:MenuBar/документація

local p = {}
local page_cache = {}

local function get_page_content(page_name)
    if page_cache[page_name] then
        return page_cache[page_name]
    end
    local title = mw.title.new(page_name)
    if not title or not title.exists then
        page_cache[page_name] = nil
        return nil
    end
    local content = title:getContent()
    page_cache[page_name] = content
    return content
end

local function get_section_content(content, section_start)
    local section_end = mw.ustring.find(content, "\n==[^=]", section_start + 1)
    if section_end then
        return mw.ustring.sub(content, section_start, section_end - 1)
    else
        return mw.ustring.sub(content, section_start)
    end
end

local function count_facts(section_content)
    local count = 0
    for line in mw.ustring.gmatch(section_content, "[^\n]+") do
        if mw.ustring.match(line, "^%*[^%*]") then
            count = count + 1
        end
    end
    return count
end

local function count_awards(section_content)
    local count = 0
    for _ in mw.ustring.gmatch(section_content, "%[%[File:[^%]]+%.png") do
        count = count + 1
    end
    return count
end

function p.section_menu(frame)
    local page_name = frame.args.page or mw.title.getCurrentTitle().text
    local content   = get_page_content(page_name)

    if not content then return "" end

    local sections = {}

    -- Перший пункт — завжди "Подія"
    table.insert(sections, { label = "Подія", anchor = "top", badge = nil })

    for line in mw.ustring.gmatch(content, "[^\n]+") do
        if mw.ustring.match(line, "^==[^=]") and mw.ustring.match(line, "[^=]==$") then
            local section_name = mw.ustring.match(line, "^==%s*(.-)%s*==$")

            if section_name and section_name ~= "" then
                local processed = frame:preprocess(section_name)
                processed = mw.text.trim(processed)
                local anchor = mw.ustring.gsub(processed, " ", "_")
                local badge  = nil

                local section_pos = mw.ustring.find(content, line, 1, true)
                if section_pos then
                    local section_content = get_section_content(content, section_pos)

                    if mw.ustring.match(processed, "Цікаві факти") then
                        local n = count_facts(section_content)
                        if n > 0 then badge = tostring(n) end

                    elseif mw.ustring.match(processed, "Нагороди") then
                        local n = count_awards(section_content)
                        if n > 0 then badge = tostring(n) end
                    end
                end

                if processed ~= "" then
                    table.insert(sections, { label = processed, anchor = anchor, badge = badge })
                end
            end
        end
    end

    if #sections == 0 then return "" end

    local parts = {}
    table.insert(parts, '<div class="l-box">')

    for i, s in ipairs(sections) do
        local last = (i == #sections) and " l-box-item-last" or ""
        local badge_html = s.badge
            and (' <span class="mcc-badge">' .. s.badge .. '</span>')
            or  ""
        table.insert(parts,
            '<div class="l-box-item' .. last .. '" data-target="' .. s.anchor .. '">'
            .. '<span>' .. s.label .. '</span>'
            .. badge_html
            .. '</div>'
        )
    end

    table.insert(parts, '</div>')
    return table.concat(parts, '\n')
end

return p