Module:Person height

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

-- This module will implement [[Template:Infobox person/height]]

local p = {}

local function clean(s)
	s = mw.ustring.gsub(s, '([^a])meter', '%1m') -- prevents "parameter" from being changed to "param"
	s = mw.ustring.gsub(s, 'metre', 'm')
	s = mw.ustring.gsub(s, 'centi', 'c')
	s = mw.ustring.gsub(s, 'feet', 'ft')
	s = mw.ustring.gsub(s, 'foot', 'ft')
	s = mw.ustring.gsub(s, 'inches', 'in')
	s = mw.ustring.gsub(s, 'inch', 'in')
	s = mw.ustring.gsub(s, 'ms', 'm')
	s = mw.ustring.gsub(s, 'ins', 'in')
	s = mw.ustring.gsub(s, '%[%[[Ii]n|in%]%]', '[[inch|in]]')
	s = mw.ustring.gsub(s, '%[%[[Cc]entim|cm%]%]', '[[Centimetre|cm]]')
	s = mw.ustring.gsub(s, '%[%[[Mm]%]%]s', '[[Metre|m]]')
	s = mw.ustring.gsub(s, '%[%[[Cc]m%]%]s', '[[Centimetre|cm]]')

	return s
end

function convert(frame, args)
	local s = clean(args[1] or '')
	local prefer_m = (args['prefer'] or '') == 'm'
	local force_m = (args['force'] or '') == 'm'
	local prefer_cm = (args['prefer'] or '') == 'cm'
	local force_cm = (args['force'] or '') == 'cm'
	
	local cleaned = mw.ustring.gsub(s, '&[Nn][Bb][Ss][Pp];', ' ')
	
	local m = mw.ustring.find(cleaned, 'm')
	local c = mw.ustring.find(cleaned, 'cm')
	local f = mw.ustring.find(cleaned, 'ft')
	local i = mw.ustring.find(cleaned, 'in')
	
	if m == nil and f == nil and i == nil then
		return s
	end
	
	if c ~= nil and f == nil and i == nil then
		local n = tonumber(mw.ustring.sub(cleaned, 1, c - 1))
		if n == nil then
			return s
		end
		if force_m then
			return frame:expandTemplate{ title = 'convert', args = {n/100,'m','ftin',0,['abbr']='on'}} .. mw.ustring.sub(cleaned, c+2)
		end
		return frame:expandTemplate{ title = 'convert', args = {n,'cm','ftin',0,['abbr']='on'}} .. mw.ustring.sub(cleaned, c+2)
	end
	
	if m ~= nil and c == nil and f == nil and i == nil then
		local n = tonumber(mw.ustring.sub(cleaned, 1, m - 1))
		if n == nil then
			return s
		end
		if force_cm then
			return frame:expandTemplate{ title = 'convert', args = {n*100,'cm','ftin',0,['abbr']='on'}} .. mw.ustring.sub(cleaned, m+1)
		end
		return frame:expandTemplate{ title = 'convert', args = {n,'m','ftin',0,['abbr']='on'}} .. mw.ustring.sub(cleaned, m+1)
	end
	
	if f ~= nil and i ~=nil and m == nil then
		local n1 = tonumber(mw.ustring.sub(cleaned, 1, f - 1))
		local n2 = tonumber(mw.ustring.sub(cleaned, f+2, i - 1))
		if n1 == nil or n2 == nil then
			return s
		end
		if force_m or prefer_m then
			return frame:expandTemplate{ title = 'convert', args = {n1,'ft',n2,'in', 'm',2,['abbr']='on'}} .. mw.ustring.sub(cleaned, i+2)
		end
		return frame:expandTemplate{ title = 'convert', args = {n1,'ft',n2,'in', 'cm',0,['abbr']='on'}} .. mw.ustring.sub(cleaned, i+2)
	end
	
	if f ~= nil and i == nil and m == nil then
		local n = tonumber(mw.ustring.sub(cleaned, 1, f - 1))
		if n == nil then
			return s
		end
		if force_m or prefer_m then
			return frame:expandTemplate{ title = 'convert', args = {n,'ft','m',2,['abbr']='on'}} .. mw.ustring.sub(cleaned, f+2)
		end
		return frame:expandTemplate{ title = 'convert', args = {n,'ft','cm',0,['abbr']='on'}} .. mw.ustring.sub(cleaned, f+2)
	end
	
	if i ~= nil and f == nil and m == nil then
		local n = tonumber(mw.ustring.sub(cleaned, 1, i - 1))
		if n == nil then
			return s
		end
		if force_m or prefer_m then
			return frame:expandTemplate{ title = 'convert', args = {n,'in','cm',2,['abbr']='on'}} .. mw.ustring.sub(cleaned, i+2)
		end
		return frame:expandTemplate{ title = 'convert', args = {n,'in','cm',0,['abbr']='on'}} .. mw.ustring.sub(cleaned, i+2)
	end
	
	return s
end

function p.main(frame)
	return convert(frame, frame.args[1] and frame.args or frame:getParent().args)
end

return p