モジュール:WeaponNavbox

このモジュールについての説明文ページを モジュール:WeaponNavbox/doc に作成できます

local p = {}

local aw = require('Module:Utility/Library')
local iu = require('Module:Utility/Image')
local nu = require('Module:Utility/Name')
local getArgs -- lazily initialized

local function createRow(tbl, text)
	return tbl:tag('tr')
		:tag('th')
			:wikitext(text)
			:done()
		:tag('td')
			:tag('ul')
				:addClass('tpl-navbox-list')
end

local function createRowForType(lang)
	return function(tbl, name, num)
		local type = nu.type(name, lang, num)
		return createRow(tbl, type)
	end
end

local function createRowForAmmo(lang)
	return function(tbl, name)
		local item
		local opts = { size = 20 }
		if name == 'special' then
			item = string.format(
				'%s <span class="text-ammo-special">%s</span>',
				iu.ammo('special_sniper', opts),
				nu.type('supply_drop', lang))
		else
			item = string.format(
				'%s <span class="text-ammo-%s">%s</span>',
				iu.ammo(name, opts),
				name,
				nu.ammo(name, lang))
		end
		
		return createRow(tbl, item)
	end
end

local function createRowForHSMultiplier(tbl, hsmul)
	return createRow(tbl, string.format('%s倍', hsmul))
end

local function renderNavbox(keys, names, fn, stat, fmt)
	fmt = fmt or function(value, name)
		return string.format('[[%s]]', value)
	end
	
	local tbl = mw.html.create('table')
			:addClass('tpl-navbox-table')
			:addClass('tpl-navbox-table-containstitle')
	tbl:tag('tr')
		:tag('th')
			:attr('colspan', 2)
			:addClass('tpl-navbox-table-title')
			:wikitext('武器')
	for _, name in ipairs(keys) do
		local list = fn(tbl, name, #names[name])
		for _, value in ipairs(names[name]) do
			list:tag('li')
				:wikitext(fmt(value, stat))
		end
	end
	return tbl:done()
end

local function formatAmmoText(value, stat)
	local ammoraw = stat[value].ammo
	local ammo
	if aw.stringstarts(ammoraw, 'special_') then
		ammo = 'special'
	else
		ammo = ammoraw
	end
	return string.format('[[%s|<span class="text-ammo text-ammo-%s">%s</span>]]', value, ammo, value)
end

local function renderNavboxByType(args)
	local stat = mw.loadData('Module:Stat/Weapon')
	
	local weaponsByTypes = {}
	for key, value in pairs(stat) do
		if weaponsByTypes[value.category] == nil then
			weaponsByTypes[value.category] = {}
		end
		table.insert(weaponsByTypes[value.category], key)
	end
	return renderNavbox(
		{ 'assault_rifle', 'sub_machine_gun', 'light_machine_gun', 'marksman_weapon', 'sniper', 'shotgun', 'pistol', 'ultimate' },
		weaponsByTypes, createRowForType(args.lang), stat, formatAmmoText)
end

local function renderNavboxByAmmo(args)
	local stat = mw.loadData('Module:Stat/Weapon')
	
	local weaponsByAmmos = {}
	for key, value in pairs(stat) do
		local target
		if aw.stringstarts(value.ammo, "special_") then
			if weaponsByAmmos.special == nil then
				weaponsByAmmos.special = {}
			end
			target = weaponsByAmmos.special
		else
			if weaponsByAmmos[value.ammo] == nil then
				weaponsByAmmos[value.ammo] = {}
			end
			target = weaponsByAmmos[value.ammo]
		end
		table.insert(target, key)
	end
	return renderNavbox(
		{ 'light', 'heavy', 'energy', 'sniper', 'shotgun', 'arrows', 'special' },
		weaponsByAmmos, createRowForAmmo(args.lang))
end

local function renderNavboxByHSMultiplier(args)
	local stat = mw.loadData('Module:Stat/Weapon')
	
	local weaponsByHSMultipliers = {}
	for key, value in pairs(stat) do
		local headshot = value.damage.headshot_charged or value.damage.headshot or 2
		if weaponsByHSMultipliers[headshot] == nil then
			weaponsByHSMultipliers[headshot] = {}
		end
		table.insert(weaponsByHSMultipliers[headshot], key)
	end
	
	local keys = aw.getTableKeys(weaponsByHSMultipliers)
	table.sort(keys)
	return renderNavbox(
		keys,
		weaponsByHSMultipliers, createRowForHSMultiplier)
end

local function preload(frame)
	if not getArgs then
		getArgs = require('Module:Arguments').getArgs
	end
	
	local args = getArgs(frame)
	args.lang = args.lang or 'ja'
	return args
end

function p.bytype(frame)
	local args = preload(frame)
	return tostring(renderNavboxByType(args))
end

function p.byammo(frame)
	local args = preload(frame)
	return tostring(renderNavboxByAmmo(args))
end

function p.byhsmul(frame)
	local args = preload(frame)
	return tostring(renderNavboxByHSMultiplier(args))
end

return p