Documentation for this module may be created at မဝ်ဂျူ:reconstruction/doc

local export = {}

-- from [[WT:POS]]
local POS_headers = require "Module:table".listToSet {
	"နာမဝိသေသန", "ကြိယာဝိသေသန", "Ambiposition", "Article", "Circumfix",
	"Circumposition", "Classifier", "Combining form", "Conjunction",
	"Contraction", "Counter", "Determiner", "Diacritical mark", "Han character",
	"Hanja", "Hanzi", "Ideophone", "Infix", "Interfix", "Interjection", "Kanji",
	"Letter", "Ligature", "နာမ်", "Number", "Numeral", "Participle", "Particle",
	"Phrase", "Postposition", "Prefix", "Preposition", "Prepositional phrase",
	"သဗ္ဗနာမ်", "နာမ်မကိတ်ညဳ", "Proverb", "Punctuation mark", "Romanization",
	"တံရိုဟ်", "အဆက်လက္ကရဴ", "Syllable", "Symbol", "ကြိယာ", 
}

-- This isn't a perfect pattern, but should work in entries that don't have
-- bad syntax.
local title_pattern = "%f[^\n%z]==+%s*(.-)%s*==+"
local function count_POS_headers(title)
	local POS_count = 0
	local content = title:getContent()
	if content then
		for header in content:gmatch(title_pattern) do
			if POS_headers[header] then
				POS_count = POS_count + 1
			end
		end
	end
	
	return POS_count
end

local function has_header(title, header_to_find)
	local content = title:getContent()
	if content then
		for header in content:gmatch(title_pattern) do
			if header == header_to_find then
				return true
			end
		end
	end
	
	return false
end

-- Track Proto-Indo-European entries with more than one part-of-speech header.
-- Invoked by {{reconstruction}}, requested by Victar.
function export.main(frame)
	local title = mw.title.getCurrentTitle()
	local cats = {}
	
	local language = title.text:match "^[^/]+"
	local langcode = require("Module:languages").getByCanonicalName(language)
	if not langcode then
		-- Can happen e.g. if used on a user page
		return
	end
	if language == "Proto-Indo-European" and count_POS_headers(title) > 1 then
 -- table.insert(cats, "Proto-Indo-European entries with more than one part of speech")
	end
	
	local has_references_header = has_header(title, "References")
	local has_further_reading_header = has_header(title, "Further reading")
	if not has_references_header then
  -- table.insert(cats, language .. " entries without References header")
	end
	
	local temp_rfref = ""
	if not (has_references_header or has_further_reading_header) then
  -- table.insert(cats, language .. " entries without References or Further reading header")
		if langcode and langcode._code=="sla-pro" then
			local book_search = "[https://google.com/search?tbm=bks&q="..mw.uri.encode("\""..mw.title.getCurrentTitle().subpageText.."\"","QUERY").." Google Books]"
			temp_rfref = frame:expandTemplate{ title = "Template:rfref", args = { langcode._code, "Check [[:Category:"..language.." reference templates]] or "..book_search } }
		end
	end
	
	return temp_rfref .. require("Module:utilities").format_categories(cats, langcode)
end

return export