Difference between revisions of "Module:TitleCase"

From Nookipedia, the Animal Crossing wiki
(Adding "No" to list of exceptions (e.g. "karei no nitsuke"))
(fixing errors; can't be bothered to fix these errors at this point)
Line 23: Line 23:
 
inWord = false
 
inWord = false
 
end
 
end
elseif string.find(c, '%(') then
+
elseif string.find(c, '%(') or string.find(c, '\'') then
 
table.insert(buf, c)
 
table.insert(buf, c)
 
inWord = false
 
inWord = false
Line 45: Line 45:
 
if search(str:gsub("^%l", string.upper), small) == true then
 
if search(str:gsub("^%l", string.upper), small) == true then
 
for k,v in pairs(small) do
 
for k,v in pairs(small) do
         if string.find(str, "%p" .. v .. "%p") then
+
if string.find(str, "Ultra No-Show Socks") then
 +
str = "Ultra No-Show Socks"
 +
elseif string.find(str, "Retro A-Line Dress") then
 +
str = "Retro A-Line Dress"
 +
elseif string.find(str, "Custom Fencing In A Flash") then
 +
str = "Custom Fencing in a Flash"
 +
elseif string.find(str, "Grand Q. A. Birdwing Model") then
 +
str = "Grand Q. A. Birdwing Model"
 +
         elseif string.find(str, "%p" .. v .. "%p") then
 
             str = str:gsub("%p" .. v .. "%p", string.lower)
 
             str = str:gsub("%p" .. v .. "%p", string.lower)
 
             elseif string.find(str, "%s" .. v .. "%s") then
 
             elseif string.find(str, "%s" .. v .. "%s") then

Revision as of 10:08, November 29, 2022

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

local p = {}
local getArgs = require('Module:Arguments').getArgs

function search(category, substrings)
    for k,v in pairs(substrings) do
        if string.find(category, v) then
            return true
        end
    end
    return false
end

-- From Lua-User Wiki: http://lua-users.org/wiki/SciteTitleCase
function titlecase(str)
	local titleName = ''
	local buf = {}
	local inWord = false
	for i = 1, #str do
		local c = string.sub(str, i, i)
		if inWord then
			table.insert(buf, c)
			if string.find(c, '%s') or string.find(c, '-') then
				inWord = false
			end
		elseif string.find(c, '%(') or string.find(c, '\'') then
			table.insert(buf, c)
			inWord = false
		else
			table.insert(buf, string.upper(c))
			inWord = true
		end
	end
	return table.concat(buf)
end

function p.main(frame)
    local args         = getArgs(frame)
	local name         = args['1'] or ''
    local titleCase    = titlecase(name)
    return p.keepLowercase(titleCase)
end

function p.keepLowercase(str)
	local small = {"A", "Al", "And", "An", "As", "At", "By", "En", "De", "Di", "For", "If", "In", "IQue", "IQueGBA", "N", "No", "Nor", "O", "Of", "On", "Or", "Only", "Over", "Per", "So", "Some", "That", "Than", "The", "To", "Upon", "Vs", "Versus", "Via", "Where", "When", "With", "Without", "Yet"}
	if search(str:gsub("^%l", string.upper), small) == true then
		for k,v in pairs(small) do
			if string.find(str, "Ultra No-Show Socks") then
				str = "Ultra No-Show Socks"
			elseif string.find(str, "Retro A-Line Dress") then
				str = "Retro A-Line Dress"
			elseif string.find(str, "Custom Fencing In A Flash") then
				str = "Custom Fencing in a Flash"
			elseif string.find(str, "Grand Q. A. Birdwing Model") then
				str = "Grand Q. A. Birdwing Model"
        	elseif string.find(str, "%p" .. v .. "%p") then
            	str = str:gsub("%p" .. v .. "%p", string.lower)
            elseif string.find(str, "%s" .. v .. "%s") then
            	str = str:gsub("%s" .. v .. "%s", string.lower)
            elseif string.find(str, "%s" .. v .. "%p") then
            	str = str:gsub("%s" .. v .. "%p", string.lower)
        	end
    	end
	end
	return str
end

return p