Модуль: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

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

local function getRecipePackDetails(packId)
    for _, pack in ipairs(recipePackData) do
        if pack.id == packId then
            return pack
        end
    end
    return nil
end

-----------------------------------------------------------
-- Функция для сбора рецептов из станка
-----------------------------------------------------------
local function getLatheRecipes(lathe)
    local recipes = {}
    local chemMapping = {}
    for id, chem in pairs(chemData) do
        chemMapping[id] = chem.name
    end

    -- Вспомогательная функция для обработки одного рецепта
    local function processRecipe(recipeId, defaultDiscipline, isEmag)
        local recipe = getRecipeDetails(recipeId)
        if recipe then
            if recipe.result then
                local info = {
                    id = recipe.id,
                    result = recipe.result,
                    completetime = recipe.completetime,
                    materials = recipe.materials,
                    discipline = defaultDiscipline,
                    tier = 0,
                    isEmag = isEmag or false
                }
                if defaultDiscipline ~= "Static" then
                    local researchInfo = findInResearch(recipeId)
                    if researchInfo then
                        info.discipline = researchInfo.discipline
                        info.tier = researchInfo.tier
                        info.researchName = researchInfo.name
                    end
                end
                table.insert(recipes, info)
            elseif recipe.resultReagents then
                for reagent, amount in pairs(recipe.resultReagents) do
                    local reagentName = chemMapping[reagent] or reagent
                    table.insert(recipes, {
                        id = recipe.id,
                        result = reagentName .. "|amount=" .. amount .. "ед.|mode-chem=1",
                        completetime = recipe.completetime,
                        materials = recipe.materials,
                        discipline = defaultDiscipline,
                        tier = 0,
                        isEmag = isEmag or false
                    })
                    break
                end
            end
        end
    end

    -- Обработка рецептов для lathe.Lathe (старый и новый формат)
    if lathe.Lathe then
        if lathe.Lathe.staticRecipes then
            for _, recipeId in ipairs(lathe.Lathe.staticRecipes) do
                processRecipe(recipeId, "Static", false)
            end
        end
        if lathe.Lathe.dynamicRecipes then
            for _, recipeId in ipairs(lathe.Lathe.dynamicRecipes) do
                processRecipe(recipeId, "Dynamic", false)
            end
        end
        if lathe.Lathe.staticPacks then
            for _, packId in ipairs(lathe.Lathe.staticPacks) do
                local pack = getRecipePackDetails(packId)
                if pack and pack.recipes then
                    local packRecipes = type(pack.recipes) ~= "table" and { pack.recipes } or pack.recipes
                    for _, recipeId in ipairs(packRecipes) do
                        processRecipe(recipeId, "Static", false)
                    end
                end
            end
        end
        if lathe.Lathe.dynamicPacks then
            for _, packId in ipairs(lathe.Lathe.dynamicPacks) do
                local pack = getRecipePackDetails(packId)
                if pack and pack.recipes then
                    local packRecipes = type(pack.recipes) ~= "table" and { pack.recipes } or pack.recipes
                    for _, recipeId in ipairs(packRecipes) do
                        processRecipe(recipeId, "Dynamic", false)
                    end
                end
            end
        end
    end

    if lathe.EmagLatheRecipes then
        if lathe.EmagLatheRecipes.emagStaticRecipes then
            for _, recipeId in ipairs(lathe.EmagLatheRecipes.emagStaticRecipes) do
                processRecipe(recipeId, "Static", true)
            end
        end
        if lathe.EmagLatheRecipes.emagDynamicRecipes then
            for _, recipeId in ipairs(lathe.EmagLatheRecipes.emagDynamicRecipes) do
                processRecipe(recipeId, "Dynamic", true)
            end
        end
        if lathe.EmagLatheRecipes.emagStaticPacks then
            for _, packId in ipairs(lathe.EmagLatheRecipes.emagStaticPacks) do
                local pack = getRecipePackDetails(packId)
                if pack and pack.recipes then
                    local packRecipes = type(pack.recipes) ~= "table" and { pack.recipes } or pack.recipes
                    for _, recipeId in ipairs(packRecipes) do
                        processRecipe(recipeId, "Static", true)
                    end
                end
            end
        end
        if lathe.EmagLatheRecipes.emagDynamicPacks then
            for _, packId in ipairs(lathe.EmagLatheRecipes.emagDynamicPacks) do
                local pack = getRecipePackDetails(packId)
                if pack and pack.recipes then
                    local packRecipes = type(pack.recipes) ~= "table" and { pack.recipes } or pack.recipes
                    for _, recipeId in ipairs(packRecipes) do
                        processRecipe(recipeId, "Dynamic", true)
                    end
                end
            end
        end
    end

    sortRecipesByPriority(recipes)
    return recipes
end

-----------------------------------------------------------
-- Общие таблицы для форматирования вывода
-----------------------------------------------------------
local disciplineMapping = {
    Arsenal = "Арсенал",
    Industrial = "Промышленность",
    Experimental = "Экспериментальное",
    CivilianServices = "Обслуживание персонала"
}
local tierColors = {
    [1] = "#54d554",
    [2] = "#ed9000",
    [3] = "#d72a2a"
}
local materialMappingGlobal = {}
for _, material in ipairs(materialData) do
    if material.id then
        materialMappingGlobal[material.id] = material.stackEntity or material.id or material.name
    end
end

-----------------------------------------------------------
-- Функция для формирования строки рецепта
-----------------------------------------------------------
local function formatRecipe(recipe, timeMultiplier, materialUseMultiplier)
    local out = ""
    local ct = recipe.completetime or 0
    local scaledTime = ct * timeMultiplier
    out = out .. '{{Шаблон:Prototypes/Машина/Станок/base|product=' .. recipe.result
    out = out .. '|complete-time={{#invoke:Code/Формат/Время|main|seconds|' .. scaledTime .. '}}|materials='

    if recipe.materials then
        local materialEntries = {}
        for material, amount in pairs(recipe.materials) do
            local stackEntity = materialMappingGlobal[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

    if recipe.isEmag then
        out = out .. '|mode-emag=1'
    end

    if recipe.discipline ~= "Static" then
        out = out .. '|mode-research=1'
    end

    out = out .. '}}'
    return out
end

-----------------------------------------------------------
-- Функция для поиска и вывода рецептов по ID станка 
-----------------------------------------------------------
function p.lathe(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 materialUseMultiplier = (lathe.Lathe and lathe.Lathe.materialUseMultiplier) or 1
    local timeMultiplier = (lathe.Lathe and lathe.Lathe.timeMultiplier) or 1
    local recipes = getLatheRecipes(lathe)
    local out = ""
    for _, recipe in ipairs(recipes) do
        out = out .. formatRecipe(recipe, timeMultiplier, materialUseMultiplier)
    end

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

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

    local recipesOutput = ""
    local foundAny = false

    for _, lathe in ipairs(latheData) do
        local recipes = getLatheRecipes(lathe)
        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
            if recipe.id == itemId or recipe.result == itemId then
                foundAny = true
                local recipeStr = formatRecipe(recipe, timeMultiplier, materialUseMultiplier)
                recipeStr = recipeStr:gsub("}}$", "|method-container=" .. lathe.id .. "}}")
                recipesOutput = recipesOutput .. recipeStr
            end
        end
    end

    if not foundAny then
        return '<div style="color:red;">Рецепт для предмета с ID "' .. itemId .. '" не найден во всех станках.</div>'
    end

    local out = '<div class="grid-item-compressed">' .. recipesOutput .. '</div>'
    return mw.getCurrentFrame():preprocess(out)
end

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

    local recipesOutput = ""
    local foundAny = false

    for _, lathe in ipairs(latheData) do
        local recipes = getLatheRecipes(lathe)
        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
            if recipe.materials then
                for matId, _ in pairs(recipe.materials) do
                    if matId == materialId then
                        foundAny = true
                        local recipeStr = formatRecipe(recipe, timeMultiplier, materialUseMultiplier)
                        recipeStr = recipeStr:gsub("}}$", "|method-container=" .. lathe.id .. "}}")
                        recipesOutput = recipesOutput .. recipeStr
                        break
                    end
                end
            end
        end
    end

    if not foundAny then
        return '<div style="color:red;">Для материала с ID "' .. materialId .. '" рецептов не найдено во всех станках.</div>'
    end

    local out = '<div class="grid-item-compressed">' .. recipesOutput .. '</div>'
    return mw.getCurrentFrame():preprocess(out)
end

-----------------------------------------------------------
-- Функция для универсального вызова из шаблона
-----------------------------------------------------------
function p.main(frame)
    local arguments = require("Модуль:Arguments").getArgs(frame, { unwrap = true })
    local mode = arguments[1] or ""
    local id   = arguments[2] or ""
    
    -- Для обеспечения работы mw.getCurrentFrame(), переиспользуем исходный frame
    local newFrame = {
        args = { id },
        getCurrentFrame = frame.getCurrentFrame or function() return frame end
    }
    
    if mode == "lathe" then
        return p.lathe(newFrame)
    elseif mode == "item" then
        return p.item(newFrame)
    elseif mode == "material" then
        return p.material(newFrame)
    else
        return '<div style="color:red;">Неверный режим вызова: "' .. mode .. '". Используйте "lathe", "item" или "material".</div>'
    end
end

return p