Space Station 14 Вики

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

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


-- Функция для загрузки данных  
-- Функция для загрузки данных исследований из JSON-файла
local loadData = function(filePath)
local function loadResearchData()
     local page = mw.title.new(filePath)
     return mw.text.jsonDecode(mw.title.new("User:IanComradeBot/prototypes/research.json"):getContent())
    local content = page:getContent()
    return content and mw.text.jsonDecode(content) or nil
end
end


-- Функция processRolls для преобразования диапазона
-- Таблица для перевода названий дисциплин
local processRolls = function(rolls)
local disciplineMapping = {
    local result = {}
     Arsenal = "Арсенал",
     if rolls and rolls.range then
    Industrial = "Промышленность",
        -- Если указан range
     Experimental = "Экспериментальное",
        local min, max = rolls.range:match("(%d+),%s*(%d+)")
     CivilianServices = "Обслуживание персонала"
        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
        -- Если указано value
        result[#result + 1] = string.format('[%d]', rolls.value)
     else
        result[#result + 1] = 'Не указан параметр rolls.'
    end
    return table.concat(result)
end


-- Локальные функции
-- Таблица для цветов по уровням
local findDataById, formatContent, getContentsOutput, processNestedSelectors, getTableOutput, getContainedOutput, getChemOutput, handleGroupSelector, handleAllSelector, handleNestedSelector
local tierColors = {
    [1] = "#54d554",
    [2] = "#ed9000",
    [3] = "#d72a2a"
}


-- Функция для создания хэш-таблицы для быстрого поиска по ID
function p.main(frame)
local function buildIndex(itemData)
     local dataCache = loadResearchData()
    if not itemData then return {} end
    local input = frame.args[1] or ""
     local index = {}
     local manualIcon = frame.args[2] or ""
    for _, item in ipairs(itemData) do
    local out = ""
        index[item.id] = item
     end
    return index
end
 
-- Поиск данных по ID через индекс
findDataById = function(itemDataIndex, id)
    return itemDataIndex and itemDataIndex[id] or nil
end


-- Форматирование содержимого
     if input == "" then
formatContent = function(content)
         return '<div style="color:red;">Не указан идентификатор исследования или дисциплина.</div>'
     if not content.id then
         return "Ошибка: отсутствует id у элемента."
     end
     end


     local name = string.format('{{#invoke:Entity Lookup|getname|%s}}', content.id)
     -- Если первый параметр соответствует дисциплине, работаем в режиме категории
    local image = string.format('%s.png', content.id)
    if disciplineMapping[input] then
local amount = (content.amount and content.amount ~= 1) and string.format(" [%d]", content.amount) or ""
        local found = false
    local prob = ""
        for _, research in ipairs(dataCache) do
            if research.technology and research.technology.discipline == input then
                found = true
                local tech = research.technology
                local iconPath = tech.icon and tech.icon.sprite or nil  -- иконка берётся из tech.icon.sprite
                local tierColor = tierColors[tech.tier] or "#FFFFFF"
                local disciplineName = disciplineMapping[tech.discipline] or "Неизвестная дисциплина"


    if content.weight then
                -- Формирование строки необходимых исследований (prerequisites)
        content.prob = content.weight / 100
                local prerequisites = ""
    end
                if tech.technologyPrerequisites and #tech.technologyPrerequisites > 0 then
                    prerequisites = '<ul>'
                    for _, prerequisiteId in ipairs(tech.technologyPrerequisites) do
                        if prerequisiteId and prerequisiteId ~= "" then
                            local prerequisiteName = ""
                            for _, r in ipairs(dataCache) do
                                if r.technology and r.technology.id == prerequisiteId then
                                    prerequisiteName = r.technology.name
                                    break
                                end
                            end
                            if prerequisiteName ~= "" then
                                prerequisites = prerequisites .. '<li>{{#invoke:Ftl|main|translation|' .. prerequisiteName .. '}}</li>'
                            end
                        end
                    end
                    prerequisites = prerequisites .. '</ul>'
                end


    if content.prob then
                -- Формирование строки открываемых исследований (unlocks)
        prob = string.format(" <div>%s%%</div>", content.prob * 100 >= 1 and math.floor(content.prob * 100) or content.prob * 100)
                local unlocks = ""
    end
                if tech.recipeUnlocks and #tech.recipeUnlocks > 0 then
                    unlocks = '<ul>'
                    for _, recipe in ipairs(tech.recipeUnlocks) do
                        if recipe and recipe ~= "" then
                            unlocks = unlocks .. '<li>{{#invoke:Entity Lookup|createimagetooltip|Файл:'
                                .. recipe .. '.png|' .. recipe
                                .. '|Мета=32x32px,link=}} {{#invoke:Entity Lookup|getname|'
                                .. recipe .. '}}</li>'
                        end
                    end
                    unlocks = unlocks .. '</ul>'
                end


    return string.format(
                local templateArgs = {
        '{{LinkСard|SideStyle=1|background-color=#cbcbff0a|image=%s|name=%s%s%s {{#invoke:Prototypes/Хранилище/Предмет|main|framing|contained|%s}} }}',
                    id = tech.id,
        image, name, amount, prob, content.id
                    icon = iconPath,
    )
                    name = tech.name,
end
                    discipline = tech.discipline,
                    tier = tech.tier,
                    tierColor = tierColor,
                    disciplineName = disciplineName,
                    cost = tech.cost,
                    unlocks = unlocks
                }
                if prerequisites ~= "" then
                    templateArgs.prerequisites = prerequisites
                end


-- Получение содержимого через таблицу
                out = out .. frame:expandTemplate({
getContentsOutput = function(contents)
                    title = 'Prototypes/Механика/Исследование',
    local result = {}
                    args = templateArgs
    for _, content in ipairs(contents) do
                })
        result[#result + 1] = formatContent(content)
            end
    end
        end
    return table.concat(result)
        if not found then
end
            out = out .. '<div style="color:red;">Не найдено исследований для дисциплины "' .. input .. '".</div>'
 
-- Обработка вложенных таблиц
processNestedSelectors = function(children, visited)
visited = visited or {}
    if not children or #children == 0 then return "" end
 
    local results = {}
 
    for _, child in ipairs(children) do
        if child.id then
if not visited[child.id] then
    results[#results + 1] = formatContent(child)
end
        elseif child["!type"] == "NestedSelector" then
            results[#results + 1] = handleNestedSelector(child, true, visited)
        elseif child["!type"] == "GroupSelector" then
            results[#results + 1] = handleGroupSelector(child, visited)
         end
         end
    end
         return mw.getCurrentFrame():preprocess(out)
 
     else
    return table.concat(results)
         -- Режим поиска по ID исследования
end
        local tech = nil
 
         for _, research in ipairs(dataCache) do
-- Обработка таблиц
            if research.technology and research.technology.id == input then
getTableOutput = function(tableId, visited)
                 tech = research.technology
    visited = visited or {}
                 break
 
    if visited[tableId] then
         return ''
    end
 
    visited[tableId] = true
 
    local tableData = loadData('User:IanComradeBot/prototypes/table.json')
    local tableDataIndex = findDataById(buildIndex(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 then
                 result[#result + 1] = processNestedSelectors(containers.entity_storage.children, visited)
            end
            if containers.entity_storage.tableId then
                 result[#result + 1] = getTableOutput(containers.entity_storage.tableId, visited)
             end
             end
         end
         end


        -- Обработка storagebase
         if not tech then
         if containers.storagebase and containers.storagebase.tableId then
             out = out .. '<div style="color:red;">Исследование с ID "' .. input .. '" не найдено.</div>'
             result[#result + 1] = getTableOutput(containers.storagebase.tableId, visited)
        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)
    if not groupSelector.children then return '' end
    local result = {}
    local wrapperStart, wrapperEnd = "", ""
 
    -- Проверка для контейнера EntityTableContainerFill
    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
         else
             result[#result + 1] = "<div>Ошибка: отсутствует id у элемента.</div>"
             local tierColor = tierColors[tech.tier] or "#FFFFFF"
        end
            local disciplineName = disciplineMapping[tech.discipline] or "Неизвестная дисциплина"
    end
            -- Если иконка задана вручную, используем её, иначе берём из tech.icon.sprite
            local iconPath = manualIcon ~= "" and manualIcon or (tech.icon and tech.icon.sprite or nil)


    return wrapperStart .. table.concat(result) .. wrapperEnd
            local prerequisites = ""
end
            if tech.technologyPrerequisites and #tech.technologyPrerequisites > 0 then
                prerequisites = '<ul>'
                for _, prerequisiteId in ipairs(tech.technologyPrerequisites) do
                    if prerequisiteId and prerequisiteId ~= "" then
                        local prerequisiteName = ""
                        for _, research in ipairs(dataCache) do
                            if research.technology and research.technology.id == prerequisiteId then
                                prerequisiteName = research.technology.name
                                break
                            end
                        end
                        if prerequisiteName ~= "" then
                            prerequisites = prerequisites .. '<li>{{#invoke:Ftl|main|translation|' .. prerequisiteName .. '}}</li>'
                        end
                    end
                end
                prerequisites = prerequisites .. '</ul>'
            end


-- Обработка NestedSelector
            local unlocks = ""
handleNestedSelector = function(nestedSelector, wrapped, visited)
            if tech.recipeUnlocks and #tech.recipeUnlocks > 0 then
visited = visited or {}
                unlocks = '<ul>'
    if not nestedSelector.tableId then return '' end
                for _, recipe in ipairs(tech.recipeUnlocks) do
                    if recipe and recipe ~= "" then
                        unlocks = unlocks .. '<li>{{#invoke:Entity Lookup|createimagetooltip|Файл:'
                            .. recipe .. '.png|' .. recipe
                            .. '|Мета=32x32px,link=}} {{#invoke:Entity Lookup|getname|'
                            .. recipe .. '}}</li>'
                    end
                end
                unlocks = unlocks .. '</ul>'
            end


    local result = {}
            local templateArgs = {
    local classesRolls, classesProb
                id = tech.id,
 
                icon = iconPath,
    if wrapped then
                name = tech.name,
        if nestedSelector.rolls and nestedSelector.rolls.range then
                discipline = tech.discipline,
             local rollsResult = processRolls(nestedSelector.rolls)
                tier = tech.tier,
             if rollsResult and #rollsResult > 0 then
                tierColor = tierColor,
                 classesRolls = ', максимум может выпасть: ' .. rollsResult
                disciplineName = disciplineName,
                cost = tech.cost,
                unlocks = unlocks
             }
             if prerequisites ~= "" then
                 templateArgs.prerequisites = prerequisites
             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
    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, visited)
    if wrapped and (classesRolls or classesProb) then
        result[#result + 1] = "}}"
    end
    return table.concat(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 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 itemData = loadData('User:IanComradeBot/prototypes/fills/Item.json')
    local itemDataIndex = 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
            out = out .. frame:expandTemplate({
            return 'Не указан ID для режима framing.'
                title = 'Prototypes/Механика/Исследование',
                args = templateArgs
            })
         end
         end
 
         return mw.getCurrentFrame():preprocess(out)
         if subMode == 'chem' then
            return frame:preprocess('{{СollapsibleMenu|' .. getChemOutput(itemDataIndex, id) .. '}}')
        elseif subMode == 'contained' then
            return frame:preprocess('{{СollapsibleMenu|' .. getContainedOutput(itemDataIndex, id) .. '}}')
        else
            return 'Неизвестный подрежим для framing: ' .. subMode
        end
    elseif mode == 'chem' then
        return frame:preprocess(getChemOutput(itemDataIndex, id))
    elseif mode == 'contained' then
        return frame:preprocess(getContainedOutput(itemDataIndex, id))
    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
end
end


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