Модуль:Песочница/Pok

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

local p = {}

-- Загружаем данные из отдельных модулей с JSON-таблицами
local itemData = mw.loadData("Модуль:User:IanComradeBot/prototypes/fills/Item.json/data")
local tableData = mw.loadData("Модуль:User:IanComradeBot/prototypes/table.json/data")

-----------------------------------------------------------
-- Функция processRolls для преобразования диапазона
local processRolls = function(rolls)
    local result = {}
    if rolls and rolls.range then
        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
        result[#result + 1] = string.format('[%d]', rolls.value)
    else
        result[#result + 1] = 'Не указан параметр rolls.'
    end
    return table.concat(result)
end

-----------------------------------------------------------
-- Создание хэш-таблицы для быстрого поиска по ID
local function buildIndex(data)
    if not data then return {} end
    local index = {}
    for _, item in ipairs(data) do
        index[item.id] = item
    end
    return index
end

-- Поиск данных по ID через хэш-таблицу
local function findDataById(dataIndex, id)
    return dataIndex and dataIndex[id] or nil
end

-----------------------------------------------------------
-- Форматирование содержимого
local function formatContent(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=#cbcbff0b|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
        result[#result + 1] = formatContent(content)
    end
    return table.concat(result)
end

-----------------------------------------------------------
-- Обработка вложенных таблиц
local function processNestedSelectors(children)
    if not children or #children == 0 then return "" end

    local results = {}
    for _, child in ipairs(children) do
        if child.id then
            results[#results + 1] = formatContent(child)
        elseif child["!type"] == "NestedSelector" then
            results[#results + 1] = handleNestedSelector(child, true)
        elseif child["!type"] == "GroupSelector" then
            results[#results + 1] = handleGroupSelector(child)
        end
    end

    return table.concat(results)
end

-----------------------------------------------------------
-- Обработка таблиц
local function getTableOutput(tableId)
    local tableIndex = buildIndex(tableData)
    local tableItem = findDataById(tableIndex, tableId)
    if not tableItem then return 'Таблица не найдена.' end

    if tableItem['!type:GroupSelector'] then
        return handleGroupSelector(tableItem['!type:GroupSelector'])
    elseif tableItem['!type:AllSelector'] then
        return processNestedSelectors(tableItem['!type:AllSelector'].children)
    end

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

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

    local result = {}

    -- Обработка 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

        -- Обработка entity_storage
        if containers.entity_storage then
            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
local function handleAllSelector(allSelector)
    if not allSelector.children then return '' end
    return processNestedSelectors(allSelector.children)
end

-----------------------------------------------------------
-- Обработка GroupSelector
local function handleGroupSelector(groupSelector)
    if not groupSelector.children then return '' end
    local result = {}
    local wrapperStart, wrapperEnd = "", ""

    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
local function handleNestedSelector(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
        result[#result + 1] = string.format('{{LinkСard/Сollapsible|name=Группа предметов%s%s|content=', 
            classesRolls or "", classesProb or "")
    end

    result[#result + 1] = getTableOutput(nestedSelector.tableId)

    if wrapped and (classesRolls or classesProb) then
        result[#result + 1] = "}}"
    end

    return table.concat(result)
end

-----------------------------------------------------------
-- Формирование списка химии
local function getChemOutput(dataIndex, id)
    local item = findDataById(dataIndex, 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

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

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

    local itemIndex = 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(itemIndex, id) .. '}}')
        elseif subMode == 'contained' then
            return frame:preprocess('{{СollapsibleMenu|' .. getContainedOutput(itemIndex, id) .. '}}')
        else
            return 'Неизвестный подрежим для framing: ' .. subMode
        end
    elseif mode == 'chem' then
        return frame:preprocess(getChemOutput(itemIndex, id))
    elseif mode == 'contained' then
        return frame:preprocess(getContainedOutput(itemIndex, id))
    elseif mode == 'rolls' then
        local entity = findDataById(itemIndex, 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

return p