Модуль:Prototypes/Объект/Торгомат

Для документации этого модуля может быть создана страница Модуль:Prototypes/Объект/Торгомат/doc

-- Загрузка данных
local vendingMachinesData = mw.loadData("Модуль:IanComradeBot/prototypes/vending machines.json/data")
local restockData = mw.loadData("Модуль:IanComradeBot/prototypes/vending machines/restock.json/data")
local inventoriesData = mw.loadData("Модуль:IanComradeBot/prototypes/vending machines/inventories.json/data")

local p = {}

-- Поиск данных по ID
local function findDataById(data, id)
	if not data then return nil end
	for _, item in ipairs(data) do
		if item.id == id then
			return item
		end
	end
	return nil
end

-- Формирование вывода инвентаря
local function formatInventoryOutput(inventoryData)
	local result = ""
	for itemId, count in pairs(inventoryData) do
		result = result .. mw.getCurrentFrame():preprocess("{{#invoke:Предмет|main|" .. itemId .. "|[" .. count .. "]|repository=|wrapper=}}")
	end
	return result ~= "" and result or "Инвентарь пуст."
end

-- Получение инвентаря автомата
local function getInventoryOutput(id, mode)
	local vendingMachine = findDataById(vendingMachinesData, id)
	if not vendingMachine then return "Автомат с таким ID не найден." end

	local inventoryId = vendingMachine.VendingMachine and vendingMachine.VendingMachine.pack
	if not inventoryId then return "Инвентарь не найден." end

	local inventory = findDataById(inventoriesData, inventoryId)
	if not inventory then return "Данные об инвентаре отсутствуют." end

	if mode == "inventory" and inventory.startingInventory then
		return formatInventoryOutput(inventory.startingInventory)
	elseif mode == "contraband" and inventory.contrabandInventory then
		return formatInventoryOutput(inventory.contrabandInventory)
	elseif mode == "emag" and inventory.emaggedInventory then
		return formatInventoryOutput(inventory.emaggedInventory)
	end

	return "Не имеет дополнительного ассортимента."
end

-- Получение информации о пополнителе автомата
local function getRestockOutput(id)
	local vendingMachine = findDataById(vendingMachinesData, id)
	if not vendingMachine then return "" end

	local packId = vendingMachine.VendingMachine and vendingMachine.VendingMachine.pack

	for _, restock in ipairs(restockData) do
	    local restockInfo = restock.VendingMachineRestock
	    if restockInfo and restockInfo.canRestock then
	        for _, restockPack in ipairs(restockInfo.canRestock) do
	            if restockPack == packId then
	                return mw.getCurrentFrame():preprocess("{{Предмет|" .. restock.id .. "|size=64px|link=|repository=|wrapper=}}")
	            end
	        end
	    end
	end

	return "Не имеет пополнителя."
end

-- Основная функция модуля
function p.main(frame)
	local mode = frame.args[1]
	local id = frame.args[2]

	if not id then return "" end

	if mode == "inventory" or mode == "contraband" or mode == "emag" then
		return getInventoryOutput(id, mode)
	elseif mode == "restock" then
		return getRestockOutput(id)
	else
		return ""
	end
end

return p