Editing Module:Sandbox/User:Alsang

require('strict')
require('Module:Mw.html extension')
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 = {
		'[[Rotten Tree Stump||Rotting Log (Hopeport)||Rock Pool||Scrap Metal Bin]]',
		'?Activity JSON = activityJSON',
		'?#- = name',
		'limit = 500'
	}
	local results = mw.smw.ask(query)
	
	results = p.screenResults(results)

	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

-- goes through a list of results and culls the ones which are not needed for this table
function p.screenResults(results)
	local resultsScreened = {}
	
	-- iterate through products
	for _, item in ipairs(results) do
		
		-- if theres only one activity, wrap in table
		if type(item.activityJSON)=='string' then
			item.activityJSON = { item.activityJSON }
		end
		
		for j,json in ipairs(item.activityJSON) do
		
--			if string.find(json,'Branches') or string.find(json,'Split') then
				
				local activityJSON = mw.text.jsonDecode(json)
				local itemNew = {}
				itemNew.name = item.name
				itemNew.XP = activityJSON.xp
				itemNew.lvl = activityJSON.level
				itemNew.duration = activityJSON.duration and tonumber(activityJSON.duration)
				itemNew.product = activityJSON.output[1].name
				itemNew.activityJSON = json
				table.insert(resultsScreened,itemNew)
--			end
		end
	end
	
	-- sort the results by recipe level
	table.sort(resultsScreened, function(item1, item2)
		local lvl1 = item1.lvl
		local lvl2 = item2.lvl
		if (lvl1 == nil) ~= (lvl2 == nil) then --one of two are empty
			return lvl2 == nil -- true if lvl2 is nil but not lvl1, false if lvl1 is nil but not lvl2
		end
		if lvl1 == nil then
			return false  -- Both empty, equivalent
		end
		return lvl1 < lvl2 -- normal comparison
	end)

	return resultsScreened

end

-- do calculations and determine strings to go in cells
function p.formatResults(results)
	
	-- iterate through products
	for _, item in ipairs(results) do
			
		item.givesCoins = string.find(item.product,'Split')

		-- profit from selling the items
		if not item.givesCoins then
			
			-- protection against missing parameters
			item.product = item.product or ''
			
			-- query is: there is a shop object page that has this item AND has buy price cannot be "N/A"
			local query = mw.smw.ask('[[Sold item::' .. item.product .. ']][[Shop sell price::!~N/A]]|?Shop sell price=data|mainlabel=-')
			if type(query)=='table' then
				-- price is returned as a number
				item.coins = tonumber(query[1].data)*0.005 or 0
			else
				-- will return nil if the item is not in a shop, or if it is in a shop but can only be sold to the shop
				item.coins =  0
			end
			
		end
		
		-- no downtime for passives
		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.coinsPerHour = item.coins and item.productPerHour and math.floor(item.coins * 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:Forager small icon.png|15px]]  Level')
			:done()
			:tag('th')
				:attr{ colspan = '2' }
				:wikitext('Forager node')
			:done()
			:tag('th')
				:attr{ colspan = '3' }
				:wikitext('Product')
			:done()
			:tag('th')
				:wikitext('Quantity/hr')
			:done()
			:tag('th')
				:attr{ colspan = '10' }
				:wikitext('Profit/hr')
			:done()
			:tag('th')
				:wikitext('XP')
			:done()
			:tag('th')
				:wikitext('XP/hr')
			: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')
		
			--level
			:IF(item.lvl)
				:tag('td')
					:css{ ['text-align'] = 'center' }
					:wikitext(item.lvl)
				:done()
			:ELSE()
				:node(unknown_value_cell)
			:END()
			
			-- activity
			:tag('td')
				:css{ ['border-right'] = '0', ['text-align'] = 'right', ['max-width'] = '100px' }
				:attr{ ['data-sort-value'] = item.name }
				:wikitext(' [[File:' .. item.name .. '.png|link=' .. item.name .. '|30x30px]]')
			:done()
			:tag('td')
				:addClass('plinkt-link no-border')
				:wikitext('[[' .. item.name .. ']]')
			:done()
			

			:tag('td')
				:css{ ['border-right'] = '0', ['text-align'] = 'right' }
				:attr{ ['data-sort-value'] = item.product }
				:wikitext('0.005 &times; ')
			:done()
			:tag('td')
				:css{ ['border-right'] = '0', ['text-align'] = 'center', ['max-width'] = '100px' }
				:addClass('plinkt-link no-border')
				:wikitext('[[File:' .. item.product .. '.png|link=' .. item.product .. '|30x30px]]')
			:done()
			:tag('td')
				:addClass('plinkt-link no-border')
				:wikitext('[[' .. item.product .. ']]')
			:done()

			
			-- products per hour (items)
			:tag('td')
				:css{ ['text-align'] = 'right' }
				:wikitext('0.9 &times; [[File:' .. item.product .. '.png|link=' .. item.product .. '|30x30px]]')
			:done()

			-- profit per hour (coins)
			:node(currency_cell(item.coinsPerHour))

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

		:done()
	end

	return out
end

return p