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)
	--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 = 0
	mw.logObject(output1Value[1]['Value'])
	if output1Value[1]['Value'] ~= nil then
		output1TotalValue = output1Value[1]['Value'] * (tonumber(args.output1qty) or 1)
	else
		output1TotalValue = 0
	end

	--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

	--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