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

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

local p = {}

-- Функция для загрузки данных станков
local function loadLatheData()
    return mw.text.jsonDecode(mw.title.new("User:IanComradeBot/prototypes/lathe.json"):getContent())
end

-- Функция для загрузки данных рецептов
local function loadRecipeData()
    return mw.text.jsonDecode(mw.title.new("User:IanComradeBot/prototypes/lathe/recipes.json"):getContent())
end

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

-- Функция для форматирования времени
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
    local seconds_part = seconds > 0 and (seconds .. " сек.") or nil

    if minutes_part and seconds_part then
        return minutes_part .. " " .. seconds_part
    elseif seconds_part then
        return seconds_part
    elseif minutes_part then
        return minutes_part
    else
        return ''
    end
end

-- Функция для сортировки рецептов
local function sortRecipesByPriority(recipes)
    table.sort(recipes, function(a, b)
        -- Приоритеты дисциплин
        local priority = { Static = 1, Unknown = 2, EMAG = 4 }
        local aPriority = priority[a.discipline] or 3
        local bPriority = priority[b.discipline] or 3

        -- EMAG рецепты всегда в конце
        if a.isEmag ~= b.isEmag then
            return not a.isEmag
        end

        -- Если приоритеты совпадают
        if aPriority == bPriority then
            -- Сравнение по уровню исследования
            if a.tier == b.tier then
                -- Если уровни совпадают, сортируем по discipline
                return a.discipline < b.discipline
            end
            return a.tier < b.tier
        end

        -- Сортируем по приоритету discipline
        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 latheData = loadLatheData()
    local recipeData = loadRecipeData()
    local researchData = loadResearchData()

    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 output = '<h3>Рецепты станка: ' .. latheId .. '</h3>'

    local recipes = {}

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

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

    -- Обработка staticRecipes
    if lathe.Lathe.staticRecipes then
        for _, recipeId in ipairs(lathe.Lathe.staticRecipes) do
            local recipe = getRecipeDetails(recipeId)
            if recipe then
                table.insert(recipes, {
                    result = recipe.result,
                    completetime = recipe.completetime,
                    materials = recipe.materials,
                    discipline = "Static",
                    tier = 0
                })
            end
        end
    end

    -- Обработка dynamicRecipes
    if lathe.Lathe.dynamicRecipes then
        for _, recipeId in ipairs(lathe.Lathe.dynamicRecipes) 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
                    })
                else
                    table.insert(recipes, {
                        result = recipe.result,
                        completetime = recipe.completetime,
                        materials = recipe.materials,
                        discipline = "Unknown",
                        tier = 0
                    })
                end
            end
        end
    end

	-- Обработка emagStaticRecipes
	if lathe.EmagLatheRecipes and lathe.EmagLatheRecipes.emagStaticRecipes then
	    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

	-- Обработка emagDynamicRecipes 
	if lathe.EmagLatheRecipes and lathe.EmagLatheRecipes.emagDynamicRecipes then
	    for _, recipeId in ipairs(lathe.EmagLatheRecipes.emagDynamicRecipes) 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, -- Discipline из исследования
	                    tier = researchInfo.tier, -- Уровень из исследования
	                    researchName = researchInfo.name,
	                    isEmag = true -- Флаг для пометки EMAG
	                })
	            else
	                table.insert(recipes, {
	                    result = recipe.result,
	                    completetime = recipe.completetime,
	                    materials = recipe.materials,
	                    discipline = "Unknown", -- Если нет информации об исследовании
	                    tier = 0,
	                    isEmag = true -- Флаг для пометки EMAG
	                })
	            end
	        end
	    end
	end
	
	-- Вывод рецептов
	for _, recipe in ipairs(recipes) do
	    output = output .. '{{Шаблон:Prototypes/Машина/Станок|product=' .. recipe.result
	    output = output .. '|complete-time=' .. format_seconds_to_short_string(recipe.completetime)
	    output = output .. '|materials='
	    for material, amount in pairs(recipe.materials) do
	        output = output .. '<li>' .. material .. ': ' .. amount .. '</li>'
	    end

	    -- Информация об исследовании
	    if recipe.discipline ~= "Static" and recipe.discipline ~= "Unknown" then
	        output = output .. '|info=Исследование: ' .. recipe.discipline .. ' - ' .. recipe.researchName
	        output = output .. ' (Уровень: ' .. recipe.tier .. ')'
	    end
	    
	    -- Пометка EMAG
	    if recipe.isEmag then
	        output = output .. '<div>Пометка: EMAG</div>'
	    end
	end

	output = output .. '}}'

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

return p