Модуль: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 = {}

-- Вспомогательная функция для поиска в gearloadout по id
function p.findInLoadoutById(id, slot)
    for _, v in pairs(gearloadout) do
        if string.lower(v.id) == string.lower(id) then
            if v.equipment then
                return v.equipment[slot], v
            end
            break
        end
    end
    return nil, nil
end

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:IanComradeBot/gear|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

    -- 2. Если в gearData ничего не найдено, ищем в gearloadout по id (jobId .. slot)
    if not itemId then
        local combinedId = jobId .. slot  -- например, "Quartermaster" + "shoes" → "Quartermastershoes"
        itemId, sourceGear = p.findInLoadoutById(combinedId, slot)
    end

    -- 3. Если и там ничего не найдено, ищем в gearloadoutGroup
    if not itemId then
        local combinedId = jobId .. slot
        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
            local firstLoadoutId = groupEntry.loadouts[1]
            itemId, sourceGear = p.findInLoadoutById(firstLoadoutId, slot)
        end
    end

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

    -- Обработка для слота "back"
    if slot == "back" then
        local storageBack = ""
        if sourceGear.storage and sourceGear.storage.back then
            for _, storageItem in ipairs(sourceGear.storage.back) do
                storageBack = storageBack .. string.format("{{#invoke:Prototypes/Предмет/Содержание|main|%s}}", storageItem)
            end
        end
        return frame:preprocess(string.format("{{#invoke:Prototypes/Предмет/Содержание|image|%s|%s}}", itemId, storageBack))
    else
        return frame:preprocess(string.format("%s", itemId))
    end
end

return p