Module:CraftingRecipe
From dos2.wiki
More actions
Documentation for this module may be created at Module:CraftingRecipe/doc
local p = {}
local cargo = mw.ext.cargo
local html = mw.html
local recipeData = mw.loadData('Module:CraftingRecipe/RecipeData')
local categoryIcons = mw.loadData('Module:CraftingRecipe/CategoryIcons')
local stationIcons = mw.loadData('Module:CraftingRecipe/StationIcons')
local propertyIcons = mw.loadData('Module:CraftingRecipe/PropertyIcons')
local function lookupItem(pageName)
if not pageName or pageName == "" or not cargo then return nil end
local tables = "Items"
local fields = "name, icon, page_name"
local where = string.format("page_name = '%s'", pageName:gsub("'", "''"))
local results = cargo.query(tables, fields, { where = where, limit = 1 })
if results and #results > 0 then
return results[1]
end
return nil
end
local function resolveData(args, prefix, index)
local suffix = index and tostring(index) or ""
local keyId = prefix .. suffix
local id = args[keyId]
if not id then return nil end
local data = {
id = id,
name = args[keyId .. "_name"],
icon = args[keyId .. "_icon"],
link = nil,
type = args[keyId .. "_type"] or "Object",
transform = args[keyId .. "_transform"] or "Consume",
amount = args[keyId .. "_amount"]
}
local searchName = data.name or id
if data.type ~= "Category" and data.type ~= "Property" and prefix ~= "station" then
local cargoData = lookupItem(searchName)
if cargoData then
data.name = data.name or cargoData.name
data.icon = data.icon or cargoData.icon
data.link = cargoData.page_name
end
end
data.name = data.name or id
if not data.icon then
if prefix == "station" then
local foundIcon = stationIcons[data.id]
if foundIcon then
data.icon = foundIcon .. "_Icon.webp"
else
data.icon = "Icon_World_Gears.png"
end
elseif data.type == "Category" then
local foundIcon = categoryIcons[data.id]
if foundIcon then
data.icon = foundIcon .. "_Icon.webp"
end
elseif data.type == "Property" then
local foundIcon = propertyIcons[data.id]
if foundIcon then
data.icon = foundIcon .. "_Icon.webp"
end
end
end
if not data.icon then
data.icon = "Inv_Unknown_A.webp"
end
if not data.link then
if data.type == "Category" then
data.link = data.name .. " (Item Category)"
elseif data.type == "Property" then
data.link = data.name .. " (Item Property)"
else
data.link = data.name
end
end
return data
end
local function getRecipeArgs(recipeId)
local recipe = recipeData[recipeId]
if not recipe then return nil end
local args = {}
if recipe.station then
args.station = recipe.station
end
if recipe.ingredients then
for i, v in ipairs(recipe.ingredients) do
local k = "mat" .. i
args[k] = v.id
args[k .. "_type"] = v.type
args[k .. "_name"] = v.name
args[k .. "_transform"] = v.transform
end
end
if recipe.results then
for i, v in ipairs(recipe.results) do
local k = "res" .. i
args[k] = v.id
args[k .. "_name"] = v.name
end
end
return args
end
function p.lookup(frame)
local recipeId = frame.args.recipe
if not recipeId then return "" end
local args = getRecipeArgs(recipeId)
if not args then return "" end
local mockFrame = {
args = args,
expandTemplate = function(self, t) return frame:expandTemplate(t) end
}
return p.row(mockFrame)
end
function p.table(frame)
local recipes = frame.args.recipes
if not recipes then return "" end
local output = frame:expandTemplate{ title = 'CraftingTable/Header' }
for recipeId in mw.text.gsplit(recipes, ",") do
recipeId = mw.text.trim(recipeId)
if recipeId ~= "" then
local args = getRecipeArgs(recipeId)
if args then
local mockFrame = {
args = args,
expandTemplate = function(self, t) return frame:expandTemplate(t) end
}
output = output .. "\n" .. p.row(mockFrame)
end
end
end
output = output .. "\n" .. frame:expandTemplate{ title = 'CraftingTable/Footer' }
return output
end
function p.row(frame)
local args = frame.args
local tr = html.create('tr')
local res1 = resolveData(args, "res", 1) or resolveData(args, "result", nil)
local productTd = tr:tag('td')
if res1 then
local templateArgs = {
[1] = res1.link,
name = res1.link,
override_icon = res1.icon
}
productTd:wikitext(frame:expandTemplate{ title = 'MdItemIcon', args = templateArgs })
if res1.amount and tonumber(res1.amount) > 1 then
productTd:wikitext(" ×" .. res1.amount)
end
end
local inputs = {}
local station = resolveData(args, "station", nil)
if station then
station.isStation = true
table.insert(inputs, station)
end
for i = 1, 5 do
local item = resolveData(args, "mat", i)
if item then table.insert(inputs, item) end
end
local maxColumns = 5
local filledColumns = 0
for _, item in ipairs(inputs) do
if filledColumns < maxColumns then
local td = tr:tag('td')
if item.isStation then
td:addClass('recipe-status-station')
elseif item.transform == "Keep" or item.transform == "None" then
td:addClass('recipe-status-keep')
elseif item.transform == "Boost" then
td:addClass('recipe-status-boost')
end
local templateArgs = {
[1] = item.link,
name = item.link,
override_icon = item.icon
}
td:wikitext(frame:expandTemplate{ title = 'MdItemIcon', args = templateArgs })
if item.amount and tonumber(item.amount) > 1 then
td:tag('small'):wikitext(" (" .. item.amount .. ")")
end
filledColumns = filledColumns + 1
end
end
if filledColumns < maxColumns then
tr:tag('td')
:attr('colspan', maxColumns - filledColumns)
:css('background-color', 'transparent')
end
return tostring(tr)
end
function p.resolveRow(frame)
local args = {}
for k, v in pairs(frame.args) do
args[k] = v
end
if args[1] then
local recipeArgs = getRecipeArgs(args[1])
if recipeArgs then
for k, v in pairs(recipeArgs) do
if args[k] == nil then
args[k] = v
end
end
end
end
local tr = html.create('tr')
local res1 = resolveData(args, "res", 1) or resolveData(args, "result", nil)
local productTd = tr:tag('td')
if res1 then
local templateArgs = {
[1] = res1.link,
name = res1.link,
override_icon = res1.icon
}
productTd:wikitext(frame:expandTemplate{ title = 'MdItemIcon', args = templateArgs })
if res1.amount and tonumber(res1.amount) > 1 then
productTd:wikitext(" ×" .. res1.amount)
end
end
local inputs = {}
local station = resolveData(args, "station", nil)
if station then
station.isStation = true
table.insert(inputs, station)
end
for i = 1, 5 do
local item = resolveData(args, "mat", i)
if item then table.insert(inputs, item) end
end
local maxColumns = 5
local filledColumns = 0
for _, item in ipairs(inputs) do
if filledColumns < maxColumns then
local td = tr:tag('td')
if item.isStation then
td:addClass('recipe-status-station')
elseif item.transform == "Keep" or item.transform == "None" then
td:addClass('recipe-status-keep')
elseif item.transform == "Boost" then
td:addClass('recipe-status-boost')
end
local templateArgs = {
[1] = item.link,
name = item.link,
override_icon = item.icon
}
td:wikitext(frame:expandTemplate{ title = 'MdItemIcon', args = templateArgs })
if item.amount and tonumber(item.amount) > 1 then
td:tag('small'):wikitext(" (" .. item.amount .. ")")
end
filledColumns = filledColumns + 1
end
end
if filledColumns < maxColumns then
tr:tag('td')
:attr('colspan', maxColumns - filledColumns)
:css('background-color', 'transparent')
end
return tostring(tr)
end
return p