モジュール:WeaponNavbox

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

local p = {}

local table = require('Module:TableExtensions')
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')
			:addClass('tpl-navbox-list')
			:tag('ul')
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 getCreateRowForNumbers(fmt)
	return function(tbl, num)
		return createRow(tbl, fmt(num))
	end
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 target = names[name]
		if target then
			local list = fn(tbl, name, #target)
			for _, value in ipairs(target) do
				list:tag('li')
					:wikitext(fmt(value, stat))
			end
		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 renderNavboxByNumbers(args, selector, formatter)
	local stat = mw.loadData('Module:Stat/Weapon')
	
	local weaponsByNumbers = {}
	for key, value in pairs(stat) do
		local num = selector(value)
		if weaponsByNumbers[num] == nil then
			weaponsByNumbers[num] = {}
		end
		table.insert(weaponsByNumbers[num], key)
	end
	
	local keys = table.keys(weaponsByNumbers)
	table.sort(keys)
	return renderNavbox(
		keys,
		weaponsByNumbers, getCreateRowForNumbers(formatter), stat, formatAmmoText)
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

local function formatMultiplier(num)
	return string.format('%s倍', num)
end

function p.byhsmul(frame)
	local args = preload(frame)
	local selector = function(stat)
		return stat.damage.headshot_charged or stat.damage_head_scale or 2
	end
	return tostring(renderNavboxByNumbers(args, selector, formatMultiplier))
end

function p.bylsmul(frame)
	local args = preload(frame)
	local selector = function(stat)
		return stat.damage.legshot_charged or stat.damage_legs_scale or 0.8
	end
	return tostring(renderNavboxByNumbers(args, selector, formatMultiplier))
end

return p