Module:Ship

Revision as of 04:29, 14 November 2017 by がか (talk | contribs)

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

local U = require("Module:Utils")
local BaseData = require('Module:BaseData')

local Ship = BaseData{
	_ships = {},
	_data_class = false,
}

function Ship:_prepareShipData()
	self._data_class = self._data_class or require('Module:ShipData')
end

function Ship:get(stat, name, model)
	return self:create(name, model)[stat]()
end

function Ship:create(name, model)
	self:_prepareShipData()
	if name == nil then
		return self._data_class()
	end
	if model == nil then
		name, model = self:process_ship_key(name)
	end
	if not model then
		model = ""
	end
	--check if we already have it
	if self._ships[name] and self._ships[name][model] then
		return self._ships[name][model]
	else
		local ship = self._data_class(self:get_table(name, model))
		if not self._ships[name] then
			self._ships[name] = {}
		end
		self._ships[name][model] = ship
		return ship
	end
end

function Ship:process_ship_key_as_reference(reference, base)
	name, model = self:process_ship_key(reference)
	if #name == 0 then
		name = base
	end
	return name, model
end

function Ship:create_from_reference(reference, base)
	name, model = self:process_ship_key(reference)
	if #name == 0 then
		name = base:base_name()
	end
	return self(name, model)
end

function Ship:get_table(name, model)
	local success, ship_table = U.requireModule(name)
	local ship_form_table
	if success then
		ship_form_table = ship_table[model]
		if not ship_form_table and ship_table.seasonals then
            ship_form_table = U.ifindBy(ship_table.seasonals, function(data) return data._suffix == model end)
        end
		if type(ship_form_table) == "string" then
			ship_form_table = self:get_table(self:process_ship_key_as_reference(ship_form_table, name))
	    end
    end
	if not ship_form_table then
		ship_form_table = { _name = name, _suffix = model }
	end
	if ship_form_table._name == nil then
	    ship_form_table._name = name
    end
	if ship_form_table._suffix == nil then
	    ship_form_table._suffix = model
    end
	return ship_form_table
end

function Ship:process_ship_key(ship_key)
	local split = mw.ustring.find(ship_key, '/')
	local ship_base_name, ship_suffix
	if split == nil then
		ship_base_name = ship_key
	else
		ship_base_name = split - 1 > 0 and mw.ustring.sub(ship_key, 1, split - 1) or ''
		ship_suffix = mw.ustring.sub(ship_key, split + 1, -1)
	end
	return ship_base_name, ship_suffix
end

function Ship:get_module(name)
	if name == nil then
		return nil
	else
		name = self:process_ship_key(name)
	end
	return mw.ustring.format('Module:%s', name)
end

function Ship:extend(data)
	data = data or {}
	setmetatable(data, data)
	data.__index = self
	data.__call = self.__call
	return data
end

Ship.__call = Ship.create
setmetatable(Ship, Ship)

return Ship