Модуль:Prototypes/Роль/Экипировка

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

-- Загрузка данных
local jobData           = mw.loadData("Модуль:IanComradeBot/job.json/data")
local gearData          = mw.loadData("Модуль:IanComradeBot/startingGear.json/data")
local gearloadout       = mw.loadData("Модуль:IanComradeBot/loadout.json/data")
local gearloadoutGroup  = mw.loadData("Модуль:IanComradeBot/loadoutGroup.json/data")

local p = {}

function p.main(frame)
    local args = frame.args
    local slot  = args[1] or ""
    local jobId = args[2] or ""
    
    if slot == "" or jobId == "" then
        return "Ошибка: Не заданы необходимые параметры. Используйте: {{#invoke:Прототипы/Роль/Экипировка|main|<slot>|<jobId>}}"
    end

    -- Поиск должности по id
    local job = nil
    for _, v in pairs(jobData) do
        if v.id == jobId then
            job = v
            break
        end
    end
    if not job then
        return "Ошибка: не найдена должность с id " .. jobId
    end

    -- Получение id набора экипировки из должности
    local gearId = job.startingGear
    if not gearId then
        return "Ошибка: у должности с id " .. jobId .. " не указан startingGear"
    end

    -- Поиск набора экипировки в gearData по id
    local gear = nil
    for _, v in pairs(gearData) do
        if v.id == gearId then
            gear = v
            break
        end
    end
    if not gear then
        return "Ошибка: не найдена экипировка с id " .. gearId
    end
    if not gear.equipment then
        return "Ошибка: в экипировке с id " .. gearId .. " отсутствует раздел equipment"
    end

    -- 1. Попытка получить экипировку из gearData
    local itemId = gear.equipment[slot]
    local sourceGear = gear  -- объект, откуда взята экипировка (для обработки, например, слота "back")
    
    -- 2. Если в gearData для данного слота ничего не найдено, переходим к gearloadoutGroup
    if not itemId then
        local combinedId = jobId .. slot  -- например, "Quartermaster" + "head" → "Quartermasterhead"
        local groupEntry = nil
        for _, v in pairs(gearloadoutGroup) do
            if string.lower(v.id) == string.lower(combinedId) then
                groupEntry = v
                break
            end
        end
        if groupEntry and groupEntry.loadouts and #groupEntry.loadouts > 0 then
            -- Берем первый id из loadouts
            local firstLoadoutId = groupEntry.loadouts[1]
            -- Ищем этот id в gearloadout
            local loadoutEntry = nil
            for _, v in pairs(gearloadout) do
                if string.lower(v.id) == string.lower(firstLoadoutId) then
                    loadoutEntry = v
                    break
                end
            end
            if loadoutEntry and loadoutEntry.equipment then
                itemId = loadoutEntry.equipment[slot]
                sourceGear = loadoutEntry
            end
        end
    end

    -- 3. Если экипировка так и не найдена, возвращаем пустую строку
    if not itemId then
        return ""
    end

    -- 4. Вывод результата
    if slot == "back" and sourceGear.storage and sourceGear.storage.back then
        local storageBack = ""
        for _, storageItem in ipairs(sourceGear.storage.back) do
            storageBack = storageBack .. string.format("{{#invoke:Прототипы/Предмет/Содержание|main|%s}}", storageItem)
        end
        return frame:preprocess(string.format("{{#invoke:Прототипы/Предмет/Содержание|image|%s|%s}}", itemId, storageBack))
    else
        return frame:preprocess(string.format("{{#invoke:Прототипы/Предмет/Содержание|image|%s}}", itemId))
    end
end

return p