Модуль: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 = {}
--- Функция для поиска объекта по id
+-- Поиск записи по id в массиве
 local function findById(tbl, id)
-     for k, v in pairs(tbl) do
+    if type(tbl) ~= "table" then return nil end
-         if k == id or (v.id and v.id == id) then
+     for _, v in ipairs(tbl) do
+         if v.id == id then
              return v
          end
@@ Строка 17: / Строка 18: @@
 end
--- Функция для преобразования первой буквы строки в заглавную
+-- Преобразование первой буквы в заглавную
 local function capitalizeFirst(str)
      if not str or str == "" then return "" end
@@ Строка 23: / Строка 24: @@
 end
-function p.main(frame)
+-- Формирование вывода предмета
-     local args = frame.args
+local function buildResult(itemId, sourceGear, slot)
-    local slot  = mw.text.trim(args[1] or "")
+     if slot == "back" and sourceGear.storage and sourceGear.storage.back then
-    local jobId = mw.text.trim(args[2] or "")
+        local storageBack = ""
+        for _, storageItem in ipairs(sourceGear.storage.back) do
-    if slot == "" or jobId == "" then
+            storageBack = storageBack .. string.format("{{Предмет|wrapper=1|repository=1|%s}}", storageItem)
-         return "Ошибка: Не заданы необходимые параметры. Используйте: {{#invoke:Prototypes/Роль/Экипировка|main|<slot>|<jobId>}}"
+        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
-    -- Приводим jobId к виду с заглавной первой буквой
+-- Обработка в режиме radio
-    jobId = capitalizeFirst(jobId)
+local function processResult(itemId, result, radioMode, frame)
+     if radioMode then
-    -- Поиск должности по id
+        return frame:preprocess(string.format("{{#invoke:Prototypes/Механика/Частоты|main|%s}}", itemId))
-    local job = findById(jobData, jobId)
+     else
-     if not job then
+         return frame:preprocess(result)
-         return "Ошибка: не найдена должность с id " .. jobId
      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 = findById(gearData, gearId)
+     if param == "special" then
-    if not gear then
+        local job = findById(jobData, capitalizeFirst(idParam))
-         return "Ошибка: не найдена экипировка с id " .. gearId
+        if not job then return "" end
+         local specials = job.special or {}
+        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
-     local gearEquipment = gear.equipment or {}
+     local radioMode = false
+    local slot = param
+    if param == "radio" then
+        radioMode = true
+        slot = "ears"
+    end
-    -- 1. Сначала пытаемся получить экипировку из gearData
+     local itemId = ""
-     local itemId = gearEquipment[slot] or ""
+     local sourceGear = nil
-     local sourceGear = gear
-     -- 2. Если для данного слота в gearData ничего не найдено – ищем через loadoutGroup / loadout
+     -- Пытаемся найти запись в roleLoadout
-     if itemId == "" then
+     local roleLoadoutId = "Job" .. capitalizeFirst(idParam)
-        -- Используем вашу логику для поиска
+    local roleLoadoutData = findById(gearRoleLoadout, roleLoadoutId)
-        local function capitalize(s)
-            return (s:gsub("^%l", string.upper))
-        end
-         -- Здесь используем jobId в качестве "role"
+    if not roleLoadoutData then
-         local groupID = jobId .. capitalize(slot)
+         -- Нет групповых лоадаутов: сначала пробуем ручной ввод gearId
+         local gear = findById(gearData, idParam)
-        -- Ищем нужную группу в gearloadoutGroup
+        if gear then
-        local groupFound = nil
+            -- ручной режим: idParam — это startingGearId
-         for _, group in ipairs(gearloadoutGroup) do
+            local gearEquipment = gear.equipment or {}
-             if group.id == groupID then
+            itemId = gearEquipment[slot] or ""
-                 groupFound = group
+            sourceGear = gear
-                 break
+         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
+        local gear = findById(gearData, gearId)
-         if not groupFound then
+         if not gear then
-             return "Группа не найдена: " .. groupID
+             return "Ошибка: не найдена экипировка с id " .. gearId
          end
+        local gearEquipment = gear.equipment or {}
-         -- Перебираем варианты loadouts из найденной группы
+         itemId = gearEquipment[slot] or ""
-         for _, loadoutID in ipairs(groupFound.loadouts) do
+        sourceGear = gear
-            for _, loadout in ipairs(gearloadout) do
+         if itemId == "" then
-                if loadout.id == loadoutID then
+            for _, groupId in ipairs(roleLoadoutData.groups or {}) do
-                    if loadout.equipment and loadout.equipment[slot] then
+                if groupId ~= "Trinkets" then
-                        itemId = loadout.equipment[slot]
+                    local group = findById(gearloadoutGroup, groupId)
-                        sourceGear = loadout
+                    if group and type(group.loadouts) == "table" then
-                         break
+                        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
-            if itemId ~= "" then break end
          end
      end
-    -- 3. Если экипировка так и не найдена, возвращаем пустую строку
      if itemId == "" then
          return ""
      end
-     -- 4. Если слот "back", добавляем содержимое из storage
+     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