Модуль:Prototypes/Хранилище/Предмет

Для документации этого модуля может быть создана страница Модуль:Prototypes/Хранилище/Предмет/doc

local p = {}

-- Кэш для данных и предварительно обработанных структур
local cache = {
    data = nil,
    allSelectors = nil,
    indexedData = {}
}

-- Функции как локальные переменные
local loadData, findDataById, formatContent, getContentsOutput, processNestedSelectors, getTableOutput, getContainedOutput, getChemOutput, handleGroupSelector, handleAllSelector, handleNestedSelector, indexData, processRolls

-- Загрузка и кэширование данных
loadData = function(filePath)
    if cache[filePath] then
        return cache[filePath]
    end

    local page = mw.title.new(filePath)
    local content = page:getContent()
    local data = content and mw.text.jsonDecode(content) or nil
    cache[filePath] = data

    return data
end

-- Индексация данных для быстрого доступа
indexData = function(data)
    cache.indexedData = {}
    if not data then return end
    for _, item in ipairs(data) do
        if item.id then
            cache.indexedData[item.id] = item
        end
    end
end

-- Поиск данных по ID (с использованием индексации)
findDataById = function(data, id)
    if not data or not id then return nil end
    return cache.indexedData[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 string.format(" [%d]", content.amount) or ""
    local prob = ""

    -- Преобразование веса в вероятность
    if content.weight then
        content.prob = content.weight / 100
    end

    if content.prob then
        local percentage = content.prob * 100
        prob = percentage >= 1 and string.format(" <div>%d%%</div>", math.floor(percentage))
              or string.format(" <div>%g%%</div>", percentage)
    end

    return string.format(
        '{{LinkСard|SideStyle=1|background-color=#cbcbff0b|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
        table.insert(result, formatContent(content))
    end
    return table.concat(result)
end

-- Обработка вложенных таблиц
processNestedSelectors = function(children)
    local result = {}
    for _, child in ipairs(children) do
        if child.id then
            table.insert(result, formatContent(child))
        elseif child["!type"] == "NestedSelector" and child.tableId then
            table.insert(result, handleNestedSelector(child, true))
        elseif child["!type"] == "GroupSelector" then
            table.insert(result, handleGroupSelector(child))
        end
    end
    return table.concat(result)
end

-- Обработка таблиц
getTableOutput = function(tableId)
    cache.allSelectors = cache.allSelectors or loadData('User:IanComradeBot/prototypes/AllSelector.json')
    local tableData = findDataById(cache.allSelectors, tableId)

    if not tableData then return 'Таблица не найдена.' end

    if tableData['!type:GroupSelector'] and tableData['!type:GroupSelector'].children then
        return handleGroupSelector(tableData['!type:GroupSelector'])
    elseif tableData['!type:AllSelector'] and tableData['!type:AllSelector'].children then
        return processNestedSelectors(tableData['!type:AllSelector'].children)
    else
        return 'Таблица не содержит элементов.'
    end
end

-- Формирование списка содержащихся предметов или таблиц
getContainedOutput = function(data, id)
    local item = findDataById(data, id)
    if not item then return '' end

    local result = {}

    if item.StorageFill and item.StorageFill.contents then
        table.insert(result, getContentsOutput(item.StorageFill.contents))
    elseif item.EntityTableContainerFill and item.EntityTableContainerFill.containers then
        local containers = item.EntityTableContainerFill.containers

        if containers.entity_storage then
            local children = containers.entity_storage.children

            if children then
                for _, child in ipairs(children) do
                    if child["!type"] == "NestedSelector" and child.tableId then
                        table.insert(result, handleNestedSelector(child, false))
                    elseif child["!type"] == "GroupSelector" then
                        table.insert(result, handleGroupSelector(child))
                    elseif child["!type"] == "AllSelector" then
                        table.insert(result, handleAllSelector(child))
                    elseif child.id then
                        table.insert(result, formatContent(child))
                    end
                end
            end

            if containers.entity_storage.tableId then
                table.insert(result, getTableOutput(containers.entity_storage.tableId))
            end
        end

        if containers.storagebase and containers.storagebase.tableId then
            table.insert(result, getTableOutput(containers.storagebase.tableId))
        end
    end

    return table.concat(result)
end

-- Обработка AllSelector
handleAllSelector = function(allSelector)
    local result = {}
    if not allSelector.children then return '' end

    for _, child in ipairs(allSelector.children) do
        if child["!type"] == "NestedSelector" and child.tableId then
            table.insert(result, handleNestedSelector(child, true))
        elseif child.id then
            table.insert(result, formatContent(child))
        end
    end

    return table.concat(result)
end

-- Обработка GroupSelector
handleGroupSelector = function(groupSelector)
    local result = ''

    -- Проверка на существование children
    if not groupSelector.children then
        return ""
    end

    local weight = groupSelector.weight or "default"

    for _, child in ipairs(groupSelector.children) do
        if child["!type"] == "GroupSelector" then
            result = result .. handleGroupSelector(child)
        elseif child["!type"] == "AllSelector" then
            result = result .. '<div class="AllSelector">'
            result = result .. handleAllSelector(child)
            result = result .. '</div>'
        elseif child.id then
            result = result .. formatContent(child)
        else
            result = result .. "<div>Ошибка: отсутствует id у элемента.</div>"
        end
    end

    -- Если weight не равен "default", оборачиваем в div
    if weight ~= "default" then
        result = string.format('<div class="together" id="%s">%s</div>', weight, result)
    end

    return result
end

-- Обработка NestedSelector
handleNestedSelector = function(nestedSelector, wrapped)
    local tableId = nestedSelector.tableId
    if not tableId then return '' end

    if wrapped then
        local classes = {}

        if nestedSelector.rolls and nestedSelector.rolls.range then
            table.insert(classes, 'rolls-' .. tostring(nestedSelector.rolls.range):gsub(',', '-'))
        end
        if nestedSelector.prob then
            table.insert(classes, 'prob-' .. tostring(math.floor(nestedSelector.prob * 100)) .. 'p')
        end

        return string.format('<div class="%s">%s</div>', table.concat(classes, ' '), getTableOutput(tableId))
    else
        return getTableOutput(tableId)
    end
end

-- Формирование списка химии
getChemOutput = function(data, id)
    local item = findDataById(data, id)
    if not item then return '' end

    local solutions = item.SolutionContainerManager and item.SolutionContainerManager.solutions
    if not solutions then return '' end

    local result = {}
    table.insert(result, '<ul>')
    for _, solution in pairs(solutions) do
        for _, reagent in ipairs(solution.reagents) do
            table.insert(result, string.format('<li>[[Химия#chem_%s|%s]] (%d ед.)</li>', reagent.ReagentId, reagent.ReagentId, reagent.Quantity))
        end
    end
    table.insert(result, '</ul>')
    return table.concat(result)
end

-- Обработка rolls
processRolls = function(rolls)
    if rolls.range then
        local min, max = rolls.range:match("(%d+),%s*(%d+)")
        min, max = tonumber(min), tonumber(max)
        return min and max and string.format('Максимальное количество предметов: от %d до %d.', min, max) or 'Некорректный формат для range.'
    elseif rolls.value then
        return string.format('Будет сгенерировано %d предметов.', rolls.value)
    else
        return 'Не указан параметр rolls.'
    end
end

-- Основная функция модуля
function p.main(frame)
    local mode = frame.args[1]
    local id = frame.args[2]

    if not id then return 'Не указан ID.' end

    cache.data = cache.data or loadData('User:IanComradeBot/prototypes/fills/Item.json')
    if not cache.data then return 'Не удалось загрузить данные.' end

    indexData(cache.data)

    if mode == 'framing' then
        local subMode = frame.args[2]
        id = frame.args[3]

        if not id then
            return 'Не указан ID для режима framing.'
        end

        if subMode == 'chem' then
            return frame:preprocess('{{СollapsibleMenu|' .. getChemOutput(cache.data, id) .. '}}')
        elseif subMode == 'contained' then
            return frame:preprocess('{{СollapsibleMenu|' .. getContainedOutput(cache.data, id) .. '}}')
        else
            return 'Неизвестный подрежим для framing: ' .. subMode
        end
    end

    if mode == 'chem' then
        return frame:preprocess(getChemOutput(cache.data, id))
    elseif mode == 'contained' then
        return frame:preprocess(getContainedOutput(cache.data, id))
    elseif mode == 'rolls' then
        local entity = findDataById(cache.data, 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 'Режим rolls не найден для этого ID.'
    else
        return 'Неизвестный режим: ' .. mode
    end
end

return p