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

Материал из Space Station 14 Вики

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

local p = {}

-- Функция для загрузки данных и создания словаря для быстрого доступа
local function loadData(filePath)
    local page = mw.title.new(filePath)
    local content = page:getContent()
    local data = content and mw.text.jsonDecode(content) or nil
    
    -- Преобразуем список в словарь для O(1) доступа
    if data then
        local dict = {}
        for _, item in ipairs(data) do
            dict[item.id] = item
        end
        return dict
    end
    return nil
end

-- Форматирование одного содержимого
local function formatContent(content)
    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.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=#d7d7ff0b|image=%s|name=%s%s%s {{#invoke:Prototypes/Хранилище/Предмет|main|framing|contained|%s}} }}',
                         image, name, amount, prob, content.id)
end

-- Получение содержимого
local function getContentsOutput(contents)
    local result = {}
    for _, content in ipairs(contents) do
        table.insert(result, formatContent(content))
    end
    return table.concat(result)
end

-- Обработка таблиц
local function getTableOutput(tableId, allSelectors)
    local tableData = allSelectors[tableId]
    local children = tableData and tableData['!type:AllSelector'] and tableData['!type:AllSelector'].children

    if not children then return 'Таблица не содержит элементов.' end

    local result = {}
    for _, child in ipairs(children) do
        table.insert(result, formatContent(child))
    end
    return table.concat(result)
end

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

    if item.StorageFill and item.StorageFill.contents then
        return getContentsOutput(item.StorageFill.contents)
    elseif item.EntityTableContainerFill and item.EntityTableContainerFill.containers then
        local tableId = item.EntityTableContainerFill.containers.storagebase and item.EntityTableContainerFill.containers.storagebase.tableId
        return tableId and getTableOutput(tableId, allSelectors) or 'Таблица не найдена.'
    end
    return ''
end

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

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

    local result = {}
    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
    return '<ul class="1">' .. table.concat(result) .. '</ul>'
end

-- Основная функция модуля
function p.main(frame)
    local mode = frame.args[1]
    local id = frame.args[2]
    
    if not id then return 'Не указан ID.' end

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

    -- Для режима framing
    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(data, id) .. '}}')
        elseif subMode == 'contained' then
            local allSelectors = loadData('User:IanComradeBot/prototypes/AllSelector.json')
            return frame:preprocess('{{СollapsibleMenu|' .. getContainedOutput(data, id, allSelectors) .. '}}')
        else
            return 'Неизвестный подрежим для framing: ' .. subMode
        end
    end

    -- Обычные режимы
    if mode == 'chem' then
        return frame:preprocess(getChemOutput(data, id))
    elseif mode == 'contained' then
        local allSelectors = loadData('User:IanComradeBot/prototypes/AllSelector.json')
        return frame:preprocess(getContainedOutput(data, id, allSelectors))
    else
        return 'Неизвестный режим: ' .. mode
    end
end

return p