模組:ProduceEncouragement

local p = {}

function p._get_level(level, count)
	return count
		and math.floor(count / 5)
		or level
end

local category_suffix = '级维基创作奖'

function p._get_category_name(level)
	if level < 10 then
		return tostring(level) .. category_suffix
	end
	-- level = f_level * (10 ^ n_level), 1 <= f_level < 10 （未對f_level執行無條件捨去前）
	local n_level = math.floor(math.log(level) / math.log(10))
	local f_level = math.floor(level / math.pow(10, n_level))
	return tostring(f_level) .. ('0'):rep(n_level) .. '—' .. tostring(f_level) .. ('9'):rep(n_level) .. category_suffix
end

function p._get_category_index(level)
	-- 2024 -> 024
	return tostring(level):sub(2)
end

local function _arg_to_level(args)
	return p._get_level(tonumber(args['1']) or 1, tonumber(args['count']))
end

-- //// ---

function p.get_level(frame)
	return tostring(_arg_to_level(frame.args))
end

function p.get_category(frame)
	local level = _arg_to_level(frame.args)
	return '[[Category:' .. p._get_category_name(level) .. '|' .. p._get_category_index(level) .. ']]'
end

return p