Модуль:Prototypes/Механика/Растение

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

local p = {}

local function kelvinToCelsius(k)
    return k - 273.15
end

local function findSeedById(id, data)
    for _, seed in ipairs(data) do
        if seed.id == id then
            return seed
        end
    end
    return nil
end

local function formatCharacteristics(seed)
    local parts = {
        ("[[Гидропоника#Потенция|Потенция]]: %s"):format(seed.potency or 0),
        ("[[Гидропоника#Урожайность|Урожайность]]: %s"):format(seed.yield or 0),
        ("[[Гидропоника#Продолжительность жизни|Время жизни]]: %s"):format(seed.lifespan or 0),
        ("[[Гидропоника#Скорость созревани|Созревания]]: %s"):format(seed.maturation or 0),
        ("[[Гидропоника#Продуктивность|Продуктивность]]: %s"):format(seed.production or 0),
        ("[[Гидропоника#Стадии роста|Стадии роста]]: %s"):format(seed.growthStages or 0),
    }
    return table.concat(parts, " <br> ")
end

local function formatConditions(seed)
    local tempC = kelvinToCelsius(seed.idealHeat or 273.15)
    local parts = {
        ("[[Гидропоника#|Вода]]: %s"):format(seed.waterConsumption or "-"),
        ("[[Гидропоника#|Удобрения]]: %s"):format(seed.nutrientConsumption or "-"),
        ("[[Гидропоника#|Температура]]: %.2f°C"):format(tempC),
        ("[[Гидропоника#|Освещения]]: %s"):format(seed.idealLight or "-"),
    }
    return table.concat(parts, " <br> ")
end

local function formatHarvestType(seed)
    return seed.harvestRepeat and "[[Гидропоника#Сбор урожая|" .. tostring(seed.harvestRepeat) .. "]]" or "-"
end

local function formatChemicals(seed)
    if not seed.chemicals then return "-" end
    local list = {}
    for chem, vals in pairs(seed.chemicals) do
        table.insert(list, ("* %s (Min: %s, Max: %s, Divisor: %s)"):format(
            chem, vals.Min or 0, vals.Max or 0, vals.PotencyDivisor or 1
        ))
    end
    return table.concat(list, '<br>')
end

local function formatMutations(seed, data)
    if not seed.mutationPrototypes then return "-" end
    local list = {}
    for _, mu in ipairs(seed.mutationPrototypes) do
        local target = findSeedById(mu, data)
        if target and target.productPrototypes then
            for _, prod in ipairs(target.productPrototypes) do
                table.insert(list, ("* {{Предмет|%s|link=}}"):format(prod))
            end
        end
    end
    return table.concat(list, '<br>')
end

local function generateHeader()
    return [[
{| id="" class="wikitable sortable mw-collapsible" style="width:100%; margin:0;"
! rowspan="2" style="width:10%;" | Плод
! rowspan="2" class="unsortable" style="width:5%;" | Семена
! rowspan="2" class="unsortable" style="width:5%;" | Растение
! colspan="3" class="unsortable" style="width:20%;" id="no-highlight" | Характеристики
! rowspan="2" class="unsortable" style="width:20%;" | Содержит вещества
! rowspan="2" style="width:20%;" | Мутации
|- 
! style="width:10%;" class="unsortable" | Рост
! style="width:10%;" class="unsortable" | Условия
! style="width:10%;" class="unsortable" | Тип сбора
]]
end

local function generateFooter()
    return "|}"
end

function p.table(frame)
    local data = mw.loadData("Модуль:IanComradeBot/prototypes/seeds.json/data")
    local rows = {}

    for _, seed in ipairs(data) do
        local fruitImg = ("{{Предмет|%s|size=64px|vertical=1|imageTooltip=1|link={{#invoke:Entity Lookup|getname|%s}}}}")
            :format(seed.productPrototypes[1], seed.packetPrototype)
        local seedImg  = ("{{Предмет|%s|size=64px|vertical=1|imageTooltip=1|l=|link={{#invoke:Entity Lookup|getname|%s}}}}")
            :format(seed.packetPrototype, seed.packetPrototype)
        local plantImg = ("{{Предмет|%s-harvest|size=64px|l=|link={{#invoke:Entity Lookup|getname|%s}}}}")
            :format(seed.packetPrototype, seed.packetPrototype)

        local colGrowth       = formatCharacteristics(seed)
        local colConditions   = formatConditions(seed)
        local colHarvest      = formatHarvestType(seed)
        local colChemicals    = formatChemicals(seed)
        local colMutations    = formatMutations(seed, data)

        local row = frame:preprocess(string.format(
            [[|-
			! %s 
			! %s 
			! %s 
			| %s 
			| %s 
			| %s 
			| %s 
			| %s ]],
            fruitImg, seedImg, plantImg,
            colGrowth, colConditions, colHarvest,
            colChemicals, colMutations
        ))
        table.insert(rows, row)
    end

    return generateHeader() .. table.concat(rows, '\n') .. '\n' .. generateFooter()
end

function p.main(frame)
    local args = frame.args
    local id = args[1]
    local mode  = mw.text.trim(args[2] or ""):lower()
    local data = mw.loadData("Модуль:IanComradeBot/prototypes/seeds.json/data")
    local seed = findSeedById(id, data)
    if not seed then return "" end

    if mode  == "growth" then
        return formatCharacteristics(seed)
    elseif mode  == "conditions" then
        return formatConditions(seed)
    elseif mode  == "harvest" then
        return formatHarvestType(seed)
    elseif mode  == "chemicals" then
        return formatChemicals(seed)
    elseif mode  == "mutations" then
        return formatMutations(seed, data)
    else
        return ""
    end
end

return p