Toggle menu
Toggle personal menu
Not logged in
Your IP address will be publicly visible if you make any edits.

Module:BookTOC: Difference between revisions

From ProleWiki, the proletarian encyclopedia
CriticalResist (talk | contribs)
No edit summary
CriticalResist (talk | contribs)
No edit summary
 
(6 intermediate revisions by the same user not shown)
Line 5: Line 5:
     local items = {}
     local items = {}
     local levels = {}
     local levels = {}
   
    -- Check if collapsed parameter is set
    local collapsed = args.collapsed == "true"
      
      
     -- Collect all TOC items and their levels
     -- Collect all TOC items and their levels
Line 10: Line 13:
     while args['item' .. i] do
     while args['item' .. i] do
         items[i] = args['item' .. i]
         items[i] = args['item' .. i]
         levels[i] = tonumber(args['level' .. i]) or 2 -- Default to level 2 (==) if not provided
         levels[i] = tonumber(args['level' .. i]) or 2 -- Default to level 2
         i = i + 1
         i = i + 1
     end
     end
      
      
    -- Generate TOC as wikitext nested list
     if #items == 0 then
     if #items == 0 then
         return ''
         return ''
     end
     end
      
      
     -- Find the minimum level to use as base
     -- Find minimum level
     local min_level = math.min(table.unpack(levels))
     local min_level = math.min(unpack(levels))
      
      
     local list = {}
     local list = {}
    local current_level = min_level
    local list_stack = {min_level}  -- Start with the base level
      
      
     for idx = 1, #items do
     for idx = 1, #items do
         local level = levels[idx]
         local normalized_level = levels[idx] - min_level + 1
         local item = items[idx]
         local item = items[idx]
        local normalized_level = level - min_level + 1
          
          
         -- Close deeper levels
         -- Add list line
         while #list_stack > 0 and list_stack[#list_stack] > normalized_level do
         table.insert(list, string.rep('#', normalized_level) .. ' ' .. item)
            table.remove(list_stack)
    end
            table.insert(list, string.rep('#', list_stack[#list_stack]))
   
        end
    -- Add mw-collapsed class if collapsed=true
       
    local collapsible_class = "mw-collapsible mw-made-collapsible"
        -- Open new levels if needed
    if collapsed then
        while normalized_level > list_stack[#list_stack] do
         collapsible_class = collapsible_class .. " mw-collapsed"
            table.insert(list, string.rep('#', normalized_level - list_stack[#list_stack]))
            table.insert(list_stack, normalized_level)
        end
       
        -- Add the item
        table.insert(list, string.rep('#', normalized_level - list_stack[#list_stack] + 1))
        table.insert(list, ' ')
         table.insert(list, item)
        table.insert(list, '\n')
       
        current_level = normalized_level
     end
     end
      
      
     return '<div class="book-toc">\n' .. table.concat(list) .. '</div>'
     return '<div class="book-toc">\n' ..
      '<div class="toc-header">\n' ..
      '<strong>Table of contents</strong>\n' ..
      '</div>\n' ..
      '<div class="' .. collapsible_class .. '">\n' ..
      table.concat(list, '\n') .. '\n' ..
      '</div>\n' ..
      '</div>'
end
end


return p
return p

Latest revision as of 11:13, 18 August 2025

local p = {}

function p.main(frame)
    local args = frame:getParent().args
    local items = {}
    local levels = {}
    
    -- Check if collapsed parameter is set
    local collapsed = args.collapsed == "true"
    
    -- Collect all TOC items and their levels
    local i = 1
    while args['item' .. i] do
        items[i] = args['item' .. i]
        levels[i] = tonumber(args['level' .. i]) or 2 -- Default to level 2
        i = i + 1
    end
    
    if #items == 0 then
        return ''
    end
    
    -- Find minimum level
    local min_level = math.min(unpack(levels))
    
    local list = {}
    
    for idx = 1, #items do
        local normalized_level = levels[idx] - min_level + 1
        local item = items[idx]
        
        -- Add list line
        table.insert(list, string.rep('#', normalized_level) .. ' ' .. item)
    end
    
    -- Add mw-collapsed class if collapsed=true
    local collapsible_class = "mw-collapsible mw-made-collapsible"
    if collapsed then
        collapsible_class = collapsible_class .. " mw-collapsed"
    end
    
    return '<div class="book-toc">\n' ..
       '<div class="toc-header">\n' ..
       '<strong>Table of contents</strong>\n' ..
       '</div>\n' ..
       '<div class="' .. collapsible_class .. '">\n' ..
       table.concat(list, '\n') .. '\n' ..
       '</div>\n' ..
       '</div>'
end

return p