Module:MyrstaSandbox

Documentation for this module may be created at Module:MyrstaSandbox/doc

local p = {}

local number_to_color_map = {
	['1'] = 'green',
	['2'] = 'yellow',
	['3'] = 'red',
	['4'] = 'purple',
}

function p.recolor_numbers(frame)
	local input_string = tostring(frame.args[1] or '')

	local output_parts = {}

	for i = 1, #input_string do
		local char = input_string:sub(i, i)

		if number_to_color_map[char] then
			local color_code = number_to_color_map[char]
			table.insert(output_parts, '<span style="color:' .. color_code .. ';">' .. char .. '</span>')
		else
			table.insert(output_parts, char)
		end
	end
	
	return table.concat(output_parts)
end

function p.main(frame)
	return p.recolor_numbers(frame)
end

return p