Редагування Модуль:FetchData7

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 next_section = mw.ustring.find(content, "\n==[^=]", section_start + 1)
    if next_section then
        return mw.ustring.sub(content, section_start, next_section - 1)
    else
        return mw.ustring.sub(content, section_start)
    end
end

-- Рахуємо рядки в таблиці
local function count_table_rows(section_content)
    local table_start = mw.ustring.find(section_content, "{|")
    if not table_start then return 0 end
    
    local table_end = mw.ustring.find(section_content, "|}", table_start)
    if not table_end then return 0 end
    
    local table_content = mw.ustring.sub(section_content, table_start, table_end)
    
    local count = 0
    for _ in mw.ustring.gmatch(table_content, "|-") do
        count = count + 1
    end
    
    -- Перший |- це заголовок
    return math.max(0, count - 1)
end

-- Рахуємо bullet points (*)
local function count_bullet_points(section_content)
    local count = 0
    for _ in mw.ustring.gmatch(section_content, "\n%*[^%*]") do
        count = count + 1
    end
    -- Перевіряємо чи секція починається з *
    if mw.ustring.match(section_content, "^%*[^%*]") or mw.ustring.match(section_content, "\n%*[^%*]") then
        -- вже пораховано
    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 = {}
    
    for line in mw.ustring.gmatch(content, "[^\n]+") do
        -- ТІЛЬКИ заголовки рівня 2: == Назва ==
        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 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(section_name, "Нагороди") then
                        local count = count_table_rows(section_content)
                        if count > 0 then
                            processed = processed .. " (" .. count .. ")"
                        end
                    
                    -- Сезони - рахуємо рядки таблиці
                    elseif mw.ustring.match(section_name, "Сезони") then
                        local count = count_table_rows(section_content)
                        if count > 0 then
                            processed = processed .. " (" .. count .. ")"
                        end
                    
                    -- Цікаві факти - рахуємо bullet points
                    elseif mw.ustring.match(section_name, "Цікаві факти") then
                        local count = count_bullet_points(section_content)
                        if count > 0 then
                            processed = processed .. " (" .. count .. ")"
                        end
                    end
                end
                
                -- Anchor (без доданих чисел для стандартних секцій)
                local anchor = frame:preprocess(section_name)
                anchor = mw.text.trim(anchor)
                anchor = mw.ustring.gsub(anchor, " ", "_")
                
                if processed ~= "" then
                    table.insert(sections, {
                        display = processed,
                        anchor = anchor
                    })
                end
            end
        end
    end
    
    if #sections == 0 then
        return ""
    end
    
    -- Будуємо HTML
    local html_parts = {}
    
    table.insert(html_parts, '<div class="l-box" style="position:fixed; top:20px; left:20px; z-index:100; border-radius:16px; background-color:#23232c; width:220px; box-sizing:border-box; overflow:hidden;">')
    
    table.insert(html_parts, '<div style="padding:15px 20px; font-weight:bold; font-size:15px; color:#fff; border-bottom:1px solid rgba(255,255,255,0.1);">Зміст</div>')
    
    for i, section in ipairs(sections) do
        local border = ""
        if i < #sections then
            border = "border-bottom:1px solid rgba(255,255,255,0.05);"
        end
        
        table.insert(html_parts, string.format(
            '<div class="l-box-item" style="padding:12px 20px; %s">[[#%s|%s]]</div>',
            border, section.anchor, section.display
        ))
    end
    
    table.insert(html_parts, '</div>')
    
    return table.concat(html_parts, '\n')
end

return p