Module:TechRecipe

Description

The recipe part of the tech pages.

If new research components are added to the game, UPLINK_REPLACEMENTS needs be extended to enable replacing the component with the related building. Otherwise it will just show the new component instead.

Usage

Example {{#invoke:TechRecipe|render|name=<techName>}}

Example

The following call:
{{#invoke:TechRecipe|render|name=Ultra-tech Framework}}

Results in:

Recipe

Fused Electrodes	IC Chip	Robotics Research		Ultra-tech Framework
1	1	3			1

Produced by

Producer	Production time
Uplink	500 seconds

Total Requirements

Fused Electrodes	IC Chip	Robotics Research		Ultra-tech Framework
20	20	60			20

---

local p = {}
local m = {} -- you can use this one for debugging without exposing functions, return this instead of p

---@type any
---@diagnostic disable-next-line: lowercase-global
mw = mw

-- Mirror tech cargo table
---@class TechCargo
---@field name string
---@field luaId string
---@field description string
---@field category string
---@field texture string
---@field requiredTech1 string
---@field requiredTech2 string
---@field requiredTech3 string
---@field progressCount integer
---@field ingredient1 string
---@field amount1 integer
---@field ingredient2 string
---@field amount2 integer
---@field ingredient3 string
---@field amount3 integer
---@field ingredient4 string
---@field amount4 integer
---@field producer1 string
---@field time1 number
---@field producer2 string
---@field time2 number
---@field producer3 string
---@field time3 number
---@field producer4 string
---@field time4 number
---@field recipeType string
---@field numProduced integer

---@param name string
function m.get_tech_cargo_data(name)
local fields = "name, luaId, description, category, texture, " ..
                   "requiredTech1, requiredTech2, requiredTech3, " ..
                   "progressCount, " ..
                   "ingredient1, amount1, " ..
                   "ingredient2, amount2, " ..
                   "ingredient3, amount3, " ..
                   "ingredient4, amount4, " ..
                   "producer1, time1, " ..
                   "producer2, time2, " ..
                   "producer3, time3, " ..
                   "producer4, time4, " ..
                   "recipeType, numProduced"

    local results = mw.ext.cargo.query(
        "tech",                    -- table name
        fields,                    -- fields list
        { where = string.format("name = '%s'", name) }
    )
    
    if not results or not results[1] then
        return nil
    end

    local row = results[1]

    local numericFields = {
        "progressCount",
        "amount1", "amount2", "amount3", "amount4",
        "time1", "time2", "time3", "time4",
        "numProduced"
    }
    for _, f in ipairs(numericFields) do
        row[f] = tonumber(row[f]) or 0
    end
    if row.numProduced == 0 then
        row.numProduced = 1
    end

    ---@cast row TechCargo
    return row
end

-- Inserts a table row cell for a single ingredient
---@param tech TechCargo The tech cargo object
---@param i integer Ingredient index (1–4)
---@param craft_count integer Amount multiplier
---@return nil 
m.ingredient_cell =  function(row, tech, i, craft_count)
    local ing = tech["ingredient" .. i]
    ---@type integer
    local amt = (tech["amount" .. i]) * craft_count

    if not ing or ing == "" then
        row:tag("td"):done()
        return
    end

    row:tag("td")
        :wikitext(string.format(
            "[[File:%s.png|64x64px|link=%s|alt=%s]] %i",
            ing, ing, ing, amt
        )):done()
end

---@param table table
---@return nil 
m.producers_header_row = function(table)
    local headerRow = table:tag("tr")
    headerRow:tag("th"):wikitext("Producer"):done()
    headerRow:tag("th"):wikitext("Production time"):done()
    headerRow:done()
end

---@param table table
---@param tech TechCargo
---@return nil 
m.producers_rows = function(table, tech)
    local local_has_producer = false
    for i = 1, 4 do
        ---@type string
        local producer = tech["producer"..i]
        if producer and producer ~= "" then
            local_has_producer = true
            local dataRow = table:tag("tr")
            dataRow:tag("td"):wikitext(string.format(
                "[[File:%s.png|64x64px|link=%s|alt=%s]][[%s]]",
                producer, producer, producer, producer
            )):done()
            dataRow:tag("td"):wikitext(string.format("%i seconds", tech["time"..i])):done()
            dataRow:done()
        end
    end

    if not local_has_producer then
         local dataRow = table:tag("tr")
        dataRow:tag("td"):wikitext("(None)"):done()
        dataRow:tag("td"):wikitext("/"):done()
        dataRow:done()
    end
end

-- Insert a 6 cells header
---@param table table
---@param tech TechCargo
---@return nil 
m.ingredients_header_row = function(table, tech)
    local headerRow = table:tag("tr")
    for i = 1, 4 do
        if tech["ingredient"..i] and tech["ingredient"..i] ~= "" then
            headerRow:tag("th"):wikitext(string.format("[[%s]]", tech["ingredient"..i])):done()
        end
    end
    -- Empty cell, for arrow under it
    headerRow:tag("th"):done()
    headerRow:tag("th"):wikitext(tech.name):done()
    headerRow:done()
end

-- Insert a 6 cells row
---@param table table
---@param tech TechCargo
---@param craft_count integer Multiply all counts by this
---@return nil 
m.ingredients_rows = function(table, tech, craft_count)
    local dataRow = table:tag("tr")
    for i = 1, 4 do
        m.ingredient_cell(dataRow, tech, i, craft_count)
    end
    dataRow:tag("td"):wikitext("[[File:Arrow_Right.png|link=Arrow Right|alt=Arrow Right|32x32px]]"):done()
    dataRow:tag("td"):wikitext(string.format("[[File:%s.png|64x64px|link=Technology/%s|alt=%s]] %i",
        tech.name, tech.name, tech.name, tech.numProduced * craft_count)):done()
    dataRow:done()
end

---@param tech TechCargo
---@return string
m.recipe_table = function(tech)
    local table = mw.html.create("table")
    table
        :addClass("wikitable")
        :tag("caption"):wikitext("Recipe"):done()

    m.ingredients_header_row(table, tech)
    m.ingredients_rows(table, tech, 1)
    return tostring(table)
end

---@param tech TechCargo
---@return string
m.produced_by_table = function(tech)
    local table = mw.html.create("table")
    table
        :addClass("wikitable")
        :tag("caption"):wikitext("Produced by"):done()

    m.producers_header_row(table)
    m.producers_rows(table, tech)

    return tostring(table)
end

---@param tech TechCargo
---@return string
m.total_requirements_table = function(tech)
    local table = mw.html.create("table")
    table
        :addClass("wikitable")
        :tag("caption"):wikitext("Total Requirements"):done()

    m.ingredients_header_row(table, tech)
    local required_count = math.ceil(tech.progressCount / tech.numProduced)
    m.ingredients_rows(table, tech, required_count) -- / could be round_up(tech.progressCount / tech.numProduced), but numProduced seems to be always 1
    return tostring(table)
end

--- Main function
---@param name string
---@return string
m.render = function(name)
    local tech = m.get_tech_cargo_data(name)
    if not tech then
        return "(Recipe not found)"
    end

    local result = 
          m.recipe_table(tech)
      ..  m.produced_by_table(tech)
      ..  m.total_requirements_table(tech)

    return result
end

p.render = function (frame)
    return m.render(frame.args.name)
end

return p

--[[
Debugging:

Return m instead of p if you want to debug those private functions

local frame = { args = { name = "Ultra-tech Framework" } }
(enter)
= p.render(frame)
(enter)

--]]