Editing Module:Infobox Recipe

local p = {}

function p._main(frame)
    local args = frame:getParent().args
    local mw = require('mw')
    local currency = require('Module:Currency').parse
    local parse = require('Module:Param Parse')
    local yn = require('Module:Yesno')
    
    --If set to true, will check if any of the items passed as a rawmaterialparam are actually intermediate materials. Will then change the display order of the infobox.
    local showFullRecipe = yn(args.showFull or 'no', false)
    
    --Get each of the rawmatX parameters from params and store their values in a new table
	local argsMaterials = p._extractRawMaterials(args)
    mw.logObject(argsMaterials)
	--empty tables to hold materials
	local rawMaterials = {}
	local intermediateMaterials = {}
	
	--counter for total price of raw materials
    local rawMaterialCost = 0
	
	--get the value for the output product
	local output1Value = args.output1 and mw.smw.ask('[[:+]][[' .. args.output1 .. ']]|?Value|limit=1')
	local output1TotalValue = output1Value[1]['Value'] * (tonumber(args.output1qty) or 1)
	
	--Check if any of the raw mats provided are intermediate products, if they are, return their own raw materials
	if showFullRecipe then
		local trueRawMaterials = p._getTrueRawMaterials(argsMaterials)
		if next(trueRawMaterials) ~= nil then
			rawMaterials = trueRawMaterials
			intermediateMaterials = argsMaterials
		end
	else
			rawMaterials = argsMaterials
	end
	
	mw.logObject(rawMaterials)

	--Simple query to get the shop buy price for the provided material, if no buy price is available returns 0
	local function getBuyPrice(material)
		local queryResult = mw.smw.ask('[[:+]][[Sold item::' .. material['name'] .. ']]|?Shop buy price|sort=Shop buy price|order=asc|limit=1') or 0
		if type(queryResult) == "table" and queryResult[1]['Shop buy price'] then
            return tonumber(queryResult[1]['Shop buy price']) or 0
        end
        return 0
	end

	--If one of the raw materials provided as a param has been identified as an intermediate material then get the facility it is created at to display in the intermediate Ingredient row
	local function getFacility(material)
		local result = mw.smw.ask('[[:+]][[' .. material .. ']]|?Uses facility|limit=1') or 'unknown'
		local pageName = ''
		if result and result[1] and result[1]["Uses facility"] then
            local usesFac = result[1]["Uses facility"]
            pageName = usesFac:gsub("%[%[", ""):gsub("%]%]", ""):gsub("|.*", ""):gsub("^:", "")
        end
        return pageName
	end

	--Set SMW properties
	if args.rawmat1 then
        mw.smw.set({
            ["Uses item"] = args.rawmat1,
        })
    end
    if args.rawmat2 then
        mw.smw.set({
            ["Uses item"] = args.rawmat2,
        })
    end
    if args.rawmat3 then
        mw.smw.set({
            ["Uses item"] = args.rawmat3,
        })
    end
    if args.facility then
		mw.smw.set({
			["Uses facility"] = args.facility
		})
	end

	--Creates a row suitable for the raw materials section of the infobox. Quantity is optional, if no value is provided it will default to 1
	local function createRawMaterialRow(item, quantity)
		local materialBuyPrice = getBuyPrice(item)
		rawMaterialCost = rawMaterialCost + materialBuyPrice
		local row = {}
		table.insert(row, '| [[File:' .. item['name'] .. '.png|30px]]')
        table.insert(row, '| [[' .. item['name'] .. ']]')
        table.insert(row, '| style="text-align: right;" | ' .. (quantity or 1))
        table.insert(row, '| style="text-align: right;" | ' .. currency(materialBuyPrice))
        table.insert(row, '|-')
        return row
	end
	
	--Creates a row suitable for the intermediate materials section of the infobox. Quantity is optional, if no value is provided it will default to 1
	local function createIntermediateMaterialRow(item, quantity)
		local row = {}
		local facility = getFacility(item)
		table.insert(row, '| [[File:' .. item .. '.png|30px]]')
    	table.insert(row, '| [[' .. item .. ']]')
        table.insert(row, '| style="text-align: right;" | ' .. (quantity or 1))
        table.insert(row, '| style="text-align: right;" | [[File:' .. facility .. '.png|30px]] [[' .. facility .. ']]')
        table.insert(row, '|-')
        return row
    end
	
-- Recipe Table Head 
    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" style="text-align:center;" | ' .. (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, '|-')
    
-- Add raw materials
    -- Headers
    table.insert(out, '! colspan="2" | Raw Ingredient')
    table.insert(out, '! Quantity')
    table.insert(out, '! Cost')
    table.insert(out, '|-')

    -- Values
    for _, material in ipairs(rawMaterials) do
    	local row = createRawMaterialRow(material)
        for _, value in ipairs(row) do
            table.insert(out, value)
        end
    end

-- Add total raw cost
    table.insert(out, '! colspan="3" | Total Raw cost')
    table.insert(out, '| style="text-align: right;" | ' .. currency(rawMaterialCost))
    table.insert(out, '|-')
    
-- Add Intermediate steps if required
	-- Headers
    if next(intermediateMaterials) ~= nil then
    	-- Headers
    	table.insert(out, '! colspan="2" | Intermediate Ingredient')
    	table.insert(out, '! Quantity')
    	table.insert(out, '! Facility')
    	table.insert(out, '|-')
    
    -- Values
    	for _, material in ipairs(intermediateMaterials) do
    		local row = createIntermediateMaterialRow(material['name'])
        	for _, value in ipairs(row) do
            	table.insert(out, value)
        	end
    	end
    end

-- Add output data
    -- Headers
    table.insert(out, '! colspan="2" | Output')
    table.insert(out, '! Quantity')
    table.insert(out, '! Value')
    table.insert(out, '|-')
    
    -- Values
    table.insert(out, '| [[File:' .. args.output1 .. '.png|30px]]')
    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;" | ' .. currency(output1TotalValue))
    table.insert(out, '|-')
    
    -- Add profit data
    table.insert(out, '! colspan="3" | Profit')
    table.insert(out, '| style="text-align: right;" | ' .. currency(output1TotalValue - rawMaterialCost))
    table.insert(out, '|}')

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

function p._extractRawMaterials(args)
    local rawMaterials = {}

    for i = 1, 9 do
        local materialKey = "rawmat" .. i
        local quantityKey = "rawmat" .. i .. "qty"

        local materialValue = args[materialKey]
        if materialValue and materialValue ~= "" then
            local quantityValue = tonumber(args[quantityKey]) or 1
            table.insert(rawMaterials, { name = materialValue, quantity = quantityValue })
        end
    end
    return rawMaterials
end

function p._getTrueRawMaterials(argsMaterials)
    local queryResult = {}
    for _, item in pairs(argsMaterials) do
        if item["name"] then
            local result = mw.smw.ask('[[:+]][[' .. item["name"] .. ']]|?Uses item')
            
            if result and result[1] and result[1]["Uses item"] then
                local usesItems = result[1]["Uses item"]

                -- If usesItems is a table (multiple items), iterate and add each to queryResult
                if type(usesItems) == "table" then
                    for _, usedItem in ipairs(usesItems) do
                        local pageName = usedItem:gsub("%[%[", ""):gsub("%]%]", ""):gsub("|.*", ""):gsub("^:", "")
                        table.insert(queryResult, { ["name"] = pageName, ["quantity"] = 1 })
                    end
                else
                    -- If usesItems is a single item, add it directly
                    local pageName = usesItems:gsub("%[%[", ""):gsub("%]%]", ""):gsub("|.*", ""):gsub("^:", "")
                    table.insert(queryResult, { ["name"] = pageName, ["quantity"] = 1 })
                end
            end
        end
    end
    return queryResult
end

return p