Module:BookTOC: Difference between revisions

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

@@ Line 5: / Line 5: @@
      local items = {}
      local levels = {}
+    -- Check if collapsed parameter is set
+    local collapsed = args.collapsed == "true"
      -- Collect all TOC items and their levels
@@ Line 10: / Line 13: @@
      while args['item' .. i] do
          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
      end
-    -- Return empty if no items
      if #items == 0 then
          return ''
      end
+    -- Find minimum level
+    local min_level = math.min(unpack(levels))
      local list = {}
-    local list_stack = {}
      for idx = 1, #items do
-         local level = levels[idx]
+         local normalized_level = levels[idx] - min_level + 1
          local item = items[idx]
-        -- Adjust for MediaWiki heading levels (== is level 2, but we want it as top-level)
+         -- Add list line
-        local depth = level - 1
+         table.insert(list, string.rep('#', normalized_level) .. ' ' .. item)
+    end
-        -- Close deeper levels
-        while #list_stack > 0 and list_stack[#list_stack] > depth do
+    -- Add mw-collapsed class if collapsed=true
-            table.remove(list_stack)
+    local collapsible_class = "mw-collapsible mw-made-collapsible"
-        end
+    if collapsed then
+        collapsible_class = collapsible_class .. " mw-collapsed"
-        -- Open new levels
-        while #list_stack < depth do
-            table.insert(list_stack, #list_stack + 1)
-        end
-         -- Add the item with the correct number of # markers
-         table.insert(list, string.rep('#', depth) .. ' ' .. item .. '\n')
      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
 return p