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

-- Функция для поиска объекта по id
local function findById(tbl, id)
    for k, v in pairs(tbl) do
        if k == id or (v.id and v.id == id) then
            return v
        end
    end
    return nil
end

-- Функция для преобразования первой буквы строки в заглавную
local function capitalizeFirst(str)
    if not str or str == "" then return "" end
    return str:sub(1,1):upper() .. str:sub(2)
end

-- Функция для преобразования первой буквы
local function capitalize(s)
    return (s:gsub("^%l", string.upper))
end

function p.main(frame)
    local args = frame.args
    local modeOrSlot = mw.text.trim(args[1] or "")
    
    -- Режим "special": вывод специальных имплантов
    if modeOrSlot == "special" then
        local jobId = mw.text.trim(args[2] or "")
        if jobId == "" then
            return "Ошибка: Не задан jobId для режима special."
        end
        
        jobId = capitalizeFirst(jobId)
        local job = findById(jobData, jobId)
        if not job then
            return "Ошибка: не найдена должность с id " .. jobId
        end
        
        if not job.special or #job.special == 0 then
            return "Нет специальных данных для должности " .. jobId
        end
        
        local implantsFound = nil
        for _, spec in ipairs(job.special) do
            if spec["!type"] == "AddImplantSpecial" then
                implantsFound = spec.implants
                break
            end
        end
        
        if not implantsFound or #implantsFound == 0 then
            return "Специальных имплантов не найдено для должности " .. jobId
        end
        
        local resultList = {}
        for _, implant in ipairs(implantsFound) do
            table.insert(resultList, string.format("{{#invoke:Entity Lookup|getname|%s}}", implant))
        end
        local resultStr = table.concat(resultList, ", ")
        return resultStr
    end

    -- Если режим не "special", то первый параметр считается слотом,
    -- а второй – id должности, и используется базовая логика поиска экипировки.
    local slot  = modeOrSlot
    local jobId = mw.text.trim(args[2] or "")
    
    if slot == "" or jobId == "" then
        return "Ошибка: Не заданы необходимые параметры. Используйте: {{#invoke:Prototypes/Роль/Экипировка|main|<slot>|<jobId>}}"
    end

    jobId = capitalizeFirst(jobId)
    local job = findById(jobData, jobId)
    if not job then
        return "Ошибка: не найдена должность с id " .. jobId
    end

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

    local gear = findById(gearData, gearId)
    if not gear then
        return "Ошибка: не найдена экипировка с id " .. gearId
    end

    local gearEquipment = gear.equipment or {}
    local itemId = gearEquipment[slot] or ""
    local sourceGear = gear

    -- Если в gearData для указанного слота ничего не найдено – ищем через loadoutGroup / loadout
    if itemId == "" then
        local groupID = jobId .. capitalize(slot)
        local groupFound = nil
        for _, group in ipairs(gearloadoutGroup) do
            if group.id == groupID then
                groupFound = group
                break
            end
        end

        if not groupFound then
            return "Группа не найдена: " .. groupID
        end

        for _, loadoutID in ipairs(groupFound.loadouts) do
            for _, loadout in ipairs(gearloadout) do
                if loadout.id == loadoutID then
                    if loadout.equipment and loadout.equipment[slot] then
                        itemId = loadout.equipment[slot]
                        sourceGear = loadout
                        break
                    end
                end
            end
            if itemId ~= "" then break end
        end
    end

    if itemId == "" then
        return ""
    end

    -- Если слот "back" и присутствует storage, добавляем обработку элементов из storage.back
    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:Prototypes/Предмет/Содержание|main|%s}}", storageItem)
        end
        return frame:preprocess(string.format("{{#invoke:Prototypes/Предмет/Содержание|image|%s|%s}}", itemId, storageBack))
    else
        return frame:preprocess(string.format("{{#invoke:Prototypes/Предмет/Содержание|image|%s}}", itemId))
    end
end

return p