Module:Tree chart: Difference between revisions

← Older edit

Module:Tree chart (view source)

Revision as of 14:45, 11 November 2025

1,664 bytes added , 11 November

no edit summary

Gurwinder

Bureaucrats, Interface administrators, Administrators, Widget editors

260

edits

@@ Line 1: / Line 1: @@
-require('strict')
+-- Module:Tree chart
+-- Defensive renderer for table-based tree pieces
+-- Avoids hard require('strict') so it won't fail on installs without that library.
+-- try to load 'strict' if available, but don't error if it's missing
+pcall(function() require('strict') end)
 local p = {}
+-- load cell definitions (Module:Tree chart/data)
 local cells = mw.loadData('Module:Tree chart/data')
+-- internal renderer: accepts an array where each entry is either:
+--  * a string key (looks up cells[key]) or
+--  * a table describing a custom cell (text, colspan, rowspan, etc)
 function p._main(cell_args)
-	local ret = mw.html.create()
+    local ret = mw.html.create()
-	local top = ret:tag('tr')
+    local tbl = ret:tag('table')
-						:css{ height = '1px',
+                 :addClass('tree-chart-table')
-								['text-align'] = 'center' }
+                 :css{ borderCollapse = 'collapse', ['vertical-align'] = 'top' }
-	local bottom = ret:tag('tr')
-						:css{ height = '1px',
+    local top = tbl:tag('tr'):css{ height = '1px', ['text-align'] = 'center' }
-								['text-align'] = 'center' }
+    local bottom = tbl:tag('tr'):css{ height = '1px', ['text-align'] = 'center' }
-	for _, v in ipairs(cell_args) do
-		if type(v) == 'string' then
+    for _, v in ipairs(cell_args) do
-			top:wikitext(cells[v].t)
+        if type(v) == 'string' then
-			bottom:wikitext(cells[v].b)
+            local celldef = cells[v]
-		else
+            if celldef then
-			top:tag('td')
+                if celldef.t then
-				:attr{ colspan = v.colspan or cell_args.colspan or 6,
+                    top:wikitext(celldef.t)
-						rowspan = v.rowspan or cell_args.rowspan or 2 }
+                else
-				:css{ padding = '0.2em',
+                    top:tag('td'):wikitext('')
-						border = (v.border or cell_args.border or '2') .. 'px solid black' }
+                end
-				:cssText(v.boxstyle or cell_args.boxstyle)
+                if celldef.b then
-				:wikitext(v.text)
+                    bottom:wikitext(celldef.b)
-		end
+                else
-	end
+                    bottom:tag('td'):wikitext('')
-	return tostring(ret)
+                end
+            else
+                -- unknown symbol => placeholder cells so columns align
+                top:tag('td'):wikitext('')
+                bottom:tag('td'):wikitext('')
+            end
+        else
+            -- custom cell (table) from unnamed params
+            local colspan = v.colspan or cell_args.colspan or 6
+            local rowspan = v.rowspan or cell_args.rowspan or 2
+            local border = (v.border or cell_args.border or '2') .. 'px solid'
+            top:tag('td')
+                :attr{ colspan = tostring(colspan), rowspan = tostring(rowspan) }
+                :css{ padding = '0.2em', border = border }
+                :cssText(v.boxstyle or cell_args.boxstyle or '')
+                :wikitext(v.text or '')
+            if tonumber(rowspan) == nil or tonumber(rowspan) < 2 then
+                bottom:tag('td'):wikitext('')
+            end
+        end
+    end
+    return tostring(tbl)
 end
 function p.main(frame)
-	local args = require('Module:Arguments').getArgs(frame, {wrappers = 'Template:Tree chart', trim = false, removeBlanks = false})
+    local args = require('Module:Arguments').getArgs(frame, {
-	local cell_args = {
+        wrappers = 'Template:Tree chart',
-		colspan = args.colspan,
+        trim = false,
-		rowspan = args.rowspan,
+        removeBlanks = false
-		border = args.border,
+    })
-		boxstyle = args.boxstyle
-	}
+    local cell_args = {
-	for _, val in ipairs(args) do
+        colspan = args.colspan,
-		local trimmedVal = val:match('^%s*(.-)%s*$')
+        rowspan = args.rowspan,
-		if trimmedVal == '' then
+        border = args.border,
-			trimmedVal = '$'
+        boxstyle = args.boxstyle
-		end
+    }
-		if cells[trimmedVal] then
-			table.insert(cell_args, trimmedVal)
+    for _, val in ipairs(args) do
-		else
+        local trimmedVal = val:match('^%s*(.-)%s*$')
-			-- Unnamed params behave weirdly
+        if trimmedVal == '' then
-			-- white space at the front counts for param_{{{1}}}, but not whitespace at the end, so remove it
+            trimmedVal = '$'
-			local rightTrimmedVal = val:gsub('%s+$','')
+        end
-			table.insert(cell_args, {
-				text = args[trimmedVal] or ('{{{'..trimmedVal..'}}}'),
+        if cells[trimmedVal] then
-				colspan = args['colspan_'..rightTrimmedVal],
+            table.insert(cell_args, trimmedVal)
-				rowspan = args['rowspan_'..rightTrimmedVal],
+        else
-				border = args['border_'..rightTrimmedVal],
+            local rightTrimmedVal = val:gsub('%s+$','')
-				boxstyle = args['boxstyle_'..rightTrimmedVal]
+            local custom = {
-			})
+                text = args[trimmedVal] or ('{{{'..trimmedVal..'}}}'),
-		end
+                colspan = args['colspan_'..rightTrimmedVal],
-	end
+                rowspan = args['rowspan_'..rightTrimmedVal],
+                border = args['border_'..rightTrimmedVal],
-	return p._main(cell_args)
+                boxstyle = args['boxstyle_'..rightTrimmedVal]
+            }
+            table.insert(cell_args, custom)
+        end
+    end
+    return p._main(cell_args)
 end
 return p