Editing Module:CarpenterPassiveList

require('strict')
require('Module:Mw.html extension')
local recipe = require('Module:Infobox Recipe') -- to make use of its extensive  material searching function
local param = require( 'Module:Paramtest' )
local currency = require('Module:Currency')
local lang = mw.getContentLanguage()

local p = {}

-- non dynamic module, no inputs
function p.main()
	-- returns only directly needed parameter needed for the row,
	-- other parameters are determined by subqueries of chained pages
	local query = {
		'[[Uses facility::Workbench]] OR [[Variant of::Stools||Tables||Chairs||Shelves||Trunks||Rocking Chairs]]',
		'?Uses facility #- = facility',
		'?Profession Level A = lvl',
		'? #- = name',
		'?Recipe JSON = recipeJSON',
		'?Activity XP = XP',
		'?Activity duration = duration',
		'?Value = sell',
		'?Activity coins = sell',
		'sort = Profession Level A',
		'limit = 500'
	}
	local results = mw.smw.ask(query)
	
	-- criterion for if an entry is a recipe, or a passive activity
	for _, item in ipairs(results) do
		item.passiveActivity = not(item.facility=='Workbench') 
	end
	
	results = p.formatResults(results)

	return p.displayTable(results)
	
	--for debugging
	--return '<pre>'..mw.text.jsonEncode(results, mw.text.JSON_PRETTY)..'</pre>'

end

-- makes the html for the cells containing currency directly
-- Replaces nil with an "unknown" cell
local function currency_cell(amount)
	if not amount then
		return mw.html.create('td')
			:addClass('table-bg-gray')
			:css{ ['text-align'] = 'center' }
			:attr{ colspan = '10' }
			:wikitext("''unknown''")
		:done()
	end
	return currency._cell(amount, { html = 'yes' })
end

-- do calculations and determine strings to go in cells
function p.formatResults(results)

	-- iterate through products
	for _, item in ipairs(results) do
		
		-- recipe in a workable format
		if item.passiveActivity then
			
			-- if it is a passive activity that is not a recipe
			-- some fields are easy to work out
    		item.outputQuantity = 1
    		item.materials = {}
    		item.intermediates = {}
    		item.buy = 0
    		item.sell = item.coins

    	else
    		
			local unpackJSON = mw.text.jsonDecode(item.recipeJSON)
			item.outputQuantity = unpackJSON.output[1].quantity
	
			-- call the module:infobox recipe to extensively search for 
			-- all raw materials (to add to prices) 
			-- all intermediate materials (for intermediate XP and duration)
			local Materials = recipe._getTrueRawMaterials(unpackJSON.materials)
			if next(Materials) ~= nil then
				item.materials = Materials.rawMaterials
				item.intermediates = Materials.intermediateMaterials
			end
			
			-- iterate through materials, adding buy price to running total (individuals not needed)
			item.buy = 0
			for _, material in ipairs(item.materials) do
			
				--shamelessley lifted from Module:Products
				local shopPriceQuery = '[[:+]][[Sold item::' .. material.name .. ']]|?Shop buy price|mainlabel=' .. material.name
				local shopPriceResult = mw.smw.ask(shopPriceQuery) or {}
				local shopPrice = 0		
				
				if shopPriceResult[1] and shopPriceResult[1]["Shop buy price"] then
					shopPrice = tonumber(shopPriceResult[1]["Shop buy price"]) or 0
				else
					item.buy = nil
				end
	
				item.buy = item.buy and item.buy + shopPrice * material.quantity
				
			end
			
			-- iterate through intermediaries to add to XP and duration parameters
			for _, intermediate in ipairs(item.intermediates) do
				
				-- look up activity XP and add it to running total
				local XPQuery = '[[Recipe output::' .. intermediate.name .. ']]|?Activity XP|mainlabel=' .. intermediate.name
				local XPResult = mw.smw.ask(XPQuery) or {}
				local XPIncrease = 0
				if XPResult[1] and XPResult[1]["Activity XP"] then
					XPIncrease =  tonumber(XPResult[1]["Activity XP"]) or 0
				else
					XPIncrease = nil
				end
				
				-- need to look up both the amount of the item used in this recipe and created in its own recipe to know how much of this xp to use
				local QTYused = intermediate.quantity		
	
				local QTYQuery = '[[Recipe output::' .. intermediate.name .. ']]|?Recipe JSON|mainlabel=' .. intermediate.name
				local QTYResult = mw.smw.ask(QTYQuery) or {}
				local QTYmade = 0
				if QTYResult[1] and QTYResult[1]["Recipe JSON"] then
					QTYmade =  mw.text.jsonDecode(QTYResult[1]["Recipe JSON"]) or {}
				end
				QTYmade = QTYmade.output[1].quantity
				
				item.XP = item.XP and XPIncrease and item.XP + XPIncrease * QTYused / QTYmade;
				
				-- look up activity duration and add it to running total
				local durationQuery = '[[Recipe output::' .. intermediate.name .. ']]|?Activity duration|mainlabel=' .. intermediate.name
				local durationResult = mw.smw.ask(durationQuery) or {}
				local durationIncrease = 0
				if durationResult[1] and durationResult[1]["Activity duration"] then
					durationIncrease =  tonumber(durationResult[1]["Activity duration"]) or 0
				else
					durationIncrease = nil
				end
				item.duration = item.duration and durationIncrease and item.duration + durationIncrease * intermediate.quantity
				
			end
			
		end
		
		-- direct values
		item.sell = item.sell and item.outputQuantity and item.sell * item.outputQuantity
		item.profit = item.sell and item.buy and item.sell - item.buy
		item.profitPerXP = item.profit and item.XP and math.floor(item.profit / item.XP * 100) / 100

		-- products made in 200 lots of 0.005, no downtime for passive activities
		local batchSize = 200
		local downtime = 0
		item.duration = item.duration and item.duration + downtime/batchSize
		item.productPerHour = item.duration and 1 / item.duration * 3600

		-- properties per hour
		item.XPPerHour = item.XP and item.productPerHour and math.floor(item.XP * item.productPerHour)
		item.profitPerHour = item.profit and item.productPerHour and math.floor(item.profit * item.productPerHour)
	end

	return results
end

-- make the table
function p.displayTable(results)
	local out = mw.html.create('table')
		:addClass('wikitable sortable')
		:tag('tr')
			:tag('th')
				:wikitext('[[File:Carpenter small icon.png|15px]]  Level')
			:done()
			:tag('th')
				:attr{ colspan = '3' }
				:wikitext('Product')
			:done()
			:tag('th')
				:wikitext('Materials')
			:done()
			:tag('th')
				:attr{ colspan = '10' }
				:wikitext('Buy Value')
			:done()
			:tag('th')
				:attr{ colspan = '10' }
				:wikitext('Sell Value')
			:done()
			:tag('th')
				:attr{ colspan = '10' }
				:wikitext('Profit')
			:done()
			:tag('th')
				:attr{ colspan = '10' }
				:wikitext('Profit/hr')
			:done()
			:tag('th')
				:wikitext('XP')
			:done()
			:tag('th')
				:wikitext('XP/hr')
			:done()
			:tag('th')
				:attr{ colspan = '10' }
				:wikitext('Coins/XP')
			:done()
		:done()

	local unknown_value_cell = mw.html.create('td')
		:addClass('table-bg-gray')
		:css{ ['text-align'] = 'center' }
		:wikitext("''unknown''")

	for i, item in ipairs(results) do
		local row = out:tag('tr')
			:IF(item.lvl)
				:tag('td')
					:css{ ['text-align'] = 'center' }
					:wikitext(item.lvl)
				:done()
			:ELSE()
				:node(unknown_value_cell)
			:END()

			:tag('td')
				:css{ ['border-right'] = '0', ['padding-right'] = '0', ['text-align'] = 'right' }
				:attr{ ['data-sort-value'] = item.name }
				:wikitext(item.outputQuantity .. ' &times;')
			:done()
			:tag('td')
				:addClass('plinkt-image no-border')
				:css{ ['border-left'] = '0', ['padding-left'] = '0' }
				:wikitext('[[File:' .. item.name .. '.png|link=' .. item.name .. '|30px]]')
			:done()
			:tag('td')
				:addClass('plinkt-link no-border')
				:wikitext('[[' .. item.name .. ']]')
			:done()
			
		:IF(not(item.passiveActivity))

			local materialCell = row:tag('td')
	
			for i, _ in ipairs(item.materials) do
				materialCell:wikitext(item.materials[i].quantity .. '&times; [[File:' .. item.materials[i].name .. '.png|18px|link=' .. item.materials[i].name .. ']]' .. '[[' .. item.materials[i].name .. ']]<br>')
			end
			
			row
				:node(currency_cell(item.buy))
				:node(currency_cell(item.sell))
				
		:ELSE()
			:tag('td')
				:addClass('table-na')
				:attr{ colspan = '21' }
				:wikitext('N/A')
			:done()
		:END()	
			
		
			:node(currency_cell(item.profit))
			:node(currency_cell(item.profitPerHour))

			:IF(item.XP)
				:tag('td')
					:wikitext(item.XP and lang:formatNum(tonumber(item.XP)))
				:done()
			:ELSE()
				:node(unknown_value_cell)
			:END()

			:IF(item.XPPerHour)
				:tag('td')
					:wikitext(item.XPPerHour and lang:formatNum(tonumber(item.XPPerHour)))
				:done()
			:ELSE()
				:node(unknown_value_cell)
			:END()

			:node(currency_cell(item.profitPerXP))

		:done()
	end

	return out
end

return p