Module:Cite book

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

local p = {}

local error = require('Module:error')

function p.main(frame)
	local args = (frame.args.book ~= nil and frame.args[1] ~= nil and frame.args) or frame:getParent().args
	local result = ''
	if not args.book then
		return error.error({
			message = '<code>book</code> parameter is required'
		})
	else 
		result = formatBookTitle(args.book)
	end
	if args.chapter then
		result = result .. ', ' .. formatChapter(args.chapter)
	end
	if args.location then
		result = result .. ', ' .. args.location
	end
	if args.page then
		result = result .. ', page ' .. args.page 
	elseif args.pages then 
		result = result .. ', pages ' .. args.pages
	elseif args.paragraph then
		if tonumber(args.paragraph, 10) then
			result = result .. ', paragraph ' .. args.paragraph 
		else
			return error.error({
				message = '<code>paragraph</code> parameter must be an integer'
			})
		end
	end
	return result
end

function formatChapter(chapter)
	if tonumber(chapter, 10) then
		return 'chapter ' .. chapter
	else
		return chapter
	end
end

function formatBookTitle(title)
	local page = mw.title.new(title, 0)
	if page and page.exists then
		return '\'\'[[' .. title .. ']]\'\''
	else
		return '\'\'' .. title .. '\'\''
	end
end

return p