Модуль:Prototypes/Машина/Станок

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

local p = {}

-----------------------------------------------------------
-- Загрузка данных
-----------------------------------------------------------
local latheData      = mw.loadData("Модуль:IanComradeBot/prototypes/lathe.json/data")
local recipeData     = mw.loadData("Модуль:IanComradeBot/prototypes/lathe/recipes.json/data")
local recipePackData = mw.loadData("Модуль:IanComradeBot/prototypes/lathe/recipe pack.json/data")
local researchData   = mw.loadData("Модуль:IanComradeBot/prototypes/research.json/data")
local materialData   = mw.loadData("Модуль:IanComradeBot/prototypes/materials.json/data")
local chemData       = mw.loadData("Модуль:IanComradeBot/chem prototypes.json/data")

-----------------------------------------------------------
-- Функция для сортировки рецептов
-----------------------------------------------------------
local function sortRecipesByPriority(recipes)
    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
            return not a.isEmag
        end

        if aPriority == bPriority then
            if a.tier == b.tier then
                return a.discipline < b.discipline
            end
            return a.tier < b.tier
        end

        return aPriority < bPriority
    end)
end

-----------------------------------------------------------
-- Основная функция модуля
-----------------------------------------------------------
function p.main(frame)
    local latheId = frame.args[1] or ""
    if latheId == "" then
        return '<div style="color:red;">Не указан ID станка.</div>'
    end

    local lathe = nil
    for _, data in ipairs(latheData) do
        if data.id == latheId then
            lathe = data
            break
        end
    end

    if not lathe then
        return '<div style="color:red;">Станок с ID "' .. latheId .. '" не найден.</div>'
    end

    local materialMapping = {}
    for _, material in ipairs(materialData) do
        if material.id then
            materialMapping[material.id] = material.stackEntity or material.id or material.name
        end
    end

    local chemMapping = {}
    for id, chem in pairs(chemData) do
        chemMapping[id] = chem.name
    end

    local out = ''
    local recipes = {}

    local function getRecipeDetails(recipeId)
        for _, recipe in ipairs(recipeData) do
            if recipe.id == recipeId then
                return recipe
            end
        end
        return nil
    end

    local function findInResearch(recipeId)
        for _, research in ipairs(researchData) do
            if research and research.recipeUnlocks then
                for _, unlock in ipairs(research.recipeUnlocks) do
                    if unlock == recipeId then
                        return {
                            name = research.name,
                            tier = research.tier,
                            discipline = research.discipline
                        }
                    end
                end
            end
        end
        return nil
    end

    -- Функция для обработки группы рецептов по packId
    local function processPack(packId, disciplineOverride, isEmag)
        local pack = nil
        for _, packData in ipairs(recipePackData) do
            if packData.id == packId then
                pack = packData
                break
            end
        end
        if not pack or not pack.recipes then
            out = out .. '<div style="color:red;">Ошибка: Группа рецептов с ID "' .. packId .. '" не найдена или не содержит рецептов.</div>'
            return
        end
        for _, recipeId in ipairs(pack.recipes) do
            local recipe = getRecipeDetails(recipeId)
            if recipe and recipe.result then
                table.insert(recipes, {
                    result = recipe.result,
                    completetime = recipe.completetime,
                    materials = recipe.materials,
                    discipline = disciplineOverride or "Static",
                    tier = 0,
                    isEmag = isEmag
                })
            elseif recipe and recipe.resultReagents then
                for reagent, amount in pairs(recipe.resultReagents) do
                    local reagentName = chemMapping[reagent] or reagent
                    table.insert(recipes, {
                        result = reagentName .. "|amount=" .. amount .. "ед.|mode-chem=1",
                        completetime = recipe.completetime,
                        materials = recipe.materials,
                        discipline = disciplineOverride or "Static",
                        tier = 0,
                        isEmag = isEmag
                    })
                    break 
                end
            else
                out = out .. '<div style="color:red;">Ошибка: Рецепт с ID "' .. recipeId .. '" не найден или поля result/resultReagents отсутствуют.</div>'
            end
        end
    end

    -- Обработка staticPacks
    if lathe.Lathe and lathe.Lathe.staticPacks then
        for _, packId in ipairs(lathe.Lathe.staticPacks) do
            processPack(packId, "Static", false)
        end
    end

    -- Обработка dynamicPacks
    if lathe.Lathe and lathe.Lathe.dynamicPacks then
        for _, packId in ipairs(lathe.Lathe.dynamicPacks) do
            local pack = nil
            for _, packData in ipairs(recipePackData) do
                if packData.id == packId then
                    pack = packData
                    break
                end
            end
            if not pack or not pack.recipes then
                out = out .. '<div style="color:red;">Ошибка: Группа рецептов с ID "' .. packId .. '" не найдена или не содержит рецептов.</div>'
            else
                for _, recipeId in ipairs(pack.recipes) do
                    local recipe = getRecipeDetails(recipeId)
                    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
                            })
                        end
                    end
                end
            end
        end
    end

    -- Обработка emagStaticPacks
    if lathe.EmagLatheRecipes and lathe.EmagLatheRecipes.emagStaticPacks then
        for _, packId in ipairs(lathe.EmagLatheRecipes.emagStaticPacks) do
            processPack(packId, "Static", true)
        end
    end

    -- Обработка emagDynamicPacks 
    if lathe.EmagLatheRecipes and lathe.EmagLatheRecipes.emagDynamicPacks then
        for _, packId in ipairs(lathe.EmagLatheRecipes.emagDynamicPacks) do
            local pack = nil
            for _, packData in ipairs(recipePackData) do
                if packData.id == packId then
                    pack = packData
                    break
                end
            end
            if not pack or not pack.recipes then
                out = out .. '<div style="color:red;">Ошибка: Группа рецептов с ID "' .. packId .. '" не найдена или не содержит рецептов.</div>'
            else
                for _, recipeId in ipairs(pack.recipes) do
                    local recipe = getRecipeDetails(recipeId)
                    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
    end

    sortRecipesByPriority(recipes)

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

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

    local materialUseMultiplier = (lathe.Lathe and lathe.Lathe.materialUseMultiplier) or 1
    local timeMultiplier = (lathe.Lathe and lathe.Lathe.timeMultiplier) or 1

    for _, recipe in ipairs(recipes) do
        local scaledTime = recipe.completetime * timeMultiplier
        out = out .. '{{Шаблон:Prototypes/Машина/Станок|product=' .. recipe.result
        out = out .. '|complete-time={{#invoke:Code/Формат/Время|main|seconds|' .. scaledTime
        out = out .. '}}|materials='

		if recipe.materials then
		    local materialEntries = {}
		    for material, amount in pairs(recipe.materials) do
		        local stackEntity = materialMapping[material] or material
		        local scaledAmount = (amount * materialUseMultiplier) / 100
		        table.insert(materialEntries, string.format('<b>[[File:%s.png|32x32px|link=]] %g {{#invoke:Entity Lookup|getname|%s}}</b>', stackEntity, scaledAmount, stackEntity))
		    end
		    out = out .. table.concat(materialEntries)
		else
		    out = out .. 'Нет данных о материалах'
		end

        -- Информация об исследовании
        if recipe.discipline ~= "Static" then
            local tierColor = tierColors[recipe.tier] or "#FFFFFF"
            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
            out = out .. ']], уровень: <span style="color: ' .. tierColor .. '">' .. recipe.tier .. '</span> </div>'
        end

        -- Пометка при взломе EMAG
        if recipe.isEmag then
            out = out .. '|mode-emag=1'
        end

        -- Пометка для исследуемой технологии
        if recipe.discipline ~= "Static" then
            out = out .. '|mode-research=1'
        end

        out = out .. '}}'
    end

    return mw.getCurrentFrame():preprocess(out)
end

return p