Space Station 14 Вики

Модуль:Песочница/Pok: различия между версиями

Интерактивная навигация по истории
← Предыдущая правкаСледующая правка →
Нет описания правки
Нет описания правки
Строка 1: Строка 1:
-- Загрузка данных
local itemData      = mw.loadData("Модуль:IanComradeBot/prototypes/fills/Item.json/data")
local itemSlotsData = mw.loadData("Модуль:IanComradeBot/prototypes/ItemSlots.json/data")
local chemData      = mw.loadData("Модуль:IanComradeBot/prototypes/fills/chem.json/data")
local p = {}
local p = {}


-----------------------------
-- Функция для загрузки данных
-- Функция для «очистки» JSON
local loadData = function(filePath)
-----------------------------
     local page = mw.title.new(filePath)
local function cleanJSON(json)
     local content = page:getContent()
     -- Удаляем запятую перед закрывающей квадратной скобкой (конец массива)
     return content and mw.text.jsonDecode(content) or nil
     json = json:gsub(",%s*%]", "]")
     -- Удаляем запятую перед закрывающей фигурной скобкой (конец объекта)
    json = json:gsub(",%s*%}", "}")
    return json
end
end


---------------------------------------------
-- Функция processRolls для преобразования диапазона
-- Функция для загрузки и парсинга JSON-файла
local processRolls = function(rolls)
---------------------------------------------
    local result = ""
local function loadJSONData(pageName)
    if rolls and rolls.range then
     local title = mw.title.new(pageName)
        -- Если указан range
     if not title then
        local min, max = rolls.range:match("(%d+),%s*(%d+)")
         return {}
        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
        -- Если указано value
        result = result .. string.format('[%d]', rolls.value)
     else
         result = result .. 'Не указан параметр rolls.'
     end
     end
     local jsonData = title:getContent() or ""
     return result
     jsonData = cleanJSON(jsonData)
end
     local success, data = pcall(mw.text.jsonDecode, jsonData)
 
    if success and type(data) == "table" then  
-- Функция для поиска первого startingItem в slots
        return data
local function getFirstStartingItem(data)
    else
    if not data or not data.ItemSlots or not data.ItemSlots.slots then
         return {}
        return nil
    end
      
     for _, slot in pairs(data.ItemSlots.slots) do
        if slot.startingItem and slot.startingItem ~= "" then
            return slot.startingItem
         end
     end
     end
   
    return nil
end
end


-----------------------------------------------------------
-- Загрузка данных: заменяем mw.loadData на loadJSONData()
-----------------------------------------------------------
local latheData    = loadJSONData("User:IanComradeBot/prototypes/lathe.json")
local recipeData  = loadJSONData("User:IanComradeBot/prototypes/lathe/recipes.json")
local researchData = loadJSONData("User:IanComradeBot/prototypes/research.json")
local materialData = loadJSONData("User:IanComradeBot/prototypes/materials.json")
local chemData    = loadJSONData("User:IanComradeBot/chem prototypes.json")


-----------------------------------------------------------
-- Локальные функции
-- Функция для форматирования времени
local findDataById, formatContent, getContentsOutput, processNestedSelectors, getTableOutput, getContainedOutput, getChemOutput, handleGroupSelector, handleAllSelector, handleNestedSelector
-----------------------------------------------------------
local function format_seconds_to_short_string(input_seconds)
    local minutes = math.floor(input_seconds / 60)
    local seconds = input_seconds % 60


    local minutes_part = minutes > 0 and (minutes .. " мин.") or nil
-- Поиск данных по ID через индекс
    local seconds_part = seconds > 0 and (seconds .. " сек.") or nil
findDataById = function(itemDataIndex, id)
    if not itemDataIndex then return nil end
for _, item in ipairs(itemDataIndex) do
if item.id == id then
return item
end
end
end


     if minutes_part and seconds_part then
-- Форматирование содержимого
         return minutes_part .. " " .. seconds_part
formatContent = function(content)
     elseif seconds_part then
     if type(content) == "table" and not content.id then
         return seconds_part
         return "Ошибка: отсутствует id у элемента."
     elseif minutes_part then
    end
         return minutes_part
 
     else
    -- Если передан объект с id
         return '0 сек.'
    local id = content.id or content
    local name = string.format('{{#invoke:Entity Lookup|getname|%s}}', id)
    local image = string.format('%s.png', 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:Песочница/Pok|main|framing|contained|%s}} {{#invoke:Песочница/Pok|main|framing|slot|%s}} {{#invoke:Песочница/Pok|main|framing|chem|%s}} }}',
        image, name, amount, prob, id, id, id
     )
end
 
-- Получение содержимого через таблицу
getContentsOutput = function(contents)
    local result = ""
    for _, content in ipairs(contents) do
         result = result .. formatContent(content)
     end
     end
    return result
end
end


-----------------------------------------------------------
-- Обработка вложенных таблиц
-- Функция для сортировки рецептов
processNestedSelectors = function(children, visited)
-----------------------------------------------------------
visited = visited or {}
local function sortRecipesByPriority(recipes)
    if not children or #children == 0 then return "" end
    table.sort(recipes, function(a, b)
        local priority = { Static = 1, EMAG = 3 }
        local aPriority = priority[a.discipline] or 2
        local bPriority = priority[b.discipline] or 2


        if a.isEmag ~= b.isEmag then
    local result = ""
            return not a.isEmag
        end


         if aPriority == bPriority then
    for _, child in ipairs(children) do
            if a.tier == b.tier then
         if child.id then
                return a.discipline < b.discipline
if not visited[child.id] then
            end
    result = result .. formatContent(child)
             return a.tier < b.tier
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


        return aPriority < bPriority
    return result
    end)
end
end


-----------------------------------------------------------
-- Обработка таблиц
-- Основная функция модуля
getTableOutput = function(tableId, visited)
-----------------------------------------------------------
     visited = visited or {}
function p.main(frame)
     -- Подключение CSS
    local cssLink = frame:extensionTag('templatestyles', '', {
        src = 'Шаблон:Prototypes/Машина/Станок/styles.css'
    })


     local latheId = frame.args[1] or ""
     if visited[tableId] then
    if latheId == "" then
         return ''
         return '<div style="color:red;">Не указан ID станка.</div>'
     end
     end


     local lathe = nil
    visited[tableId] = true
     for _, data in ipairs(latheData) do
 
        if data.id == latheId then
     local tableData = loadData('User:IanComradeBot/prototypes/table.json')
            lathe = data
     local tableDataIndex = findDataById(tableData, tableId)
            break
 
         end
    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
     end


     if not lathe then
    return 'Таблица не содержит элементов.'
         return '<div style="color:red;">Станок с ID "' .. latheId .. '" не найден.</div>'
end
 
-- Формирование списка содержащихся предметов или таблиц
getContainedOutput = function(itemData, id, visited)
    visited = visited or {}
 
     if visited[id] then
         return ''
     end
     end


     local materialMapping = {}
    visited[id] = true
     for _, material in ipairs(materialData) do
 
        if material.id then
    local item = findDataById(itemData, id)
            materialMapping[material.id] = material.stackEntity or material.id or material.name
    if not item then return '' end
        end
 
    end
     local result = {}
 
     -- Обработка StorageFill
    if item.StorageFill and item.StorageFill.contents then
        result[#result + 1] = getContentsOutput(item.StorageFill.contents)


     local chemMapping = {}
     -- Обработка EntityTableContainerFill
     for id, chem in pairs(chemData) do
     elseif item.EntityTableContainerFill and item.EntityTableContainerFill.containers then
         chemMapping[id] = chem.name
         local containers = item.EntityTableContainerFill.containers
    end


    local out = cssLink
        -- Обработка entity_storage
    local recipes = {}
        if containers.entity_storage then
            -- Если есть просто элементы с id, то форматируем их как обычные предметы
            if containers.entity_storage.children then
                for _, child in ipairs(containers.entity_storage.children) do
                    if child.id then
                        result[#result + 1] = formatContent(child) -- Теперь одиночные id не игнорируются
                    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


    local function getRecipeDetails(recipeId)
            -- Если есть таблица, то загружаем её
        for _, recipe in ipairs(recipeData) do
             if containers.entity_storage.tableId then
             if recipe.id == recipeId then
                 result[#result + 1] = getTableOutput(containers.entity_storage.tableId, visited)
                 return recipe
             end
             end
         end
         end
        return nil
    end


    local function findInResearch(recipeId)
        -- Обработка storagebase (аналогично entity_storage)
         for _, research in ipairs(researchData) do
         if containers.storagebase then
             if research.technology and research.technology.recipeUnlocks then
             if containers.storagebase.children then
                 for _, unlock in ipairs(research.technology.recipeUnlocks) do
                 for _, child in ipairs(containers.storagebase.children) do
                     if unlock == recipeId then
                     if child.id then
                         return {
                        result[#result + 1] = formatContent(child)
                            name = research.technology.name,
                    elseif child["!type"] == "GroupSelector" then
                            tier = research.technology.tier,
                         result[#result + 1] = handleGroupSelector(child, visited)
                            discipline = research.technology.discipline
                    elseif child["!type"] == "AllSelector" then
                        }
                        result[#result + 1] = processNestedSelectors(child.children, visited)
                     end
                     end
                 end
                 end
            end
            if containers.storagebase.tableId then
                result[#result + 1] = getTableOutput(containers.storagebase.tableId, visited)
             end
             end
         end
         end
        return nil
     end
     end


     -- Обработка staticRecipes
     return table.concat(result)
     if lathe.Lathe and lathe.Lathe.staticRecipes then
end
        for _, recipeId in ipairs(lathe.Lathe.staticRecipes) do
 
            local recipe = getRecipeDetails(recipeId)
-- Обработка AllSelector
            if recipe and recipe.result then
handleAllSelector = function(allSelector)
                table.insert(recipes, {
     if not allSelector.children then return '' end
                    result = recipe.result,
    return processNestedSelectors(allSelector.children)
                    completetime = recipe.completetime,
end
                    materials = recipe.materials,
 
                    discipline = "Static",
-- Обработка GroupSelector
                    tier = 0
handleGroupSelector = function(groupSelector)
                })
    if not groupSelector.children then return '' end
            elseif recipe and recipe.resultReagents then
    local result = ""
                for reagent, amount in pairs(recipe.resultReagents) do
    local wrapperStart, wrapperEnd = "", ""
                    local reagentName = chemMapping[reagent] or reagent
 
                    table.insert(recipes, {
    -- Проверка для контейнера EntityTableContainerFill
                        result = reagentName .. "|amount=" .. amount .. "ед.|mode-chem=1",
    if groupSelector.weight and groupSelector.weight ~= "default" then
                        completetime = recipe.completetime,
        wrapperStart = string.format('{{LinkСard/Сollapsible|name=Группа предметов %s%%|content=', groupSelector.weight)
                        materials = recipe.materials,
        wrapperEnd = "}}"
                        discipline = "Static",
    elseif groupSelector["!type"] == "GroupSelector" and not groupSelector.weight then
                        tier = 0
        wrapperStart = '{{LinkСard/Сollapsible|name=Может выпасть лишь один из:|content='
                    })
        wrapperEnd = "}}"
                    break
    end
                end
 
             else
    for _, child in ipairs(groupSelector.children) do
                out = out .. '<div style="color:red;">Ошибка: Рецепт с ID "' .. recipeId .. '" не найден или поля result/resultReagents отсутствуют.</div>'
        if child["!type"] == "GroupSelector" then
            end
            result = result .. handleGroupSelector(child)
        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


     -- Обработка dynamicRecipes
     return wrapperStart .. result .. wrapperEnd
     if lathe.Lathe and lathe.Lathe.dynamicRecipes then
end
        for _, recipeId in ipairs(lathe.Lathe.dynamicRecipes) do
 
            local recipe = getRecipeDetails(recipeId)
-- Обработка NestedSelector
            if recipe then
handleNestedSelector = function(nestedSelector, wrapped, visited)
                local researchInfo = findInResearch(recipeId)
visited = visited or {}
                if researchInfo then
     if not nestedSelector.tableId then return '' end
                    table.insert(recipes, {
 
                        result = recipe.result,
    local result = ""
                        completetime = recipe.completetime,
    local classesRolls, classesProb
                        materials = recipe.materials,
 
                        discipline = researchInfo.discipline,
    if wrapped then
                        tier = researchInfo.tier,
        if nestedSelector.rolls and nestedSelector.rolls.range then
                        researchName = researchInfo.name
            local rollsResult = processRolls(nestedSelector.rolls)
                    })
            if rollsResult and #rollsResult > 0 then
                end
                classesRolls = ', максимум может выпасть: ' .. rollsResult
             end
             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
     end
     end


    -- Обработка emagStaticRecipes
     if wrapped and (classesRolls or classesProb) then
     if lathe.EmagLatheRecipes and lathe.EmagLatheRecipes.emagStaticRecipes then
         result = result .. string.format('{{LinkСard/Сollapsible|name=Группа предметов%s%s|content=', classesRolls or "", classesProb or "")
         for _, recipeId in ipairs(lathe.EmagLatheRecipes.emagStaticRecipes) do
            local recipe = getRecipeDetails(recipeId)
            if recipe then
                table.insert(recipes, {
                    result = recipe.result,
                    completetime = recipe.completetime,
                    materials = recipe.materials,
                    discipline = "Static",
                    tier = 0,
                    isEmag = true
                })
            end
        end
     end
     end


     -- Обработка emagDynamicRecipes
     result = result .. getTableOutput(nestedSelector.tableId, visited)
    if lathe.EmagLatheRecipes and lathe.EmagLatheRecipes.emagDynamicRecipes then
 
        for _, recipeId in ipairs(lathe.EmagLatheRecipes.emagDynamicRecipes) do
    if wrapped and (classesRolls or classesProb) then
            local recipe = getRecipeDetails(recipeId)
        result = result .. "}}"
            if recipe then
                local researchInfo = findInResearch(recipeId)
                if researchInfo then
                    table.insert(recipes, {
                        result = recipe.result,
                        completetime = recipe.completetime,
                        materials = recipe.materials,
                        discipline = researchInfo.discipline,
                        tier = researchInfo.tier,
                        researchName = researchInfo.name,
                        isEmag = true
                    })
                end
            end
        end
     end
     end


     sortRecipesByPriority(recipes)
     return result
end


    -- Таблица для перевода названий дисциплин
-- Формирование списка химии
     local disciplineMapping = {
getChemOutput = function(itemData, id)
        Arsenal = "Арсенал",
     local item = findDataById(itemData, id)
        Industrial = "Промышленность",
    if not item
        Experimental = "Экспериментальное",
      or not item.SolutionContainerManager
         CivilianServices = "Обслуживание персонала"
      or not item.SolutionContainerManager.solutions then
     }
         return ''
     end


     -- Таблица для цветов по уровням
     local foodSolution = item.SolutionContainerManager.solutions["food"]
     local tierColors = {
     if not foodSolution or not foodSolution.reagents then
         [1] = "#54d554",
         return ''
         [2] = "#ed9000",
    end
         [3] = "#d72a2a"
 
     }
    local result = ""
    for _, reagent in ipairs(foodSolution.reagents) do
         result = result .. string.format(
            '<li>[[Химия#chem_%s|%s]] (%d ед.)</li>',
            reagent.ReagentId, reagent.ReagentId, reagent.Quantity
         )
     end


     local materialUseMultiplier = (lathe.Lathe and lathe.Lathe.materialUseMultiplier) or 1
     return string.format('<ul class="1">%s</ul>', result)
    local timeMultiplier = (lathe.Lathe and lathe.Lathe.timeMultiplier) or 1
end


    for _, recipe in ipairs(recipes) do
-- Основная функция модуля
        local scaledTime = format_seconds_to_short_string(recipe.completetime * timeMultiplier)
function p.main(frame)
        out = out .. '{{Шаблон:Prototypes/Машина/Станок|product=' .. recipe.result
    local mode = frame.args[1]
        out = out .. '|complete-time=' .. scaledTime
    local id = frame.args[2]
        out = out .. '|materials='


        if type(recipe.materials) == "table" and next(recipe.materials) then
    if not id then return 'Не указан ID.' end
            for material, amount in pairs(recipe.materials) do
   
                local stackEntity = materialMapping[material] or material
    local itemDataIndex = itemData
                local scaledAmount = (amount * materialUseMultiplier) / 100
   
                out = out .. '<b>[[File:' .. stackEntity .. '.png|32x32px|link=]] ' .. scaledAmount .. ' {{#invoke:Entity Lookup|getname|' .. stackEntity .. '}}</b>'
    if not itemData then return 'Не удалось загрузить данные.' end
            end
        else
            out = out .. 'Нет данных о материалах'
        end


        -- Информация об исследовании
    if mode == 'framing' then
        if recipe.discipline ~= "Static" then
        local subMode = frame.args[2]
            local tierColor = tierColors[recipe.tier] or "#FFFFFF"
        local id = frame.args[3]
            local disciplineName = disciplineMapping[recipe.discipline] or "Неизвестная дисциплина"


            out = out .. '|info=<div style="font-weight:600;"><span style="margin:8px;">[[File:' .. recipe.discipline .. '.png|16x16px|link=]]</span> [[Руководство по исследованию и разработке|' .. disciplineName
        if not id then
             out = out .. ']], уровень: <span style="color: ' .. tierColor .. '">' .. recipe.tier .. '</span> </div>'
             return 'Не указан ID для режима framing.'
         end
         end


        -- Пометка при взломе EMAG
         if subMode == 'chem' then
         if recipe.isEmag then
             return frame:preprocess('{{СollapsibleMenu|color=#3e7c82|' .. getChemOutput(chemData, id) .. '}}')
             out = out .. '|mode-emag=1'
        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 = getFirstStartingItem(itemDataEntry)
    if not startingItem then return "" end
    return frame:preprocess('{{СollapsibleMenu|color=#71702a|' .. formatContent(startingItem) .. '}}')
        else
            return 'Неизвестный подрежим для framing: ' .. subMode
         end
         end
    elseif mode == 'chem' then
        return frame:preprocess(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 = getFirstStartingItem(itemDataEntry)
    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
         if recipe.discipline ~= "Static" then
             local containers = entity.EntityTableContainerFill.containers
             out = out .. '|mode-research=1'
            if containers.entity_storage and containers.entity_storage.rolls then
                return processRolls(containers.entity_storage.rolls)
            end
         end
         end


         out = out .. '}}'
         return ''
    else
        return 'Неизвестный режим: ' .. mode
     end
     end
    return mw.getCurrentFrame():preprocess(out)
end
end


return p
return p
Источник — https://station14.ru/wiki/Модуль:Песочница/Pok