Module:eo-pron

local export = {}
local lang = require("Module:languages").getByCode("eo")

local consonants = {
	 = "b",
	 = "t͡s",
	 = "t͡ʃ",
	 = "d",
	 = "f",
	 = "ɡ",
	 = "d͡ʒ",
	 = "h",
	 = "x",
	 = "j",
	 = "ʒ",
	 = "k",
	 = "l",
	 = "m",
	 = "n",
	 = "p",
	 = "r",
	 = "s",
	 = "ʃ",
	 = "t",
	 = "v",
	 = "z",
	 = "u̯"}

local vowels = {
	 = "a",
	 = "e",
	 = "i",
	 = "o",
	 = "u",

}

local letters_phonemes = {}

-- combine into single table
for k, v in pairs(vowels) do letters_phonemes = v end
for k, v in pairs(consonants) do letters_phonemes = v end

local function has_vowel(term)
	return mw.ustring.lower(term):find("") ~= nil
end

local function letters_to_syllables(letters)
	if not letters then
		return { = letters}
	end
	local l_r_exceptions = { = true,  = true,  = true,  = true}

	local result = { = ""}
	local j = 1
	for i = 1, #letters - 2 do
		result = result .. letters
		local letter = mw.ustring.lower(letters)
		local letter1 = mw.ustring.lower(letters)
		local letter2 = mw.ustring.lower(letters)
		
		if vowels then
			if consonants and vowels then
				-- single consonant goes with following vowel
				if has_vowel(result) and (letter1 ~= 'ŭ') then
					j = j + 1
					result = ""
				end
				
			elseif consonants and not l_r_exceptions and (letter2 == 'l' or letter2 == 'r') and (letter1 ~= 'l' and letter1 ~= 'r') then
				-- consonant followed by l or r goes with l or r
				if has_vowel(result) then
					j = j + 1
					result = ""
				end

			elseif vowels then
				-- two vowels
				if has_vowel(result) then
					j = j + 1
					result = ""
				end
			end
		elseif consonants then
			if consonants and vowels then
				if (mw.ustring.len(result) ~= 1) then
					-- single consonant goes with following vowel
					if has_vowel(result) then
						j = j + 1
						result = ""
					end
				end
			elseif consonants and not l_r_exceptions and (letter2 == 'l' or letter2 == 'r') and (letter1 ~= 'l' and letter1 ~= 'r') then
				-- consonant followed by l or r goes with l or r
				if has_vowel(result) then
					j = j + 1
					result = ""
				end

			elseif vowels then
				-- two vowels
				if has_vowel(result) then
					j = j + 1
					result = ""
				end
			end
		end
	end
	
	-- add last two letters
	if letters then
		local c1 = letters
		local c2 = letters
			
		if c1 ~= 'ŭ' then
            if vowels and vowels then
                result = result .. c1
                j = j + 1
                result = c2
            elseif has_vowel(result) and has_vowel(c1 .. c2) then
            	j = j + 1
            	result = c1 .. c2
        	else
        		result = result .. c1 .. c2
    		end
        	
        else
            if vowels] and vowels then
            	result = result .. c1
                j = j + 1
                result = c2
            elseif has_vowel(result) and has_vowel(c1 .. c2) then
            	j = j + 1
            	result = c1 .. c2
        	else
        		result = result .. c1 .. c2
    		end
        end
	end
	
	local result2 = {}
	for i, j in pairs(result) do
		if j and j ~= "" then
			table.insert(result2, j)
		end
	end
	return result2
end

local function string_to_letters(term)
	return mw.text.split(term, "")
end

local function string_to_syllables(term)
	-- split if given artificial syllable breaks
	local split_input = mw.text.split(term, '‧', true)
	local result = {}
	for _, split in pairs(split_input) do
		for j, syllable in pairs(letters_to_syllables(string_to_letters(split))) do
			table.insert(result, syllable)
		end
	end
	
	return result
end

local function string_to_ipa(syllable)
	local syllable_letters = string_to_letters(syllable)
	local syllable_ipa = ""
	
	for k, letter in pairs(syllable_letters) do
		if letters_phonemes then
			syllable_ipa = syllable_ipa .. letters_phonemes
		end

	end
	return syllable_ipa
end

local function strip_initial_consonants(term)
	local letters = string_to_letters(term)
	local result = {}
	
	local gate = false
	for i, j in pairs(letters) do
		if vowels then
			gate = true
		end
		
		if gate then
			table.insert(result, j)
		end
	end
	
	return table.concat(result)
end

local function term_to_words(term)
	-- split by spaces, hyphens, or periods
	return mw.text.split(term, '')
end

local function term_to_IPA_and_rhyme(term)
	local words = term_to_words(term)
	local result = {}
	local rhyme_letters

	for i, word in pairs(words) do
		if word ~= "" then
			-- add /o/ if word is a single character and a consonant
			if mw.ustring.len(word) == 1 then
				if consonants then
					word = word .. 'o'
				end
			end
			
			-- break into syllables and make each into IPA
			local hyphenated = string_to_syllables(word)
			local word_result = {}
			for j, syllable in pairs(hyphenated) do
				local syllable_ipa = string_to_ipa(syllable)
				word_result = syllable_ipa
			end

			-- add stress to penultimate syllable, and set rhyme to last two syllables
			if word_result then
				rhyme_letters = strip_initial_consonants(hyphenated .. hyphenated)
				
				word_result = "ˈ" .. word_result
				
			end
			
			result = table.concat(word_result)
		end
	end

	-- rhyme to ipa
	local rhyme = nil
	if rhyme_letters then
		rhyme = string_to_ipa(rhyme_letters)
	end
	
	return { = result,  = rhyme}
end

local function parse_input(input)
	-- no input -> use page title
	return input or mw.title.getCurrentTitle().text
end

function export.IPA(IPA_input)
	IPA_input = mw.ustring.lower(parse_input(IPA_input))
	return  ""
end

function export.rhyme(rhyme_input)
	rhyme_input = parse_input(rhyme_input)
	return term_to_IPA_and_rhyme(rhyme_input).rhyme
end

function export.hyphenation(hyphenation_input)
	hyphenation_input = parse_input(hyphenation_input)
	
	local words = term_to_words(hyphenation_input)
	local result = {}
	local hyphenated
	
	for i, word in pairs(words) do
		hyphenated = string_to_syllables(word)
		table.insert(result, table.concat(hyphenated, "‧"))
	end
	
	return table.concat(result, ' ')
end

function export.letters(letters_input)
	letters_input = parse_input(letters_input)
	return table.concat(string_to_letters(letters_input), '-')
end

local function get_artificial_breaks(frame)
	-- override for syllable breaks
	local args = frame:getParent().args
	if not args then
		return nil
	end
	
	local param = 1
	local result = {}
	while true do
		if not args then
			if not result then
				return nil
			end
			
			return table.concat(result, "‧")

		end
		table.insert(result, args)
		param = param + 1
	end
	
end

function export.pronunciation_section(frame)
	local args = frame:getParent().args
	local artificial_breaks = get_artificial_breaks(frame)
	
	local IPA_override = args
	if not IPA_override then
		IPA_override = export.IPA(artificial_breaks)
	end

	local hyphenation_override = args
	if not hyphenation_override then
		hyphenation_override = export.hyphenation(artificial_breaks)
	end
	
	local rhyme_override = args
	if not rhyme_override then
		rhyme_override = export.rhyme(artificial_breaks)
	end
		
	local audio = args

	if rhyme_override then
		rhyme_override = require("Module:rhymes").format_rhymes { lang = lang, rhymes = {{ rhyme = rhyme_override }} }
	end
	IPA_override = require("Module:IPA").format_IPA_full { lang = lang, items = {{pron = IPA_override }} }

	local result = "<ul>"
	result = result .. "<li>" .. IPA_override .. "</li>"
	
	if audio then
		audio = require("Module:audio").format_audio { lang = lang, file = audio }
		result = result .. "<li>" .. audio .. "</li>"
	end
	
	if rhyme_override then
		result = result .. "<li>" .. rhyme_override .. "</li>"
	end
	
	result = result .. "<li>Hyphenation: " .. hyphenation_override .. "</li>"
	result = result .. "</ul>"
	
	return result
end

return export