Editing Module:Sandbox/User:The Gaffer/Modules/Infobox Recipe

local p = {}

function p._main(frame)
    local args = frame:getParent().args
    local mw = require('mw')
    local currency = require('Module:Currency').parse

     -- Define raw material costs based on Semantic MediaWiki queries
    local function getCost(queryResult)
        if type(queryResult) == "table" and queryResult[1]['Shop buy price'] then
            return tonumber(queryResult[1]['Shop buy price']) or 0
        end
        return 0
    end

    local rawmat1cost = args.rawmat1 and getCost(mw.smw.ask('[[:+]][[Sold item::' .. args.rawmat1 .. ']]|?Shop buy price|sort=Shop buy price|order=asc|limit=1')) or 0
    local rawmat2cost = args.rawmat2 and getCost(mw.smw.ask('[[:+]][[Sold item::' .. args.rawmat2 .. ']]|?Shop buy price|sort=Shop buy price|order=asc|limit=1')) or 0
    local rawmat3cost = args.rawmat3 and getCost(mw.smw.ask('[[:+]][[Sold item::' .. args.rawmat3 .. ']]|?Shop buy price|sort=Shop buy price|order=asc|limit=1')) or 0

    -- Calculate total cost and output cost
    local totalCost = (rawmat1cost * (tonumber(args.rawmat1qty) or 1)) +
                      (rawmat2cost * (tonumber(args.rawmat2qty) or 1)) +
                      (rawmat3cost * (tonumber(args.rawmat3qty) or 1))
    local output1cost = args.output1 and getCost(mw.smw.ask('[[:+]][[' .. args.output1 .. ']]|?Value|limit=1')) or 0

    -- Calculate output cost total and profit
    local output1costTotal = output1cost * (tonumber(args.output1qty) or 1)
    local profit = output1costTotal - totalCost

    -- Generate the output table
    local out = {}
    table.insert(out, '{| class="wikitable"')
    table.insert(out, '! colspan="4" | Requirements')
    table.insert(out, '|-')
    table.insert(out, '! colspan="2" | Facility')
    table.insert(out, '! colspan="2" | ' .. (args.facility and '[[' .. args.facility .. ']]' or 'Unknown'))
    table.insert(out, '|-')
    table.insert(out, '! colspan="2" | Profession')
    table.insert(out, '! Level')
    table.insert(out, '! XP')
    table.insert(out, '|-')
    table.insert(out, '| colspan="2" | ' .. (args.profession and '[[' .. args.profession .. ']]' or 'Unknown'))
    table.insert(out, '| style="text-align: center;" | ' .. (args.level or 'Unknown'))
    table.insert(out, '| style="text-align: center;" | ' .. (args.exp or 'Unknown'))
    table.insert(out, '|-')
    table.insert(out, '! colspan="2" | Raw Ingredient')
    table.insert(out, '! Quantity')
    table.insert(out, '! Cost')
    table.insert(out, '|-')

    -- Add each raw material to the table
    if args.rawmat1 then
        table.insert(out, '| [[File:' .. args.rawmat1 .. '.png|30px]]')
        table.insert(out, '| [[' .. args.rawmat1 .. ']]')
        table.insert(out, '| style="text-align: right;" | ' .. (args.rawmat1qty or 1))
        table.insert(out, '| style="text-align: right;" | ' .. currency(rawmat1cost))
        table.insert(out, '|-')
    end
    if args.rawmat2 then
        table.insert(out, '| [[File:' .. args.rawmat2 .. '.png|30px]]')
        table.insert(out, '| [[' .. args.rawmat2 .. ']]')
        table.insert(out, '| style="text-align: right;" | ' .. (args.rawmat2qty or 1))
        table.insert(out, '| style="text-align: right;" | ' .. currency(rawmat2cost))
        table.insert(out, '|-')
    end
    if args.rawmat3 then
        table.insert(out, '| [[File:' .. args.rawmat3 .. '.png|30px]]')
        table.insert(out, '| [[' .. args.rawmat3 .. ']]')
        table.insert(out, '| style="text-align: right;" | ' .. (args.rawmat3qty or 1))
        table.insert(out, '| style="text-align: right;" | ' .. currency(rawmat3cost))
        table.insert(out, '|-')
    end

    -- Add total raw cost
    table.insert(out, '! colspan="3" | Total Raw cost')
    table.insert(out, '| style="text-align: right;" | ' .. totalCost)
    table.insert(out, '|-')
    
    -- Add Intermediate steps if required
    -- Headers
    table.insert(out, '! colspan="2" | Intermediate Ingredient')
    table.insert(out, '! Quantity')
    table.insert(out, '! Facility')
    table.insert(out, '|-')
    
    -- Values
    if args.preparedmat1 then
    	table.insert(out, '| [[File:' .. args.preparedmat1 .. '.png|30px]] [[' .. args.preparedmat1 .. ']]')
        table.insert(out, '| style="text-align: right;" | ' .. (args.preparedmat1qty or 1))
        table.insert(out, '| style="text-align: right;" | [[File:' .. args.preparedmat1fac .. '.png|30px]] [[' .. args.preparedmat1fac .. ']]')
        table.insert(out, '|-')
    end

    -- Add output data and profit
    table.insert(out, '! colspan="2" | Output')
    table.insert(out, '! Quantity')
    table.insert(out, '! Value')
    table.insert(out, '|-')
    table.insert(out, '| ' .. (args.output1 and '[[' .. args.output1 .. ']]' or 'Unknown'))
    table.insert(out, '| style="text-align: right;" | ' .. (args.output1qty or 1))
    table.insert(out, '| style="text-align: right;" | ' .. output1costTotal)
    table.insert(out, '|-')
    table.insert(out, '! colspan="3" | Profit')
    table.insert(out, '| style="text-align: right;" | ' .. profit)
    table.insert(out, '|}')

    -- Add debug information to the output as a simple paragraph
    local debugMessage = 'Debug Info:<br>' ..
        'rawmat1query: ' .. (rawmat1query or 'nil') .. '<br>' ..
        'rawmat1cost: ' .. tostring(rawmat1cost) .. '<br>' ..
        'rawmat2query: ' .. (rawmat2query or 'nil') .. '<br>' ..
        'rawmat2cost: ' .. tostring(rawmat2cost) .. '<br>' ..
        'rawmat3query: ' .. (rawmat3query or 'nil') .. '<br>' ..
        'rawmat3cost: ' .. tostring(rawmat3cost) .. '<br>' ..
        'totalCost: ' .. tostring(totalCost) .. '<br>' ..
        'output1query: ' .. (output1query or 'nil') .. '<br>' ..
        'output1cost: ' .. tostring(output1cost) .. '<br>' ..
        'output1costTotal: ' .. tostring(output1costTotal) .. '<br>' ..
        'profit: ' .. tostring(profit)

    table.insert(out, debugMessage)

    return table.concat(out, '\n')
end

return p