WorldPills

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

local p = {} --p stands for package

local lang = mw.language.getContentLanguage()


function subworld(world, subworld)
	input = mw.ustring.lower(world .. subworld)
	subworlds = {"husky", "beagle", "akita", "corgi", "labrador", "shepherd"}
	for key,value in ipairs(subworlds) do 
		if (mw.ustring.find(input, value, 1, true) ~= nil) then
			return lang:ucfirst(value)
		end
	end
	return ''--return 'not found'
end


function world(world, subworld)
	input = mw.ustring.lower(world .. subworld)
	worlds = {"survival 1", "survival 2", "survival 3", "survival 4", "survival 5", "creative", "patreon", "hub", "amplified", "MCMMO", "skyblock"}
	for key,value in ipairs(worlds) do 
		if (mw.ustring.find(input, value, 1, true) ~= nil) then
			return lang:ucfirst(value)
		end
	end
	if (mw.ustring.find(input, 'survival', 1, true) ~= nil) then
		return 'Survival 1'
	end
	return ''--return 'not found'
end

function pill (text, color)
	return ('[[' .. text .. '|<span class="world-pill" style="background:' .. color .. 
			'; padding: 3px 7px; color: white; font-weight: bold; border-radius: 7.5px; display: inline-block;">' .. text .. '</span>]]')
end


function p.single(frame)
	color = frame.args[color] or 'lightslategray'
	world = world(frame.args[1], frame.args[2])
	subworld = subworld(frame.args[1], frame.args[2])
	
	if (subworld ~= '') then
		return (pill(world, color) .. ' ' .. pill(subworld, color))
	elseif (world ~= '') then
		return (pill(world, color))
	else 
		return ''
	end
end


function p.multi(frame)
	color = frame.args[color] or 'lightslategray'
	frame.args[2] = frame.args[2] or ''
	input = mw.ustring.lower(frame.args[1] .. frame.args[2])
	worlds = {"survival 1", "survival 2", "survival 3", "survival 4", "survival 5", "creative", "patreon", "hub", "amplified", "MCMMO", "skyblock"}
	out = ''
	
	for key,value in ipairs(worlds) do 
		if (mw.ustring.find(input, value, 1, true) ~= nil) then
			value = lang:ucfirst(value)
			out = out .. pill(value, color) .. ' '
		end
	end
	
	return out
end


return p