Модуль: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
      return str:sub(1,1):upper() .. str:sub(2)
+end
+-- Формирование вывода предмета
+local function buildResult(itemId, sourceGear, slot)
+    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("{{Предмет|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
+end
+-- Обработка в режиме 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
 function p.main(frame)
      local args = frame.args
-     local slot  = args[1] or ""
+     local param   = mw.text.trim(args[1] or "")
-     local jobId = args[2] or ""
+     local idParam = mw.text.trim(args[2] or "")
+     if param == "" or idParam == "" then
-     if slot == "" or jobId == "" then
+         return "Ошибка: Не заданы необходимые параметры. Используйте: {{#invoke:Прототип/Роль/Экипировка|main|<slot/special/radio>|<jobId или startingGearId>}}"
-         return "Ошибка: Не заданы необходимые параметры. Используйте: {{#invoke:Prototypes/Роль/Экипировка|main|<slot>|<jobId>}}"
      end
-     -- Поиск должности по id
+     -- Режим special: импланты
-     local job = findById(jobData, jobId)
+     if param == "special" then
-    if not job then
+        local job = findById(jobData, capitalizeFirst(idParam))
-         return "Ошибка: не найдена должность с id " .. jobId
+        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
-     -- Получение id набора экипировки из должности
+     local radioMode = false
-     local gearId = job.startingGear
+     local slot = param
-     if not gearId then
+     if param == "radio" then
-         return "Ошибка: у должности с id " .. jobId .. " не указан startingGear"
+         radioMode = true
+        slot = "ears"
      end
-    -- Поиск набора экипировки в gearData по id
+     local itemId = ""
-     local gear = findById(gearData, gearId)
+     local sourceGear = nil
-    if not gear then
-        return "Ошибка: не найдена экипировка с id " .. gearId
-     end
-     local gearEquipment = gear.equipment or {}
+    -- Пытаемся найти запись в roleLoadout
+     local roleLoadoutId = "Job" .. capitalizeFirst(idParam)
+    local roleLoadoutData = findById(gearRoleLoadout, roleLoadoutId)
-     -- 1. Попытка получить экипировку из gearData
+     if not roleLoadoutData then
-    local itemId = gearEquipment[slot] or ""
+        -- Нет групповых лоадаутов: сначала пробуем ручной ввод gearId
-    local sourceGear = gear
+        local gear = findById(gearData, idParam)
+        if gear then
-    -- 2. Если в gearData для данного слота ничего не найдено – ищем в gearloadoutGroup
+            -- ручной режим: idParam — это startingGearId
-    if itemId == "" then
+            local gearEquipment = gear.equipment or {}
-         local combinedId = jobId .. capitalizeFirst(slot)
+            itemId = gearEquipment[slot] or ""
-         local groupEntry = findById(gearloadoutGroup, combinedId)
+            sourceGear = gear
-         if groupEntry and groupEntry.loadouts and #groupEntry.loadouts > 0 then
+        else
-             local firstLoadoutId = groupEntry.loadouts[1]
+            -- fallback: берём startingGear из job
-             local loadoutEntry = findById(gearloadout, firstLoadoutId)
+            local job = findById(jobData, capitalizeFirst(idParam))
-            if loadoutEntry and loadoutEntry.equipment then
+            if not job or not job.startingGear then
-                itemId = loadoutEntry.equipment[slot] or ""
+                return "Ошибка: не найдена экипировка или должность с id " .. idParam
-                 sourceGear = loadoutEntry
+            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
+        local gearEquipment = gear.equipment or {}
+        itemId = gearEquipment[slot] or ""
+        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
-    -- 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