Editing Module:PotionList

require('Module:Mw.html extension')
local param = require( 'Module:Paramtest' )
local currency = require('Module:Currency')
local lang = mw.getContentLanguage()
require("Module:Mw.html extension")


local p = {}

-- non dynamic module, no inputs
function p.main()

	-- returns almost every parameter needed for the row, except buy values for reagents
	local query = {
		'[[Category:Potions]]',
		'[[Uses facility::Standard Potion Station||Potent Potion Station]]',
		'?Profession Level A = lvl',
		'? #- = name',
		'?Uses item.Uses item #- = reagents',
		'?Value = sell',
		'?Activity XP = brewXP',
		'?Uses item.Activity XP = prepXP',
		'?Activity KP = brewKP',
		'?Uses item.Activity KP = prepKP',
		'?Activity duration = brewDuration',
		'?Uses item.Activity duration = prepDuration',
		'sort = Profession Level A'
	}
	local results = mw.smw.ask(query)

	results = p.formatResults(results)

	return p.displayTable(results)

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 tostring(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 potions
	for i, item in ipairs(results) do

		--in case of single reagent potions, make table of 1 element
		if type(item.reagents) ~= 'table' then
			item.reagents = {item.reagents}
		end

		-- iterate through reagents, adding buy price to running total (individuals not needed)
		-- starting value 20 is for bottle
		item.buy = 20
		for j, reagent in ipairs(item.reagents) do

			--shamelessley lifted from Module:Products
			local shopPriceQuery = '[[:+]][[Sold item::' .. reagent .. ']]|?Shop buy price|mainlabel=' .. reagent
			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
			end

			item.buy = item.buy + shopPrice
		end

		-- sanitise data, set to 0 if its not there
		local lvl = item.lvl or '?'
		local buy = item.buy or 0
		local sell = item.sell or 0
		local brewXP = item.brewXP or 0
		local prepXP = item.prepXP or 0
		local brewKP = item.brewKP or 0
		local prepKP = item.prepKP or 0
		local brewDuration = item.brewDuration or 0
		local prepDuration = item.prepDuration or 0

		-- direct values
		item.profit = item.sell and item.buy and item.sell - item.buy
		item.XP = item.brewXP and item.prepXP and item.brewXP + item.prepXP
		item.KP = item.brewKP and item.prepKP and item.brewKP + item.prepKP
		item.profitPerXP = item.profit and item.XP and math.floor(item.profit / item.XP * 100) / 100

		-- 40 seconds to buy new supplies and travel and deposit and everything, per 12 potions
		local batchSize = 12
		local downtime = 40
		item.duration = item.prepDuration and item.brewDuration and item.prepDuration + item.brewDuration + downtime/batchSize
		item.potionPerHour = item.duration and 1 / item.duration * 3600

		-- properties per hour
		item.XPPerHour = item.XP and item.potionPerHour and math.floor(item.XP * item.potionPerHour)
		item.KPPerHour = item.KP and item.potionPerHour and math.floor(item.KP * item.potionPerHour) / 100
		item.profitPerHour = item.profit and item.potionPerHour and math.floor(item.profit * item.potionPerHour)
	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:Alchemist small icon.png|15px]]  Level')
			:done()
			:tag('th')
				:wikitext('Potion')
			:done()
			:tag('th')
				:wikitext('Reagents (plus bottle)')
			: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()


	for i,item in ipairs(results) do

		-- need to generate the text for the reagent cell before starting the row	
		local reagentCell = ''
		for j, reagent in ipairs(item.reagents) do
			reagentCell = reagentCell .. '[[File:' .. reagent .. '.png|30px|link=' .. reagent .. ']] [[' .. reagent .. ']]<br>'
		end

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

	out
		:tag('tr')
			:IF(item.lvl)
				:tag('td')
					:css{ ['text-align'] = 'center' }
					:wikitext(item.lvl)
				:done()
			:ELSE()
				:node(unknown_value_cell)
			:END()

			:tag('td')
				:wikitext('[[File:' .. item.name .. '.png|30px|link=' .. item.name .. ']] [[' .. item.name .. ']]')
			:done()

			:tag('td')
				:wikitext(reagentCell)
			:done()

			:wikitext(currency_cell(item.buy))

			:wikitext(currency_cell(item.sell))

			:wikitext(currency_cell(item.profit))

			:wikitext(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()

			:wikitext(currency_cell(item.profitPerXP))

		:done()

	end

	return out
end

return p