モジュール:WeaponNavbox

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

local p = {}

local aw = require('Module:Utility/Library')
local formatter -- lazily initialized

local function stringStarts(str, start)
	return string.sub(str, 1, string.len(start)) == start
end

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

local TypeDict = {
	assault_rifle = "アサルトライフル",
	sub_machine_gun = "サブマシンガン",
	light_machine_gun = "ライトマシンガン",
	sniper = "スナイパーライフル",
	shotgun = "ショットガン",
	pistol = "ピストル",
}
local function createRowForType(tbl, name)
	local type = TypeDict[name]
	return createRow(tbl, type)
end

local AmmoDict = {
	light = "ライトアモ",
	heavy = "ヘビーアモ",
	energy = "エネルギーアモ",
	sniper = "スナイパーアモ",
	shotgun = "ショットガンアモ",
}
local function createRowForAmmo(tbl, name)
	local item
	if name == 'special' then
		item = formatter:ammo('専用スナイパーアモ', 20) .. ' 物資投下'
	else
		local ammo = AmmoDict[name]
		item = string.format('%s %s', formatter:ammo(ammo, 20), ammo)
	end
	
	return createRow(tbl, item)
end

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

local function renderNavbox(keys, names, fn)
	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)
		for _, value in ipairs(names[name]) do
			list:tag('li')
				:wikitext(string.format('[[%s]]', value))
		end
	end
	return tbl:done()
end

local function renderNavboxByType()
	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', 'sniper', 'shotgun', 'pistol' },
		weaponsByTypes, createRowForType)
end

local function renderNavboxByAmmo()
	local stat = mw.loadData('Module:Stat/Weapon')
	
	local weaponsByAmmos = {}
	for key, value in pairs(stat) do
		local target
		if 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', 'special' },
		weaponsByAmmos, createRowForAmmo)
end

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

function p.bytype(frame)
	formatter = require('Module:Utility/Formatter').new(frame)
	return tostring(renderNavboxByType(args))
end

function p.byammo(frame)
	formatter = require('Module:Utility/Formatter').new(frame)
	return tostring(renderNavboxByAmmo(args))
end

function p.byhsmul(frame)
	formatter = require('Module:Utility/Formatter').new(frame)
	return tostring(renderNavboxByHSMultiplier(args))
end

return p