|
|
| Строка 4: |
Строка 4: |
| local p = {} | | local p = {} |
|
| |
|
| -- Функция для загрузки данных | | -- Функция для поиска данных по ID |
| local loadData = function(filePath) | | local function findDataById(itemData, id) |
| local page = mw.title.new(filePath)
| | if not itemData then return nil end |
| local content = page:getContent()
| | for _, item in ipairs(itemData) do |
| return content and mw.text.jsonDecode(content) or nil
| | if item.id == id then |
| end
| | return item |
| | |
| -- Функция processRolls для преобразования диапазона
| |
| local processRolls = function(rolls)
| |
| local result = {}
| |
| if rolls and rolls.range then
| |
| -- Если указан range
| |
| local min, max = rolls.range:match("(%d+),%s*(%d+)")
| |
| min, max = tonumber(min), tonumber(max)
| |
| if min and max then
| |
| result[#result + 1] = string.format('[%d-%d]', min + 1, max + 1)
| |
| else
| |
| result[#result + 1] = 'Некорректный формат для range.'
| |
| end
| |
| elseif rolls and rolls.value then
| |
| -- Если указано value
| |
| result[#result + 1] = string.format('[%d]', rolls.value)
| |
| else
| |
| result[#result + 1] = 'Не указан параметр rolls.'
| |
| end
| |
| return table.concat(result)
| |
| end
| |
| | |
| -- Локальные функции
| |
| local findDataById, formatContent, getContentsOutput, processNestedSelectors, getTableOutput, getContainedOutput, getChemOutput, handleGroupSelector, handleAllSelector, handleNestedSelector
| |
| | |
| -- Исправленная функция buildIndex, поддерживающая как одиночный объект, так и массив объектов
| |
| local function buildIndex(itemData)
| |
| if not itemData then return {} end | |
| local index = {} | |
| if itemData.id then
| |
| -- Если данные представляют собой одиночный объект
| |
| index[itemData.id] = itemData
| |
| else
| |
| -- Если данные представляют собой массив объектов
| |
| for _, item in ipairs(itemData) do
| |
| if item.id then
| |
| index[item.id] = item
| |
| end
| |
| end | | end |
| end | | end |
| return index | | return nil |
| end
| |
| | |
| -- Поиск данных по ID через индекс
| |
| findDataById = function(itemDataIndex, id)
| |
| return itemDataIndex and itemDataIndex[id] or nil
| |
| end
| |
| | |
| -- Форматирование содержимого
| |
| formatContent = function(content)
| |
| if not content.id then
| |
| return "Ошибка: отсутствует id у элемента."
| |
| end
| |
| | |
| local name = string.format('{{#invoke:Entity Lookup|getname|%s}}', content.id)
| |
| local image = string.format('%s.png', content.id)
| |
| local amount = (content.amount and content.amount ~= 1) and string.format(" [%d]", content.amount) or ""
| |
| local prob = ""
| |
| | |
| if content.weight then
| |
| content.prob = content.weight / 100
| |
| end
| |
| | |
| if content.prob then
| |
| prob = string.format(" <div>%s%%</div>", content.prob * 100 >= 1 and math.floor(content.prob * 100) or content.prob * 100)
| |
| end
| |
| | |
| return string.format(
| |
| '{{LinkСard|SideStyle=1|background-color=#cbcbff0a|image=%s|name=%s%s%s {{#invoke:Prototypes/Хранилище/Предмет|main|framing|contained|%s}} }}',
| |
| image, name, amount, prob, content.id
| |
| )
| |
| end
| |
| | |
| -- Получение содержимого через таблицу
| |
| getContentsOutput = function(contents)
| |
| local result = {}
| |
| for _, content in ipairs(contents) do
| |
| result[#result + 1] = formatContent(content)
| |
| end
| |
| return table.concat(result)
| |
| end | | end |
|
| |
|
| -- Обработка вложенных таблиц | | -- Функция обработки содержимого |
| processNestedSelectors = function(children) | | local function processNestedSelectors(children) |
| if not children or #children == 0 then return "" end | | if not children or #children == 0 then return "" end |
|
| |
| local results = {} | | local results = {} |
|
| |
|
| for _, child in ipairs(children) do | | for _, child in ipairs(children) do |
| if child.id then | | if child.id then |
| results[#results + 1] = formatContent(child) | | results[#results + 1] = string.format("Обнаружен элемент: %s", child.id) |
| elseif child["!type"] == "NestedSelector" then
| |
| results[#results + 1] = handleNestedSelector(child, true)
| |
| elseif child["!type"] == "GroupSelector" then | | elseif child["!type"] == "GroupSelector" then |
| results[#results + 1] = handleGroupSelector(child) | | results[#results + 1] = processNestedSelectors(child.children) |
| end | | end |
| end | | end |
|
| |
|
| return table.concat(results) | | return table.concat(results, ", ") |
| end
| |
| | |
| -- Обработка таблиц
| |
| getTableOutput = function(tableId)
| |
| local tableData = loadData('User:IanComradeBot/prototypes/table.json')
| |
| local tableDataIndex = findDataById(buildIndex(tableData), tableId)
| |
| | |
| if not tableDataIndex then return 'Таблица не найдена.' end
| |
| | |
| if tableDataIndex['!type:GroupSelector'] then
| |
| return handleGroupSelector(tableDataIndex['!type:GroupSelector'])
| |
| elseif tableDataIndex['!type:AllSelector'] then
| |
| return processNestedSelectors(tableDataIndex['!type:AllSelector'].children)
| |
| end
| |
| | |
| return 'Таблица не содержит элементов.'
| |
| end | | end |
|
| |
|
| -- Формирование списка содержащихся предметов или таблиц | | -- Обработка контейнеров |
| getContainedOutput = function(itemData, id) | | local function getContainedOutput(id) |
| local item = findDataById(itemData, id) | | local item = findDataById(itemData, id) |
| if not item then return '' end | | if not item then return "ID не найден." end |
|
| |
|
| local result = {}
| | if item.EntityTableContainerFill and item.EntityTableContainerFill.containers then |
| | |
| -- Обработка StorageFill
| |
| if item.StorageFill and item.StorageFill.contents then | |
| result[#result + 1] = getContentsOutput(item.StorageFill.contents)
| |
| | |
| -- Обработка EntityTableContainerFill
| |
| elseif item.EntityTableContainerFill and item.EntityTableContainerFill.containers then
| |
| local containers = item.EntityTableContainerFill.containers | | local containers = item.EntityTableContainerFill.containers |
| | | |
| -- Обработка entity_storage | | if containers.entity_storage and containers.entity_storage.children then |
| if containers.entity_storage then | | return processNestedSelectors(containers.entity_storage.children) |
| if containers.entity_storage.children then
| |
| result[#result + 1] = processNestedSelectors(containers.entity_storage.children)
| |
| end
| |
| | |
| if containers.entity_storage.tableId then
| |
| result[#result + 1] = getTableOutput(containers.entity_storage.tableId)
| |
| end
| |
| end
| |
| | |
| -- Обработка storagebase
| |
| if containers.storagebase and containers.storagebase.tableId then
| |
| result[#result + 1] = getTableOutput(containers.storagebase.tableId)
| |
| end
| |
| end
| |
| | |
| return table.concat(result)
| |
| end
| |
| | |
| -- Обработка AllSelector
| |
| handleAllSelector = function(allSelector)
| |
| if not allSelector.children then return '' end
| |
| return processNestedSelectors(allSelector.children)
| |
| end
| |
| | |
| -- Обработка GroupSelector
| |
| handleGroupSelector = function(groupSelector)
| |
| if not groupSelector.children then return '' end
| |
| local result = {}
| |
| local wrapperStart, wrapperEnd = "", ""
| |
| | |
| -- Проверка для контейнера EntityTableContainerFill
| |
| if groupSelector.weight and groupSelector.weight ~= "default" then
| |
| wrapperStart = string.format('{{LinkСard/Сollapsible|name=Группа предметов %s%%|content=', groupSelector.weight)
| |
| wrapperEnd = "}}"
| |
| elseif groupSelector["!type"] == "GroupSelector" and not groupSelector.weight then
| |
| wrapperStart = '{{LinkСard/Сollapsible|name=Может выпасть лишь один из:|content='
| |
| wrapperEnd = "}}"
| |
| end
| |
| | |
| for _, child in ipairs(groupSelector.children) do
| |
| if child["!type"] == "GroupSelector" then
| |
| result[#result + 1] = handleGroupSelector(child)
| |
| elseif child["!type"] == "AllSelector" then
| |
| result[#result + 1] = string.format('{{LinkСard/Сollapsible|name=Выпадают только вместе:|content=%s}}', handleAllSelector(child))
| |
| elseif child.id then
| |
| result[#result + 1] = formatContent(child)
| |
| else
| |
| result[#result + 1] = "<div>Ошибка: отсутствует id у элемента.</div>"
| |
| end
| |
| end
| |
| | |
| return wrapperStart .. table.concat(result) .. wrapperEnd
| |
| end
| |
| | |
| -- Обработка NestedSelector
| |
| handleNestedSelector = function(nestedSelector, wrapped)
| |
| if not nestedSelector.tableId then return '' end
| |
| | |
| local result = {}
| |
| | |
| local classesRolls, classesProb
| |
| | |
| if wrapped then
| |
| if nestedSelector.rolls and nestedSelector.rolls.range then
| |
| local rollsResult = processRolls(nestedSelector.rolls)
| |
| if rollsResult and #rollsResult > 0 then
| |
| classesRolls = ', максимум может выпасть: ' .. rollsResult
| |
| end
| |
| end
| |
| if nestedSelector.prob then
| |
| classesProb = string.format(" <div>%s%%</div>", nestedSelector.prob * 100 >= 1 and math.floor(nestedSelector.prob * 100) or nestedSelector.prob * 100)
| |
| end
| |
| end
| |
| | |
| if wrapped and (classesRolls or classesProb) then
| |
| if (classesRolls and #classesRolls > 0) or (classesProb and #classesProb > 0) then
| |
| result[#result + 1] = string.format('{{LinkСard/Сollapsible|name=Группа предметов%s%s|content=', classesRolls or "", classesProb or "")
| |
| end
| |
| end
| |
| | |
| result[#result + 1] = getTableOutput(nestedSelector.tableId)
| |
| | |
| if wrapped and (classesRolls or classesProb) then
| |
| if (classesRolls and #classesRolls > 0) or (classesProb and #classesProb > 0) then
| |
| result[#result + 1] = "}}"
| |
| end | | end |
| end | | end |
|
| |
|
| return table.concat(result) | | return "Нет вложенных элементов." |
| end
| |
| | |
| -- Формирование списка химии
| |
| getChemOutput = function(itemData, id)
| |
| local item = findDataById(itemData, id)
| |
| if not item or not item.SolutionContainerManager or not item.SolutionContainerManager.solutions then return '' end
| |
| | |
| local result = {}
| |
| for _, solution in pairs(item.SolutionContainerManager.solutions) do
| |
| for _, reagent in ipairs(solution.reagents) do
| |
| result[#result + 1] = string.format('<li>[[Химия#chem_%s|%s]] (%d ед.)</li>', reagent.ReagentId, reagent.ReagentId, reagent.Quantity)
| |
| end
| |
| end
| |
| return string.format('<ul class="1">%s</ul>', table.concat(result))
| |
| end | | end |
|
| |
|
| -- Основная функция модуля | | -- Основная функция |
| function p.main(frame) | | function p.main(frame) |
| local mode = frame.args[1] | | local id = frame.args[1] |
| local id = frame.args[2]
| | if not id then return "Не указан ID." end |
| | | return getContainedOutput(id) |
| if not id then return 'Не указан ID.' end | |
|
| |
| local itemDataIndex = buildIndex(itemData)
| |
|
| |
| if not itemData then return 'Не удалось загрузить данные.' end
| |
| | |
| if mode == 'framing' then
| |
| local subMode = frame.args[2]
| |
| local id = frame.args[3]
| |
| | |
| if not id then
| |
| return 'Не указан ID для режима framing.'
| |
| end
| |
| | |
| if subMode == 'chem' then
| |
| return frame:preprocess('{{СollapsibleMenu|' .. getChemOutput(itemDataIndex, id) .. '}}')
| |
| elseif subMode == 'contained' then
| |
| return frame:preprocess('{{СollapsibleMenu|' .. getContainedOutput(itemDataIndex, id) .. '}}')
| |
| else
| |
| return 'Неизвестный подрежим для framing: ' .. subMode
| |
| end
| |
| elseif mode == 'chem' then | |
| return frame:preprocess(getChemOutput(itemDataIndex, id))
| |
| elseif mode == 'contained' then
| |
| return frame:preprocess(getContainedOutput(itemDataIndex, id))
| |
| elseif mode == 'rolls' then
| |
| local entity = findDataById(itemDataIndex, id)
| |
| if not entity then return 'ID не найден в данных.' end
| |
| | |
| if entity.EntityTableContainerFill then
| |
| local containers = entity.EntityTableContainerFill.containers
| |
| if containers.entity_storage and containers.entity_storage.rolls then
| |
| return processRolls(containers.entity_storage.rolls)
| |
| end
| |
| end
| |
| | |
| return ''
| |
| else
| |
| return 'Неизвестный режим: ' .. mode
| |
| end
| |
| end | | end |
|
| |
|
| return p | | return p |