Module:Mormonverse

local p = {};

--possible book inputs
local book_aliases = {
	['1 Nephi'] = {'1nephi', '1ne', '1nep'},
	['2 Nephi'] = {'2nephi', '2ne', '2nep'},
	['3 Nephi'] = {'3nephi', '3ne', '3nep'},
	['4 Nephi'] = {'4nephi', '4ne', '4nep'},
	['Words of Mormon'] = {'wordsofmormon', 'wofm', 'wom'},
	['Jacob']   = {'jac'},
	['Enos']    = {'en'},
	['Jarom']   = {'jar'},
	['Omni']    = {'om'},
--	['Mosiah']  = {},
	['Alma']    = {'al'},
	['Helaman'] = {'hel', 'helam'},
	['Mormon']  = {'morm'},
	['Ether']   = {'eth'},
	['Moroni']  = {'moro'},
	['Doctrine and Covenants'] = {'doctrineandcovenants', 'd&c', 'dc'},
--	['Moses'] = {},
	['Abraham'] = {'abr'},
	['Joseph Smith–History'] = {'josephsmithhistory', 'jsh'},
	['Joseph Smith–Matthew'] = {'josephsmithmatthew', 'jsm'},
	['Articles of Faith'] = {'articlesoffaith', 'aoff', 'aof'},
}

--these books only have one chapter, have to be handled differently
local no_chapters = {
	['Enos'] = true,
	['Jarom'] = true,
	['Omni'] = true,
	['Words of Mormon'] = true,
	['4 Nephi'] = true,
}

--pattern for the url of each site using _book etc.
-- (underscore then letters [a-z]) for variables
local url_path_tbl = {
	wikisource = 's:Bible (_version)/_book#_schap:_svers',
	lds_org = 'https://www.churchofjesuschrist.org/study/scriptures/_book/_schap?lang=eng&id=_vrange#p_svers',
	coc_net = 'http://www.communityofchrist.net/Scriptures/'
}

--changes book name to use in url for each site, only if necessary
local site_book_tbl = {
	lds_org = {
		['1 Nephi']         = 'bofm/1-ne',
        ['2 Nephi']         = 'bofm/2-ne',
        ['Jacob']           = 'bofm/jacob',
        ['Enos']            = 'bofm/enos',
        ['Jarom']           = 'bofm/jarom',
        ['Omni']            = 'bofm/omni',
        ['Words of Mormon'] = 'bofm/w-of-m',
        ['Mosiah']          = 'bofm/mosiah',
        ['Alma']            = 'bofm/alma',
        ['Helaman']         = 'bofm/hel',
        ['3 Nephi']         = 'bofm/3-ne',
        ['4 Nephi']         = 'bofm/4-ne',
        ['Mormon']          = 'bofm/morm',
        ['Ether']           = 'bofm/ether',
        ['Moroni']          = 'bofm/moro',
        ['Doctrine and Covenants'] = 'dc-testament/dc',
        ['Moses']                = 'pgp/moses',
        ['Abraham']              = 'pgp/abr',
        ['Joseph Smith–Matthew'] = 'pgp/js-m',
        ['Joseph Smith–History'] = 'pgp/js-h',
        ['Articles of Faith']    = 'pgp/a-of-f',
    }
}

local function lookupTableReplace(arg, lookupTable)
	local abbrTitle = string.lower(arg)
	local fullTitle = arg
	for k, v in pairs(lookupTable) do
	    if type(v) == "table" then
	        for i = 1, #v do
	            if v[i] == abbrTitle then
	                fullTitle = k
	                break --this is optional
	            end
	        end
	    end
	end
	return fullTitle
end

local function processBookName(arg)
	arg = arg:gsub('[-–— .]', '') -- removing dashes, spaces, periods
	arg = lookupTableReplace(arg, book_aliases)
	return arg
end

local function prepareURL(book, ref, site)
	local split_ref = mw.text.split(ref, '[-–—]')   --split the ref into the part before and after the dash/hyphen
	local s_ref = mw.text.split(split_ref[1], '%p') --any punctuation can be used to separate chapter from verse
	local e_ref = split_ref[2] or split_ref[1]
	e_ref = mw.text.split(e_ref, '%p')
	for i, v in ipairs(s_ref) do s_ref[i] = v:gsub('%D', '') end  --remove any non-numeric character (such as f)
	for i, v in ipairs(e_ref) do e_ref[i] = v:gsub('%D', '') end

	local e_chap, e_vers, s_chap, s_vers
	local chapter_only = not s_ref[2]
	if no_chapters[book] then
		chapter_only = false
		s_chap = 1
		s_vers = s_ref[2] or s_ref[1] or 1   --verse 3 can be specified as "3" or "1:3"
		e_chap = 1
		e_vers = e_ref[2] or e_ref[1] or 1
	else
		s_chap = s_ref[1] or 1
		s_vers = s_ref[2] or 1
		if e_ref[2] or not s_ref[2] then     --chapter-chapter or chapter(:verse)?-chapter:verse
			e_chap = e_ref[1] or s_chap
		else                                 --chapter:verse-verse
			e_chap = s_chap
		end
		e_vers = e_ref[2] or e_ref[1] or s_vers
	end

	if site == 'lds_org' then
		book = site_book_tbl[site][book] or book
	end

	local v_range
	if e_chap == s_chap and e_vers == s_vers then
		v_range = s_vers
	else
		v_range = s_vers .. '-' .. e_vers
	end

	local url_path = url_path_tbl[site]

	local url = url_path:gsub('_%l+', {  --get the components into the url
					_book      = book,
					_schap     = s_chap,
					_svers     = s_vers,
					_echap     = e_chap,
					_evers     = e_vers,
					_vrange    = v_range,
					_version   = version,
				})
	
	-- cut off verse details
	if (chapter_only == true) then
		local tmp = mw.text.split(url, '&')
		url = tmp[1]
	end
	return url
end

local function trimArg(text)
	if type(text) == 'string' then
		text = text:match('(%S.-)%s*$')  --trimmed text or nil if empty
	end
	return text
end

function p.main(frame)
	local targs = frame:getParent().args
	local args = {}
	for _, param in ipairs({1, 2, 3, 4}) do
		args[param] = trimArg(targs[param])
	end
	
	local book = frame.args[1]
	local ref = frame.args[2]
	local version = frame.args[3] or 'lds_org'
	
	book = processBookName(book)
	local url = prepareURL(book, ref, version)

	return '[' .. url .. ' '
			.. book .. ' ' .. ref:gsub('[-–—]', '–') .. ']'
end

return p