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:

---

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",
        fields,
        { 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.researched_with_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 table table
---@param tech TechCargo
---@return nil
m.researched_with_rows = function(table, tech)
    -- Research are different than usual recipes in that they only have one shared recipe, only the speed changes
    -- We list the speed for each exiting uplinks component
    local cargo_getters  = require("Module:CargoGetters")
    for _, uplink in ipairs(cargo_getters.uplinks()) do
        local name = uplink.name
        local production_time = uplink.uplinkRate * tech.time1

        local text = string.format(
            "[[File:%s.png|64x64px|link=%s|alt=%s]][[%s]]",
            name, name, name, name
        ) 
        local dataRow = table:tag("tr")
        dataRow:tag("td"):wikitext(text):done()
        if production_time ~= 0 then
            dataRow:tag("td"):wikitext(string.format("%i seconds", production_time)):done()
        else
            dataRow:tag("td"):done()
        end
        dataRow:done()
    end

end

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

    m.researched_with_header_row(table)
    m.researched_with_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.researched_with_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)

--]]