Module:TreeList

La documentation pour ce module peut être créée à Module:TreeList/doc

local p = {}

local function split(text, sep)
	if not sep then sep = ">" end
	local result = {}
	for token in mw.text.gsplit(text, sep, true) do
		table.insert(result, mw.text.trim(token))
	end
	return result
end

local function insertPath(tree, path)
	local node = tree
	for _, part in ipairs(path) do
		node.children = node.children or {}
		node.children[part] = node.children[part] or {}
		node = node.children[part]
	end
end

local function renderNode(node, prefix, isLast, level, colorByLevel)
	local output = ""
	local keys = {}

	for k in pairs(node.children or {}) do table.insert(keys, k) end
	table.sort(keys)

	for i, name in ipairs(keys) do
		local child = node.children[name]
		local isLastChild = (i == #keys)

		local line = prefix .. (isLast and "    " or "│   ")
		line = line .. (isLastChild and "└── " or "├── ")

		local color = colorByLevel[level] or ""
		local text = name:match("^%[%[.+%]%]$") and name or "[[" .. name .. "]]"
		if color ~= "" then
			line = line .. string.format('<span style="color:%s;">%s</span>\n', color, text)
		else
			line = line .. text .. "\n"
		end

		output = output .. line
		output = output .. renderNode(child, prefix .. (isLast and "    " or "│   "), isLastChild, level + 1, colorByLevel)
	end

	return output
end

function p.render(frame)
	local args = frame:getParent().args
	local paths = {}
	local root = {}

	for _, v in ipairs(args) do
		if v and v ~= "" then
			table.insert(paths, split(v))
		end
	end

	for _, path in ipairs(paths) do
		insertPath(root, path)
	end

	local colorByLevel = {
		[1] = "#993333",
		[2] = "#209a57",
		[3] = "#2080a5",
		[4] = "#5555cc",
		[5] = "#cc3366"
	}

	-- ✅ Rendu propre : pas de fond, pas de cadre, mais alignement préservé
	local result = '<div style="font-family:Menlo; font-size:0.95em; line-height:1.6em; background:none; border:none; padding:0; margin:0; color:#222; white-space:pre;">\n'
	result = result .. renderNode(root, "", true, 1, colorByLevel)
	result = result .. '</div>'

	return result
end

return p