• Welcome to the Kancolle Wiki!
  • If you have any questions regarding site content, account registration, etc., please visit the KanColle Wiki Discord

Difference between revisions of "Module:Equipment"

From Kancolle Wiki
Jump to navigation Jump to search
com>Ckwng
(add __call metamethod)
 
(18 intermediate revisions by 3 users not shown)
Line 1: Line 1:
 +
local U = require('Module:Core')
 +
local EquipmentData = require('Module:EquipmentData')
 +
 +
local EquipmentCollection = mw.loadData('Module:Data/Equipment')
 +
 
local Equipment = {}
 
local Equipment = {}
  
 
local equipments = {}
 
local equipments = {}
  
local EquipmentData = require('Module:EquipmentData')
+
local function requireEquipmentModule(name, is_enemy)
 +
  local success, data
 +
  if not is_enemy then
 +
success, data = U.requireModule('Data/Equipment/' .. name)
 +
    if not success then
 +
      success, data = U.requireModule('Data/Item/' .. name)
 +
    end
 +
    if not success then
 +
      success, data = U.requireModule('Data/PseudoItem/' .. name)
 +
    end
 +
    if not success then
 +
      data = U.find(EquipmentCollection, name, '_name')
 +
      -- success = true
 +
    end
 +
  end
 +
  if not success then
 +
    success, data = U.requireModule('Data/EnemyEquipment/' .. name)
 +
  end
 +
  if not success then
 +
  return {_name = name}
 +
  end
 +
  data._is_enemy = is_enemy
 +
  return data
 +
end
 +
 
 +
function Equipment:create(name, is_enemy)
 +
  if not name then
 +
    return EquipmentData()
 +
  end
 +
  if not equipments[name] then
 +
    equipments[name] = EquipmentData(requireEquipmentModule(name, is_enemy))
 +
  end
 +
  return equipments[name]
 +
end
  
 
function Equipment:get(stat, name)
 
function Equipment:get(stat, name)
return self:create(name)[stat]()
+
  return self:create(name)[stat]()
 +
end
 +
 
 +
function Equipment:get_module(name, typeName)
 +
  return name and string.format('Module:Data/%s/%s', typeName or 'Equipment', name) or nil
 
end
 
end
  
function Equipment:create(name)
+
function Equipment:extend(data)
if equipments[name] then
+
  data = data or {}
return equipments[name]
+
  setmetatable(data, data)
else
+
  data.__index = self
local equipment_table = require(mw.ustring.format('Module:%s', name))
+
  data.__call = self.__call
local equipment = EquipmentData:create(equipment_table)
+
  return data
equipments[name] = equipment
 
return equipment
 
end
 
 
end
 
end
  

Latest revision as of 15:13, 30 April 2023

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

local U = require('Module:Core')
local EquipmentData = require('Module:EquipmentData')

local EquipmentCollection = mw.loadData('Module:Data/Equipment')

local Equipment = {}

local equipments = {}

local function requireEquipmentModule(name, is_enemy)
  local success, data
  if not is_enemy then
	success, data = U.requireModule('Data/Equipment/' .. name)
    if not success then
      success, data = U.requireModule('Data/Item/' .. name)
    end
    if not success then
      success, data = U.requireModule('Data/PseudoItem/' .. name)
    end
    if not success then
      data = U.find(EquipmentCollection, name, '_name')
      -- success = true
    end
  end
  if not success then
    success, data = U.requireModule('Data/EnemyEquipment/' .. name)
  end
  if not success then
  	return {_name = name}
  end
  data._is_enemy = is_enemy
  return data
end

function Equipment:create(name, is_enemy)
  if not name then
    return EquipmentData()
  end
  if not equipments[name] then
    equipments[name] = EquipmentData(requireEquipmentModule(name, is_enemy))
  end
  return equipments[name]
end

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

function Equipment:get_module(name, typeName)
  return name and string.format('Module:Data/%s/%s', typeName or 'Equipment', name) or nil
end

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

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

return Equipment