Module:ShipCategoriesKai

Documentation for this module may be created at Module:ShipCategoriesKai/doc

local BaseData = require('Module:BaseData')
local Formatting = require('Module:Formatting')
local Ship = require('Module:Ship')

local format = require('Module:Utils').format

local ShipCategoriesKai = BaseData{
	_type_category_template = "Category:${ship_type}s",
	_stock_equipment_category_template = "Category:Ships with ${equipment_name} as stock equipment",
}

function ShipCategoriesKai:Categories(args)
	return self{
		_args = args
	}:get_categories()
end

function ShipCategoriesKai:create_ship()
	local name = self._args.name or self._args[1]
	if name then
		self._ship = Ship(name, self._args.model)
	end
end

function ShipCategoriesKai:create_static_categories()
	table.insert(self._categories, "Category:Player ships")
end

function ShipCategoriesKai:create_type_categories()
	local ship_type = self._ship:type()
	if ship_type then
		table.insert(self._categories, format{self._type_category_template, ship_type = Formatting:format_ship_type(ship_type)})
	end
end

function ShipCategoriesKai:create_class_categories()
	local ship_class = self._ship:class()
	if ship_class then
		table.insert(self._categories, format{"Category:${ship_class}", ship_class = ship_class:name()})
	end
end

function ShipCategoriesKai:create_rarity_categories()
	local rarity = self._ship:rarity()
	if rarity then
		table.insert(self._categories, format{"Category:${rarity} rarity ships", rarity = Formatting:format_ship_rarity(rarity)})
	end
end

function ShipCategoriesKai:create_construction_categories()
	local buildable, buildable_lsc = self._ship:buildable(), self._ship:buildable_lsc()
	if buildable then
		table.insert(self._categories, format{"Category:Ships buildable in Normal Construction"})
	end
	if buildable_lsc then
		table.insert(self._categories, format{"Category:Ships buildable in Large Ship Construction"})
	end
end

function ShipCategoriesKai:create_remodel_categories()
	local remodel_blueprint = self._ship:remodel_blueprint()
	if remodel_blueprint then
		table.insert(self._categories, format{"Category:Ships that require a blueprint for remodelling"})
	end
end

function ShipCategoriesKai:create_stock_equipment_categories()
	local slots = self._ship:slots_length()
	if slots then
		for i = 1, slots do
			local equipment, size = self._ship:slot(i)
			if equipment then
				table.insert(self._categories, format{self._stock_equipment_category_template, equipment_name = equipment:link()})
			end
		end
	end
end

function ShipCategoriesKai:write_categories()
	local links = {}
	for _, category in ipairs(self._categories) do
		table.insert(links, Formatting:format_link(category))
	end
	return table.concat(links)
end

function ShipCategoriesKai:get_categories()
	self._categories = {}
	self:create_ship()
	if self._ship then
		self:create_categories()
	end
	return self:write_categories()
end

function ShipCategoriesKai:create_categories()
	self:create_static_categories()
	self:create_type_categories()
	self:create_class_categories()
	--self:create_rarity_categories()
	self:create_construction_categories()
	self:create_remodel_categories()
	self:create_stock_equipment_categories()
end

return ShipCategoriesKai