Module:MediaWiki message helper

local m_str_utils = require("Module:string utilities")

local dump = mw.dumpObject
local get_current_title = mw.title.getCurrentTitle
local gsplit = m_str_utils.gsplit
local make_title = mw.title.makeTitle
local new_title = mw.title.new
local pattern_escape = m_str_utils.pattern_escape
local php_trim = require("Module:Scribunto").php_trim
local ufind = m_str_utils.find
local ugsub = m_str_utils.gsub
local ulower = m_str_utils.lower
local uupper = m_str_utils.upper

local export = {}

local function get_title(frame)
	local args = frame.args
	local title = args and args.title or nil
	return title == nil and get_current_title() or
		new_title(title) or
		error(("%s is not a valid title"):format(dump(title)))
end

local function print_suggestions(suggestions)
	if #suggestions == 0 then
		return ""
	else
		local prefix = "* Did you mean "
		local suffix
		if #suggestions > 1 then
			prefix = prefix .. " any of these?\n"
			suggestions = suggestions:map(function(link) return "** " .. link end)
			suffix = ""
		else
			suffix = "?"
		end
		return prefix .. suggestions:concat "\n" .. suffix
	end
end

-- For ] on uncreated category pages.
function export.category_suggestions(frame)
	local title = ugsub(get_title(frame).text, "^.", uupper)
	local output = require("Module:array")()
	
	local function make_suggestion(title, suffix)
		output:insert("''']'''" .. (suffix or ""))
	end
	
	local function check_for_page_from_function(func)
		local suggestion = func(title)
		if suggestion then
			local suggestion_title = make_title(14, suggestion)
			if suggestion_title and suggestion_title.exists then
				make_suggestion(suggestion)
				return true
			end
		end
		return false
	end
	
	local function check_for_page_with_suffix(suffix)
		return check_for_page_from_function(function(title)
			return title .. " " .. suffix
		end)
	end
	
	local function check_for_page_with_prefix_removed(prefix)
		return check_for_page_from_function(function(title)
			return title:gsub(pattern_escape(prefix), "")
		end)
	end
	
	check_for_page_with_prefix_removed("List of ")
	check_for_page_with_prefix_removed("list of ")
	
	local has_language_category = check_for_page_with_suffix("language")
	check_for_page_with_suffix("Language")
	check_for_page_with_suffix("languages")
	local has_script_category = check_for_page_with_suffix("script")
	
	local function check_other_names_of_languages(language_name)
		for code, data in pairs(require("Module:languages/data/all")) do
			local function check_name_list(list)
				if list then
					for _, name in ipairs(list) do
						-- The aliases and varieties are recursive,
						-- with subtables that themselves contain names.
						if type(name) == "table" then
							check_name_list(name)
						else
							if name == language_name then
								local object = require("Module:languages").makeObject(code, data)
								make_suggestion(object:getCategoryName())
							end
						end
					end
				end
			end
			check_name_list(data.otherNames)
			check_name_list(data.aliases)
			check_name_list(data.varieties)
		end
	end
	
	-- If title looks like a language category, then check if the language name
	-- in it is a valid canonical name, or one of the otherNames for some
	-- language.
	-- If the title looks like a language code, check for a language or a script
	-- with that code.
	local function check_language_name(language_name, is_language_category, has_language_category)
		local ret = false
		if not has_language_category then
			if require("Module:languages/canonical names") then
				if not is_language_category then
					make_suggestion(language_name .. " language")
				else
					output:insert("* '''" .. language_name .. "''' is a valid Wiktionary language name.")
				end
				ret = true
			end
		end
		
		-- Some otherNames are the canonical name of another language.
		check_other_names_of_languages(language_name)
		
		return ret
	end
	local language_name = title:match "^(.+) language$" or title:match "^(.+ Language)$"
	check_language_name(language_name or title, language_name ~= nil, has_language_category)
	
	-- Most languages (7965/8085 by last count) have uppercase letters at
	-- beginning of the name and after whitespace and punctuation characters,
	-- and lowercase everywhere else. Exceptions include languages
	-- with apostrophes, such as Yup'ik, and languages with tone letters,
	-- such as ǃXóõ.
	local fixed_capitalization = ugsub(ulower(language_name or title), "%f%a", uupper)
	if fixed_capitalization ~= (language_name or title) then
		check_language_name(fixed_capitalization)
	end
	
	if title:find "^+$" then
		local function check_for_valid_code(code, ...)
			for _, module_name in ipairs { ... } do
				local object = require("Module:" .. module_name).getByCode(code)
				if object then
					make_suggestion(object:getCategoryName(), " (code <code>" .. code .. "</code>)")
				end
			end
		end
		local code = title:lower()
		check_for_valid_code(code, "languages", "etymology languages", "scripts", "families")
		check_for_valid_code(code .. "-pro", "languages", "etymology languages")
	end
	
	local function check_script_name(script_name, is_script_category, has_script_category)
		if not has_script_category then
			local object = require("Module:scripts").getByCanonicalName(script_name)
			if object then
				if is_script_category then
					output:insert("* " .. script_name .. " is a valid Wiktionary script name.")
				else
					make_suggestion(object:getCategoryName())
				end
			end
		end
		for code, data in pairs(require("Module:scripts/data")) do
			local function check_other_names_of_script(list)
				if list then
					for _, name in ipairs(list) do
						if type(name) == "table" then
							check_other_names_of_script(name)
						elseif script_name == name then
							local object = require("Module:scripts").makeObject(code, data)
							make_suggestion(object:getCategoryName())
						end
					end
				end
			end
			check_other_names_of_script(data.otherNames)
			check_other_names_of_script(data.varieties)
			check_other_names_of_script(data.aliases)
		end
	end
	
	local script_name = title:match "^(.+) script$"
	check_script_name(script_name or title, script_name ~= nil, has_script_category)
	
	return print_suggestions(output)
end

function export.template_suggestions(frame)
	local title = get_title(frame).text
	local output = require("Module:array")()
	
	local function make_suggestion(title, suffix)
		output:insert("''']'''" .. (suffix or ""))
	end
	
	local function check_for_page_with_prefix(prefix)
		local suggestion = prefix .. title
		local suggestion_title = make_title(10, suggestion)
		if suggestion_title and suggestion_title.exists then
			make_suggestion(suggestion)
			return true
		end
		return false
	end
	
	if title:find(" ", 1, true) then
		local with_hyphen = title:gsub(" ", "-")
		local suggestion_title = make_title(10, with_hyphen)
		if suggestion_title and suggestion_title.exists then
			make_suggestion(with_hyphen)
		end
	end
	
	local prefixes = frame.args.prefixes
	if prefixes then
		for prefix in gsplit(prefixes, ",", true, true) do
			check_for_page_with_prefix(prefix)
		end
	end
	
	local prefix, rest = title:match "^(+) *:(.+)$"
	if prefix then
		prefix = prefix:upper()
	else
		prefix, rest = "", title
	end
	
	if prefix == "" or prefix == "R" or prefix == "RQ" then
		local templates = require("Module:MediaWiki message helper/R: and RQ: templates")
		local rest_pattern = pattern_escape(rest)
			:gsub("%l", function(letter)
				return ""
			end)
		for suggestion in templates:gmatch("%f? ?:*?"
				.. rest_pattern
				.. "?%d?%d?%d?%d??%d?%d?%d?%d?%l?%f") do
			local suggestion_title = make_title(10, suggestion)
			if suggestion_title and suggestion_title.exists then
				make_suggestion(suggestion)
			end
		end
	end
	
	return print_suggestions(output)
end

function export.module_suggestions(frame)
	local title = get_title(frame).text
	local output = require("Module:array")()
	
	local function make_suggestion(title)
		output:insert("''']'''")
	end
	
	local pronunciation_suffixes = require("Module:array"){"IPA", "pr", "pron", "pronunc", "pronunciation"}
	
	if pronunciation_suffixes:some(function(suffix)
		return title:find("%-" .. suffix .. "$")
	end) then
		for _, suggestion in ipairs(pronunciation_suffixes:map(function(suffix)
			return title:gsub("%-%w+$", "-" .. suffix)
		end)) do
			local suggestion_title = make_title(828, suggestion)
			if suggestion_title and suggestion_title.exists then
				make_suggestion(suggestion)
			end
		end
	end
	
	-- Look for the modules actually invoked by the template with the same name
	-- as the module (accounting for "Module:Template:..." cases).
	local template_title = new_title(title, 10)
	if template_title then
		local template_text = template_title:getContent()
		if template_text then
			for template in require("Module:template parser").find_templates(template_text) do
				if template:get_name() == "#INVOKE:" then
					local args = template:get_arguments()
					if args then -- args is the function name, so #INVOKE: will throw an error if not present
						make_suggestion(php_trim(args)) -- args is the module name
					end
				end
			end
		end
	end
	
	return print_suggestions(output)
end

function export.is_data_module_not_documentation(frame)
	local title = get_title(frame)
	if require("Module:pages").get_pagetype(title) == "module" then
		return ufind(title.text, "^" .. (frame.args or "") .. "$")
	end
end

return export