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

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

local findDataById, formatContent, getContentsOutput, processNestedSelectors, getTableOutput, getContainedOutput, getChemOutput, handleGroupSelector, handleAllSelector, handleNestedSelector, processRolls, getStackCount

---------------------------------------------------------------------
-- Функция поиска данных по ID
findDataById = function(data, id)
    if not data then 
        return nil 
    end

    if data[id] then
        return data[id]
    end

    if type(data) == "table" then
        if #data > 0 then
            for _, item in ipairs(data) do
                if type(item) == "table" and item.id == id then
                    return item
                end
            end
        else
            for _, item in pairs(data) do
                if type(item) == "table" and item.id == id then
                    return item
                end
            end
        end
    end
    return nil
end

---------------------------------------------------------------------
-- Форматирование содержимого
formatContent = function(content)
    if type(content) == "table" and not content.id then
        return "Ошибка: отсутствует id у элемента."
    end

    local id = content.id or content
    
    local amountNumber = 1
    if content.amount then
        if type(content.amount) == "table" then
            amountNumber = content.amount.value or 1
        else
            amountNumber = content.amount
        end
    end
    local amount = (amountNumber and amountNumber ~= 1) and string.format(" [%d]", amountNumber) or ""
    
    local prob = ""
    local probValue = nil
    if content.weight then
        probValue = content.weight / 100
    end

    if probValue then
        prob = string.format(" <div>%s%%</div>", probValue * 100 >= 1 and math.floor(probValue * 100) or probValue * 100)
    end

    return string.format(
        '{{#invoke:Prototypes/Предмет/Содержание|main|%s|%s%s}}',
        id, amount, prob
    )
end

---------------------------------------------------------------------
-- Получение содержимого через таблицу
getContentsOutput = function(contents)
    local result = ""
    if type(contents) == "table" then
        if #contents > 0 then
            for _, content in ipairs(contents) do
                result = result .. formatContent(content)
            end
        else
            for _, content in pairs(contents) do
                result = result .. formatContent(content)
            end
        end
    end
    return result
end

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

    local result = ""
    if type(children) == "table" then
        if #children > 0 then
            for _, child in ipairs(children) do
                if child.id then
                    if not visited[child.id] then
                        result = result .. formatContent(child)
                    end
                elseif child["!type"] == "NestedSelector" then
                    result = result .. handleNestedSelector(child, true, visited)
                elseif child["!type"] == "GroupSelector" then
                    result = result .. handleGroupSelector(child, visited)
                end
            end
        else
            for _, child in pairs(children) do
                if child.id then
                    if not visited[child.id] then
                        result = result .. formatContent(child)
                    end
                elseif child["!type"] == "NestedSelector" then
                    result = result .. handleNestedSelector(child, true, visited)
                elseif child["!type"] == "GroupSelector" then
                    result = result .. handleGroupSelector(child, visited)
                end
            end
        end
    end
    return result
end

---------------------------------------------------------------------
-- Обработка таблиц (для table.json)
getTableOutput = function(tableId, visited)
    visited = visited or {}

    if visited[tableId] then
        return ''
    end

    visited[tableId] = true

    local tableData = mw.loadData("Модуль:IanComradeBot/prototypes/table.json/data")
    local tableDataIndex = findDataById(tableData, tableId)

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

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

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

---------------------------------------------------------------------
-- Формирование списка содержащихся предметов или таблиц
getContainedOutput = function(itemData, id, visited)
    visited = visited or {} 

    if visited[id] then
        return ''
    end

    visited[id] = true

    local item = findDataById(itemData, 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 and type(containers.entity_storage.children) == "table" then
                if #containers.entity_storage.children > 0 then
                    for _, child in ipairs(containers.entity_storage.children) do
                        if child.id then
                            result[#result + 1] = formatContent(child)
                        elseif child["!type"] == "GroupSelector" then
                            result[#result + 1] = handleGroupSelector(child, visited)
                        elseif child["!type"] == "AllSelector" then
                            result[#result + 1] = processNestedSelectors(child.children, visited)
                        end
                    end
                else
                    for _, child in pairs(containers.entity_storage.children) do
                        if child.id then
                            result[#result + 1] = formatContent(child)
                        elseif child["!type"] == "GroupSelector" then
                            result[#result + 1] = handleGroupSelector(child, visited)
                        elseif child["!type"] == "AllSelector" then
                            result[#result + 1] = processNestedSelectors(child.children, visited)
                        end
                    end
                end
            end

            if containers.entity_storage.tableId then
                result[#result + 1] = getTableOutput(containers.entity_storage.tableId, visited)
            end
        end

        -- Обработка storagebase (аналогично entity_storage)
        if containers.storagebase then
            if containers.storagebase.children and type(containers.storagebase.children) == "table" then
                if #containers.storagebase.children > 0 then
                    for _, child in ipairs(containers.storagebase.children) do
                        if child.id then
                            result[#result + 1] = formatContent(child)
                        elseif child["!type"] == "GroupSelector" then
                            result[#result + 1] = handleGroupSelector(child, visited)
                        elseif child["!type"] == "AllSelector" then
                            result[#result + 1] = processNestedSelectors(child.children, visited)
                        end
                    end
                else
                    for _, child in pairs(containers.storagebase.children) do
                        if child.id then
                            result[#result + 1] = formatContent(child)
                        elseif child["!type"] == "GroupSelector" then
                            result[#result + 1] = handleGroupSelector(child, visited)
                        elseif child["!type"] == "AllSelector" then
                            result[#result + 1] = processNestedSelectors(child.children, visited)
                        end
                    end
                end
            end

            if containers.storagebase.tableId then
                result[#result + 1] = getTableOutput(containers.storagebase.tableId, visited)
            end
        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, visited)
    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

    if type(groupSelector.children) == "table" then
        if #groupSelector.children > 0 then
            for _, child in ipairs(groupSelector.children) do
                if child["!type"] == "GroupSelector" then
                    result = result .. handleGroupSelector(child, visited)
                elseif child["!type"] == "AllSelector" then
                    result = result .. string.format('{{LinkСard/Сollapsible|name=Выпадают только вместе:|content=%s}}', handleAllSelector(child))
                elseif child.id then
                    result = result .. formatContent(child)
                else
                    result = result .. "<div>Ошибка: отсутствует id у элемента.</div>"
                end
            end
        else
            for _, child in pairs(groupSelector.children) do
                if child["!type"] == "GroupSelector" then
                    result = result .. handleGroupSelector(child, visited)
                elseif child["!type"] == "AllSelector" then
                    result = result .. string.format('{{LinkСard/Сollapsible|name=Выпадают только вместе:|content=%s}}', handleAllSelector(child))
                elseif child.id then
                    result = result .. formatContent(child)
                else
                    result = result .. "<div>Ошибка: отсутствует id у элемента.</div>"
                end
            end
        end
    end

    return wrapperStart .. result .. wrapperEnd
end

---------------------------------------------------------------------
-- Обработка NestedSelector
handleNestedSelector = function(nestedSelector, wrapped, visited)
    visited = visited or {}
    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 .. string.format('{{LinkСard/Сollapsible|name=Группа предметов%s%s|content=', classesRolls or "", classesProb or "")
    end

    result = result .. getTableOutput(nestedSelector.tableId, visited)

    if wrapped and (classesRolls or classesProb) then
        result = result .. "}}"
    end

    return result
end

---------------------------------------------------------------------
-- Функция для преобразования диапазона (rolls)
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 .. string.format('[%d-%d]', min + 1, max + 1)
        else
            result = result .. 'Некорректный формат для range.'
        end
    elseif rolls and rolls.value then
        result = result .. string.format('[%d]', rolls.value)
    else
        result = result .. 'Не указан параметр rolls.'
    end
    return result
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 allReagents = {}
    for _, solution in pairs(item.SolutionContainerManager.solutions) do
        if solution and solution.reagents and type(solution.reagents) == "table" then
            if #solution.reagents > 0 then
                for _, reagent in ipairs(solution.reagents) do
                    table.insert(allReagents, reagent)
                end
            else
                for _, reagent in pairs(solution.reagents) do
                    table.insert(allReagents, reagent)
                end
            end
        end
    end

    local hasNonFiber = false
    if type(allReagents) == "table" then
        if #allReagents > 0 then
            for _, reagent in ipairs(allReagents) do
                if reagent.ReagentId ~= "Fiber" then
                    hasNonFiber = true
                    break
                end
            end
        else
            for _, reagent in pairs(allReagents) do
                if reagent.ReagentId ~= "Fiber" then
                    hasNonFiber = true
                    break
                end
            end
        end
    end

    if not hasNonFiber then
        return ''
    end

    local function processSolution(solution)
        local output = ""
        if solution and solution.reagents and type(solution.reagents) == "table" then
            if #solution.reagents > 0 then
                for _, reagent in ipairs(solution.reagents) do
                    local chemInfo = chemTranslateData[reagent.ReagentId]
                    local displayName = chemInfo and chemInfo.name or reagent.ReagentId
                    output = output .. string.format(
                        '<li>[[Химия#chem_%s|%s]] (%d ед.)</li>',
                        reagent.ReagentId, displayName, reagent.Quantity
                    )
                end
            else
                for _, reagent in pairs(solution.reagents) do
                    local chemInfo = chemTranslateData[reagent.ReagentId]
                    local displayName = chemInfo and chemInfo.name or reagent.ReagentId
                    output = output .. string.format(
                        '<li>[[Химия#chem_%s|%s]] (%d ед.)</li>',
                        reagent.ReagentId, displayName, reagent.Quantity
                    )
                end
            end
        end
        return output
    end

    local result = ""
    result = result .. processSolution(item.SolutionContainerManager.solutions["drink"])
    result = result .. processSolution(item.SolutionContainerManager.solutions["food"])
    result = result .. processSolution(item.SolutionContainerManager.solutions["beaker"])
    result = result .. processSolution(item.SolutionContainerManager.solutions["injector"])
    result = result .. processSolution(item.SolutionContainerManager.solutions["pen"])
    
    return result ~= "" and string.format('<ul class="1">%s</ul>', result) or ''
end

---------------------------------------------------------------------
-- Функция поиска count в itemStackData
getStackCount = function(id)
    local item = findDataById(itemStackData, id)
    if item and item.Stack and item.Stack.count then
        return item.Stack.count
    end
    return nil
end

---------------------------------------------------------------------
-- Загрузка данных
local itemData          = mw.loadData("Модуль:IanComradeBot/prototypes/fills/Item.json/data")
local itemSlotsData     = mw.loadData("Модуль:IanComradeBot/prototypes/ItemSlots.json/data")
local itemStackData     = mw.loadData("Модуль:IanComradeBot/prototypes/fills/stack.json/data")
local chemData          = mw.loadData("Модуль:IanComradeBot/prototypes/fills/chem.json/data")
local chemTranslateData = mw.loadData("Модуль:IanComradeBot/chem prototypes.json/data")

local p = {}

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

    if not id then return 'Не указан ID.' end
    
    local itemDataIndex = 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|color=#3e7c82|' .. getChemOutput(chemData, id) .. '}}')
        elseif subMode == 'contained' then
            return frame:preprocess('{{СollapsibleMenu|' .. getContainedOutput(itemDataIndex, id) .. '}}')
        elseif subMode == "slot" then
            local itemDataEntry = findDataById(itemSlotsData, id)
            if not itemDataEntry then return "" end
            local startingItem = nil
            if itemDataEntry.ItemSlots and itemDataEntry.ItemSlots.slots and type(itemDataEntry.ItemSlots.slots) == "table" then
                if #itemDataEntry.ItemSlots.slots > 0 then
                    for _, slot in ipairs(itemDataEntry.ItemSlots.slots) do
                        if slot.startingItem and slot.startingItem ~= "" then
                            startingItem = slot.startingItem
                            break
                        end
                    end
                else
                    for _, slot in pairs(itemDataEntry.ItemSlots.slots) do
                        if slot.startingItem and slot.startingItem ~= "" then
                            startingItem = slot.startingItem
                            break
                        end
                    end
                end
            end
            if not startingItem then return "" end
            return frame:preprocess('{{СollapsibleMenu|color=#71702a|' .. formatContent(startingItem) .. '}}')
        elseif subMode == "stack" then
            local count = getStackCount(id)
            return count and '(' .. count .. ')' or ""
        else
            return 'Неизвестный подрежим для framing: ' .. subMode
        end
    elseif mode == 'stack' then
        local count = getStackCount(id)
        return count or ""
    elseif mode == 'chem' then
        return getChemOutput(chemData, id)
    elseif mode == 'contained' then
        return frame:preprocess(getContainedOutput(itemDataIndex, id))
    elseif mode == "slot" then
        local itemDataEntry = findDataById(itemSlotsData, id)
        if not itemDataEntry then return "" end
        local startingItem = nil
        if itemDataEntry.ItemSlots and itemDataEntry.ItemSlots.slots and type(itemDataEntry.ItemSlots.slots) == "table" then
            if #itemDataEntry.ItemSlots.slots > 0 then
                for _, slot in ipairs(itemDataEntry.ItemSlots.slots) do
                    if slot.startingItem and slot.startingItem ~= "" then
                        startingItem = slot.startingItem
                        break
                    end
                end
            else
                for _, slot in pairs(itemDataEntry.ItemSlots.slots) do
                    if slot.startingItem and slot.startingItem ~= "" then
                        startingItem = slot.startingItem
                        break
                    end
                end
            end
        end
        if not startingItem then return "" end
        return frame:preprocess(formatContent(startingItem))
    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

return p