Модуль:Prototypes/Роль/Экипировка: различия между версиями

мНет описания правки
мНет описания правки
 
(не показаны 62 промежуточные версии этого же участника)
Строка 1: Строка 1:
-- Загрузка данных
local jobData           = mw.loadData("Модуль:IanComradeBot/job.json/data")
local jobData         = mw.loadData("Модуль:IanComradeBot/job.json/data")
local gearData         = mw.loadData("Модуль:IanComradeBot/startingGear.json/data")
local gearData         = mw.loadData("Модуль:IanComradeBot/startingGear.json/data")
local gearRoleLoadout  = mw.loadData("Модуль:IanComradeBot/roleLoadout.json/data")
local gearloadout     = mw.loadData("Модуль:IanComradeBot/loadout.json/data")
local gearloadout       = mw.loadData("Модуль:IanComradeBot/loadout.json/data")
local gearloadoutGroup = mw.loadData("Модуль:IanComradeBot/loadoutGroup.json/data")
local gearloadoutGroup = mw.loadData("Модуль:IanComradeBot/loadoutGroup.json/data")


local p = {}
local p = {}


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


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


    -- Поиск должности по id
-- Формирование вывода предмета
     local job = nil
local function buildResult(itemId, sourceGear, slot)
    for _, v in pairs(jobData) do
     if slot == "back" and sourceGear.storage and sourceGear.storage.back then
        if v.id == jobId then
        local storageBack = ""
            job = v
        for _, storageItem in ipairs(sourceGear.storage.back) do
            break
            storageBack = storageBack .. string.format("{{Предмет|wrapper=1|repository=1|%s}}", storageItem)
         end
         end
        return string.format("<div>{{Предмет|%s|repository=1|size=64px|label=|imageTooltip=1}} {{СollapsibleMenu|%s}}</div>", itemId, storageBack)
    else
        return string.format("{{Предмет|%s|repository=1|size=64px|label=|imageTooltip=1}}", itemId)
     end
     end
     if not job then
end
         return "Ошибка: не найдена должность с id " .. jobId
 
-- Обработка в режиме radio
local function processResult(itemId, result, radioMode, frame)
     if radioMode then
         return frame:preprocess(string.format("{{#invoke:Prototypes/Механика/Частоты|main|%s}}", itemId))
    else
        return frame:preprocess(result)
     end
     end
end


     -- Получение id набора экипировки из должности
function p.main(frame)
     local gearId = job.startingGear
    local args = frame.args
     if not gearId then
     local param  = mw.text.trim(args[1] or "")
         return "Ошибка: у должности с id " .. jobId .. " не указан startingGear"
     local idParam = mw.text.trim(args[2] or "")
     if param == "" or idParam == "" then
         return "Ошибка: Не заданы необходимые параметры. Используйте: {{#invoke:Прототип/Роль/Экипировка|main|<slot/special/radio>|<jobId или startingGearId>}}"
     end
     end


     -- Поиск набора экипировки в gearData по id
     -- Режим special: импланты
     local gear = nil
     if param == "special" then
    for _, v in pairs(gearData) do
        local job = findById(jobData, capitalizeFirst(idParam))
        if v.id == gearId then
        if not job then return "" end
             gear = v
        local specials = job.special or {}
             break
        local implantsList = {}
        for _, special in ipairs(specials) do
            if special["!type"] == "AddImplantSpecial" and special.implants then
                for _, implant in ipairs(special.implants) do
                    table.insert(implantsList, string.format("{{#invoke:Entity Lookup|getname|%s}}", implant))
                end
             end
        end
        if #implantsList > 0 then
            return frame:preprocess(table.concat(implantsList, ", "))
        else
             return ""
         end
         end
     end
     end
     if not gear then
 
        return "Ошибка: не найдена экипировка с id " .. gearId
     local radioMode = false
     end
     local slot = param
     if not gear.equipment then
     if param == "radio" then
         return "Ошибка: в экипировке с id " .. gearId .. " отсутствует раздел equipment"
         radioMode = true
        slot = "ears"
     end
     end


    -- 1. Попытка получить экипировку из gearData
     local itemId = ""
     local itemId = gear.equipment[slot]
     local sourceGear = nil
     local sourceGear = gear


     -- 2. Если в gearData ничего не найдено, ищем в gearloadout по id (jobId .. slot)
     -- Пытаемся найти запись в roleLoadout
     if not itemId then
     local roleLoadoutId = "Job" .. capitalizeFirst(idParam)
        local combinedId = jobId .. slot  -- например, "Quartermaster" + "shoes" → "Quartermastershoes"
    local roleLoadoutData = findById(gearRoleLoadout, roleLoadoutId)
        itemId, sourceGear = p.findInLoadoutById(combinedId, slot)
    end


     -- 3. Если и там ничего не найдено, ищем в gearloadoutGroup
     if not roleLoadoutData then
    if not itemId then
        -- Нет групповых лоадаутов: сначала пробуем ручной ввод gearId
        local combinedId = jobId .. slot
        local gear = findById(gearData, idParam)
         local groupEntry = nil
        if gear then
        for _, v in pairs(gearloadoutGroup) do
            -- ручной режим: idParam — это startingGearId
             if string.lower(v.id) == string.lower(combinedId) then
            local gearEquipment = gear.equipment or {}
                 groupEntry = v
            itemId = gearEquipment[slot] or ""
                break
            sourceGear = gear
         else
            -- fallback: берём startingGear из job
            local job = findById(jobData, capitalizeFirst(idParam))
             if not job or not job.startingGear then
                 return "Ошибка: не найдена экипировка или должность с id " .. idParam
             end
             end
            local gear2 = findById(gearData, job.startingGear)
            if not gear2 then
                return "Ошибка: не найдена экипировка с id " .. job.startingGear
            end
            local gearEquipment2 = gear2.equipment or {}
            itemId = gearEquipment2[slot] or ""
            sourceGear = gear2
        end
    else
        -- Есть групповые лоадауты: классический режим через roleLoadout
        local jobId = capitalizeFirst(idParam)
        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
         end
         if groupEntry and groupEntry.loadouts and #groupEntry.loadouts > 0 then
         local gearEquipment = gear.equipment or {}
            local firstLoadoutId = groupEntry.loadouts[1]
        itemId = gearEquipment[slot] or ""
            itemId, sourceGear = p.findInLoadoutById(firstLoadoutId, slot)
        sourceGear = gear
        if itemId == "" then
            for _, groupId in ipairs(roleLoadoutData.groups or {}) do
                if groupId ~= "Trinkets" then
                    local group = findById(gearloadoutGroup, groupId)
                    if group and type(group.loadouts) == "table" then
                        for _, loadoutID in ipairs(group.loadouts) do
                            local loadout = findById(gearloadout, loadoutID)
                            if loadout and not loadout.effects and loadout.equipment and loadout.equipment[slot] then
                                itemId = loadout.equipment[slot]
                                sourceGear = loadout
                                break
                            end
                        end
                    end
                end
                if itemId ~= "" then break end
            end
         end
         end
     end
     end


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


     -- Обработка для слота "back"
     local result = buildResult(itemId, sourceGear, slot)
    if slot == "back" then
    return processResult(itemId, result, radioMode, frame)
        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("{{#invoke:Prototypes/Предмет/Содержание|image|%s}}", itemId))
    end
end
end


return p
return p