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

Интерактивная навигация по истории

← Предыдущая правка

@@ Строка 1: / Строка 1: @@
--- Загрузка данных
 local jobData           = mw.loadData("Модуль:IanComradeBot/job.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 gearloadoutGroup  = mw.loadData("Модуль:IanComradeBot/loadoutGroup.json/data")
@@ Строка 7: / Строка 7: @@
 local p = {}
-function p.main(frame)
+-- Поиск записи по id в массиве
-     local args = frame.args
+local function findById(tbl, id)
-    local slot  = args[1] or ""
+     if type(tbl) ~= "table" then return nil end
-     local jobId = args[2] or ""
+     for _, v in ipairs(tbl) do
+        if v.id == id then
-    if slot == "" or jobId == "" then
+            return v
-         return "Ошибка: Не заданы необходимые параметры. Используйте: {{#invoke:Prototypes/Роль/Экипировка|main|<slot>|<jobId>}}"
+         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
-    -- Поиск должности по 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 string.lower(v.id) == string.lower(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
+        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
-     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
-     -- Получение 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
-     -- Поиск набора экипировки в gearData по id
+     -- Режим special: импланты
-     local gear = nil
+     if param == "special" then
-    for _, v in pairs(gearData) do
+        local job = findById(jobData, capitalizeFirst(idParam))
-        if string.lower(v.id) == string.lower(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
-     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
-    -- 1. Попытка получить экипировку из gearData
+     local itemId = ""
-     local itemId = gear.equipment[slot]
+     local sourceGear = nil
-     local sourceGear = gear  -- объект, откуда взята экипировка (для слота "back")
+    -- Пытаемся найти запись в roleLoadout
-    -- Проверяем, что значение не nil и не пустое
+    local roleLoadoutId = "Job" .. capitalizeFirst(idParam)
-     if itemId == nil or itemId == "" then
+     local roleLoadoutData = findById(gearRoleLoadout, roleLoadoutId)
-         -- 2. Если в gearData для данного слота ничего не найдено, переходим к gearloadoutGroup
-         local combinedId = jobId .. slot
+     if not roleLoadoutData then
-         local groupEntry = nil
+         -- Нет групповых лоадаутов: сначала пробуем ручной ввод gearId
-        for _, v in pairs(gearloadoutGroup) do
+        local gear = findById(gearData, idParam)
-             if string.lower(v.id) == string.lower(combinedId) then
+         if gear then
-                 groupEntry = v
+            -- ручной режим: idParam — это startingGearId
-                break
+            local gearEquipment = gear.equipment or {}
+            itemId = gearEquipment[slot] or ""
+            sourceGear = gear
+         else
+            -- fallback: берём startingGear из job
+            local job = findById(jobData, capitalizeFirst(idParam))
+             if not job or not job.startingGear then
+                return "Ошибка: не найдена экипировка или должность с id " .. idParam
+            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
-         if groupEntry and groupEntry.loadouts and #groupEntry.loadouts > 0 then
+        local gear = findById(gearData, gearId)
-             -- Берем первый id из loadouts
+         if not gear then
-            local firstLoadoutId = groupEntry.loadouts[1]
+             return "Ошибка: не найдена экипировка с id " .. gearId
-            -- Ищем этот id в gearloadout
+        end
-            local loadoutEntry = nil
+        local gearEquipment = gear.equipment or {}
-             for _, v in pairs(gearloadout) do
+        itemId = gearEquipment[slot] or ""
-                 if string.lower(v.id) == string.lower(firstLoadoutId) then
+        sourceGear = gear
-                    loadoutEntry = v
+        if itemId == "" then
-                     break
+             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
-            end
+                if itemId ~= "" then break end
-            if loadoutEntry and loadoutEntry.equipment then
-                itemId = loadoutEntry.equipment[slot]
-                sourceGear = loadoutEntry
              end
          end
      end
-    -- 3. Если экипировка так и не найдена, возвращаем пустую строку
+     if itemId == "" then
-     if itemId == nil or itemId == "" then
          return ""
      end
-     -- 4. Вывод результата
+     local result = buildResult(itemId, sourceGear, slot)
-    if slot == "back" and sourceGear.storage and sourceGear.storage.back then
+    return processResult(itemId, result, radioMode, frame)
-        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