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

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

local p = {}

-- Функция для загрузки данных исследований из JSON-файла
local function loadResearchData()
    return mw.text.jsonDecode(mw.title.new("User:IanComradeBot/Песочница.json"):getContent())
end

-- Функция для поиска исследований по дисциплине
local function findResearchByDiscipline(dataCache, discipline)
    local results = {}
    for _, research in ipairs(dataCache) do
        if research.technology and research.technology.discipline == discipline then
            table.insert(results, research.technology)
        end
    end
    return results
end

-- Таблица для перевода названий дисциплин
local disciplineMapping = {
    Arsenal = "Арсенал",
    Industrial = "Промышленность",
    Experimental = "Экспериментальное",
    CivilianServices = "Обслуживание персонала"
}

-- Таблица для цветов по уровням
local tierColors = {
    [1] = "#54d554",
    [2] = "#ed9000",
    [3] = "#d72a2a"
}

function p.main(frame)
    -- Подключение CSS
    local cssLink = frame:extensionTag('templatestyles', '', {
        src = 'Шаблон:Research/styles.css'
    })

    local dataCache = loadResearchData()
    local discipline = frame.args[1] or ""

    if discipline and discipline ~= "" then
        -- Инициализация строки вывода
        local out = cssLink .. '<div class="research-group">'

        -- Получаем список исследований по дисциплине
        local researches = findResearchByDiscipline(dataCache, discipline)

        if #researches == 0 then
            out = out .. '<div style="color:red;">Нет исследований для дисциплины "' .. discipline .. '"</div>'
        end

        for _, tech in ipairs(researches) do
            local disciplineName = disciplineMapping[tech.discipline] or "Неизвестная дисциплина"
            local tierColor = tierColors[tech.tier] or "#FFFFFF"
            local iconPath = tech.icon and tech.icon.sprite or nil

            -- Формирование строки prerequisites
            local prerequisitesStr = ""
            if tech.technologyPrerequisites and #tech.technologyPrerequisites > 0 then
                prerequisitesStr = '<ul>'
                for _, prerequisite in ipairs(tech.technologyPrerequisites) do
                    if prerequisite and prerequisite ~= "" then 
                        prerequisitesStr = prerequisitesStr .. '<li>{{#invoke:Entity Lookup|createimagetooltip|Файл:' 
                            .. prerequisite .. '.png|' .. prerequisite 
                            .. '|Мета=32x32px,link=}}' 
                            .. prerequisite .. '</li>'
                    end
                end
                prerequisitesStr = prerequisitesStr .. '</ul>'
            end

            -- Формирование строки unlocks
            local unlocksStr = ""
            if tech.recipeUnlocks and #tech.recipeUnlocks > 0 then
                unlocksStr = '<ul>'
                for _, recipe in ipairs(tech.recipeUnlocks) do
                    if recipe and recipe ~= "" then 
                        unlocksStr = unlocksStr .. '<li>{{#invoke:Entity Lookup|createimagetooltip|Файл:' 
                            .. recipe .. '.png|' .. recipe 
                            .. '|Мета=32x32px,link=}} {{#invoke:Entity Lookup|getname|' 
                            .. recipe .. '}}</li>'
                    end
                end
                unlocksStr = unlocksStr .. '</ul>'
            end

            -- Формируем вывод с использованием шаблона
            out = out .. mw.getCurrentFrame():preprocess(
		    	'{{Prototypes/Механика/Исследование' ..
		    	'|id=' .. tech.id ..
		        '|icon=' .. iconPath ..
		        '|name=' .. tech.name ..
		        '|discipline=' .. tech.discipline ..
		        '|tier=' .. tech.tier ..
		        '|tierColor=' .. tierColor ..
		        '|disciplineName=' .. disciplineName ..
		        '|cost=' .. tech.cost ..
		        '|prerequisites=' .. prerequisitesStr ..
		        '|unlocks=' .. unlocksStr ..
		        '}}'
		    )
        end

        out = mw.getCurrentFrame():preprocess(out)
        
        return out
    else
        return cssLink .. '<div style="color:red;">Дисциплина "' .. discipline .. '" не найдена.</div>'
    end
end

return p