模組:Utilities/templates

local insert = table.insert

local export = {}

-- Used by {{categorize}}
function export.template_categorize(frame)
	local args = frame:getParent().args
	local format = frame.args["format"]
	
	args = require("Module:parameters").process(args, {
		[1] = {required = true, type = "language", default = "und"},
		[2] = {required = true, list = true, allow_holes = true},
		["sort"] = {list = true, separate_no_index = true, allow_holes = true},
	})
	
	local lang = args[1]
	if not lang then
		return ""
	end
	local raw_cats = args[2]
	local sort_keys = args.sort
	local default_sort = sort_keys.default
	local cats = {}
	local prefix = format == "pos" and lang:getFullName() or
		format == "topic" and lang:getFullName() .. " " or ""
	
	local cats_with_sort_keys = {}
	for i = 1, raw_cats.maxindex do
		local cat = raw_cats[i]
		if cat then
			if format == "pos" then
				-- 如果cat里有'的'就把prefix插到最后一个'的'字的后面，否则把prefix插到cat的前面
				rindex_de = mw.ustring.find(cat, '的[^的]*$')
				if rindex_de and not no_prefix_fix then
					cat = mw.ustring.sub(cat, 1, rindex_de) .. prefix .. mw.ustring.sub(cat, rindex_de + 1)
				else
					cat = prefix .. cat
				end
			elseif format == "topic" then
				cat = prefix .. cat
			end
			if individual_sort_key then
				individual_sort_keys[#categories] = individual_sort_key
				has_individual_sort_keys = true
			end
			
			insert(cats, cat)
			local sort_key = sort_keys[i]
			if #cats_with_sort_keys > 0 then
				insert(cats_with_sort_keys, {
					category = cat,
					sort_key = sort_key
				})
			elseif sort_key then
				for j = 1, #cats - 1 do
					insert(cats_with_sort_keys, {category = cats[j]})
				end
				insert(cats_with_sort_keys, {
					category = cat,
					sort_key = sort_key
				})
			end
		end
	end
	
	if #cats_with_sort_keys > 0 then
		return require("Module:utilities/format_categories_with_sort_keys")(cats_with_sort_keys, lang, default_sort)
	else
		return require("Module:utilities").format_categories(cats, lang, default_sort)
	end
end

return export